feat: add tool_type metadata to all tools

pyproject.toml

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

src/uipath_langchain/agent/tools/context_tool.py

@@ -106,6 +106,10 @@ async def context_tool_fn(query: str) -> dict[str, Any]:
         args_schema=input_model,
         coroutine=context_tool_fn,
         output_type=output_model,
+        metadata={
+            "tool_type": "context",
+            "display_name": resource.name,
+        },
     )
 
 
@@ -183,6 +187,10 @@ async def context_tool_fn(query: str) -> dict[str, Any]:
         args_schema=input_model,
         coroutine=context_tool_fn,
         output_type=output_model,
+        metadata={
+            "tool_type": "context",
+            "display_name": resource.name,
+        },
     )
 
 
@@ -302,6 +310,10 @@ async def context_tool_fn(
         args_schema=input_model,
         coroutine=context_tool_fn,
         output_type=output_model,
+        metadata={
+            "tool_type": "context",
+            "display_name": resource.name,
+        },
     )
 
 

src/uipath_langchain/agent/tools/integration_tool.py

@@ -179,6 +179,10 @@ async def integration_tool_fn(**kwargs: Any):
         args_schema=input_model,
         coroutine=integration_tool_fn,
         output_type=output_model,
+        metadata={
+            "tool_type": "integration",
+            "display_name": resource.name,
+        },
     )
     tool.set_tool_wrappers(awrapper=wrapper)
 

src/uipath_langchain/agent/tools/mcp_tool.py

@@ -66,6 +66,8 @@ async def init_session(
         async with semaphore:
             await session.initialize()
             tools = await load_mcp_tools(session)
+            for tool in tools:
+                tool.metadata = {"tool_type": "mcp", "display_name": tool.name}
             return _filter_tools(tools, cfg)
 
     async def create_session(