update

phidatahq · Nov 13, 2024 · 986aee5 · 986aee5
1 parent 2a753fe
commit 986aee5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/phi/model/google/gemini.py b/phi/model/google/gemini.py
@@ -498,7 +498,7 @@ def response(self, messages: List[Message]) -> ModelResponse:
             model_response.content = assistant_message.get_content_string()
 
         # -*- Remove parts from messages
-        [setattr(m, 'parts', None) for m in messages if hasattr(m, 'parts')]
+        [setattr(m, "parts", None) for m in messages if hasattr(m, "parts")]  # type: ignore
 
         logger.debug("---------- Gemini Response End ----------")
         return model_response
@@ -619,5 +619,6 @@ def response_stream(self, messages: List[Message]) -> Iterator[ModelResponse]:
             yield from self.response_stream(messages=messages)
 
         # -*- Remove parts from messages
-        [setattr(m, 'parts', None) for m in messages if hasattr(m, 'parts')]
+        [setattr(m, "parts", None) for m in messages if hasattr(m, "parts")]  # type: ignore
+
         logger.debug("---------- Gemini Response End ----------")