uipath-langchain 0.0.132__py3-none-any.whl → 0.0.134__py3-none-any.whl

uipath_langchain/_cli/cli_eval.py

@@ -1,16 +1,19 @@
 import asyncio
+import uuid
 from typing import List, Optional
 
 from openinference.instrumentation.langchain import (
     LangChainInstrumentor,
     get_current_span,
 )
+from uipath._cli._evals._progress_reporter import StudioWebProgressReporter
 from uipath._cli._evals._runtime import UiPathEvalContext, UiPathEvalRuntime
 from uipath._cli._runtime._contracts import (
     UiPathRuntimeFactory,
 )
 from uipath._cli._utils._eval_set import EvalHelpers
 from uipath._cli.middlewares import MiddlewareResult
+from uipath._events._event_bus import EventBus
 from uipath.eval._helpers import auto_discover_entrypoint
 
 from uipath_langchain._cli._runtime._context import LangGraphRuntimeContext
@@ -31,14 +34,16 @@ def langgraph_eval_middleware(
             should_continue=True
         )  # Continue with normal flow if no langgraph.json
 
-    eval_context = UiPathEvalContext.with_defaults(**kwargs)
-    eval_context.eval_set = eval_set or EvalHelpers.auto_discover_eval_set()
-    eval_context.eval_ids = eval_ids
-
     try:
         _instrument_traceable_attributes()
 
