feat: bindings and bindings overwrites support (#4)

Author: cotovanu-cristian

src/uipath_lowcode/_cli/cli_run.py

@@ -1,5 +1,6 @@
 import asyncio
 import os
+from pathlib import Path
 from typing import Optional
 
 from openinference.instrumentation.langchain import (
@@ -28,7 +29,10 @@
     _instrument_traceable_attributes,
 )
 
-from ..lowcode_agent_graph_builder import build_lowcode_agent_graph
+from ..agent_graph_builder import build_agent_graph
+from .constants import AGENT_FILENAME
+from .input_loader import load_agent_configuration
+from .runtime import AgentLangGraphRuntime
 
 
 def lowcode_run_middleware(
@@ -51,9 +55,13 @@ def lowcode_run_middleware(
 
         def generate_runtime(ctx: LangGraphRuntimeContext) -> LangGraphRuntime:
             async def graph_builder():
-                return await build_lowcode_agent_graph(input_data=ctx.input)
+                agent_json_path = Path.cwd() / AGENT_FILENAME
+                agent_definition = load_agent_configuration(agent_json_path)
+
+                return await build_agent_graph(agent_definition, input_data=ctx.input)
+
+            runtime = AgentLangGraphRuntime(ctx, graph_builder)
 
-            runtime = LangGraphRuntime(ctx, graph_builder)
             # If not resuming and no job id, delete the previous state file
             if not ctx.resume and ctx.job_id is None:
                 if os.path.exists(runtime.state_file_path):

src/uipath_lowcode/_cli/constants.py

@@ -0,0 +1 @@
+AGENT_FILENAME = "agent.json"

…owcode_agent_graph_builder/exceptions.py src/uipath_lowcode/_cli/exceptions.pysrc/uipath_lowcode/lowcode_agent_graph_builder/exceptions.py renamed to src/uipath_lowcode/_cli/exceptions.py src/uipath_lowcode/lowcode_agent_graph_builder/exceptions.py renamed to src/uipath_lowcode/_cli/exceptions.py

@@ -0,0 +1,35 @@
+from pathlib import Path
+
+from pydantic import ValidationError
+from uipath.agent.models.agent import LowCodeAgentDefinition
+
+from .constants import (
+    AGENT_FILENAME,
+)
+from .exceptions import (
+    ConfigurationError,
+    InputValidationError,
+)
+
+
+def load_agent_configuration(file_path: Path) -> LowCodeAgentDefinition:
+    """Load and validate agent.json configuration.
+
+    Raises:
+        ConfigurationError: If file missing or has invalid structure
+        InputValidationError: If validation against schema fails
+    """
+    if not file_path.exists():
+        raise ConfigurationError(f"{AGENT_FILENAME} not found at {file_path}")
+
+    try:
+        return LowCodeAgentDefinition.model_validate_json(file_path.read_text())
+    except ValidationError as e:
+        raise InputValidationError(
+            f"{AGENT_FILENAME} failed schema validation. Error: {e}",
+            validation_errors=e.errors(),
+        ) from e
+    except (ValueError, TypeError) as e:
+        raise ConfigurationError(
+            f"Invalid {AGENT_FILENAME} structure. Error: {e}"
+        ) from e

src/uipath_lowcode/_cli/input_loader.py

@@ -0,0 +1,7 @@
+from uipath_langchain._cli._runtime._runtime import (
+    LangGraphRuntime,
+)
+
+
+class AgentLangGraphRuntime(LangGraphRuntime):
+    pass

src/uipath_lowcode/_cli/runtime.py

@@ -0,0 +1,7 @@
+"""Agent Graph Builder - Translates agent.json to LangGraph StateGraph."""
+
+from .graph import build_agent_graph
+
+__all__ = ["build_agent_graph"]
+
+__version__ = "0.1.0"

src/uipath_lowcode/agent_graph_builder/__init__.py

@@ -1,9 +1,9 @@
-"""LowCode Agent graph construction - wrapper delegating to uipath_langchain.agent.graph."""
+"""Agent graph construction - wrapper delegating to uipath_langchain.agent.graph."""
 
-from pathlib import Path
 from typing import Any, Optional, Union
 
 from langgraph.graph import StateGraph
+from uipath.agent.models.agent import LowCodeAgentDefinition
 from uipath_langchain.agent.react import (
     AgentGraphConfig,
     AgentGraphState,
@@ -12,25 +12,20 @@
 )
 from uipath_langchain.agent.tools import create_tools_from_resources
 
-from .constants import (
-    AGENT_CONFIG_FILENAME,
-    AGENT_LOOP_RECURSION_LIMIT,
-)
-from .input_loader import load_agent_configuration, load_input_arguments
+from .input_utils import validate_input_data
 from .llm_utils import create_llm
 from .message_utils import build_agent_messages
 
+AGENT_LOOP_RECURSION_LIMIT = 50
+
 
-async def build_lowcode_agent_graph(
+async def build_agent_graph(
+    agent_definition: LowCodeAgentDefinition,
     input_data: Optional[Union[str, dict[str, Any]]] = None,
 ) -> StateGraph[AgentGraphState]:
     """Build LangGraph agent from agent.json configuration and optional input data."""
 
-    agent_json_path = Path.cwd() / AGENT_CONFIG_FILENAME
-    agent_definition = load_agent_configuration(agent_json_path)
-    input_arguments = load_input_arguments(
-        agent_definition.input_schema, input_data=input_data
-    )
+    input_arguments = validate_input_data(agent_definition.input_schema, input_data)
 
     tools = await create_tools_from_resources(agent_definition)
     llm = create_llm(

…ode/lowcode_agent_graph_builder/graph.py …ath_lowcode/agent_graph_builder/graph.pysrc/uipath_lowcode/lowcode_agent_graph_builder/graph.py renamed to src/uipath_lowcode/agent_graph_builder/graph.py src/uipath_lowcode/lowcode_agent_graph_builder/graph.py renamed to src/uipath_lowcode/agent_graph_builder/graph.py

@@ -0,0 +1,33 @@
+from typing import Any, Optional, Union
+
+from jsonschema_pydantic import jsonschema_to_pydantic
+
+
+def validate_input_data(
+    input_schema: dict[str, Any],
+    input_data: Optional[Union[str, dict[str, Any]]] = None,
+) -> dict[str, Any]:
+    """Load and validate input arguments against schema.
+
+    Raises:
+        InputValidationError: If input doesn't match schema
+        ConfigurationError: If input has invalid structure
+    """
+    try:
+        if input_data is None or input_data == "":
+            return {}
+
+        pydantic_input_model = jsonschema_to_pydantic(input_schema)
+
+        if isinstance(input_data, str):
+            parsed_data = pydantic_input_model.model_validate_json(input_data)
+        else:
+            parsed_data = pydantic_input_model.model_validate(input_data)
+
+        return parsed_data.model_dump()
+
+    except Exception as e:
+        raise Exception(
+            "Agent input failed schema validation",
+            validation_errors=e.errors(),
+        ) from e

src/uipath_lowcode/agent_graph_builder/input_utils.py

@@ -1,6 +1,5 @@
 from uipath_langchain.chat.models import UiPathChat
 
-# LLM configuration defaults
 LLM_TIMEOUT_SECONDS = 300
 LLM_MAX_RETRIES = 2