Upload free_ask_internet.py

free_ask_internet.py

@@ -222,17 +222,39 @@ def chat(prompt, model:str, stream=True, debug=False):
  
     
 def ask_internet(query:str, model:str, debug=False):
-  
     content_list = search_web_ref(query,debug=debug)
     if debug:
         print(content_list)
     prompt = gen_prompt(query,content_list,context_length_limit=6000,debug=debug)
-    total_token =  ""
- 
+    total_token = ""
+    
+    # 收集所有回答内容
+    response_content = ""
     for token in chat(prompt=prompt, model=model):
         if token:
             total_token += token
+            response_content += token
             yield token
+            
+    # 处理引用链接
+    if content_list:
+        # 创建引用到URL的映射
+        citation_map = {f"[citation:{i+1}]": content_list[i].get('url') 
+                       for i in range(len(content_list))}
+        
+        # 替换所有引用为链接
+        modified_content = response_content
+        for citation, url in citation_map.items():
+            if url:
+                modified_content = modified_content.replace(
+                    citation, 
+                    f"[{citation}]({url})"
+                )
+        
+        # 输出修改后的内容
+        yield "\n\n修改后的引用链接：\n"
+        yield modified_content
+    
     yield "\n\n"
     # 是否返回参考资料
     if True:
@@ -242,7 +264,6 @@ def ask_internet(query:str, model:str, debug=False):
         count = 1
         for url_content in content_list:
            url = url_content.get('url')
-           yield "*[{}. {}]({})*".format(str(count),url,url )  
+           yield "*[{}. {}]({})*".format(str(count),url,url)  
            yield "\n"
-           count += 1
- 
+           count += 1