fix(ollama.py): fix returned error message for streaming error

This commit is contained in:
Krrish Dholakia 2024-01-08 23:52:57 +05:30
parent 9527a9770f
commit e990c70beb

View file

@ -234,7 +234,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=str(response)
)
streamwrapper = litellm.CustomStreamWrapper(