uipath-langchain 0.0.123__py3-none-any.whl → 0.0.125__py3-none-any.whl

uipath_langchain/_cli/_runtime/_runtime.py

@@ -18,6 +18,7 @@ from uipath._cli._runtime._contracts import (
 
 from ..._utils import _instrument_traceable_attributes
 from ...tracers import AsyncUiPathTracer
+from .._utils._graph import LangGraphConfig
 from ._context import LangGraphRuntimeContext
 from ._exception import LangGraphRuntimeError
 from ._input import LangGraphInputProcessor
@@ -56,11 +57,6 @@ class LangGraphRuntime(UiPathBaseRuntime):
         tracer = None
 
         try:
-            if self.context.resume is False and self.context.job_id is None:
-                # Delete the previous graph state file at debug time
-                if os.path.exists(self.state_file_path):
-                    os.remove(self.state_file_path)
-
             async with AsyncSqliteSaver.from_conn_string(
                 self.state_file_path
             ) as memory:
@@ -85,9 +81,11 @@ class LangGraphRuntime(UiPathBaseRuntime):
 
                 graph_config: RunnableConfig = {
                     "configurable": {
-                        "thread_id": self.context.job_id
-                        if self.context.job_id
-                        else "default"
+                        "thread_id": (
+                            self.context.execution_id
+                            or self.context.job_id
+                            or "default"
+                        )
                     },
                     "callbacks": callbacks,
                 }
@@ -190,12 +188,14 @@ class LangGraphRuntime(UiPathBaseRuntime):
             ) from e
 
         if self.context.langgraph_config is None:
-            raise LangGraphRuntimeError(
-                "CONFIG_MISSING",
-                "Invalid configuration",
-                "Failed to load configuration",
-                UiPathErrorCategory.DEPLOYMENT,
-            )
+            self.context.langgraph_config = LangGraphConfig()
+            if not self.context.langgraph_config.exists:
+                raise LangGraphRuntimeError(
+                    "CONFIG_MISSING",
+                    "Invalid configuration",
+                    "Failed to load configuration",
+                    UiPathErrorCategory.DEPLOYMENT,
+                )
 
         try:
             self.context.langgraph_config.load_config()

uipath_langchain/_cli/cli_dev.py

@@ -0,0 +1,41 @@
+import asyncio
+from typing import Optional
+
+from openinference.instrumentation.langchain import (
+    LangChainInstrumentor,
+    get_current_span,
+)
+from uipath._cli._dev._terminal import UiPathDevTerminal
+from uipath._cli._runtime._contracts import UiPathRuntimeFactory
+from uipath._cli._utils._console import ConsoleLogger
+from uipath._cli.middlewares import MiddlewareResult
+
+from ._runtime._context import LangGraphRuntimeContext
+from ._runtime._runtime import LangGraphRuntime
+
+console = ConsoleLogger()
+
+
+def langgraph_dev_middleware(interface: Optional[str]) -> MiddlewareResult:
+    """Middleware to launch the developer terminal"""
+
+    try:
+        if interface == "terminal":
+            runtime_factory = UiPathRuntimeFactory(
+                LangGraphRuntime, LangGraphRuntimeContext
+            )
+            runtime_factory.add_instrumentor(LangChainInstrumentor, get_current_span)
+            app = UiPathDevTerminal(runtime_factory)
+            asyncio.run(app.run_async())
+        else:
+            console.error(f"Unknown interface: {interface}")
+    except KeyboardInterrupt:
+        console.info("Debug session interrupted by user")
+    except Exception as e:
+        console.error(f"Error occurred: {e}")
+        return MiddlewareResult(
+            should_continue=False,
+            should_include_stacktrace=True,
+        )
+
+    return MiddlewareResult(should_continue=False)

uipath_langchain/_cli/cli_run.py

@@ -1,4 +1,5 @@
 import asyncio
+import os
 from os import environ as env
 from typing import Optional
 