-        runtime_entrypoint = entrypoint or auto_discover_entrypoint()
+        event_bus = EventBus()
+
+        if not kwargs.get("no_report", False):
+            progress_reporter = StudioWebProgressReporter(
+                spans_exporter=LangChainExporter()
+            )
+            asyncio.run(progress_reporter.subscribe_to_eval_runtime_events(event_bus))
 
         def generate_runtime_context(
             context_entrypoint: str, langgraph_config: LangGraphConfig, **context_kwargs
@@ -48,6 +53,14 @@ def langgraph_eval_middleware(
             context.entrypoint = context_entrypoint
             return context
 
+        runtime_entrypoint = entrypoint or auto_discover_entrypoint()
+
+        eval_context = UiPathEvalContext.with_defaults(
+            entrypoint=runtime_entrypoint, execution_id=str(uuid.uuid4()), **kwargs
+        )
+        eval_context.eval_set = eval_set or EvalHelpers.auto_discover_eval_set()
+        eval_context.eval_ids = eval_ids
+
         runtime_factory = UiPathRuntimeFactory(
             LangGraphRuntime,
             LangGraphRuntimeContext,
@@ -65,9 +78,10 @@ def langgraph_eval_middleware(
 
         async def execute():
             async with UiPathEvalRuntime.from_eval_context(
-                factory=runtime_factory, context=eval_context
+                factory=runtime_factory, context=eval_context, event_bus=event_bus
             ) as eval_runtime:
                 await eval_runtime.execute()
+                await event_bus.wait_for_all()
 
         asyncio.run(execute())
         return MiddlewareResult(should_continue=False)

uipath_langchain/tools/preconfigured.py

@@ -0,0 +1,191 @@
+import json
+import logging
+from typing import Iterable, Optional, Type
+
+import httpx
+from jsonschema_pydantic import jsonschema_to_pydantic as create_model  # type: ignore
+from langchain_core.caches import BaseCache
+from langchain_core.runnables.utils import Output
+from langchain_core.tools import BaseTool, StructuredTool
+from langgraph.types import interrupt
+from pydantic import BaseModel
+from uipath import UiPath
+from uipath.agent.models.agent import (
+    AgentDefinition,
+    AgentEscalationChannel,
+    AgentEscalationResourceConfig,
+    AgentIntegrationToolResourceConfig,
+    AgentProcessToolResourceConfig,
+    AgentResourceConfig,
+)
+from uipath.models import CreateAction, InvokeProcess
+
+logger = logging.getLogger(__name__)
+
+
+def create_process_tool(resource: AgentProcessToolResourceConfig) -> Iterable[BaseTool]:
+    async def process(**kwargs) -> BaseModel:
+        return interrupt(
+            InvokeProcess(
+                name=resource.name,
+                input_arguments=kwargs,
+                process_folder_path=resource.properties.folder_path,
+            )
+        )
+
+    input_schema = create_model(resource.input_schema)
+
+    class ProcessTool(StructuredTool):
+        @property
+        def OutputType(self) -> type[Output]:
+            return create_model(resource.output_schema)
+
+    yield ProcessTool(
+        name=resource.name,
+        args_schema=input_schema,
+        description=resource.description,
+        coroutine=process,
+    )
+
+
+def create_escalation_tool_from_channel(channel: AgentEscalationChannel) -> BaseTool:
+    async def escalate(**kwargs) -> BaseModel:
+        recipients = channel.recipients
+        if len(recipients) > 1:
+            logger.warning(
+                "Received more than one recipient. Defaulting to first recipient."
+            )
+        assignee = recipients[0].value if recipients else None
+        return interrupt(
+            CreateAction(
+                title=channel.description,
+                data=kwargs,
+                assignee=assignee,
+                app_name=channel.properties.app_name,
+                app_folder_path=None,  # Channels specify folder name but not folder path.
+                app_folder_key=channel.properties.resource_key,
+                app_key=channel.properties.resource_key,
+                app_version=channel.properties.app_version,
+            )
+        )
+
+    input_schema = create_model(channel.input_schema)
+
+    class EscalationTool(StructuredTool):
+        @property
+        def OutputType(self) -> type[Output]:
+            return create_model(channel.output_schema)
+
+    return EscalationTool(
+        name=channel.name,
+        args_schema=input_schema,
+        description=channel.description,
+        coroutine=escalate,
+    )
+
+
+def create_escalation_tool(
+    resource: AgentEscalationResourceConfig,
+) -> Iterable[BaseTool]:
+    for channel in resource.channels:
+        yield create_escalation_tool_from_channel(channel)
+
+
+def create_integration_tool(
+    resource: AgentIntegrationToolResourceConfig,
+) -> Iterable[BaseTool]:
+    async def integration(**kwargs) -> BaseModel:
+        uipath = UiPath()
+        remote_connection = await uipath.connections.retrieve_async(
+            resource.properties.connection.id
+        )
+        token = await uipath.connections.retrieve_token_async(
+            resource.properties.connection.id
+        )
+        tool_url = f"{remote_connection.api_base_uri}/v3/element/instances/{remote_connection.element_instance_id}{resource.properties.tool_path}"
+
+        response = await httpx.AsyncClient().request(
+            resource.properties.method,
+            tool_url,
+            headers={"Authorization": f"Bearer {token.access_token}"},
+            content=json.dumps(kwargs),
+        )
+        return response.json()
+
+    input_schema = create_model(resource.input_schema)
+
+    class IntegrationTool(StructuredTool):
+        @property
+        def OutputType(self) -> type[Output]:
+            return create_model({})
+
+    yield IntegrationTool(
+        name=resource.name,
+        args_schema=input_schema,
+        description=resource.description,
+        coroutine=integration,
+    )
+
+
+def create_cached_wrapper_from_tool(
+    wrapped: BaseTool, cache: Optional[BaseCache]
+) -> BaseTool:
+    if cache is None:
+        return wrapped
+    else:
+
+        async def cached_invocation(**kwargs) -> BaseModel:
+            namespace = f"{wrapped.name}.tool_invoke"
+            key = str(kwargs)
+            cached = cache.lookup(namespace, key)
+            if cached:
+                return cached[0]
+            response = await wrapped.ainvoke(input=kwargs)
+            cache.update(namespace, key, [response])
+            return response
+
+        input_schema = wrapped.args_schema
+
+        class CachedTool(StructuredTool):
+            OutputType: Type[BaseModel] = wrapped.OutputType
+
+        return CachedTool(
+            name=wrapped.name,
+            args_schema=input_schema,
+            description=wrapped.description,
+            coroutine=cached_invocation,
+        )
+
+
+def create_cached_wrapper(
+    tools: Iterable[BaseTool], cache: Optional[BaseCache]
+) -> Iterable[BaseTool]:
+    for wrapped in tools:
+        yield create_cached_wrapper_from_tool(wrapped, cache)
+
+
+def create_resource_tool(
+    resource: AgentResourceConfig, cache: Optional[BaseCache] = None
+) -> Iterable[BaseTool]:
+    match resource:
+        case AgentProcessToolResourceConfig():
+            return create_cached_wrapper(create_process_tool(resource), cache)
+        case AgentIntegrationToolResourceConfig():
+            return create_cached_wrapper(create_integration_tool(resource), cache)
+        case AgentEscalationResourceConfig():
+            return create_cached_wrapper(create_escalation_tool(resource), cache)
+        case _:
+            raise NotImplementedError()
+
+
+def safe_extract_tools(
+    agent_definition: AgentDefinition, cache: Optional[BaseCache] = None
+) -> list[BaseTool]:
+    tools = []
+    for resource in agent_definition.resources:
+        try:
+            for structured_tool in create_resource_tool(resource, cache):
+                tools.append(structured_tool)
+        except NotImplementedError:
+            logger.warning(f"Unable to convert {resource.name} into a tool.")
+    return tools

{uipath_langchain-0.0.132.dist-info → uipath_langchain-0.0.134.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.132
+Version: 0.0.134
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-langchain-python
@@ -15,6 +15,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Software Development :: Build Tools
 Requires-Python: >=3.10
 Requires-Dist: httpx>=0.27.0
+Requires-Dist: jsonschema-pydantic>=0.6
 Requires-Dist: langchain-community>=0.3.21
 Requires-Dist: langchain-core>=0.3.34
 Requires-Dist: langchain-openai>=0.3.3
@@ -25,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.56
+Requires-Dist: uipath<2.2.0,>=2.1.64
 Provides-Extra: langchain
 Description-Content-Type: text/markdown
 

{uipath_langchain-0.0.132.dist-info → uipath_langchain-0.0.134.dist-info}/RECORD

@@ -2,7 +2,7 @@ uipath_langchain/__init__.py,sha256=VBrvQn7d3nuOdN7zEnV2_S-uhmkjgEIlXiFVeZxZakQ,
 uipath_langchain/middlewares.py,sha256=6ljfbtWekrYc5G9KWDLSaViJ1DVIaNM-4qeB1BfHywE,731
 uipath_langchain/_cli/__init__.py,sha256=juqd9PbXs4yg45zMJ7BHAOPQjb7sgEbWE9InBtGZhfo,24
 uipath_langchain/_cli/cli_dev.py,sha256=3e9RldNGirIk9184NdLK6kDuGeeqZjekTxbSZRtXjBE,1505
-uipath_langchain/_cli/cli_eval.py,sha256=mvaKRF-pvxdNa2mk_ude9SYSd_kmMBo-hChlXokXOcw,2795
+uipath_langchain/_cli/cli_eval.py,sha256=OBU8ADax8eF6KKOC8vxqdDR8YbuXiIe_r632p7SzveA,3386
 uipath_langchain/_cli/cli_init.py,sha256=xhxJ8tuMSrVUNHvltgyPpOrvgMA-wq9shHeYYwvHILs,8199
 uipath_langchain/_cli/cli_new.py,sha256=KKLxCzz7cDQ__rRr_a496IHWlSQXhmrBNgmKHnXAnTY,2336
 uipath_langchain/_cli/cli_run.py,sha256=hRcoXJgOIFceCswzTfZKyzqVee3j-oSh-13EfuQmmE8,2614
@@ -30,10 +30,12 @@ uipath_langchain/embeddings/__init__.py,sha256=QICtYB58ZyqFfDQrEaO8lTEgAU5NuEKlR
 uipath_langchain/embeddings/embeddings.py,sha256=45gKyb6HVKigwE-0CXeZcAk33c0mteaEdPGa8hviqcw,4339
 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/vectorstores/__init__.py,sha256=w8qs1P548ud1aIcVA_QhBgf_jZDrRMK5Lono78yA8cs,114
 uipath_langchain/vectorstores/context_grounding_vectorstore.py,sha256=TncIXG-YsUlO0R5ZYzWsM-Dj1SVCZbzmo2LraVxXelc,9559
-uipath_langchain-0.0.132.dist-info/METADATA,sha256=ta5yg3ODDXTdagrZVgeevO7wzwjwDeIIR3dAC_wmctM,4235
-uipath_langchain-0.0.132.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath_langchain-0.0.132.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
-uipath_langchain-0.0.132.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
-uipath_langchain-0.0.132.dist-info/RECORD,,
+uipath_langchain-0.0.134.dist-info/METADATA,sha256=jqUMjAKUcTSiZQNCXpW2OF5jhoSzvkziuuG2nE7Vtu4,4275
+uipath_langchain-0.0.134.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath_langchain-0.0.134.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
+uipath_langchain-0.0.134.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
+uipath_langchain-0.0.134.dist-info/RECORD,,