fix: add output schema for context tool on batch transform attachment (#664)

gcuip · web-flow · commit dd572e49eac3 · 2026-03-04T15:02:34.000Z
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.8.4"
+version = "0.8.5"
 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"
diff --git a/src/uipath_langchain/agent/tools/context_tool.py b/src/uipath_langchain/agent/tools/context_tool.py
@@ -368,6 +368,7 @@ async def context_batch_transform_wrapper(
             "display_name": resource.name,
             "index_name": resource.index_name,
             "context_retrieval_mode": resource.settings.retrieval_mode,
+            "output_schema": output_model,
         },
     )
     tool.set_tool_wrappers(awrapper=context_batch_transform_wrapper)
diff --git a/uv.lock b/uv.lock

Original file line number	Diff line number	Diff line change
`@@ -368,6 +368,7 @@ async def context_batch_transform_wrapper(`
`368`	`368`	`"display_name": resource.name,`
`369`	`369`	`"index_name": resource.index_name,`
`370`	`370`	`"context_retrieval_mode": resource.settings.retrieval_mode,`
	`371`	`+ "output_schema": output_model,`
`371`	`372`	`},`
`372`	`373`	`)`
`373`	`374`	`tool.set_tool_wrappers(awrapper=context_batch_transform_wrapper)`