@@ -40,7 +41,8 @@ def langgraph_run_middleware(
             context.langgraph_config = config
             context.debug = kwargs.get("debug", False)
             context.logs_min_level = env.get("LOG_LEVEL", "INFO")
-            context.job_id = env.get("UIPATH_JOB_KEY")
+            context.job_id = env.get("UIPATH_JOB_KEY", None)
+            context.execution_id = env.get("UIPATH_JOB_KEY", None)
             context.trace_id = env.get("UIPATH_TRACE_ID")
             context.is_eval_run = kwargs.get("is_eval_run", False)
             context.tracing_enabled = tracing
@@ -64,6 +66,10 @@ def langgraph_run_middleware(
             env["UIPATH_REQUESTING_FEATURE"] = "langgraph-agent"
 
             async with LangGraphRuntime.from_context(context) as runtime:
+                if context.resume is False and context.job_id is None:
+                    # Delete the previous graph state file at debug time
+                    if os.path.exists(runtime.state_file_path):
+                        os.remove(runtime.state_file_path)
                 await runtime.execute()
 
         asyncio.run(execute())

uipath_langchain/_utils/_request_mixin.py

@@ -9,7 +9,7 @@ import httpx
 import openai
 from langchain_core.embeddings import Embeddings
 from langchain_core.language_models.chat_models import _cleanup_llm_representation
-from pydantic import BaseModel, Field, SecretStr
+from pydantic import BaseModel, ConfigDict, Field, SecretStr
 from tenacity import (
     AsyncRetrying,
     Retrying,
@@ -37,8 +37,7 @@ def get_from_uipath_url():
 
 
 class UiPathRequestMixin(BaseModel):
-    class Config:
-        arbitrary_types_allowed = True
+    model_config = ConfigDict(arbitrary_types_allowed=True)
 
     default_headers: Optional[Mapping[str, str]] = {
         "X-UiPath-Streaming-Enabled": "false",

uipath_langchain/builder/agent_config.py

@@ -0,0 +1,191 @@
+from enum import Enum
+from typing import Annotated, Any, Dict, List, Literal, Optional, Union
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class AgentMessageRole(str, Enum):
+    """Enum for message roles"""
+
+    SYSTEM = "System"
+    USER = "User"
+
+
+class AgentMessage(BaseModel):
+    """Message model for agent conversations"""
+
+    role: AgentMessageRole
+    content: str
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentSettings(BaseModel):
+    """Settings for agent configuration"""
+
+    engine: str = Field(..., description="Engine type, e.g., 'basic-v1'")
+    model: str = Field(..., description="LLM model identifier")
+    max_tokens: int = Field(
+        ..., alias="maxTokens", description="Maximum number of tokens"
+    )
+    temperature: float = Field(..., description="Temperature for response generation")
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentResourceType(str, Enum):
+    """Enum for resource types"""
+
+    TOOL = "tool"
+    CONTEXT = "context"
+    ESCALATION = "escalation"
+
+
+class AgentBaseResourceConfig(BaseModel):
+    """Base resource model with common properties"""
+
+    name: str
+    description: str
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentUnknownResourceConfig(AgentBaseResourceConfig):
+    """Fallback for unknown or future resource types"""
+
+    resource_type: str = Field(alias="$resourceType")
+
+    model_config = ConfigDict(extra="allow")
+
+
+class AgentToolSettings(BaseModel):
+    """Settings for tool configuration"""
+
+    max_attempts: int = Field(0, alias="maxAttempts")
+    retry_delay: int = Field(0, alias="retryDelay")
+    timeout: int = Field(0)
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentToolProperties(BaseModel):
+    """Properties specific to tool configuration"""
+
+    folder_path: Optional[str] = Field(None, alias="folderPath")
+    process_name: Optional[str] = Field(None, alias="processName")
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentToolResourceConfig(AgentBaseResourceConfig):
+    """Tool resource with tool-specific properties"""
+
+    resource_type: Literal[AgentResourceType.TOOL] = Field(alias="$resourceType")
+    type: str = Field(..., description="Tool type")
+    arguments: Dict[str, Any] = Field(
+        default_factory=dict, description="Tool arguments"
+    )
+    input_schema: Dict[str, Any] = Field(
+        ..., alias="inputSchema", description="Input schema for the tool"
+    )
+    output_schema: Dict[str, Any] = Field(
+        ..., alias="outputSchema", description="Output schema for the tool"
+    )
+    properties: AgentToolProperties = Field(..., description="Tool-specific properties")
+    settings: AgentToolSettings = Field(
+        default_factory=AgentToolSettings, description="Tool settings"
+    )
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentContextSettings(BaseModel):
+    """Settings for context configuration"""
+
+    result_count: int = Field(alias="resultCount")
+    retrieval_mode: Literal["Semantic", "Structured"] = Field(alias="retrievalMode")
+    threshold: float = Field(default=0)
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentContextResourceConfig(AgentBaseResourceConfig):
+    """Context resource with context-specific properties"""
+
+    resource_type: Literal[AgentResourceType.CONTEXT] = Field(alias="$resourceType")
+    folder_path: str = Field(alias="folderPath")
+    index_name: str = Field(alias="indexName")
+    settings: AgentContextSettings = Field(..., description="Context settings")
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentEscalationResourceConfig(AgentBaseResourceConfig):
+    """Escalation resource with escalation-specific properties"""
+
+    resource_type: Literal[AgentResourceType.ESCALATION] = Field(alias="$resourceType")
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+# Discriminated union for known types
+KnownAgentResourceConfig = Annotated[
+    Union[
+        AgentToolResourceConfig,
+        AgentContextResourceConfig,
+        AgentEscalationResourceConfig,
+    ],
+    Field(discriminator="resource_type"),
+]
+
+# Final union includes unknowns as a catch-all
+AgentResourceConfig = Union[
+    KnownAgentResourceConfig,
+    AgentUnknownResourceConfig,
+]
+
+
+class AgentConfig(BaseModel):
+    """Main agent model"""
+
+    id: str = Field(..., description="Agent id or project name")
+    name: str = Field(..., description="Agent name or project name")
+    input_schema: Dict[str, Any] = Field(
+        ..., alias="inputSchema", description="JSON schema for input arguments"
+    )
+    output_schema: Dict[str, Any] = Field(
+        ..., alias="outputSchema", description="JSON schema for output arguments"
+    )
+    messages: List[AgentMessage] = Field(
+        ..., description="List of system and user messages"
+    )
+    features: List[Any] = Field(
+        default_factory=list, description="Currently empty feature list"
+    )
+    version: str = Field("1.0.0", description="Agent version")
+    settings: AgentSettings = Field(..., description="Agent settings configuration")
+    resources: List[AgentResourceConfig] = Field(
+        ..., description="List of tools, context, and escalation resources"
+    )
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )

uipath_langchain/middlewares.py

@@ -1,5 +1,6 @@
 from uipath._cli.middlewares import Middlewares
 
+from ._cli.cli_dev import langgraph_dev_middleware
 from ._cli.cli_init import langgraph_init_middleware
 from ._cli.cli_new import langgraph_new_middleware
 from ._cli.cli_run import langgraph_run_middleware
@@ -10,3 +11,4 @@ def register_middleware():
     Middlewares.register("init", langgraph_init_middleware)
     Middlewares.register("run", langgraph_run_middleware)
     Middlewares.register("new", langgraph_new_middleware)
+    Middlewares.register("dev", langgraph_dev_middleware)

{uipath_langchain-0.0.123.dist-info → uipath_langchain-0.0.125.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.123
+Version: 0.0.125
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-langchain-python
@@ -22,9 +22,10 @@ Requires-Dist: langchain>=0.3.4
 Requires-Dist: langgraph-checkpoint-sqlite>=2.0.3
 Requires-Dist: langgraph<0.7.0,>=0.5.0
 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.10
+Requires-Dist: uipath<2.2.0,>=2.1.33
 Provides-Extra: langchain
 Description-Content-Type: text/markdown
 

{uipath_langchain-0.0.123.dist-info → uipath_langchain-0.0.125.dist-info}/RECORD

@@ -1,21 +1,23 @@
 uipath_langchain/__init__.py,sha256=VBrvQn7d3nuOdN7zEnV2_S-uhmkjgEIlXiFVeZxZakQ,80
-uipath_langchain/middlewares.py,sha256=tre7o9DFMgWk1DJiEEUmT6_wiP-PPkWtKmG0iOyvr9c,509
+uipath_langchain/middlewares.py,sha256=EOWVTr52SFLEf5lzGAvcDN2qbtPsy4_EsFoXlV-cqdY,618
 uipath_langchain/_cli/__init__.py,sha256=juqd9PbXs4yg45zMJ7BHAOPQjb7sgEbWE9InBtGZhfo,24
+uipath_langchain/_cli/cli_dev.py,sha256=VlI8qgCw-63I97tp_9lbXs-CVcNSjpd2sC13YNZAIuU,1401
 uipath_langchain/_cli/cli_init.py,sha256=xhxJ8tuMSrVUNHvltgyPpOrvgMA-wq9shHeYYwvHILs,8199
 uipath_langchain/_cli/cli_new.py,sha256=dL8-Rri6u67ZZdbb4nT38A5xD_Q3fVnG0UK9VSeKaqg,2563
-uipath_langchain/_cli/cli_run.py,sha256=Jr-nxzBaSPQoVrdV1nQBSDVFElFV3N0a0uNDDrdSc8I,3375
+uipath_langchain/_cli/cli_run.py,sha256=R-cUi3lO3Qd4ysTXD7PW4sa1RsB427v_Y6xUQxWijfQ,3725
 uipath_langchain/_cli/_runtime/_context.py,sha256=yyzYJDmk2fkH4T5gm4cLGRyXtjLESrpzHBT9euqluTA,817
 uipath_langchain/_cli/_runtime/_exception.py,sha256=USKkLYkG-dzjX3fEiMMOHnVUpiXJs_xF0OQXCCOvbYM,546
 uipath_langchain/_cli/_runtime/_input.py,sha256=vZ8vfVxvPSaPWmIPghvNx1VRKzbalHsKUMBPiKDvJWM,5492
 uipath_langchain/_cli/_runtime/_output.py,sha256=yJOZPWv2FRUJWv1NRs9JmpB4QMTDXu8jrxoaKrfJvzw,9078
-uipath_langchain/_cli/_runtime/_runtime.py,sha256=XTlPaMuHyv-4DjQMoEh76D2Z18M3S5wm4tDP-UAiE5Y,14505
+uipath_langchain/_cli/_runtime/_runtime.py,sha256=dRM28l1k3qFf8NLWytRKlOcPwZJ__qUjqmV04wN9JRY,14504
 uipath_langchain/_cli/_templates/langgraph.json.template,sha256=eeh391Gta_hoRgaNaZ58nW1LNvCVXA7hlAH6l7Veous,107
 uipath_langchain/_cli/_templates/main.py.template,sha256=nMJQIYPlRk90iANfNVpkJ2EQX20Dxsyq92-BucEz_UM,1189
 uipath_langchain/_cli/_utils/_graph.py,sha256=JPShHNl0UQvl4AdjLIqLpNt_JAjpWH9WWF22Gs47Xew,7445
 uipath_langchain/_utils/__init__.py,sha256=WoY66enCygRXTh6v5B1UrRcFCnQYuPJ8oqDkwomXzLc,194
-uipath_langchain/_utils/_request_mixin.py,sha256=Tr57358_dwb7SQ1OU75XjdFVWEIr05-QJifeJfHHZgc,19680
+uipath_langchain/_utils/_request_mixin.py,sha256=ddKFs_0mjoFCmvPTiOTPJh1IIqYUo5CUka-B7zAZphE,19695
 uipath_langchain/_utils/_settings.py,sha256=2fExMQJ88YptfldmzMfZIpsx-m1gfMkeYGf5t6KIe0A,3084
 uipath_langchain/_utils/_sleep_policy.py,sha256=e9pHdjmcCj4CVoFM1jMyZFelH11YatsgWfpyrfXzKBQ,1251
+uipath_langchain/builder/agent_config.py,sha256=b9WODKPjvufj41Ow_dQn5CnaTAjAZyQoNhuAl8vfiso,5809
 uipath_langchain/chat/__init__.py,sha256=WDcvy91ixvZ3Mq7Ae94g5CjyQwXovDBnEv1NlD5SXBE,116
 uipath_langchain/chat/models.py,sha256=m5PRAFXzUamt6-1K9uSlWUvZg_NfVyYHkgoQDJ-1rGs,10527
 uipath_langchain/embeddings/__init__.py,sha256=QICtYB58ZyqFfDQrEaO8lTEgAU5NuEKlR7iIrS0OBtc,156
@@ -29,8 +31,8 @@ uipath_langchain/tracers/_instrument_traceable.py,sha256=0e841zVzcPWjOGtmBx0GeHb
 uipath_langchain/tracers/_utils.py,sha256=JOT1tKMdvqjMDtj2WbmbOWMeMlTXBWavxWpogX7KlRA,1543
 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.123.dist-info/METADATA,sha256=YfxavLE3bBdZlDjaKgBnX410AeHdIg6tEFZpQInaBjA,4172
-uipath_langchain-0.0.123.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath_langchain-0.0.123.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
-uipath_langchain-0.0.123.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
-uipath_langchain-0.0.123.dist-info/RECORD,,
+uipath_langchain-0.0.125.dist-info/METADATA,sha256=-qGeRLnvPNVGjyNyQRn5SQ0bgd8Kn07sgfgmVCYfihg,4235
+uipath_langchain-0.0.125.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath_langchain-0.0.125.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
+uipath_langchain-0.0.125.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
+uipath_langchain-0.0.125.dist-info/RECORD,,