uipath-langchain 0.0.135__py3-none-any.whl → 0.0.137__py3-none-any.whl

uipath_langchain/_tracing/_oteladapter.py

@@ -1,10 +1,8 @@
 import json
 import logging
-from typing import Any, Dict, List
+from typing import Any, Dict, List, Optional
 
-from opentelemetry.sdk.trace.export import (
-    SpanExportResult,
-)
+from opentelemetry.sdk.trace.export import SpanExportResult
 from uipath.tracing import LlmOpsHttpExporter
 
 logger = logging.getLogger(__name__)
@@ -71,6 +69,11 @@ class LangChainExporter(LlmOpsHttpExporter):
         # Add more mappings as needed
     }
 
+    class Status:
+        SUCCESS = 1
+        ERROR = 2
+        INTERRUPTED = 3
+
     def __init__(self, *args: Any, **kwargs: Any) -> None:
         super().__init__(*args, **kwargs)
 
@@ -161,13 +164,18 @@ class LangChainExporter(LlmOpsHttpExporter):
 
         return result
 
+    def _determine_status(self, error: Optional[str]) -> int:
+        if error:
+            if error and error.startswith("GraphInterrupt("):
+                return self.Status.INTERRUPTED
+            return self.Status.ERROR
+        return self.Status.SUCCESS
+
     def _process_span_attributes(self, span_data: Dict[str, Any]) -> Dict[str, Any]:
         """Extracts, transforms, and maps attributes for a span."""
         if "Attributes" not in span_data:
             return span_data
 
-        logger.debug(f"Processing span: {span_data}")
-
         attributes_val = span_data["Attributes"]
         if isinstance(attributes_val, str):
             try:
@@ -206,7 +214,11 @@ class LangChainExporter(LlmOpsHttpExporter):
 
         span_data["Attributes"] = json.dumps(processed_attributes)
 
