Update app.py

app.py

@@ -5,7 +5,6 @@ from flask import Flask, request, jsonify
 from datetime import datetime, timezone, timedelta
 import asyncio
 import re
-import random
 
 app = Flask(__name__)
 
@@ -14,7 +13,6 @@ 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') # 默认值是你的 URL
 
 if not all([TELEGRAM_BOT_TOKEN, AI_API_ENDPOINT, AI_API_KEY, AI_MODEL]):
     raise ValueError("请设置所有必要的环境变量")
@@ -67,44 +65,6 @@ TOOLS = [
                 "required": []
             }
         }
-    },
-    {
-        "type": "function",
-        "function": {
-            "name": "generate_random_number",
-            "description": "生成指定范围内的随机整数",
-            "parameters": {
-                "type": "object",
-                "properties": {
-                    "min": {
-                        "type": "integer",
-                        "description": "随机数的最小值"
-                    },
-                     "max": {
-                        "type": "integer",
-                        "description": "随机数的最大值"
-                    }
-                },
-                "required": ["min", "max"]
-            }
-        }
-    },
-    {
-       "type": "function",
-        "function": {
-            "name": "search_searxng",
-            "description": "使用SearXNG搜索查询",
-            "parameters": {
-                "type": "object",
-                "properties": {
-                    "query": {
-                        "type": "string",
-                        "description": "搜索查询词"
-                    }
-                },
-                "required": ["query"]
-            }
-        }
     }
 ]
 
@@ -113,28 +73,6 @@ def get_current_datetime():
     local_time = utc_time.astimezone(timezone(timedelta(hours=8)))
     return local_time.strftime("%Y-%m-%d %H:%M:%S")
 
-def generate_random_number(min, max):
-    return random.randint(min, max)
-
-def search_searxng(query):
-  search_url = f"{SEARXNG_URL}/search"
-  params = {"q": query, "format": "json"}
-  try:
-    response = requests.get(search_url, params=params)
-    response.raise_for_status()
-    search_results = response.json()
-    if search_results and search_results.get("results"):
-       results = search_results["results"][:3]
-       formatted_results = []
-       for result in results:
-            formatted_results.append(f"[{result['title']}]({result['url']})")
-       return "\n".join(formatted_results)
-    else:
-       return "没有找到相关结果"
-  except requests.exceptions.RequestException as e:
-    return f"搜索失败: {e}"
-
-
 def make_telegram_request(method, data=None):
     url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/{method}"
     if PHP_PROXY_URL:
@@ -349,47 +287,41 @@ async def handleAiResponse(ai_data, chatId, history, thinking_message_id):
         if choice.get('message'):
             message = choice['message']
             if message.get('tool_calls'):
-                for tool_call in message['tool_calls']:
-                    function_name = tool_call['function']['name']
-                    arguments = json.loads(tool_call['function']['arguments'])
-                    
-                    tool_result = None
-                    if function_name == 'get_current_datetime':
-                        tool_result = get_current_datetime()
-                    elif function_name == 'generate_random_number':
-                         tool_result = generate_random_number(arguments.get('min'), arguments.get('max'))
-                    elif function_name == 'search_searxng':
-                         tool_result = search_searxng(arguments.get('query'))
-
-                    if tool_result is not None:
-                        history.append({
-                            'tool_call_id': tool_call['id'],
-                            'role': 'tool',
-                            'name': function_name,
-                            'content': str(tool_result),
-                        })
-                    else:
-                         return '工具调用失败'
-
-                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,
+                tool_call = message['tool_calls'][0]
+                function_name = tool_call['function']['name']
+                
+                if function_name == 'get_current_datetime':
+                    current_datetime = get_current_datetime()
+                    history.append({
+                        'tool_call_id': tool_call['id'],
+                        'role': 'tool',
+                        'name': function_name,
+                        'content': current_datetime,
                     })
-                    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 响应失败，请稍后再试'
+                    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 '不支持的工具调用'
             elif message.get('content'):
               return message['content']