From f167d33f6fb2dd02845a25aba9b88fef744f795e Mon Sep 17 00:00:00 2001 From: jacob weiss Date: Thu, 3 Oct 2024 16:42:00 -0600 Subject: [PATCH] assert content is str --- phi/llm/ollama/chat.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/phi/llm/ollama/chat.py b/phi/llm/ollama/chat.py index 11ae11d46..2f76ef700 100644 --- a/phi/llm/ollama/chat.py +++ b/phi/llm/ollama/chat.py @@ -113,7 +113,7 @@ def response(self, messages: List[Message], current_user_query: str | None = Non if current_user_query is None: for m in reversed(messages): - if m.role == "user": + if m.role == "user" and isinstance(m.content, str): current_user_query = m.content break @@ -457,7 +457,11 @@ def response_stream(self, messages: List[Message], current_user_query: str | Non if any(item.tool_call_error for item in function_call_results): messages = self.add_tool_call_error_message(messages) else: - messages = self.add_original_user_message(messages, original_user_message_content) + # Ensure original_user_message_content is a string or None + user_message = ( + original_user_message_content if isinstance(original_user_message_content, str) else None + ) + messages = self.add_original_user_message(messages, user_message) # Deactivate tool calls by turning off JSON mode after 1 tool call if self.deactivate_tools_after_use: