diff --git a/libs/partners/ollama/langchain_ollama/chat_models.py b/libs/partners/ollama/langchain_ollama/chat_models.py index 1afab281089..36f0c4f320c 100644 --- a/libs/partners/ollama/langchain_ollama/chat_models.py +++ b/libs/partners/ollama/langchain_ollama/chat_models.py @@ -1122,7 +1122,9 @@ class ChatOllama(BaseChatModel): usage_metadata=_get_usage_metadata_from_generation_info( stream_resp ), - tool_calls=_get_tool_calls_from_response(stream_resp), + tool_calls=_get_tool_calls_from_response( + stream_resp, model_name=self.model + ), ), generation_info=generation_info, ) @@ -1565,3 +1567,4 @@ class ChatOllama(BaseChatModel): +