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

fix(utils.py): log user_id to langfuse

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-23 12:14:09 +05:30
parent 2df5ce4b7c
commit e620d2f219
5 changed files with 71 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/integrations/langfuse.py
View file

@ -39,6 +39,7 @@ class LangFuseLogger:
# Method definition
try:
print(f"result in langfuse logging: {response_obj}")
print_verbose(
f"Langfuse Logging - Enters logging function for model {kwargs}"
)
@ -64,7 +65,7 @@ class LangFuseLogger:
# end of processing langfuse ########################
input = prompt
output = response_obj["choices"][0]["message"].json()
print(f"OUTPUT IN LANGFUSE: {output}; original: {response_obj['choices'][0]['message']}")
self._log_langfuse_v2(
user_id,
metadata,
@ -135,6 +136,7 @@ class LangFuseLogger:
)
)
print(f"LANGFUSE OUTPUT: {output}")
trace.generation(
CreateGeneration(
name=metadata.get("generation_name", "litellm-completion"),
@ -171,6 +173,7 @@ class LangFuseLogger:
user_id=user_id,
)
trace.generation(
name=metadata.get("generation_name", "litellm-completion"),
startTime=start_time,