added maximum number of retries to structured LLM chat

Deeplearn-PeD · Sep 22, 2024 · 47e5cb8 · 47e5cb8
1 parent 1494621
commit 47e5cb8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/base_agent/llminterface.py b/base_agent/llminterface.py
@@ -171,6 +171,7 @@ def get_response(self, question: str, context: str = "", response_model: BaseMod
             model=self.model,
             messages=messages,
             response_model=response_model,
+            max_retries=3
         )
         self.chat_history.enqueue(response)