VietCat commited on
Commit
2d5241b
·
1 Parent(s): 5c04d26

fix max token error

Browse files
Files changed (1) hide show
  1. app/message_processor.py +2 -3
app/message_processor.py CHANGED
@@ -300,9 +300,8 @@ class MessageProcessor:
300
  asyncio.create_task(self.facebook.send_message(message=f"{get_random_message(SUMMARY_STATUS_MESSAGES)}"))
301
 
302
  try:
303
- from google.generativeai.types import GenerationConfig
304
- generation_config = GenerationConfig(max_output_tokens=2048, temperature=0.5)
305
- answer = await self.channel.llm.generate_text(prompt, generation_config=generation_config)
306
 
307
  if answer and answer.strip():
308
  logger.info(f"LLM trả về câu trả lời thành công: \n\tanswer: {_safe_truncate(answer)}")
 
300
  asyncio.create_task(self.facebook.send_message(message=f"{get_random_message(SUMMARY_STATUS_MESSAGES)}"))
301
 
302
  try:
303
+ # Đã bỏ cấu hình tường minh để sử dụng cài đặt mặc định của thư viện Gemini.
304
+ answer = await self.channel.llm.generate_text(prompt)
 
305
 
306
  if answer and answer.strip():
307
  logger.info(f"LLM trả về câu trả lời thành công: \n\tanswer: {_safe_truncate(answer)}")