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

refactor: logging class to use continue instead of break for streaming logging

Browse source
This commit is contained in:
alisalim17 2024-05-14 21:21:21 +04:00
parent 68637997c7
commit ed7c9e4e87
1 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
litellm/utils.py
View file

@ -1617,7 +1617,7 @@ class Logging:
# this only logs streaming once, complete_streaming_response exists i.e when stream ends # this only logs streaming once, complete_streaming_response exists i.e when stream ends
if self.stream: if self.stream:
if "complete_streaming_response" not in kwargs: if "complete_streaming_response" not in kwargs:
return continue
else: else:
print_verbose("reaches supabase for streaming logging!") print_verbose("reaches supabase for streaming logging!")
result = kwargs["complete_streaming_response"] result = kwargs["complete_streaming_response"]
@ -1651,7 +1651,7 @@ class Logging:
print_verbose("reaches langsmith for logging!") print_verbose("reaches langsmith for logging!")
if self.stream: if self.stream:
if "complete_streaming_response" not in kwargs: if "complete_streaming_response" not in kwargs:
break continue
else: else:
print_verbose( print_verbose(
"reaches langsmith for streaming logging!" "reaches langsmith for streaming logging!"
@ -1677,7 +1677,7 @@ class Logging:
# this only logs streaming once, complete_streaming_response exists i.e when stream ends # this only logs streaming once, complete_streaming_response exists i.e when stream ends
if self.stream: if self.stream:
if "complete_streaming_response" not in kwargs: if "complete_streaming_response" not in kwargs:
break continue
else: else:
print_verbose("reaches logfire for streaming logging!") print_verbose("reaches logfire for streaming logging!")
result = kwargs["complete_streaming_response"] result = kwargs["complete_streaming_response"]
@ -1707,7 +1707,7 @@ class Logging:
# this only logs streaming once, complete_streaming_response exists i.e when stream ends # this only logs streaming once, complete_streaming_response exists i.e when stream ends
if self.stream: if self.stream:
if "complete_streaming_response" not in kwargs: if "complete_streaming_response" not in kwargs:
break continue
else: else:
result = kwargs["complete_streaming_response"] result = kwargs["complete_streaming_response"]
@ -1852,7 +1852,7 @@ class Logging:
f"is complete_streaming_response in kwargs: {kwargs.get('complete_streaming_response', None)}" f"is complete_streaming_response in kwargs: {kwargs.get('complete_streaming_response', None)}"
) )
if complete_streaming_response is None: if complete_streaming_response is None:
break continue
else: else:
print_verbose("reaches langfuse for streaming logging!") print_verbose("reaches langfuse for streaming logging!")
result = kwargs["complete_streaming_response"] result = kwargs["complete_streaming_response"]
@ -1881,7 +1881,7 @@ class Logging:
f"is complete_streaming_response in kwargs: {kwargs.get('complete_streaming_response', None)}" f"is complete_streaming_response in kwargs: {kwargs.get('complete_streaming_response', None)}"
) )
if complete_streaming_response is None: if complete_streaming_response is None:
break continue
else: else:
print_verbose( print_verbose(
"reaches clickhouse for streaming logging!" "reaches clickhouse for streaming logging!"
@ -1910,7 +1910,7 @@ class Logging:
f"is complete_streaming_response in kwargs: {kwargs.get('complete_streaming_response', None)}" f"is complete_streaming_response in kwargs: {kwargs.get('complete_streaming_response', None)}"
) )
if complete_streaming_response is None: if complete_streaming_response is None:
break continue
else: else:
print_verbose( print_verbose(
"reaches greenscale for streaming logging!" "reaches greenscale for streaming logging!"