Update app.py

app.py

@@ -5,6 +5,7 @@ from flask import Flask, request, jsonify
 from datetime import datetime, timezone, timedelta
 import asyncio
 import re
+from urllib.parse import quote
 
 app = Flask(__name__)
 
@@ -13,6 +14,7 @@ AI_API_ENDPOINT = os.environ.get('AI_API_ENDPOINT')
 AI_API_KEY = os.environ.get('AI_API_KEY')
 AI_MODEL = os.environ.get('AI_MODEL')
 PHP_PROXY_URL = os.environ.get('PHP_PROXY_URL')
+SEARXNG_URL = os.environ.get('SEARXNG_URL', 'https://sousuo.emoe.top')  # 使用环境变量，默认值
 
 if not all([TELEGRAM_BOT_TOKEN, AI_API_ENDPOINT, AI_API_KEY, AI_MODEL]):
     raise ValueError("请设置所有必要的环境变量")
@@ -65,6 +67,23 @@ TOOLS = [
                 "required": []
             }
         }
+    },
+        {
+        "type": "function",
+        "function": {
+            "name": "search_with_searxng",
+            "description": "使用 Searxng 搜索信息，并返回搜索结果的摘要",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "query": {
+                        "type": "string",
+                        "description": "要搜索的关键词"
+                    }
+                },
+                "required": ["query"]
+            }
+        }
     }
 ]
 
@@ -73,6 +92,31 @@ def get_current_datetime():
     local_time = utc_time.astimezone(timezone(timedelta(hours=8)))
     return local_time.strftime("%Y-%m-%d %H:%M:%S")
 
+def search_with_searxng(query):
+    """使用 Searxng 搜索信息，并返回搜索结果的摘要"""
+    try:
+        search_url = f"{SEARXNG_URL}/search?q={quote(query)}&format=json"
+        response = requests.get(search_url)
+        response.raise_for_status()
+        results = response.json().get('results', [])
+        if results:
+             # 提取标题和链接，返回格式化的结果
+            formatted_results = []
+            for result in results:
+                title = result.get('title', '无标题')
+                url = result.get('url', '无链接')
+                formatted_results.append(f"- [{title}]({url})")
+            return "\n".join(formatted_results)
+        else:
+            return "没有找到相关的搜索结果。"
+    except requests.exceptions.RequestException as e:
+        print(f"Searxng 请求失败: {e}")
+        return "搜索请求失败，请稍后再试。"
+    except Exception as e:
+        print(f"处理搜索结果时发生错误：{e}")
+        return "处理搜索结果时发生错误。"
+
+
 def make_telegram_request(method, data=None):
     url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/{method}"
     if PHP_PROXY_URL:
@@ -318,7 +362,35 @@ async def handleAiResponse(ai_data, chatId, history, thinking_message_id):
                         print(f'AI API 响应失败: {e}')
                         await editTelegramMessage(chatId, thinking_message_id, 'AI API 响应失败，请稍后再试')
                         return 'AI API 响应失败，请稍后再试'
-                    
+                elif function_name == 'search_with_searxng':
+                   query = tool_call['function']['arguments']['query']
+                   search_results = search_with_searxng(query)
+                   history.append({
+                        'tool_call_id': tool_call['id'],
+                        'role': 'tool',
+                        'name': function_name,
+                        'content': search_results,
+                    })
+                   messages = [
+                       {'role': 'system', 'content': PROMPT_TEMPLATES.get(USER_SETTINGS.get(chatId, {}).get('prompt_index', CURRENT_PROMPT_INDEX), "")},
+                        *history
+                    ]
+
+                   try:
+                        ai_response = requests.post(AI_API_ENDPOINT, headers=AI_API_HEADERS, json={
+                            'model': AI_MODEL,
+                            'messages': messages,
+                            'max_tokens': MAX_TOKENS,
+                            'temperature': USER_SETTINGS.get(chatId, {}).get('temperature', DEFAULT_TEMP),
+                            'tools': TOOLS,
+                        })
+                        ai_response.raise_for_status()
+                        ai_data = ai_response.json()
+                        return await handleAiResponse(ai_data, chatId, history, thinking_message_id)
+                   except requests.exceptions.RequestException as e:
+                       print(f'AI API 响应失败: {e}')
+                       await editTelegramMessage(chatId, thinking_message_id, 'AI API 响应失败，请稍后再试')
+                       return 'AI API 响应失败，请稍后再试'
                 
                 else:
                      return '不支持的工具调用'