From a81823da2d11c63ab5d0edc5285e243401cbba5d Mon Sep 17 00:00:00 2001 From: Mason Daugherty Date: Fri, 22 Aug 2025 11:10:05 -0400 Subject: [PATCH] fix(ollama): enhance image data handling in content blocks for v1 support --- libs/partners/ollama/langchain_ollama/chat_models.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/libs/partners/ollama/langchain_ollama/chat_models.py b/libs/partners/ollama/langchain_ollama/chat_models.py index 5ad18987b6c..2607ff3d31c 100644 --- a/libs/partners/ollama/langchain_ollama/chat_models.py +++ b/libs/partners/ollama/langchain_ollama/chat_models.py @@ -269,8 +269,12 @@ def _lc_tool_call_to_openai_tool_call(tool_call_: ToolCall) -> dict: def _get_image_from_data_content_block(block: dict) -> str: """Format standard data content block to format expected by Ollama.""" if block["type"] == "image": - if block["source_type"] == "base64": + if block.get("source_type") == "base64": + # v0 return block["data"] + if block.get("base64") is not None: + # v1 + return block["base64"] error_message = "Image data only supported through in-line base64 format." raise ValueError(error_message)