From 837ce269aefa8599c23a637b5bbb22786782c0f1 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 25 Dec 2023 23:45:27 +0530 Subject: [PATCH] (feat) ollama_chat add async stream --- litellm/llms/ollama_chat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/llms/ollama_chat.py b/litellm/llms/ollama_chat.py index cfcc883fa83..16d234f7a4e 100644 --- a/litellm/llms/ollama_chat.py +++ b/litellm/llms/ollama_chat.py @@ -263,7 +263,7 @@ async def ollama_async_streaming(url, data, model_response, encoding, logging_ob streamwrapper = litellm.CustomStreamWrapper( completion_stream=response.aiter_lines(), model=data["model"], - custom_llm_provider="ollama", + custom_llm_provider="ollama_chat", logging_obj=logging_obj, ) async for transformed_chunk in streamwrapper: