Use _full_model_name in llm trace if available

This commit is contained in:
vipyne
2025-11-26 14:35:14 -06:00
parent ebda94ca98
commit 33eeb8ce44

View File

@@ -483,7 +483,9 @@ def traced_llm(func: Optional[Callable] = None, *, name: Optional[str] = None) -
# Add all available attributes to the span
attribute_kwargs = {
"service_name": service_class_name,
"model": getattr(self, "model_name", "unknown"),
"model": getattr(
self, getattr(self, "_full_model_name", "model_name"), "unknown"
),
"stream": True, # Most LLM services use streaming
"parameters": params,
}