From 7d26eaf4812471bce4171cf4ddc551d67ae69279 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 29 Feb 2024 11:42:24 -0800 Subject: [PATCH] fix(proxy_server.py): fix if block --- litellm/proxy/proxy_server.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 7578f3809f1..ecf1d4ae593 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -941,8 +941,7 @@ async def _PROXY_track_cost_callback( raise Exception("User API key missing from custom callback.") else: if kwargs["stream"] != True or ( - kwargs["stream"] == True - and kwargs.get("complete_streaming_response") in kwargs + kwargs["stream"] == True and "complete_streaming_response" in kwargs ): raise Exception( f"Model not in litellm model cost map. Add custom pricing - https://docs.litellm.ai/docs/proxy/custom_pricing"