phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(utils.py): handle ollama yielding a dict

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-22 12:23:42 +05:30
parent eb2d13e2fb
commit f1270a7c78
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/utils.py
View file

@ -5811,7 +5811,10 @@ class CustomStreamWrapper:
def handle_ollama_stream(self, chunk):
try:
json_chunk = json.loads(chunk)
if isinstance(chunk, dict):
json_chunk = chunk
else:
json_chunk = json.loads(chunk)
if "error" in json_chunk:
raise Exception(f"Ollama Error - {json_chunk}")