From 1568b162f524127235c0fad46681f7f339b0f7cd Mon Sep 17 00:00:00 2001
From: Chris Van Pelt <vanpelt@wandb.com>
Date: Thu, 1 Feb 2024 16:14:07 -0800
Subject: [PATCH] Re-raise exception in async ollama streaming

---
 litellm/llms/ollama.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/litellm/llms/ollama.py b/litellm/llms/ollama.py
index c8847117e..8e9c12660 100644
--- a/litellm/llms/ollama.py
+++ b/litellm/llms/ollama.py
@@ -267,6 +267,7 @@ async def ollama_async_streaming(url, data, model_response, encoding, logging_ob
                 yield transformed_chunk
     except Exception as e:
         traceback.print_exc()
+        raise e
 
 
 async def ollama_acompletion(url, data, model_response, encoding, logging_obj):