feat: add input schema to state (#31)

Author: andreitava-uip

pyproject.toml

@@ -6,7 +6,7 @@ readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
   "uipath>=2.2.8",
-  "uipath-langchain>=0.1.6",
+  "uipath-langchain>=0.1.13",
   "langgraph>=1.0.0",
   "azure-monitor-opentelemetry>=1.7.0",
   "opentelemetry-instrumentation-aiohttp-client>=0.49.0",

src/uipath_agents/_services/flags_service.py

@@ -5,8 +5,8 @@
 from pydantic import BaseModel, ConfigDict, Field
 from uipath._config import Config
 from uipath._execution_context import ExecutionContext
-from uipath._services._base_service import BaseService
 from uipath._utils import Endpoint, RequestSpec
+from uipath.platform.common._base_service import BaseService
 
 
 class FeatureFlagsRequest(BaseModel):

src/uipath_agents/agent_graph_builder/graph.py

@@ -6,6 +6,7 @@
 from uipath_langchain.agent.react import (
     AgentGraphConfig,
     create_agent,
+    resolve_input_model,
     resolve_output_model,
 )
 from uipath_langchain.agent.tools import create_tools_from_resources
@@ -42,6 +43,7 @@ async def build_agent_graph(
     agent_messages = build_agent_messages(
         agent_definition.messages, input_data, agent_definition.name or ""
     )
+    input_model = resolve_input_model(agent_definition.input_schema)
     output_model = resolve_output_model(agent_definition.output_schema)
 
     # Create agent config with feature flags
@@ -53,7 +55,7 @@ async def build_agent_graph(
         model=llm,
         tools=tools,
         messages=agent_messages,
-        input_schema=None,
+        input_schema=input_model,
         output_schema=output_model,
         config=agent_config,
     )