fix(ollama.py): correctly raise ollama streaming error

Fixes #4974
BerriAI · Jul 30, 2024 · 311521e · 311521e
1 parent 185a685
commit 311521e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/litellm/llms/ollama.py b/litellm/llms/ollama.py
@@ -258,7 +258,7 @@ def get_ollama_response(
                 logging_obj=logging_obj,
             )
         return response
-    elif stream == True:
+    elif stream is True:
         return ollama_completion_stream(url=url, data=data, logging_obj=logging_obj)
 
     response = requests.post(
@@ -326,7 +326,7 @@ def ollama_completion_stream(url, data, logging_obj):
         try:
             if response.status_code != 200:
                 raise OllamaError(
-                    status_code=response.status_code, message=response.text
+                    status_code=response.status_code, message=response.read()
                 )
 
             streamwrapper = litellm.CustomStreamWrapper(