feat(StructuredToolWithOutputType): support dynamic output types

`StructuredTool`s are `Runnable`s, but `OutputType` is always `Any` unless overridden.
This commit adds the override since we have strongly typed output types.

Additionally, removing asterisks from IS output schemas (bugfix).

Author: akshaylive

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.1.14"
+version = "0.1.15"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

src/uipath_langchain/agent/tools/context_tool.py

@@ -10,6 +10,7 @@
 
 from uipath_langchain.retrievers import ContextGroundingRetriever
 
+from .structured_tool_with_output_type import StructuredToolWithOutputType
 from .utils import sanitize_tool_name
 
 
@@ -43,9 +44,10 @@ class ContextOutputSchemaModel(BaseModel):
     async def context_tool_fn(query: str) -> dict[str, Any]:
         return {"documents": await retriever.ainvoke(query)}
 
-    return StructuredTool(
+    return StructuredToolWithOutputType(
         name=tool_name,
         description=resource.description,
         args_schema=input_model,
         coroutine=context_tool_fn,
+        output_type=output_model,
     )

src/uipath_langchain/agent/tools/process_tool.py

@@ -9,6 +9,7 @@
 from uipath.eval.mocks import mockable
 from uipath.platform.common import InvokeProcess
 
+from .structured_tool_with_output_type import StructuredToolWithOutputType
 from .utils import sanitize_tool_name
 
 
@@ -37,11 +38,12 @@ async def process_tool_fn(**kwargs: Any):
             )
         )
 
-    tool = StructuredTool(
+    tool = StructuredToolWithOutputType(
         name=tool_name,
         description=resource.description,
         args_schema=input_model,
         coroutine=process_tool_fn,
+        output_type=output_model,
     )
 
     return tool

src/uipath_langchain/agent/tools/structured_tool_with_output_type.py

@@ -0,0 +1,14 @@
+from typing import Any
+
+from langchain_core.tools import StructuredTool
+from pydantic import Field
+from typing_extensions import override
+
+
+class StructuredToolWithOutputType(StructuredTool):
+    output_type: Any = Field(Any, description="Output type.")
+
+    @override
+    @property
+    def OutputType(self) -> type[Any]:
+        return self.output_type