github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-04 20:46:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ollama): Remove redundant message from response_metadata (#30778)

Browse Source
This commit is contained in:
Jacob Lee
2025-04-10 23:12:57 -07:00
committed by GitHub
parent f3c3ec9aec
commit e72f3c26a0
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
libs/partners/ollama/langchain_ollama/chat_models.py
View File

@@ -740,7 +740,9 @@ class ChatOllama(BaseChatModel):
tool_calls=_get_tool_calls_from_response(stream_resp),
),
generation_info=(
dict(stream_resp) if stream_resp.get("done") is True else None
dict(stream_resp).pop("message", None)
if stream_resp.get("done") is True
else None
),
)
if chunk.generation_info and (
@@ -792,7 +794,9 @@ class ChatOllama(BaseChatModel):
tool_calls=_get_tool_calls_from_response(stream_resp),
),
generation_info=(
dict(stream_resp) if stream_resp.get("done") is True else None
dict(stream_resp).pop("message", None)
if stream_resp.get("done") is True
else None
),
)
if chunk.generation_info and (