feat: add tool metadata (#527)

Author: cristian-groza

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.5.31"
+version = "0.5.32"
 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"

src/uipath_langchain/agent/tools/internal_tools/analyze_files_tool.py

@@ -102,7 +102,10 @@ async def analyze_file_tool_wrapper(
         output_type=output_model,
         argument_properties=resource.argument_properties,
         metadata={
+            "tool_type": resource.type.lower(),
+            "display_name": tool_name,
             "args_schema": input_model,
+            "output_schema": output_model,
         },
     )
     tool.set_tool_wrappers(awrapper=analyze_file_tool_wrapper)

src/uipath_langchain/agent/tools/internal_tools/batch_transform_tool.py

@@ -154,6 +154,12 @@ async def batch_transform_tool_wrapper(
         coroutine=batch_transform_tool_fn,
         output_type=output_model,
         argument_properties=resource.argument_properties,
+        metadata={
+            "tool_type": resource.type.lower(),
+            "display_name": tool_name,
+            "args_schema": input_model,
+            "output_schema": output_model,
+        },
     )
     tool.set_tool_wrappers(awrapper=batch_transform_tool_wrapper)
     return tool

src/uipath_langchain/agent/tools/internal_tools/deeprag_tool.py

@@ -137,6 +137,12 @@ async def deeprag_tool_wrapper(
         coroutine=deeprag_tool_fn,
         output_type=output_model,
         argument_properties=resource.argument_properties,
+        metadata={
+            "tool_type": resource.type.lower(),
+            "display_name": tool_name,
+            "args_schema": input_model,
+            "output_schema": output_model,
+        },
     )
     tool.set_tool_wrappers(awrapper=deeprag_tool_wrapper)
     return tool

src/uipath_langchain/agent/tools/process_tool.py

@@ -76,6 +76,7 @@ async def process_tool_wrapper(
             "display_name": process_name,
             "folder_path": folder_path,
             "args_schema": input_model,
+            "output_schema": output_model,
         },
         argument_properties=resource.argument_properties,
     )