randydev commited on
Commit
c5534ed
·
verified ·
1 Parent(s): 246cc78

Update Akeno/plugins/chatgpt.py

Browse files
Files changed (1) hide show
  1. 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
- output = response_data.text
 
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))