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

fix - errant prints on langfuse

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-01 19:07:56 -07:00
parent b3f5ff4d11
commit 4e3ca5a26e
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/integrations/langfuse.py
View file

@ -111,7 +111,7 @@ class LangFuseLogger:
pass pass
# end of processing langfuse ######################## # end of processing langfuse ########################
print(f"response obj type: {type(response_obj)}") print_verbose(f"response obj type: {type(response_obj)}")
if ( if (
level == "ERROR" level == "ERROR"
and status_message is not None and status_message is not None
@ -144,7 +144,7 @@ class LangFuseLogger:
trace_id = None trace_id = None
generation_id = None generation_id = None
if self._is_langfuse_v2(): if self._is_langfuse_v2():
print("INSIDE V2 LANGFUSE") print_verbose("INSIDE V2 LANGFUSE")
trace_id, generation_id = self._log_langfuse_v2( trace_id, generation_id = self._log_langfuse_v2(
user_id, user_id,
metadata, metadata,
@ -373,7 +373,7 @@ class LangFuseLogger:
print_verbose(f"trace_params: {trace_params}") print_verbose(f"trace_params: {trace_params}")
print(f"trace_params: {trace_params}") print_verbose(f"trace_params: {trace_params}")
trace = self.Langfuse.trace(**trace_params) trace = self.Langfuse.trace(**trace_params)
generation_id = None generation_id = None
@ -427,7 +427,7 @@ class LangFuseLogger:
generation_client = trace.generation(**generation_params) generation_client = trace.generation(**generation_params)
print(f"LANGFUSE TRACE ID - {generation_client.trace_id}") print_verbose(f"LANGFUSE TRACE ID - {generation_client.trace_id}")
return generation_client.trace_id, generation_id return generation_client.trace_id, generation_id
except Exception as e: except Exception as e:
verbose_logger.debug(f"Langfuse Layer Error - {traceback.format_exc()}") verbose_logger.debug(f"Langfuse Layer Error - {traceback.format_exc()}")