cdryzun/tg_bot_collections
1
0
Fork 0
mirror of https://github.com/cdryzun/tg_bot_collections.git synced 2025-11-04 16:56:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

support streaming response for claude_photo_handler/gemini_photo_handler

Browse Source
This commit is contained in:
F4ria
2024-04-11 17:02:39 +08:00
parent fdc9d00eb0
commit 82972d1cbd
2 changed files with 27 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
handlers/claude.py
View File

@ -191,8 +191,18 @@ def claude_photo_handler(message: Message, bot: TeleBot) -> None:
},
],
model=ANTHROPIC_MODEL,
stream=True,
)
bot_reply_markdown(reply_id, who, r.content[0].text, bot)
s = ""
start = time.time()
for e in r:
if e.type == "content_block_delta":
s += e.delta.text
if time.time() - start > 1.7:
start = time.time()
bot_reply_markdown(reply_id, who, s, bot, split_text=False)
bot_reply_markdown(reply_id, who, s, bot)
except Exception as e:
print(e)
bot_reply_markdown(reply_id, who, "answer wrong", bot)