Update app.py

app.py

@@ -20,6 +20,7 @@ TELEGRAM_BOT_TOKEN = os.environ.get("TELEGRAM_BOT_TOKEN")
 AI_API_ENDPOINT = os.environ.get("AI_API_ENDPOINT", "https://new.yangtb2024.me/v1/chat/completions")
 AI_API_KEY = os.environ.get("AI_API_KEY")
 AI_MODEL = os.environ.get("AI_MODEL", "gemini-2.0-flash-exp")
+CF_WORKER_URL = os.environ.get("CF_WORKER_URL")
 
 AI_API_HEADERS = {
     "Content-Type": "application/json",
@@ -122,13 +123,25 @@ class EventEmitter:
 
 async def set_bot_commands(bot: Bot):
     try:
-        await bot.delete_my_commands()
+        url = f"{CF_WORKER_URL}/bot{TELEGRAM_BOT_TOKEN}/deleteMyCommands"
+        async with aiohttp.ClientSession() as session:
+           async with session.post(url) as response:
+                if not response.ok:
+                    error_text = await response.text()
+                    logging.error(f"Telegram delete commands failed: {response.status}, {error_text}")
+                    return
         logging.info("Telegram commands deleted successfully")
         commands = [
             (command[0], command[1])
             for command in BOT_COMMANDS
         ]
-        await bot.set_my_commands(commands=commands)
+        url = f"{CF_WORKER_URL}/bot{TELEGRAM_BOT_TOKEN}/setMyCommands"
+        async with aiohttp.ClientSession() as session:
+           async with session.post(url, json={"commands": commands}) as response:
+                if not response.ok:
+                    error_text = await response.text()
+                    logging.error(f"Telegram set commands failed: {response.status}, {error_text}")
+                    return
         logging.info("Telegram commands set successfully")
     except Exception as e:
         logging.error(f"Error setting Telegram commands: {e}")
@@ -160,37 +173,38 @@ async def handle_telegram_update(update: Update, context):
 
         if command == "clearall":
             chat_histories.pop(chat_id, None)
-            await context.bot.send_message(chat_id=chat_id, text="聊天记录已清空。")
+            await send_telegram_message(chat_id=chat_id, text="聊天记录已清空。", context=context)
             return
         if command == "help":
-            await context.bot.send_message(chat_id=chat_id, text=get_help_message())
+            await send_telegram_message(chat_id=chat_id, text=get_help_message(), context=context)
             return
         if command == "start":
-            await context.bot.send_message(
+            await send_telegram_message(
                 chat_id=chat_id,
                 text="欢迎使用！请选择操作：",
                 reply_markup={
                     "inline_keyboard": [[{"text": "清空聊天记录", "callback_data": "clearall"}]],
                 },
+                context=context
             )
             return
         if is_group_chat:
             if command == "enableai":
                 GROUP_SETTINGS.setdefault(chat_id, {}).update({"aiEnabled": True})
-                await context.bot.send_message(chat_id=chat_id, text="已在群组中启用 AI 回复。")
+                await send_telegram_message(chat_id=chat_id, text="已在群组中启用 AI 回复。", context=context)
                 return
             if command == "disableai":
                 GROUP_SETTINGS.setdefault(chat_id, {}).update({"aiEnabled": False})
-                await context.bot.send_message(chat_id=chat_id, text="已在群组中禁用 AI 回复。")
+                await send_telegram_message(chat_id=chat_id, text="已在群组中禁用 AI 回复。", context=context)
                 return
             if user_message.startswith("/setprefix "):
                 prefix = user_message[len("/setprefix ") :].strip()
                 GROUP_SETTINGS.setdefault(chat_id, {}).update({"prefix": prefix})
-                await context.bot.send_message(chat_id=chat_id, text=f"已设置群组触发前缀为: {prefix}")
+                await send_telegram_message(chat_id=chat_id, text=f"已设置群组触发前缀为: {prefix}", context=context)
                 return
             if command == "getprefix":
                 prefix = GROUP_SETTINGS.get(chat_id, {}).get("prefix", "无")
-                await context.bot.send_message(chat_id=chat_id, text=f"当前群组触发前缀为: {prefix}")
+                await send_telegram_message(chat_id=chat_id, text=f"当前群组触发前缀为: {prefix}", context=context)
                 return
         else:
             if user_message.startswith("/settemp "):
@@ -198,15 +212,15 @@ async def handle_telegram_update(update: Update, context):
                     temp = float(user_message[len("/settemp ") :].strip())
                     if 0 <= temp <= 2:
                         USER_SETTINGS.setdefault(from_user_id, {}).update({"temperature": temp})
-                        await context.bot.send_message(chat_id=chat_id, text=f"已设置AI回复温度为: {temp}")
+                        await send_telegram_message(chat_id=chat_id, text=f"已设置AI回复温度为: {temp}", context=context)
                     else:
-                        await context.bot.send_message(chat_id=chat_id, text="温度设置无效，请输入0到2之间的数字。")
+                        await send_telegram_message(chat_id=chat_id, text="温度设置无效，请输入0到2之间的数字。", context=context)
                 except ValueError:
-                    await context.bot.send_message(chat_id=chat_id, text="温度设置无效，请输入0到2之间的数字。")
+                    await send_telegram_message(chat_id=chat_id, text="温度设置无效，请输入0到2之间的数字。", context=context)
                 return
             if command == "gettemp":
                 temp = USER_SETTINGS.get(from_user_id, {}).get("temperature", DEFAULT_TEMP)
-                await context.bot.send_message(chat_id=chat_id, text=f"当前AI回复温度为: {temp}")
+                await send_telegram_message(chat_id=chat_id, text=f"当前AI回复温度为: {temp}", context=context)
                 return
             if user_message.startswith("/promat "):
                 try:
@@ -214,24 +228,24 @@ async def handle_telegram_update(update: Update, context):
                     if index in PROMPT_TEMPLATES:
                         global CURRENT_PROMPT_INDEX
                         CURRENT_PROMPT_INDEX = index
-                        await context.bot.send_message(chat_id=chat_id, text=f"已切换到提示词 {index}。")
+                        await send_telegram_message(chat_id=chat_id, text=f"已切换到提示词 {index}。", context=context)
                     else:
-                        await context.bot.send_message(
-                            chat_id=chat_id, text="提示词索引无效。请使用 /getpromat 查看可用的索引。"
+                        await send_telegram_message(
+                            chat_id=chat_id, text="提示词索引无效。请使用 /getpromat 查看可用的索引。", context=context
                         )
                 except ValueError:
-                    await context.bot.send_message(
-                        chat_id=chat_id, text="提示词索引无效。请使用 /getpromat 查看可用的索引。"
+                    await send_telegram_message(
+                        chat_id=chat_id, text="提示词索引无效。请使用 /getpromat 查看可用的索引。", context=context
                     )
                 return
             if command == "getpromat":
-                await context.bot.send_message(
-                    chat_id=chat_id, text=f"当前使用的提示词索引是: {CURRENT_PROMPT_INDEX}"
+                await send_telegram_message(
+                    chat_id=chat_id, text=f"当前使用的提示词索引是: {CURRENT_PROMPT_INDEX}", context=context
                 )
                 return
             if command == "resetuser":
                 USER_SETTINGS.pop(from_user_id, None)
-                await context.bot.send_message(chat_id=chat_id, text="已重置您的个人设置。")
+                await send_telegram_message(chat_id=chat_id, text="已重置您的个人设置。", context=context)
                 return
 
     if is_group_chat:
@@ -286,7 +300,7 @@ async def process_ai_message(chat_id, user_message, from_user_id, context):
                 if not ai_response.ok:
                     error_text = await ai_response.text()
                     logging.error(f"AI API 响应失败: {ai_response.status}, {error_text}")
-                    await context.bot.send_message(chat_id=chat_id, text="AI API 响应失败，请稍后再试")
+                    await send_telegram_message(chat_id=chat_id, text="AI API 响应失败，请稍后再试", context=context)
                     return
 
                 ai_data = await ai_response.json()
@@ -295,11 +309,11 @@ async def process_ai_message(chat_id, user_message, from_user_id, context):
                 history.append({"role": "assistant", "content": ai_reply})
                 chat_histories[chat_id] = history
 
-                await context.bot.send_message(chat_id=chat_id, text=ai_reply)
+                await send_telegram_message(chat_id=chat_id, text=ai_reply, context=context)
 
     except Exception as error:
         logging.error(f"处理消息时发生错误: {error}")
-        await context.bot.send_message(chat_id=chat_id, text="处理消息时发生错误，请稍后再试")
+        await send_telegram_message(chat_id=chat_id, text="处理消息时发生错误，请稍后再试", context=context)
 
 
 async def handle_ai_response(ai_data, chat_id, history, event_emitter, context):
@@ -455,22 +469,39 @@ async def handle_callback_query(callback_query, context):
 
     if data == "clearall":
         chat_histories.pop(chat_id, None)
-        await context.bot.send_message(chat_id=chat_id, text="聊天记录已清空。")
+        await send_telegram_message(chat_id=chat_id, text="聊天记录已清空。", context=context)
 
-    await context.bot.send_message(
+    await send_telegram_message(
         chat_id=chat_id,
         text="请选择操作：",
         reply_markup={
             "inline_keyboard": [[{"text": "清空聊天记录", "callback_data": "clearall"}]],
         },
+        context=context
     )
 
 
 async def send_event_message(chat_id, event, context):
     if event["type"] == "status":
-        await context.bot.send_message(chat_id=chat_id, text=f"状态更新: {event['data']['description']}")
+        await send_telegram_message(chat_id=chat_id, text=f"状态更新: {event['data']['description']}", context=context)
     elif event["type"] == "citation":
-        await context.bot.send_message(chat_id=chat_id, text=f"引用信息: {event['data']['metadata'][0]['name']}")
+        await send_telegram_message(chat_id=chat_id, text=f"引用信息: {event['data']['metadata'][0]['name']}", context=context)
+
+
+async def send_telegram_message(chat_id, text, context, options=None):
+    url = f"{CF_WORKER_URL}/bot{TELEGRAM_BOT_TOKEN}/sendMessage"
+    json_data = {"chat_id": chat_id, "text": text}
+    if options:
+        json_data.update(options)
+    try:
+        async with aiohttp.ClientSession() as session:
+            async with session.post(url, json=json_data) as response:
+                if not response.ok:
+                    error_text = await response.text()
+                    logging.error(f"发送 Telegram 消息失败: {response.status}, {error_text}")
+    except Exception as error:
+        logging.error(f"发送 Telegram 消息时发生错误: {error}")
+
 
 def get_help_message():
     return f"""
@@ -501,6 +532,7 @@ def get_help_message():
   - 机器人具有攻击性，请谨慎使用。
 """
 
+
 async def main():
     logging.basicConfig(level=logging.INFO)
     application = Application.builder().token(TELEGRAM_BOT_TOKEN).build()
@@ -531,5 +563,6 @@ async def main():
     # Start the bot
     await application.run_polling()
 
+
 if __name__ == "__main__":
-    asyncio.run(main())
+    asyncio.run(main())