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

fix(litellm_logging.py): fix condition check

Browse source 
Fixes https://github.com/BerriAI/litellm/issues/4633
This commit is contained in:
Krrish Dholakia 2024-07-12 09:22:19 -07:00
parent 88eb25da5c
commit f5b3cc6c02
2 changed files with 57 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

28
litellm/litellm_core_utils/litellm_logging.py
View file

@ -1105,19 +1105,19 @@ class Logging:
and self.model_call_details.get("litellm_params", {}).get(
"acompletion", False
)
== False
is not True
and self.model_call_details.get("litellm_params", {}).get(
"aembedding", False
)
== False
is not True
and self.model_call_details.get("litellm_params", {}).get(
"aimage_generation", False
)
== False
is not True
and self.model_call_details.get("litellm_params", {}).get(
"atranscription", False
)
== False
is not True
):
global openMeterLogger
if openMeterLogger is None:
@ -1150,19 +1150,19 @@ class Logging:
and self.model_call_details.get("litellm_params", {}).get(
"acompletion", False
)
== False
is not True
and self.model_call_details.get("litellm_params", {}).get(
"aembedding", False
)
== False
is not True
and self.model_call_details.get("litellm_params", {}).get(
"aimage_generation", False
)
== False
is not True
and self.model_call_details.get("litellm_params", {}).get(
"atranscription", False
)
== False
is not True
): # custom logger class
if self.stream and complete_streaming_response is None:
callback.log_stream_event(
@ -1190,19 +1190,19 @@ class Logging:
and self.model_call_details.get("litellm_params", {}).get(
"acompletion", False
)
== False
is not True
and self.model_call_details.get("litellm_params", {}).get(
"aembedding", False
)
== False
is not True
and self.model_call_details.get("litellm_params", {}).get(
"aimage_generation", False
)
== False
is not True
and self.model_call_details.get("litellm_params", {}).get(
"atranscription", False
)
== False
is not True
): # custom logger functions
print_verbose(
f"success callbacks: Running Custom Callback Function"
@ -1634,11 +1634,11 @@ class Logging:
and self.model_call_details.get("litellm_params", {}).get(
"acompletion", False
)
== False
is not True
and self.model_call_details.get("litellm_params", {}).get(
"aembedding", False
)
== False
is not True
): # custom logger class
callback.log_failure_event(