phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(proxy_server.py): fix if block

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-29 11:42:24 -08:00
parent 3fea348e4e
commit 7d26eaf481
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/proxy_server.py
View file

@ -941,8 +941,7 @@ async def _PROXY_track_cost_callback(
raise Exception("User API key missing from custom callback.") raise Exception("User API key missing from custom callback.")
else: else:
if kwargs["stream"] != True or ( if kwargs["stream"] != True or (
kwargs["stream"] == True kwargs["stream"] == True and "complete_streaming_response" in kwargs
and kwargs.get("complete_streaming_response") in kwargs
): ):
raise Exception( raise Exception(
f"Model not in litellm model cost map. Add custom pricing - https://docs.litellm.ai/docs/proxy/custom_pricing" f"Model not in litellm model cost map. Add custom pricing - https://docs.litellm.ai/docs/proxy/custom_pricing"