Update Akeno/plugins/chatgpt.py
Browse files- Akeno/plugins/chatgpt.py +8 -7
Akeno/plugins/chatgpt.py
CHANGED
@@ -115,11 +115,12 @@ async def chatbot_talk(client: Client, message: Message):
|
|
115 |
genai.types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: genai.types.HarmBlockThreshold.BLOCK_NONE,
|
116 |
}
|
117 |
)
|
118 |
-
backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
|
119 |
-
backup_chat.append({"role": "user", "parts": [{"text": query}]})
|
120 |
-
chat_session = model_flash.start_chat(history=backup_chat)
|
121 |
-
response_data = chat_session.send_message(query)
|
122 |
-
|
|
|
123 |
if len(output) > 4096:
|
124 |
with open("chat.txt", "w+", encoding="utf8") as out_file:
|
125 |
out_file.write(output)
|
@@ -130,8 +131,8 @@ async def chatbot_talk(client: Client, message: Message):
|
|
130 |
os.remove("chat.txt")
|
131 |
else:
|
132 |
await message.reply_text(output)
|
133 |
-
backup_chat.append({"role": "model", "parts": [{"text": output}]})
|
134 |
-
await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
|
135 |
except Exception as e:
|
136 |
LOGS.error(str(e))
|
137 |
return await message.reply_text(str(e))
|
|
|
115 |
genai.types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: genai.types.HarmBlockThreshold.BLOCK_NONE,
|
116 |
}
|
117 |
)
|
118 |
+
# backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
|
119 |
+
# backup_chat.append({"role": "user", "parts": [{"text": query}]})
|
120 |
+
# chat_session = model_flash.start_chat(history=backup_chat)
|
121 |
+
# response_data = chat_session.send_message(query)
|
122 |
+
response = model_flash.generate_content(query)
|
123 |
+
output = response.text
|
124 |
if len(output) > 4096:
|
125 |
with open("chat.txt", "w+", encoding="utf8") as out_file:
|
126 |
out_file.write(output)
|
|
|
131 |
os.remove("chat.txt")
|
132 |
else:
|
133 |
await message.reply_text(output)
|
134 |
+
# backup_chat.append({"role": "model", "parts": [{"text": output}]})
|
135 |
+
# await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
|
136 |
except Exception as e:
|
137 |
LOGS.error(str(e))
|
138 |
return await message.reply_text(str(e))
|