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)