fix: keep workspace mapping dependency-safe

Author: radu-mocanu

src/uipath_langchain/_client_side_tool_types.py

@@ -0,0 +1,13 @@
+"""Shared client-side tool types."""
+
+from typing import Any, TypedDict
+
+
+class ClientSideToolInfo(TypedDict):
+    """Schemas exposed for a client-side tool."""
+
+    input_schema: dict[str, Any] | None
+    output_schema: dict[str, Any] | None
+
+
+__all__ = ["ClientSideToolInfo"]

src/uipath_langchain/agent/tools/client_side_tool.py

@@ -2,13 +2,16 @@
 
 import json
 from contextvars import ContextVar
-from typing import Annotated, Any, TypedDict
+from typing import Annotated, Any
 
 from langchain_core.messages import ToolMessage
 from langchain_core.tools import InjectedToolCallId, StructuredTool
 from uipath.agent.models.agent import AgentClientSideToolResourceConfig
 from uipath.eval.mocks import mockable
 
+from uipath_langchain._client_side_tool_types import (
+    ClientSideToolInfo as ClientSideToolInfo,
+)
 from uipath_langchain._utils.durable_interrupt import durable_interrupt
 from uipath_langchain.agent.react.jsonschema_pydantic_converter import (
     create_model as create_model_from_schema,
@@ -26,11 +29,6 @@
 UIPATH_CLIENT_SIDE_TOOLS_INPUT_KEY = "uipath__client_side_tools"
 
 
-class ClientSideToolInfo(TypedDict):
-    input_schema: dict[str, Any] | None
-    output_schema: dict[str, Any] | None
-
-
 def apply_tool_filter(
     declared_tools: list[str | dict[str, Any]],
     agent_tools: dict[str, ClientSideToolInfo],

src/uipath_langchain/runtime/messages.py

@@ -40,7 +40,7 @@
 )
 from uipath.runtime import UiPathRuntimeStorageProtocol
 
-from uipath_langchain.agent.tools.client_side_tool import ClientSideToolInfo
+from uipath_langchain._client_side_tool_types import ClientSideToolInfo
 from uipath_langchain.chat.hitl import IS_CONVERSATIONAL_CLIENT_SIDE_TOOL
 
 from ._citations import (
@@ -188,9 +188,7 @@ def _map_messages_internal(
                                 )
                             )
                     elif isinstance(data, UiPathExternalValue):
-                        part_metadata = getattr(
-                            uipath_content_part, "metadata", None
-                        )
+                        part_metadata = getattr(uipath_content_part, "metadata", None)
                         has_workspace_metadata = (
                             isinstance(part_metadata, dict)
                             and part_metadata.get("fileKind") == "workspace"

src/uipath_langchain/runtime/runtime.py

@@ -32,7 +32,7 @@
 )
 from uipath.runtime.schema import UiPathRuntimeSchema
 
-from uipath_langchain.agent.tools.client_side_tool import ClientSideToolInfo
+from uipath_langchain._client_side_tool_types import ClientSideToolInfo
 from uipath_langchain.chat.hitl import (
     IS_CONVERSATIONAL_CLIENT_SIDE_TOOL,
     get_confirmation_schema,

tests/runtime/test_chat_message_mapper_workspace.py

@@ -41,7 +41,7 @@ def test_assistant_file_parts_are_skipped() -> None:
                 data=UiPathInlineValue(inline="done, see the plan"),
                 citations=[],
             ),
-            _file_part("p2", "plan/todo.md", metadata={"fileKind": "workspace"}),
+            _file_part("ws-a1-0", "plan/todo.md", metadata={"fileKind": "workspace"}),
         ],
         tool_calls=[],
     )