Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import json
 import requests
 from flask import Flask, request, jsonify
-from datetime import datetime
+from datetime import datetime, timezone, timedelta
 import asyncio
 
 app = Flask(__name__)
@@ -52,6 +52,17 @@ BOT_COMMANDS = [
 ]
 DEFAULT_TEMP = 1.5
 
+def get_current_datetime_utc8():
+    utc_plus_8 = timezone(timedelta(hours=8))
+    now = datetime.now(utc_plus_8)
+    return now.strftime("%Y年%m月%d日 %H时%M分%S秒")
+
+TOOL_DEFINITIONS = {
+    "get_current_datetime_utc8": {
+        "description": "获取当前的日期和时间（UTC+8时区）。",
+    }
+}
+
 def make_telegram_request(method, data=None):
     url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/{method}"
     if PHP_PROXY_URL:
@@ -182,7 +193,6 @@ def parseCommand(userMessage):
         command = command.split('@')[0]
     return command[1:]
 
-
 async def handlePrivateCommand(chatId, userMessage, fromUserId, isGroupChat):
     command = parseCommand(userMessage)
     if userMessage.startswith('/settemp '):
@@ -237,18 +247,57 @@ async def processAiMessage(chatId, userMessage, fromUserId, message_id):
     
     thinking_message = await sendTelegramMessage(chatId, "正在思考，请等待...", options={'reply_to_message_id': message_id})
     thinking_message_id = thinking_message.get('result', {}).get('message_id')
-
+    
+    tools = [
+        {
+            "type": "function",
+            "function": {
+                "name": "get_current_datetime_utc8",
+                "description": "获取当前的日期和时间（UTC+8时区）。",
+            }
+        }
+    ]
+    
     try:
         ai_response = requests.post(AI_API_ENDPOINT, headers=AI_API_HEADERS, json={
             'model': AI_MODEL,
             'messages': messages,
             'max_tokens': MAX_TOKENS,
             'temperature': userTemp,
+            'tools': tools,
+            'tool_choice': 'auto',
         })
         ai_response.raise_for_status()
         ai_data = ai_response.json()
         ai_reply = await handleAiResponse(ai_data, chatId, history)
-
+        
+        if isinstance(ai_reply, dict) and ai_reply.get('tool_calls'):
+            tool_calls = ai_reply['tool_calls']
+            for tool_call in tool_calls:
+                tool_call_id = tool_call.get('id')
+                function_name = tool_call['function']['name']
+                
+                if function_name == 'get_current_datetime_utc8':
+                    tool_result = get_current_datetime_utc8()
+                    
+                    messages.append({
+                        'tool_call_id': tool_call_id,
+                        'role': 'tool',
+                        'name': function_name,
+                        'content': tool_result
+                    })
+                    
+                    ai_response = requests.post(AI_API_ENDPOINT, headers=AI_API_HEADERS, json={
+                        'model': AI_MODEL,
+                        'messages': messages,
+                        'max_tokens': MAX_TOKENS,
+                        'temperature': userTemp,
+                    })
+                    ai_response.raise_for_status()
+                    ai_data = ai_response.json()
+                    ai_reply = await handleAiResponse(ai_data, chatId, history)
+                    break
+        
         history.append({'role': 'assistant', 'content': ai_reply})
         chatHistories[chatId] = history
 
@@ -263,8 +312,11 @@ async def processAiMessage(chatId, userMessage, fromUserId, message_id):
 async def handleAiResponse(ai_data, chatId, history):
     if ai_data and ai_data.get('choices') and len(ai_data['choices']) > 0:
         choice = ai_data['choices'][0]
-        if choice.get('message') and choice['message'].get('content'):
-            return choice['message']['content']
+        if choice.get('message'):
+            if choice['message'].get('content'):
+                return choice['message']['content']
+            elif choice['message'].get('tool_calls'):
+                return {'tool_calls': choice['message']['tool_calls']}
     return 'AI 返回了无法识别的格式'
 
 async def handleCallbackQuery(callbackQuery):
@@ -318,7 +370,6 @@ async def editTelegramMessage(chatId, message_id, text, options={}):
     except requests.exceptions.RequestException as e:
         print(f'编辑 Telegram 消息失败: {e}')
 
-
 def getHelpMessage():
     return f"""
 可用指令：