💻 Code: 1. Refactor the code, simplify the function call code.

2. Upgrade ModelMerge version to 0.4.7
yym68686 · May 21, 2024 · 7e76714 · 7e76714
1 parent fdc1e48
commit 7e76714
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/bot.py b/bot.py
@@ -15,6 +15,7 @@
     WEB_HOOK,
     PORT,
     BOT_TOKEN,
+    Users,
     update_first_buttons_message,
     update_model_buttons,
     get_current_lang,
@@ -181,7 +182,7 @@ async def getChatGPT(update, context, title, robot, message, chatid, messageid):
     image_has_send = 0
 
     try:
-        for data in robot.ask_stream(text, convo_id=str(chatid), pass_history=pass_history):
+        for data in robot.ask_stream(text, convo_id=str(chatid), pass_history=pass_history, model=Users.get_config(chatid, "engine")):
             if "🌐" not in data:
                 result = result + data
             tmpresult = result

diff --git a/requirements.txt b/requirements.txt
@@ -2,7 +2,7 @@
 pytz
 python-dotenv
 md2tgmd==0.1.9
-ModelMerge==0.5.4
+ModelMerge==0.5.5
 fake_useragent
 oauth2client==3.0.0
 python-telegram-bot[webhooks,rate-limiter]==21.0.1