handled case where is

langchain-ai · Nov 23, 2024 · b454c42 · b454c42
1 parent 8abb2c0
commit b454c42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/partners/ollama/langchain_ollama/chat_models.py b/libs/partners/ollama/langchain_ollama/chat_models.py
@@ -66,7 +66,7 @@ def _get_tool_calls_from_response(
     """Get tool calls from ollama response."""
     tool_calls = []
     if response.get("message", None):
-        if "tool_calls" in response["message"]:
+        if response.get("tool_calls", None):
             for tc in response["message"]["tool_calls"]:
                 tool_calls.append(
                     tool_call(