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
+import random
 
 app = Flask(__name__)
 
@@ -65,6 +66,44 @@ 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_bing",
+            "description": "使用必应搜索查询",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "query": {
+                        "type": "string",
+                        "description": "搜索查询词"
+                    }
+                },
+                "required": ["query"]
+            }
+        }
     }
 ]
 
@@ -73,6 +112,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 generate_random_number(min, max):
+    return random.randint(min, max)
+
+def search_bing(query):
+  search_api_url = "https://api.bing.microsoft.com/v7.0/search"
+  bing_api_key = os.environ.get("BING_API_KEY")
+  if not bing_api_key:
+      return "请设置 BING_API_KEY 环境变量"
+  headers = {"Ocp-Apim-Subscription-Key": bing_api_key}
+  params = {"q": query}
+  try:
+    response = requests.get(search_api_url, headers=headers, params=params)
+    response.raise_for_status()
+    search_results = response.json()
+    if search_results and search_results.get("webPages") and search_results.get("webPages").get("value"):
+       results = search_results["webPages"]["value"][:3]
+       formatted_results = []
+       for result in results:
+            formatted_results.append(f"[{result['name']}]({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:
@@ -287,41 +351,47 @@ async def handleAiResponse(ai_data, chatId, history, thinking_message_id):
         if choice.get('message'):
             message = choice['message']
             if message.get('tool_calls'):
-                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,
-                    })
-                    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 响应失败，请稍后再试'
+                for tool_call in message['tool_calls']:
+                    function_name = tool_call['function']['name']
+                    arguments = json.loads(tool_call['function']['arguments'])
                     
-                
-                else:
-                     return '不支持的工具调用'
+                    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_bing':
+                         tool_result = search_bing(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,
+                    })
+                    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 响应失败，请稍后再试'
             elif message.get('content'):
               return message['content']