github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-17 16:39:52 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Show Stats 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), tool_calls=_get_tool_calls_from_response(stream_resp),
), ),
generation_info=( 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 ( if chunk.generation_info and (
@ -792,7 +794,9 @@ class ChatOllama(BaseChatModel):
tool_calls=_get_tool_calls_from_response(stream_resp), tool_calls=_get_tool_calls_from_response(stream_resp),
), ),
generation_info=( 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 ( if chunk.generation_info and (