feat: Addition of DF tool

src/uipath_langchain/agent/react/__init__.py

@@ -1,7 +1,7 @@
 """UiPath ReAct Agent implementation"""
 
 from .agent import create_agent
-from .types import AgentGraphConfig, AgentGraphNode, AgentGraphState
+from .types import AgentGraphConfig, AgentGraphNode, AgentGraphState, AgentResources
 from .utils import resolve_input_model, resolve_output_model
 
 __all__ = [
@@ -11,4 +11,5 @@
     "AgentGraphNode",
     "AgentGraphState",
     "AgentGraphConfig",
+    "AgentResources",
 ]

src/uipath_langchain/agent/react/agent.py

@@ -36,6 +36,7 @@
     AgentGraphConfig,
     AgentGraphNode,
     AgentGraphState,
+    AgentResources,
 )
 from .utils import create_state_with_input
 
@@ -53,6 +54,7 @@ def create_agent(
     output_schema: Type[OutputT] | None = None,
     config: AgentGraphConfig | None = None,
     guardrails: Sequence[tuple[BaseGuardrail, GuardrailAction]] | None = None,
+    resources: AgentResources | None = None,
 ) -> StateGraph[AgentGraphState, None, InputT, OutputT]:
     """Build agent graph with INIT -> AGENT (subgraph) <-> TOOLS loop, terminated by control flow tools.
 
@@ -74,8 +76,12 @@ def create_agent(
     )
     llm_tools: list[BaseTool] = [*agent_tools, *flow_control_tools]
 
-    init_node = create_init_node(messages, input_schema, config.is_conversational)
-
+    init_node = create_init_node(
+        messages,
+        input_schema,
+        config.is_conversational,
+        resources_for_init=resources,
+    )
     tool_nodes = create_tool_node(
         agent_tools, handle_tool_errors=config.is_conversational
     )

src/uipath_langchain/agent/react/init_node.py

@@ -1,5 +1,6 @@
 """State initialization node for the ReAct Agent graph."""
 
+import logging
 from typing import Any, Callable, Sequence
 
 from langchain_core.messages import HumanMessage, SystemMessage
@@ -10,20 +11,51 @@
     get_job_attachments,
     parse_attachments_from_conversation_messages,
 )
+from .types import AgentResources
+
+logger = logging.getLogger(__name__)
 
 
 def create_init_node(
     messages: Sequence[SystemMessage | HumanMessage]
-    | Callable[[Any], Sequence[SystemMessage | HumanMessage]],
+    | Callable[..., Sequence[SystemMessage | HumanMessage]],
     input_schema: type[BaseModel] | None,
     is_conversational: bool = False,
+    resources_for_init: AgentResources | None = None,
 ):
-    def graph_state_init(state: Any) -> Any:
+    async def graph_state_init(state: Any) -> Any:
+        # --- Data Fabric schema fetch (INIT-time) ---
+        schema_context: str | None = None
+        if resources_for_init:
+            from uipath_langchain.agent.tools.datafabric_tool import (
+                fetch_entity_schemas,
+                format_schemas_for_context,
+                get_datafabric_entity_identifiers_from_resources,
+            )
+
+            entity_identifiers = get_datafabric_entity_identifiers_from_resources(
+                resources_for_init
+            )
+            if entity_identifiers:
+                logger.info(
+                    "Fetching Data Fabric schemas for %d identifier(s)",
+                    len(entity_identifiers),
+                )
+                entities = await fetch_entity_schemas(entity_identifiers)
+                schema_context = format_schemas_for_context(entities)
+
+        # --- Resolve messages ---
         resolved_messages: Sequence[SystemMessage | HumanMessage] | Overwrite
         if callable(messages):
-            resolved_messages = list(messages(state))
+            if schema_context:
+                resolved_messages = list(
+                    messages(state, additional_context=schema_context)
+                )
+            else:
+                resolved_messages = list(messages(state))
         else:
             resolved_messages = list(messages)
+
         if is_conversational:
             # For conversational agents we need to reorder the messages so that the system message is first, followed by
             # the initial user message. When resuming the conversation, the state will have the entire message history,

src/uipath_langchain/agent/react/types.py

@@ -1,9 +1,10 @@
 from enum import StrEnum
-from typing import Annotated, Any, Hashable, Literal, Optional
+from typing import Annotated, Any, Hashable, Literal, Optional, Sequence
 
 from langchain_core.messages import AnyMessage
 from langgraph.graph.message import add_messages
 from pydantic import BaseModel, Field
+from uipath.agent.models.agent import BaseAgentResourceConfig
 from uipath.agent.react import END_EXECUTION_TOOL, RAISE_ERROR_TOOL
 from uipath.platform.attachments import Attachment
 
@@ -14,6 +15,8 @@
 
 FLOW_CONTROL_TOOLS = [END_EXECUTION_TOOL.name, RAISE_ERROR_TOOL.name]
 
+AgentResources = Sequence[BaseAgentResourceConfig]
+
 
 class InnerAgentGraphState(BaseModel):
     job_attachments: Annotated[dict[str, Attachment], merge_dicts] = {}

src/uipath_langchain/agent/tools/__init__.py

@@ -1,6 +1,13 @@
 """Tool creation and management for LowCode agents."""
 
 from .context_tool import create_context_tool
+from .datafabric_tool import (
+    create_datafabric_tools,
+    fetch_entity_schemas,
+    format_schemas_for_context,
+    get_datafabric_contexts,
+    get_datafabric_entity_identifiers_from_resources,
+)
 from .escalation_tool import create_escalation_tool
 from .extraction_tool import create_ixp_extraction_tool
 from .integration_tool import create_integration_tool
@@ -16,12 +23,17 @@
     "create_tools_from_resources",
     "create_tool_node",
     "create_context_tool",
+    "create_datafabric_tools",
     "open_mcp_tools",
     "create_process_tool",
     "create_integration_tool",
     "create_escalation_tool",
     "create_ixp_extraction_tool",
     "create_ixp_escalation_tool",
+    "fetch_entity_schemas",
+    "format_schemas_for_context",
+    "get_datafabric_contexts",
+    "get_datafabric_entity_identifiers_from_resources",
     "UiPathToolNode",
     "ToolWrapperMixin",
 ]

src/uipath_langchain/agent/tools/context_tool.py

@@ -142,6 +142,11 @@ def create_context_tool(resource: AgentContextResourceConfig) -> StructuredTool:
         return handle_deep_rag(tool_name, resource)
     elif retrieval_mode == AgentContextRetrievalMode.BATCH_TRANSFORM.value.lower():
         return handle_batch_transform(tool_name, resource)
+    elif retrieval_mode == AgentContextRetrievalMode.DATA_FABRIC.value.lower():
+        raise ValueError(
+            "Data Fabric context should be handled via create_datafabric_tools(), "
+            "not create_context_tool()"
+        )
     else:
         return handle_semantic_search(tool_name, resource)
 

src/uipath_langchain/agent/tools/datafabric_tool/__init__.py

@@ -0,0 +1,17 @@
+"""Data Fabric tool module for entity-based SQL queries."""
+
+from .datafabric_tool import (
+    create_datafabric_tools,
+    fetch_entity_schemas,
+    format_schemas_for_context,
+    get_datafabric_contexts,
+    get_datafabric_entity_identifiers_from_resources,
+)
+
+__all__ = [
+    "create_datafabric_tools",
+    "fetch_entity_schemas",
+    "format_schemas_for_context",
+    "get_datafabric_contexts",
+    "get_datafabric_entity_identifiers_from_resources",
+]