uipath-langchain 0.0.101__py3-none-any.whl → 0.0.102__py3-none-any.whl

uipath_langchain/_cli/_runtime/_runtime.py

@@ -80,9 +80,6 @@ class LangGraphRuntime(UiPathBaseRuntime):
 
                 if self.context.job_id and self.context.tracing_enabled:
                     tracer = AsyncUiPathTracer(context=self.context.trace_context)
-                    await tracer.init_trace(
-                        self.context.entrypoint, self.context.job_id
-                    )
                     callbacks = [tracer]
 
                 graph_config: RunnableConfig = {

uipath_langchain/tracers/AsyncUiPathTracer.py

@@ -2,7 +2,6 @@ import asyncio
 import json
 import logging
 import queue
-import re
 import uuid
 import warnings
 from os import environ as env
@@ -43,9 +42,7 @@ class AsyncUiPathTracer(AsyncBaseTracer):
 
         self.context = context or UiPathTraceContext()
 
-        llm_ops_pattern = self._get_base_url() + "{orgId}/llmops_"
-
-        self.url = llm_ops_pattern.format(orgId=self.context.org_id).rstrip("/")
+        self.base_url = self._get_base_url()
 
         auth_token = env.get("UNATTENDED_USER_ACCESS_TOKEN") or env.get(
             "UIPATH_ACCESS_TOKEN"
@@ -103,44 +100,6 @@ class AsyncUiPathTracer(AsyncBaseTracer):
                     )
                     self._send_span(previous_run)
 
-    async def init_trace(self, run_name, trace_id=None) -> None:
-        if self.context.trace_id:
-            # trace id already set no need to do anything
-            return
-
-        # no trace id, start a new trace
-        await self.start_trace(run_name, trace_id)
-
-    async def start_trace(self, run_name, trace_id=None) -> None:
-        self.context.trace_id = str(uuid.uuid4())
-
-        run_name = run_name or f"Job Run: {self.context.trace_id}"
-        trace_data = {
-            "id": self.context.trace_id,
-            "name": re.sub(
-                r"[!@#$<>\.]", "", run_name
-            ),  # if we use these characters the Agents UI throws some error (but llmops backend seems fine)
-            "referenceId": self.context.reference_id,
-            "attributes": "{}",
-            "organizationId": self.context.org_id,
-            "tenantId": self.context.tenant_id,
-        }
-
-        for attempt in range(self.retries):
-            response = await self.client.post(
-                f"{self.url}/api/Agent/trace/", headers=self.headers, json=trace_data
-            )
-
-            if response.is_success:
-                break
-
-            await asyncio.sleep(0.5 * (2**attempt))  # Exponential backoff
-
-        if 400 <= response.status_code < 600:
-            logger.warning(
-                f"Error when sending trace: {response}. Body is: {response.text}"
-            )
-
     async def wait_for_all_tracers(self) -> None:
         """
         Wait for all pending log requests to complete
@@ -159,11 +118,13 @@ class AsyncUiPathTracer(AsyncBaseTracer):
 
                 span_data = self.log_queue.get_nowait()
 
+                url = self._build_url(self.context.trace_id)
+
                 for attempt in range(self.retries):
                     response = await self.client.post(
-                        f"{self.url}/api/Agent/span/",
+                        url,
                         headers=self.headers,
-                        json=span_data,
+                        json=[span_data],  # api expects a list of spans
                         timeout=10,
                     )
 
@@ -189,11 +150,12 @@ class AsyncUiPathTracer(AsyncBaseTracer):
                     break
 
                 span_data = self.log_queue.get_nowait()
+                url = self._build_url(self.context.trace_id)
 
                 response = await self.client.post(
-                    f"{self.url}/api/Agent/span/",
+                    url,
                     headers=self.headers,
-                    json=span_data,
+                    json=[span_data],  # api expects a list of spans
                     timeout=10,
                 )
             except Exception as e:
@@ -239,6 +201,7 @@ class AsyncUiPathTracer(AsyncBaseTracer):
                 "jobKey": self.context.job_id,
                 "folderKey": self.context.folder_key,
                 "processKey": self.context.folder_key,
+                "expiryTimeUtc": None,
             }
 
             self.log_queue.put(span_data)
@@ -293,16 +256,11 @@ class AsyncUiPathTracer(AsyncBaseTracer):
         uipath_url = (
             env.get("UIPATH_URL") or "https://cloud.uipath.com/dummyOrg/dummyTennant/"
         )
-        uipath_url = uipath_url.rstrip("/")
 
-        # split by "//" to get ['', 'https:', 'alpha.uipath.com/ada/byoa']
-        parts = uipath_url.split("//")
-
-        # after splitting by //, the base URL will be at index 1 along with the rest,
-        # hence split it again using "/" to get ['https:', 'alpha.uipath.com', 'ada', 'byoa']
-        base_url_parts = parts[1].split("/")
+        uipath_url = uipath_url.rstrip("/")
 
-        # combine scheme and netloc to get the base URL
-        base_url = parts[0] + "//" + base_url_parts[0] + "/"
+        return uipath_url
 
-        return base_url
+    def _build_url(self, trace_id: Optional[str]) -> str:
+        """Construct the URL for the API request."""
+        return f"{self.base_url}/llmopstenant_/api/Traces/spans?traceId={trace_id}&source=Robots"

{uipath_langchain-0.0.101.dist-info → uipath_langchain-0.0.102.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.101
+Version: 0.0.102
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-langchain-python

{uipath_langchain-0.0.101.dist-info → uipath_langchain-0.0.102.dist-info}/RECORD

@@ -9,7 +9,7 @@ uipath_langchain/_cli/_runtime/_escalation.py,sha256=oA5NvZvCo8ngELFJRyhZNM69DxV
 uipath_langchain/_cli/_runtime/_exception.py,sha256=USKkLYkG-dzjX3fEiMMOHnVUpiXJs_xF0OQXCCOvbYM,546
 uipath_langchain/_cli/_runtime/_input.py,sha256=gKzPaGW-EzgeAskWJjbCWnfZRLu_BM7lCXkq0XkVGLU,5614
 uipath_langchain/_cli/_runtime/_output.py,sha256=WQSrsvGaaclZ6GLWEh6Nk1Mz1iGaIB45PgIX3DS3AN4,16130
-uipath_langchain/_cli/_runtime/_runtime.py,sha256=wj6e6tQdP2GjuSM-6G6Ug-FMqwOHwfUq-DGbd1gMfIg,11825
+uipath_langchain/_cli/_runtime/_runtime.py,sha256=ai3qxNmdPACxowq7WcNBn9m8-KXe4vFwvpBn0lh6kmQ,11689
 uipath_langchain/_cli/_templates/langgraph.json.template,sha256=eeh391Gta_hoRgaNaZ58nW1LNvCVXA7hlAH6l7Veous,107
 uipath_langchain/_cli/_templates/main.py.template,sha256=9JEyPxwc4Ce8Dfd2UAgHgpwkkjuXwWXOQZ-65P53QuM,1195
 uipath_langchain/_cli/_utils/_graph.py,sha256=WLBSJfPc3_C07SqJhePRe17JIc5wcBvEqLviMcNOdTA,6950
@@ -25,7 +25,7 @@ uipath_langchain/embeddings/__init__.py,sha256=QICtYB58ZyqFfDQrEaO8lTEgAU5NuEKlR
 uipath_langchain/embeddings/embeddings.py,sha256=gntzTfwO1pHbgnXiPdfETJaaurvQWqxVUCH75VMah54,4274
 uipath_langchain/retrievers/__init__.py,sha256=rOn7PyyHgZ4pMnXWPkGqmuBmx8eGuo-Oyndo7Wm9IUU,108
 uipath_langchain/retrievers/context_grounding_retriever.py,sha256=CeVSMEz5xTQIladkzDLeQXGC1_ycW72gb0RB41JZeYA,2000
-uipath_langchain/tracers/AsyncUiPathTracer.py,sha256=gheNa679YnM9HrP0VFr-oYfWmpS5-dKOIG8ItzU4AEQ,10477
+uipath_langchain/tracers/AsyncUiPathTracer.py,sha256=VLqoegxa8J3VwMCkGicO626nSzyCQv-Zj14lZrWXA2Y,8900
 uipath_langchain/tracers/UiPathTracer.py,sha256=V5g1nlB0TI1wYvUIkfCEcAdhy4thqeMBmjflWtxc-_M,5601
 uipath_langchain/tracers/__init__.py,sha256=MwrQh6iuPXMS72S5EX0JdCAX0TKe-l7fIPGV3EG0Ssk,256
 uipath_langchain/tracers/_events.py,sha256=CJri76SSdu7rGJIkXurJ2C5sQahfSK4E5UWwWYdEAtE,922
@@ -36,8 +36,8 @@ uipath_langchain/utils/_request_mixin.py,sha256=WFyTDyAthSci1DRwUwS21I3hLntD7HdV
 uipath_langchain/utils/_settings.py,sha256=MhwEVj4gVRSar0RBf2w2hTjO-5Qm-HpCuufqN3gSWjA,3390
 uipath_langchain/utils/_sleep_policy.py,sha256=e9pHdjmcCj4CVoFM1jMyZFelH11YatsgWfpyrfXzKBQ,1251
 uipath_langchain/vectorstores/context_grounding_vectorstore.py,sha256=eTa5sX43-ydB1pj9VNHUPbB-hC36fZK_CGrNe5U2Nrw,9393
-uipath_langchain-0.0.101.dist-info/METADATA,sha256=_DMJPrVOEPDT_0kJm6aLPYMzME2NfuQYx55jAd3qeT8,3937
-uipath_langchain-0.0.101.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath_langchain-0.0.101.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
-uipath_langchain-0.0.101.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
-uipath_langchain-0.0.101.dist-info/RECORD,,
+uipath_langchain-0.0.102.dist-info/METADATA,sha256=1NAjdRLlmcErrOc_DuOUJ8I3sJIeqg2UL5I4-EiRTeQ,3937
+uipath_langchain-0.0.102.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath_langchain-0.0.102.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
+uipath_langchain-0.0.102.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
+uipath_langchain-0.0.102.dist-info/RECORD,,