phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) langfuse - use exact same response_id as s3 logging

Browse source
This commit is contained in:
ishaan-jaff 2024-02-02 14:29:23 -08:00
parent 553349cb18
commit 9947b0db96
1 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/integrations/langfuse.py
View file

@ -207,7 +207,15 @@ class LangFuseLogger:
if generation_name is None: if generation_name is None:
# just log `litellm-{call_type}` as the generation name # just log `litellm-{call_type}` as the generation name
generation_name = f"litellm-{kwargs.get('call_type', 'completion')}" generation_name = f"litellm-{kwargs.get('call_type', 'completion')}"
response_id = response_obj.get("id", None) + "-time-" + str(end_time) response_id = None
if response_obj.get("id", None) is not None:
response_id = (
"time-"
+ start_time.strftime("%H-%M-%S-%f")
+ "_"
+ response_obj.get("id")
)
trace_params = { trace_params = {
"name": generation_name, "name": generation_name,
"input": input, "input": input,