diff --git a/libs/arcade-core/arcade_core/catalog.py b/libs/arcade-core/arcade_core/catalog.py index 6ba78ddb..2af42384 100644 --- a/libs/arcade-core/arcade_core/catalog.py +++ b/libs/arcade-core/arcade_core/catalog.py @@ -255,11 +255,17 @@ class ToolCatalog(BaseModel): output_model=output_model, ) - def add_module(self, module: ModuleType) -> None: + def add_module(self, module: ModuleType, name: str | None = None) -> None: """ Add all the tools in a module to the catalog. + + Args: + module: The module to add. + name: Optionally override the name of the toolkit with this parameter """ toolkit = Toolkit.from_module(module) + if name: + toolkit.name = name self.add_toolkit(toolkit) def add_toolkit(self, toolkit: Toolkit) -> None: diff --git a/libs/arcade-core/pyproject.toml b/libs/arcade-core/pyproject.toml index 7c2bec1d..d6aefb65 100644 --- a/libs/arcade-core/pyproject.toml +++ b/libs/arcade-core/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "arcade-core" -version = "3.2.0" +version = "3.3.0" description = "Arcade Core - Core library for Arcade platform" readme = "README.md" license = {text = "MIT"} diff --git a/libs/arcade-mcp-server/arcade_mcp_server/mcp_app.py b/libs/arcade-mcp-server/arcade_mcp_server/mcp_app.py index 5dcc1432..70c4a03d 100644 --- a/libs/arcade-mcp-server/arcade_mcp_server/mcp_app.py +++ b/libs/arcade-mcp-server/arcade_mcp_server/mcp_app.py @@ -193,7 +193,7 @@ class MCPApp: def add_tools_from_module(self, module: ModuleType) -> None: """Add all the tools in a module to the catalog.""" - self._catalog.add_module(module) + self._catalog.add_module(module, self._toolkit_name) def tool( self, diff --git a/libs/arcade-mcp-server/pyproject.toml b/libs/arcade-mcp-server/pyproject.toml index f7335b96..bccb6a7a 100644 --- a/libs/arcade-mcp-server/pyproject.toml +++ b/libs/arcade-mcp-server/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "hatchling.build" [project] name = "arcade-mcp-server" -version = "1.6.0" +version = "1.6.1" description = "Model Context Protocol (MCP) server framework for Arcade.dev" readme = "README.md" authors = [{ name = "Arcade.dev" }] @@ -21,7 +21,7 @@ classifiers = [ ] requires-python = ">=3.10" dependencies = [ - "arcade-core>=3.2.0,<4.0.0", + "arcade-core>=3.3.0,<4.0.0", "arcade-serve>=3.0.0,<4.0.0", "arcade-tdk>=3.0.0,<4.0.0", "arcadepy>=1.5.0",