arcade-core 3.2.0__tar.gz → 3.3.1__tar.gz

{arcade_core-3.2.0 → arcade_core-3.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arcade-core
-Version: 3.2.0
+Version: 3.3.1
 Summary: Arcade Core - Core library for Arcade platform
 Author-email: Arcade <dev@arcade.dev>
 License: MIT

{arcade_core-3.2.0 → arcade_core-3.3.1}/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:
@@ -277,7 +283,6 @@ class ToolCatalog(BaseModel):
                     module = import_module(module_name)
                     tool_func = getattr(module, tool_name)
                     self.add_tool(tool_func, toolkit, module)
-
                 except ToolDefinitionError as e:
                     raise e.with_context(tool_name) from e
                 except ToolkitLoadError as e:

{arcade_core-3.2.0 → arcade_core-3.3.1}/arcade_core/toolkit.py

@@ -1,7 +1,9 @@
+import contextlib
 import importlib.metadata
 import importlib.util
 import logging
 import os
+import sys
 import types
 from collections import defaultdict
 from pathlib import Path, PurePosixPath, PureWindowsPath
@@ -293,9 +295,27 @@ class Toolkit(BaseModel):
                 f"Failed to locate Python files in package directory for '{package_name}'."
             ) from e
 
+        # Get the currently executing file (the entrypoint file) so that we can skip it when loading tools.
+        # Skipping this file is necessary because tools are discovered via AST parsing, but those tools
+        # aren't in the module's namespace yet since the file is still executing.
+        current_file = None
+        main_module = sys.modules.get("__main__")
+        if main_module and hasattr(main_module, "__file__") and main_module.__file__:
+            with contextlib.suppress(Exception):
+                current_file = Path(main_module.__file__).resolve()
+
         tools: dict[str, list[str]] = {}
 
         for module_path in modules:
+            # Skip adding tools from the currently executing file
+            if current_file:
+                try:
+                    module_path_resolved = module_path.resolve()
+                    if module_path_resolved == current_file:
+                        continue
+                except Exception:  # noqa: S110
+                    pass
+
             relative_path = module_path.relative_to(package_dir)
             cls.validate_file(module_path)
             # Build import path and avoid duplicating the package prefix if it already exists

{arcade_core-3.2.0 → arcade_core-3.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "arcade-core"
-version = "3.2.0"
+version = "3.3.1"
 description = "Arcade Core - Core library for Arcade platform"
 readme = "README.md"
 license = {text = "MIT"}