Use _full_model_name in llm trace if available
This commit is contained in:
@@ -483,7 +483,9 @@ def traced_llm(func: Optional[Callable] = None, *, name: Optional[str] = None) -
|
|||||||
# Add all available attributes to the span
|
# Add all available attributes to the span
|
||||||
attribute_kwargs = {
|
attribute_kwargs = {
|
||||||
"service_name": service_class_name,
|
"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
|
"stream": True, # Most LLM services use streaming
|
||||||
"parameters": params,
|
"parameters": params,
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user