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

(feat) langfuse send metadata as tags

Browse source
This commit is contained in:
ishaan-jaff 2024-01-17 17:29:46 -08:00
parent 46f84bec69
commit 5c1ae3d412
1 changed files with 18 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

25
litellm/integrations/langfuse.py
View file

@ -166,13 +166,24 @@ class LangFuseLogger:
input,
response_obj,
):
trace = self.Langfuse.trace(
name=metadata.get("generation_name", "litellm-completion"),
input=input,
output=output,
user_id=metadata.get("trace_user_id", user_id),
id=metadata.get("trace_id", None),
)
import langfuse
tags = []
supports_tags = Version(langfuse.version.__version__) >= Version("2.6.3")
trace_params = {
"name": metadata.get("generation_name", "litellm-completion"),
"input": input,
"output": output,
"user_id": metadata.get("trace_user_id", user_id),
"id": metadata.get("trace_id", None),
}
if supports_tags:
for key, value in metadata.items():
tags.append(f"{key}:{value}")
trace_params.update({"tags": tags})
trace = self.Langfuse.trace(**trace_params)
trace.generation(
name=metadata.get("generation_name", "litellm-completion"),