Upload free_ask_internet.py

free_ask_internet.py

@@ -279,7 +279,7 @@ def ask_internet(query:str, model:str, debug=False):
         citation_map = {f"citation:{i+1}": (chr(0x245F + i+1), content_list[i].get('url')) 
                        for i in range(len(content_list))}
         
-        # 替换引用标记并添加新标签页打开属性
+        # 替换引用标记
         modified_content = response_content
         for citation, (symbol, url) in citation_map.items():
             if url:
@@ -287,12 +287,55 @@ def ask_internet(query:str, model:str, debug=False):
                     f"[{citation}]", 
                     f"[{symbol}]({url} target=\"_blank\")"
                 )
+
+        # 输出修改后的内容
+        yield modified_content
+
+    yield "\n\n"
+    # 是否返回参考资料
+    if True:
+        yield "---"
+        yield "\n### 参考资料\n"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 = ""
+    
+    # 收集所有回答内容
+    response_content = ""
+    for token in chat(prompt=prompt, model=model):
+        if token:
+            total_token += token
+            response_content += token
+        
+    # 处理引用链接
+    if content_list:
+        # 创建符号到URL的映射（①对应第一个结果，②对应第二个...）
+        symbol_url_map = {
+            chr(0x2460 + i): content_list[i].get('url') 
+            for i in range(len(content_list))
+        }
         
+        # 替换所有符号为带新标签页打开的链接
+        modified_content = response_content
+        for symbol, url in symbol_url_map.items():
+            if url:
+                # 使用 HTML 语法实现新标签页打开
+                modified_content = modified_content.replace(
+                    symbol, 
+                    f'<a href="{url}" target="_blank">{symbol}</a>'
+                )
         yield modified_content
-    # ... 修改参考资料部分的显示 ...
+    yield "\n\n"
+    # 参考资料部分
     if True:
-        yield "---\n参考资料:\n"
+        yield "---\n### 参考资料\n"
+        for idx, url_content in enumerate(content_list):
+            url = url_content.get('url')
+            symbol = chr(0x2460 + idx)
+            # 使用带符号的链接列表
+            yield f"* {symbol} <a href=\"{url}\" target=\"_blank\">{url}</a>\n"
         for idx, url_content in enumerate(content_list):
             url = url_content.get('url')
-            # 使用带圆圈数字编号
-            yield f"* {chr(0x245F + idx+1)} [{url}]({url} target=\"_blank\")*\n"
+            yield f"* {chr(0x245F + idx+1)} [{url}]({url})\n"