fix: fix the llm chat completions span (#1191)

Author: AAgnihotry

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath"
-version = "2.5.37"
+version = "2.5.38"
 description = "Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools."
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

src/uipath/platform/chat/_llm_gateway_service.py

@@ -18,6 +18,7 @@
 
 from typing import Any
 
+from opentelemetry import trace
 from pydantic import BaseModel
 
 from ..._utils import Endpoint
@@ -202,7 +203,7 @@ async def embeddings(
 
         return TextEmbedding.model_validate(response.json())
 
-    @traced(name="llm_chat_completions", run_type="uipath")
+    @traced(name="LLM call", run_type="uipath")
     async def chat_completions(
         self,
         messages: list[dict[str, str]],
@@ -288,6 +289,10 @@ class Country(BaseModel):
             When using a Pydantic BaseModel as response_format, it will be automatically
             converted to the appropriate JSON schema format for the LLM Gateway.
         """
+        span = trace.get_current_span()
+        span.set_attribute("model", model)
+        span.set_attribute("uipath.custom_instrumentation", True)
+
         endpoint = EndpointManager.get_passthrough_endpoint().format(
             model=model, api_version=api_version
         )
@@ -347,7 +352,7 @@ def __init__(
     ) -> None:
         super().__init__(config=config, execution_context=execution_context)
 
-    @traced(name="llm_chat_completions", run_type="uipath")
+    @traced(name="LLM call", run_type="uipath")
     async def chat_completions(
         self,
         messages: list[dict[str, str]] | list[tuple[str, str]],
@@ -480,6 +485,10 @@ class Country(BaseModel):
             This service uses UiPath's normalized API format which provides consistent
             behavior across different underlying model providers and enhanced enterprise features.
         """
+        span = trace.get_current_span()
+        span.set_attribute("model", model)
+        span.set_attribute("uipath.custom_instrumentation", True)
+
         converted_messages = []
 
         for message in messages: