fix: deep rag output model (#570)

Author: radu-mocanu

pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-langchain"
-version = "0.5.47"
+version = "0.5.48"
 description = "Python SDK that enables developers to build and deploy LangGraph agents to the UiPath Cloud Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
-    "uipath>=2.8.18,<2.9.0",
+    "uipath>=2.8.21,<2.9.0",
     "uipath-runtime>=0.8.0, <0.9.0",
     "langgraph>=1.0.0, <2.0.0",
     "langchain-core>=1.2.11, <2.0.0",

src/uipath_langchain/agent/tools/context_tool.py

@@ -6,7 +6,7 @@
 from langchain_core.documents import Document
 from langchain_core.tools import StructuredTool
 from langgraph.types import interrupt
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, create_model
 from uipath.agent.models.agent import (
     AgentContextResourceConfig,
     AgentContextRetrievalMode,
@@ -17,7 +17,7 @@
     BatchTransformOutputColumn,
     BatchTransformResponse,
     CitationMode,
-    DeepRagResponse,
+    DeepRagContent,
 )
 
 from uipath_langchain.retrievers import ContextGroundingRetriever
@@ -127,7 +127,11 @@ def handle_deep_rag(
         raise ValueError("Citation mode is required for Deep RAG")
     citation_mode = CitationMode(resource.settings.citation_mode.value)
 
-    output_model = DeepRagResponse
+    output_model = create_model(
+        "DeepRagOutputModel",
+        __base__=DeepRagContent,
+        deep_rag_id=(str, Field(alias="deepRagId")),
+    )
 
     if is_static_query(resource):
         # Static query - no input parameter needed

tests/agent/tools/test_context_tool.py

@@ -15,7 +15,7 @@
 from uipath.platform.context_grounding import (
     BatchTransformResponse,
     CitationMode,
-    DeepRagResponse,
+    DeepRagContent,
 )
 
 from uipath_langchain.agent.tools.context_tool import (
@@ -80,7 +80,10 @@ def test_successful_deep_rag_creation(self, base_resource_config):
         assert result.name == "test_deep_rag"
         assert result.description == "Test Deep RAG tool"
         assert result.args_schema is None
-        assert result.output_type == DeepRagResponse
+        assert issubclass(result.output_type, DeepRagContent)
+        schema = result.output_type.model_json_schema()
+        assert "deepRagId" in schema["properties"]
+        assert schema["properties"]["deepRagId"]["type"] == "string"
 
     def test_missing_query_object_raises_error(self, base_resource_config):
         """Test that missing query object raises ValueError."""
@@ -227,7 +230,7 @@ def test_dynamic_query_deep_rag_creation(self, base_resource_config):
         assert result.name == "test_deep_rag"
         assert result.description == "Test Deep RAG tool"
         assert result.args_schema is not None  # Dynamic has input schema
-        assert result.output_type == DeepRagResponse
+        assert issubclass(result.output_type, DeepRagContent)
 
     def test_dynamic_query_deep_rag_has_query_parameter(self, base_resource_config):
         """Test that dynamic Deep RAG tool has query parameter in schema."""
@@ -329,7 +332,7 @@ def test_create_deep_rag_tool(self, deep_rag_config):
         assert isinstance(result, StructuredToolWithOutputType)
         assert result.name == "test_deep_rag"
         assert result.args_schema is None  # Deep RAG has no input schema
-        assert result.output_type == DeepRagResponse
+        assert issubclass(result.output_type, DeepRagContent)
 
     def test_case_insensitive_retrieval_mode(self, deep_rag_config):
         """Test that retrieval mode matching is case-insensitive."""