handled case where or is

langchain-ai · Nov 23, 2024 · ef718b6 · ef718b6
1 parent a83357d
commit ef718b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/partners/ollama/langchain_ollama/chat_models.py b/libs/partners/ollama/langchain_ollama/chat_models.py
@@ -65,8 +65,8 @@ def _get_tool_calls_from_response(
 ) -> List[ToolCall]:
     """Get tool calls from ollama response."""
     tool_calls = []
-    if "message" in response:
-        if "tool_calls" in response["message"]:
+    if response.get("message", None):
+        if response["message"].get("tool_calls", None):
             for tc in response["message"]["tool_calls"]:
                 tool_calls.append(
                     tool_call(