diff --git a/phi/model/google/gemini.py b/phi/model/google/gemini.py index f0975cb49..621343a86 100644 --- a/phi/model/google/gemini.py +++ b/phi/model/google/gemini.py @@ -504,9 +504,9 @@ def response(self, messages: List[Message]) -> ModelResponse: model_response.content = assistant_message.get_content_string() # -*- Remove parts from messages - for m in messages: - if hasattr(m, "parts"): - m.parts = None + # for m in messages: + # if hasattr(m, "parts"): + # m.parts = None logger.debug("---------- Gemini Response End ----------") return model_response @@ -621,8 +621,8 @@ def response_stream(self, messages: List[Message]) -> Iterator[ModelResponse]: yield from self.response_stream(messages=messages) # -*- Remove parts from messages - for m in messages: - if hasattr(m, "parts"): - m.parts = None + # for m in messages: + # if hasattr(m, "parts"): + # m.parts = None logger.debug("---------- Gemini Response End ----------")