scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-13 03:59:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 66

fix(langfuse): fold custom_tool_call into the function_call branch

Browse source 
Keeps the payload build to a single dict literal so the type-discipline
gate's LIT002 total stays at its ceiling.
This commit is contained in:
Muspi Merol 2026-08-19 11:24:38 +08:00
parent 329f61460c
commit 36a226cda9
1 changed files with 15 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
litellm/integrations/langfuse/langfuse_otel.py
View file

@ -196,32 +196,27 @@ class LangfuseOtelLogger(OpenTelemetry):
"content": getattr(getattr(item, "content", [{}])[0], "text", ""),
}
)
elif item_type == "function_call":
arguments_str = getattr(item, "arguments", "{}")
arguments_obj = (
safe_json_loads(arguments_str, default={})
if isinstance(arguments_str, str)
else arguments_str
)
elif item_type in ("function_call", "custom_tool_call"):
# A custom (freeform) tool carries its payload as a raw string under `input`,
# where a function call carries a JSON string under `arguments`.
if item_type == "custom_tool_call":
payload_key, payload_value = "input", getattr(item, "input", "")
else:
arguments_str = getattr(item, "arguments", "{}")
payload_key = "arguments"
payload_value = (
safe_json_loads(arguments_str, default={})
if isinstance(arguments_str, str)
else arguments_str
)
langfuse_tool_call = {
"id": getattr(item, "id", ""),
"name": getattr(item, "name", ""),
"call_id": getattr(item, "call_id", ""),
"type": "function_call",
"arguments": arguments_obj,
"type": item_type,
payload_key: payload_value,
}
output_items_data.append(langfuse_tool_call)
elif item_type == "custom_tool_call":
# Custom (freeform) tools carry a raw string `input` instead of JSON `arguments`
output_items_data.append(
{
"id": getattr(item, "id", ""),
"name": getattr(item, "name", ""),
"call_id": getattr(item, "call_id", ""),
"type": "custom_tool_call",
"input": getattr(item, "input", ""),
}
)
if output_items_data:
safe_set_attribute(
span,