From acfc990b87ae57de8f35d5abfc1048046fcae65e Mon Sep 17 00:00:00 2001
From: feder-cr <85809106+feder-cr@users.noreply.github.com>
Date: Fri, 13 Sep 2024 23:44:34 +0200
Subject: [PATCH] check gemini

---
 src/llm/llm_manager.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/llm/llm_manager.py b/src/llm/llm_manager.py
index df5681b..5c48c55 100644
--- a/src/llm/llm_manager.py
+++ b/src/llm/llm_manager.py
@@ -84,7 +84,7 @@ class GeminiModel(AIModel):
         HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
         HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
         HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE
-        },max_output_tokens=3000)
+        })
 
     def invoke(self, prompt: str) -> BaseMessage:
         response = self.model.invoke(prompt)