From c22e0fe17359fbbfdd8c8627ec0aed352b7341d5 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 17 Aug 2023 10:49:48 -0700 Subject: [PATCH] fix tg ai stream check --- litellm/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/main.py b/litellm/main.py index ef8a61758c..24d0faf61d 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -324,7 +324,7 @@ def completion( ## LOGGING logging(model=model, input=prompt, custom_llm_provider=custom_llm_provider, logger_fn=logger_fn) - if stream == True or optional_params['stream_tokens'] == True: + if stream == True: return together_ai_completion_streaming({ "model": model, "prompt": prompt,