Ollama part2 (#356)

Co-authored-by: Tedfulk <[email protected]>
instructor-ai · Jan 23, 2024 · 23a3abb · 23a3abb
1 parent bd837bb
commit 23a3abb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/instructor/patch.py b/instructor/patch.py
@@ -51,10 +51,10 @@ def dump_message(message: ChatCompletionMessage) -> ChatCompletionMessageParam:
         "role": message.role,
         "content": message.content or "",
     }
-    if message.tool_calls is not None:
+    if hasattr(message, "tool_calls") and message.tool_calls is not None:
         ret["tool_calls"] = message.model_dump()["tool_calls"]
         ret["content"] += json.dumps(message.model_dump()["tool_calls"])
-    if message.function_call is not None:
+    if hasattr(message, "function_call") and message.function_call is not None:
         ret["content"] += json.dumps(message.model_dump()["function_call"])
     return ret