-        logger.info(f"Transformed span: {span_data}")
+        # Determine status based on error information
+        error = attributes.get("error") or attributes.get("exception.message")
+        status = self._determine_status(error)
+        span_data["Status"] = status
+
         return span_data
 
     def _send_with_retries(

uipath_langchain/tools/preconfigured.py

@@ -1,6 +1,6 @@
 import json
 import logging
-from typing import Iterable, Optional, Type
+from typing import Any, Iterable, Optional, Type
 
 import httpx
 from jsonschema_pydantic import jsonschema_to_pydantic as create_model  # type: ignore
@@ -14,11 +14,13 @@ from uipath.agent.models.agent import (
     AgentDefinition,
     AgentEscalationChannel,
     AgentEscalationResourceConfig,
+    AgentIntegrationToolParameter,
     AgentIntegrationToolResourceConfig,
     AgentProcessToolResourceConfig,
     AgentResourceConfig,
 )
 from uipath.models import CreateAction, InvokeProcess
+from uipath.models.connections import ConnectionTokenType
 
 logger = logging.getLogger(__name__)
 
@@ -91,6 +93,27 @@ def create_escalation_tool(
         yield create_escalation_tool_from_channel(channel)
 
 
+METHOD_MAP = {"GETBYID": "GET"}
+
+
+def build_query_params(parameters: list[AgentIntegrationToolParameter]):
+    query_params = [
+        x for x in parameters if x.field_location == "query" and x.value is not None
+    ]
+    if query_params:
+        return "?" + "&".join(f"{q.name}={q.value}" for q in query_params)
+    return ""
+
+
+def filter_query_params(
+    kwargs: dict[str, Any], parameters: list[AgentIntegrationToolParameter]
+):
+    query_params = {x.name for x in parameters if x.field_location == "query"}
+    non_query_params = {x.name for x in parameters if x.field_location != "query"}
+    fields_to_ignore = query_params - non_query_params
+    return {k: v for k, v in kwargs.items() if k not in fields_to_ignore}
+
+
 def create_integration_tool(
     resource: AgentIntegrationToolResourceConfig,
 ) -> Iterable[BaseTool]:
@@ -100,15 +123,21 @@ def create_integration_tool(
             resource.properties.connection.id
         )
         token = await uipath.connections.retrieve_token_async(
-            resource.properties.connection.id
+            resource.properties.connection.id, ConnectionTokenType.BEARER
         )
         tool_url = f"{remote_connection.api_base_uri}/v3/element/instances/{remote_connection.element_instance_id}{resource.properties.tool_path}"
+        tool_url = f"{tool_url}{build_query_params(resource.properties.parameters)}"
+        tool_url = tool_url.format(**kwargs)
 
+        authorization = f"{token.token_type} {token.access_token}"
+        method = METHOD_MAP.get(resource.properties.method, resource.properties.method)
         response = await httpx.AsyncClient().request(
-            resource.properties.method,
+            method,
             tool_url,
-            headers={"Authorization": f"Bearer {token.access_token}"},
-            content=json.dumps(kwargs),
+            headers={"Authorization": authorization},
+            content=json.dumps(
+                filter_query_params(kwargs, resource.properties.parameters)
+            ),
         )
         return response.json()
 

{uipath_langchain-0.0.135.dist-info → uipath_langchain-0.0.137.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.135
+Version: 0.0.137
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-langchain-python
@@ -26,7 +26,7 @@ Requires-Dist: openai>=1.65.5
 Requires-Dist: openinference-instrumentation-langchain>=0.1.50
 Requires-Dist: pydantic-settings>=2.6.0
 Requires-Dist: python-dotenv>=1.0.1
-Requires-Dist: uipath<2.2.0,>=2.1.64
+Requires-Dist: uipath<2.2.0,>=2.1.65
 Provides-Extra: langchain
 Description-Content-Type: text/markdown
 

{uipath_langchain-0.0.135.dist-info → uipath_langchain-0.0.137.dist-info}/RECORD

@@ -17,7 +17,7 @@ uipath_langchain/_cli/_templates/main.py.template,sha256=GpSblGH2hwS9ibqQmX2iB2n
 uipath_langchain/_cli/_utils/_graph.py,sha256=nMJWy8FmaD9rqPUY2lHc5uVpUzbXD1RO12uJnhe0kdo,6803
 uipath_langchain/_tracing/__init__.py,sha256=UqrLc_WimpzKY82M0LJsgJ-HFQUQFjOmOlD1XQ8V-R4,181
 uipath_langchain/_tracing/_instrument_traceable.py,sha256=8f9FyAKWE6kH1N8ErbpwqZHAzNjGwbLjQn7jdX5yAgA,4343
-uipath_langchain/_tracing/_oteladapter.py,sha256=6PLyJxLhvtkIlLQoOE2ZFxWys1N3iYZZH9DzpzrwKkk,8434
+uipath_langchain/_tracing/_oteladapter.py,sha256=PD0gsC39ZNvrm0gsfnt1ti6DEy56sBA9sIoxaAbHFFM,8887
 uipath_langchain/_tracing/_utils.py,sha256=r_fiSk3HDDAcePY_UbbEYiSbNqzn5gFeMPYBDvGrFx0,902
 uipath_langchain/_utils/__init__.py,sha256=-w-4TD9ZnJDCpj4VIPXhJciukrmDJJbmnOFnhAkAaEU,81
 uipath_langchain/_utils/_request_mixin.py,sha256=ddKFs_0mjoFCmvPTiOTPJh1IIqYUo5CUka-B7zAZphE,19695
@@ -31,11 +31,11 @@ uipath_langchain/embeddings/embeddings.py,sha256=45gKyb6HVKigwE-0CXeZcAk33c0mtea
 uipath_langchain/retrievers/__init__.py,sha256=rOn7PyyHgZ4pMnXWPkGqmuBmx8eGuo-Oyndo7Wm9IUU,108
 uipath_langchain/retrievers/context_grounding_retriever.py,sha256=YLCIwy89LhLnNqcM0YJ5mZoeNyCs5UiKD3Wly8gnW1E,2239
 uipath_langchain/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-uipath_langchain/tools/preconfigured.py,sha256=Ucl9mta_su66j-5uOC-vZzyc3V2-Em6q9WO1qsmQv48,6291
+uipath_langchain/tools/preconfigured.py,sha256=xCP0hiQuFKIv45PTvMsoWlwsxJDs7goyZujKflYBngY,7476
 uipath_langchain/vectorstores/__init__.py,sha256=w8qs1P548ud1aIcVA_QhBgf_jZDrRMK5Lono78yA8cs,114
 uipath_langchain/vectorstores/context_grounding_vectorstore.py,sha256=TncIXG-YsUlO0R5ZYzWsM-Dj1SVCZbzmo2LraVxXelc,9559
-uipath_langchain-0.0.135.dist-info/METADATA,sha256=CE88C26k5ph4JltUBD19qwcSbySU_4mb-EhSigDnl9E,4275
-uipath_langchain-0.0.135.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath_langchain-0.0.135.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
-uipath_langchain-0.0.135.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
-uipath_langchain-0.0.135.dist-info/RECORD,,
+uipath_langchain-0.0.137.dist-info/METADATA,sha256=tYkBQ1NZGqzxGefhPDnDELJi2McbQnAZHiNdtbNVyxE,4275
+uipath_langchain-0.0.137.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath_langchain-0.0.137.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
+uipath_langchain-0.0.137.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
+uipath_langchain-0.0.137.dist-info/RECORD,,