Upload free_ask_internet.py

free_ask_internet.py

@@ -121,7 +121,8 @@ def gen_prompt(question, content_list, lang="zh-CN", context_length_limit=11000,
     }
     if len(ref_content) > 0:
         prompts = '''
-        You are a large language AI assistant. You are given a user question, and please write clean, concise and accurate answer to the question. You will be given a set of related contexts to the question, each starting with a reference symbol.
+        You are a large language AI assistant. You are given a user question, and please write clean, concise and accurate answer to the question. You will be given a set of related contexts to the question, each starting with a reference symbol. Please use the context and cite the context at the end of each sentence if applicable.
+        Your answer must be correct, accurate and written by an expert using an unbiased and professional tone. Please limit to 1024 tokens. Do not give any information that is not related to the question, and do not repeat. Say "information is missing on" followed by the related topic, if the given context do not provide sufficient information.
         
         IMPORTANT INSTRUCTIONS FOR CITATIONS:
         1. Use ONLY the exact reference symbols provided (like ¹, ², ³) to cite sources
@@ -130,9 +131,8 @@ def gen_prompt(question, content_list, lang="zh-CN", context_length_limit=11000,
         4. Each citation should correspond to exactly one source
         5. Do not use citation symbols that weren't defined in the context
         
-        Your answer must be correct, accurate and written by an expert using an unbiased and professional tone. Please limit to 1024 tokens. Do not give any information that is not related to the question, and do not repeat. Say "information is missing on" followed by the related topic, if the given context do not provide sufficient information.
-        
-        Here are the set of contexts:
+        Please cite the contexts with the reference numbers, in the format (like ¹, ², ³) . If a sentence comes from multiple contexts, please list all applicable citations, like(¹ ²). Other than code and specific names and citations, your answer must be written in the same language as the question.
+            Here are the set of contexts:
         '''  + "\n\n" + "```" 
         
         # 使用 Unicode 上标数字作为引用标记
@@ -275,22 +275,15 @@ def ask_internet(query:str, model:str, debug=False):
     else:
         yield response_content
     
-    # 参考资料部分
-    yield "\n\n---\n"
-    yield "参考资料:\n"
-    if content_list:
-        for i, url_content in enumerate(content_list):
-            if i >= 10:  # 限制最多10个引用
-                break
-            url = url_content.get('url')
-            if url:
-                # 提取域名作为显示名称
-                domain = tldextract.extract(url).domain
-                subdomain = tldextract.extract(url).subdomain
-                site_name = subdomain + "." + domain if subdomain else domain
-                
-                # 使用上标数字作为引用编号
-                num = str(i + 1)
-                sup_num = ''.join(superscript_digits.get(c, c) for c in num)
-                
-                yield f"{sup_num} [{site_name}]({url})\n"
+    yield "\n\n"
+    # 是否返回参考资料
+    if True:
+        yield "---"
+        yield "\nSearxng"
+        yield "参考资料:\n"
+        count = 1
+        for url_content in content_list:
+           url = url_content.get('url')
+           yield "*[{}. {}]({})*".format(str(count),url,url )  
+           yield "\n"
+           count += 1