Disable response validation

vojay-dev · Apr 8, 2024 · 581d1a4 · 581d1a4
1 parent b99bec2
commit 581d1a4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gemini_movie_detectives_api/gemini.py b/gemini_movie_detectives_api/gemini.py
@@ -37,15 +37,15 @@ def __init__(self, project_id: str, location: str, credentials: Credentials, mod
     def start_chat(self) -> ChatSession:
         # noinspection PyBroadException
         try:
-            return self.model.start_chat()
+            return self.model.start_chat(response_validation=False)
         except Exception as e:
             logger.warning(
                 'error while using model %s, using fallback model %s, error: %s',
                 self.model,
                 self.FALLBACK_MODEL,
                 e
             )
-            return self.fallback_model.start_chat()
+            return self.fallback_model.start_chat(response_validation=False)
 
     @staticmethod
     def get_chat_response(chat: ChatSession, prompt: str) -> str: