Update app.py

app.py

@@ -3,6 +3,7 @@ import json
 import requests
 from flask import Flask, request, jsonify
 from datetime import datetime
+import asyncio
 
 app = Flask(__name__)
 
@@ -113,7 +114,6 @@ class EventEmitter:
             'metadata': [{'name': title, 'source': url, 'html': False}],
         })
 
-
 def make_telegram_request(method, data=None):
     url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/{method}"
     if PHP_PROXY_URL:  # 使用代理
@@ -265,6 +265,13 @@ def parseCommand(userMessage):
         command = command.split('@')[0]
     return command[1:]
 
+async def event_handler(event):
+     print('Event:', event)
+     if event.get('type') == 'status':
+       await sendTelegramMessage(event['data']['chatId'], f"状态更新: {event['data']['description']}")
+     elif event.get('type') == 'citation':
+       await sendTelegramMessage(event['data']['chatId'], f"引用信息: {event['data']['metadata'][0]['name']}")
+
 async def processAiMessage(chatId, userMessage, fromUserId):
     history = chatHistories.get(chatId, [])
     userTemp = USER_SETTINGS.get(fromUserId, {}).get('temperature', DEFAULT_TEMP)
@@ -279,13 +286,7 @@ async def processAiMessage(chatId, userMessage, fromUserId):
         *history
     ]
     
-    eventEmitter = EventEmitter(async (event) => {
-        print('Event:', event)
-        if event.get('type') == 'status':
-          await sendTelegramMessage(chatId, f"状态更新: {event['data']['description']}")
-        elif event.get('type') == 'citation':
-          await sendTelegramMessage(chatId, f"引用信息: {event['data']['metadata'][0]['name']}")
-    })
+    eventEmitter = EventEmitter(event_handler)
 
     try:
         ai_response = requests.post(AI_API_ENDPOINT, headers=AI_API_HEADERS, json={
@@ -319,7 +320,7 @@ async def handleAiResponse(ai_data, chatId, history, eventEmitter):
             toolCalls = choice['message']['tool_calls']
             toolResults = []
             for toolCall in toolCalls:
-                toolResult = await executeToolCall(toolCall, eventEmitter)
+                toolResult = await executeToolCall(toolCall, eventEmitter,chatId)
                 toolResults.append(toolResult)
 
             newMessages = [
@@ -340,7 +341,7 @@ async def handleAiResponse(ai_data, chatId, history, eventEmitter):
             return 'AI 返回了无法识别的格式'
     return 'AI 返回了无法识别的格式'
 
-async def executeToolCall(toolCall, eventEmitter):
+async def executeToolCall(toolCall, eventEmitter,chatId):
     name = toolCall['function']['name']
     args = toolCall['function'].get('arguments', {})
     
@@ -510,8 +511,5 @@ async def handle_webhook():
 def health_check():
     return 'OK'
 
-
-import asyncio
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))
-
+    app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 8080)))