scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

Add null check to parent_observation_id assignment

Browse source
This commit is contained in:
Hannes Burrichter 2024-05-21 18:24:18 +02:00
parent b89b3d8c44
commit 82391d270c
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/integrations/langfuse.py
View file

@ -470,7 +470,6 @@ class LangFuseLogger:
generation_params = {
"name": generation_name,
"id": clean_metadata.pop("generation_id", generation_id),
"parent_observation_id": metadata.get("parent_observation_id"),
"start_time": start_time,
"end_time": end_time,
"model": kwargs["model"],
@ -483,6 +482,10 @@ class LangFuseLogger:
"version": clean_metadata.pop("version", None),
}
parent_observation_id = metadata.get("parent_observation_id", None)
if parent_observation_id is not None:
generation_params["parent_observation_id"] = parent_observation_id
if supports_prompt:
user_prompt = clean_metadata.pop("prompt", None)
if user_prompt is None: