github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-02-03 15:55:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

llm fix

Browse Source
This commit is contained in:
isaac hershenson
2024-07-29 08:13:46 -07:00
parent eed9a19c2b
commit 1f364dcd81
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -305,9 +305,7 @@ class OllamaLLM(BaseLLM):
if not isinstance(stream_resp, str):
chunk = GenerationChunk(
text=(
stream_resp["message"]["content"]
if "message" in stream_resp
else ""
stream_resp.get("response","")
),
generation_info=(
dict(stream_resp) if stream_resp.get("done") is True else None
@@ -331,9 +329,7 @@ class OllamaLLM(BaseLLM):
if not isinstance(stream_resp, str):
chunk = GenerationChunk(
text=(
stream_resp["message"]["content"]
if "message" in stream_resp
else ""
stream_resp.get("response","")
),
generation_info=(
dict(stream_resp) if stream_resp.get("done") is True else None