fix tg tunnel bug.

This commit is contained in:
Liu Zhicong
2023-09-29 17:22:22 -07:00
parent 99f6f12227
commit 55b2701faa
2 changed files with 11 additions and 4 deletions
+1 -1
View File
@@ -1,4 +1,4 @@
#!/usr/bin/bash #!/bin/bash
# Build the docker image # Build the docker image
docker build -t aios . docker build -t aios .
docker tag aios:latest paios/aios:latest docker tag aios:latest paios/aios:latest
+10 -3
View File
@@ -3,6 +3,7 @@ import threading
import asyncio import asyncio
import uuid import uuid
import time import time
import aiofiles
from telegram import Update from telegram import Update
from telegram import Bot from telegram import Bot
@@ -153,12 +154,12 @@ class TelegramTunnel(AgentTunnel):
reomte_user_name = contact.name reomte_user_name = contact.name
if not contact.is_family_member: if not contact.is_family_member:
if self.allow_group != "contact" and self.allow_group !="guest": if self.allow_group != "contact" and self.allow_group !="guest":
await update.message.reply_text(f"You are not allowed to talk to me! Please contact my father~") await update.message.reply_text(f"You're not supposed to talk to me! Please contact my father~")
return return
else: else:
if self.allow_group != "guest": if self.allow_group != "guest":
await update.message.reply_text(f"You are not allowed to talk to me! Please contact my father~") await update.message.reply_text(f"You're not supposed to talk to me! Please contact my father~")
return return
if cm.is_auto_create_contact_from_telegram: if cm.is_auto_create_contact_from_telegram:
@@ -187,7 +188,13 @@ class TelegramTunnel(AgentTunnel):
if knowledge_object is not None: if knowledge_object is not None:
if knowledge_object.get_object_type() == ObjectType.Image: if knowledge_object.get_object_type() == ObjectType.Image:
image = KnowledgeBase().bytes_from_object(knowledge_object) image = KnowledgeBase().bytes_from_object(knowledge_object)
await update.message.reply_photo(image) try:
async with aiofiles.open("tg_send_temp.png", mode='wb') as local_file:
if local_file:
await local_file.write(image)
await update.message.reply_photo("tg_send_temp.png")
except Exception as e:
logger.error(f"save image error: {e}")
return return
else: else:
pos = resp_msg.body.find("audio file") pos = resp_msg.body.find("audio file")