Ollama part2 (#356)

Co-authored-by: Tedfulk <teddyfulk@gmail.com>
2026-06-05 22:50:18 +00:00 · 2024-01-23 12:36:35 -05:00
parent bd837bb809
commit 23a3abb294
1 changed files with 2 additions and 2 deletions
@@ -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