backend.ai-plugin 25.6.10__py3-none-any.whl → 25.17.0rc1__py3-none-any.whl

ai/backend/plugin/VERSION

@@ -1 +1 @@
-25.6.10
+25.17.0rc1

ai/backend/plugin/cli.py

@@ -5,7 +5,7 @@ import itertools
 import json
 import logging
 from collections import defaultdict
-from typing import Any, Self
+from typing import Self
 
 import click
 import colorama
@@ -36,14 +36,7 @@ class CLIContext:
         self.log_level = log_level
 
     def __enter__(self) -> Self:
-        log_config: dict[str, Any] = {}
-        if self.log_level != LogLevel.NOTSET:
-            log_config["level"] = self.log_level
-            log_config["pkg-ns"] = {
-                "": LogLevel.WARNING,
-                "ai.backend": self.log_level,
-            }
-        self._logger = LocalLogger(log_config)
+        self._logger = LocalLogger(log_level=self.log_level)
         self._logger.__enter__()
         return self
 

ai/backend/plugin/entrypoint.py

@@ -78,27 +78,81 @@ _default_glob_excluded_patterns = [
     "ai/backend/runner",
     "ai/backend/kernel",
     "wheelhouse",
+    "tools",
 ]
 
+_optimized_glob_search_patterns = {
+    # These patterns only apply to scanning BUILD files in dev setups and pex distributions.
+    # They do not affect standard package entrypoint searches.
+    # NOTE: most entrypoint declaration in BUILD files are in the package's top-level only!
+    "backendai_cli_v10": ["ai/backend/*", "ai/backend/appproxy/*"],
+    "backendai_network_manager_v1": ["ai/backend/*"],
+    "backendai_event_dispatcher_v20": ["ai/backend/*", "ai/backend/appproxy/*"],
+    "backendai_stats_monitor_v20": ["ai/backend/*", "ai/backend/appproxy/*"],
+    "backendai_error_monitor_v20": ["ai/backend/*", "ai/backend/appproxy/*"],
+    "backendai_hook_v20": ["ai/backend/*"],
+    "backendai_webapp_v20": ["ai/backend/*"],
+    "backendai_scheduler_v10": ["ai/backend/manager"],
+    "backendai_agentselector_v10": ["ai/backend/manager"],
+}
 
-def _glob(base_path: Path, filename: str, excluded_patterns: Iterable[str]) -> Iterator[Path]:
-    q: collections.deque[Path] = collections.deque()
-    q.append(base_path)
+
+def _glob(
+    base_path: Path,
+    filename: str,
+    excluded_patterns: Iterable[str],
+    match_patterns: Iterable[str] | None = None,
+) -> Iterator[Path]:
+    q: collections.deque[tuple[Path, bool]] = collections.deque()
+    assert base_path.is_dir()
+    q.append((base_path, False))
     while q:
-        search_path = q.pop()
-        assert search_path.is_dir()
+        search_path, suffix_match = q.pop()
+
+        # Check if current directory matches any pattern and we should yield files from it
+        current_matches = False
+        if match_patterns is not None:
+            current_matches = any(search_path.match(pattern) for pattern in match_patterns)
+
         for item in search_path.iterdir():
             if item.is_dir():
-                if search_path.name == "__pycache__":
+                if item.name == "__pycache__":
                     continue
-                if search_path.name.startswith("."):
+                if item.name.startswith("."):
                     continue
-                if any(search_path.match(pattern) for pattern in excluded_patterns):
+                if any(item.match(pattern) for pattern in excluded_patterns):
                     continue
-                q.append(item)
+
+                # Determine if we should queue this directory
+                should_queue = False
+                new_suffix_match = False
+
+                if match_patterns is None:
+                    # No patterns specified - queue all non-excluded directories
+                    should_queue = True
+                    new_suffix_match = False
+                elif not suffix_match:
+                    # Haven't found a matching directory yet - check if this one matches
+                    if any(item.match(pattern) for pattern in match_patterns):
+                        should_queue = True
+                        new_suffix_match = True
+                    else:
+                        # Keep searching - queue without suffix match
+                        should_queue = True
+                        new_suffix_match = False
+                else:
+                    # Already found a matching directory - only queue if this also matches
+                    if any(item.match(pattern) for pattern in match_patterns):
+                        should_queue = True
+                        new_suffix_match = True
+
+                if should_queue:
+                    q.append((item, new_suffix_match))
             else:
                 if item.name == filename:
-                    yield item
+                    # Yield file if no patterns or current directory matches
+                    if match_patterns is None or current_matches:
+                        yield item
 
 
 def scan_entrypoint_from_buildscript(group_name: str) -> Iterator[EntryPoint]:
@@ -108,11 +162,17 @@ def scan_entrypoint_from_buildscript(group_name: str) -> Iterator[EntryPoint]:
     log.debug(
         "scan_entrypoint_from_buildscript(%r): Namespace path: %s", group_name, ai_backend_ns_path
     )
-    for buildscript_path in _glob(ai_backend_ns_path, "BUILD", _default_glob_excluded_patterns):
+    match_patterns = _optimized_glob_search_patterns.get(group_name, None)
+    # First, it is invoked in PEX or temporary test environment generated by Pantsbuild.
+    # In the test environment, BUILD files are NOT copied, so the plugin discovery will rely on the
+    # followed build-root search below.
+    for buildscript_path in _glob(
+        ai_backend_ns_path, "BUILD", _default_glob_excluded_patterns, match_patterns
+    ):
         for entrypoint in extract_entrypoints_from_buildscript(group_name, buildscript_path):
             entrypoints[entrypoint.name] = entrypoint
     if os.environ.get("SCIE", None) is None:
-        # Override with the entrypoints found in the current source directories,
+        # Override with the entrypoints found in the current build-root directory.
         try:
             build_root = find_build_root()
         except ValueError:
@@ -120,7 +180,9 @@ def scan_entrypoint_from_buildscript(group_name: str) -> Iterator[EntryPoint]:
         else:
             src_path = build_root / "src"
             log.debug("scan_entrypoint_from_buildscript(%r): current src: %s", group_name, src_path)
-            for buildscript_path in _glob(src_path, "BUILD", _default_glob_excluded_patterns):
+            for buildscript_path in _glob(
+                src_path, "BUILD", _default_glob_excluded_patterns, match_patterns
+            ):
                 for entrypoint in extract_entrypoints_from_buildscript(
                     group_name, buildscript_path
                 ):

{backend_ai_plugin-25.6.10.dist-info → backend_ai_plugin-25.17.0rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: backend.ai-plugin
-Version: 25.6.10
+Version: 25.17.0rc1
 Summary: Backend.AI Plugin Subsystem
 Home-page: https://github.com/lablup/backend.ai
 Author: Lablup Inc. and contributors
@@ -15,15 +15,17 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Environment :: No Input/Output (Daemon)
 Classifier: Topic :: Scientific/Engineering
 Classifier: Topic :: Software Development
-Classifier: Development Status :: 5 - Production/Stable
+Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python :: 3.13
 Classifier: License :: OSI Approved :: MIT License
 Requires-Python: >=3.13,<3.14
 Description-Content-Type: text/markdown
-Requires-Dist: backend.ai-logging==25.6.10
+Requires-Dist: backend.ai-common==25.17.0rc1
+Requires-Dist: backend.ai-logging==25.17.0rc1
 Requires-Dist: click~=8.1.7
 Requires-Dist: colorama>=0.4.6
 Requires-Dist: tabulate~=0.8.9
+Requires-Dist: types-colorama
 Requires-Dist: types-tabulate
 Dynamic: author
 Dynamic: classifier

backend_ai_plugin-25.17.0rc1.dist-info/RECORD

@@ -0,0 +1,11 @@
+ai/backend/plugin/VERSION,sha256=vhxW9UTYhT2p3VipuobjwM2Le06BsZH08HmwBPKZ0B0,11
+ai/backend/plugin/__init__.py,sha256=HKBIEWtrpEk2KY3fB3Xb72N0xz5zoYUyn2HfZ75bTsc,96
+ai/backend/plugin/cli.py,sha256=7SLUJy-8uPd_tSBOEZPYw1566QwrJJmCngoVyqrPAWE,5182
+ai/backend/plugin/entrypoint.py,sha256=lFqz-QNcWfYBlFoDTGEU5sY0CgYD0NWEHhhtmXs4840,12083
+ai/backend/plugin/py.typed,sha256=L3M0nPxGMCVTGcbI38G0aomWrOnRTY4HVjsWWRWRjsI,12
+backend_ai_plugin-25.17.0rc1.dist-info/METADATA,sha256=19MgsKG27y3JOa59l7viutcez-ZEnbJatdvOXwECP6o,1524
+backend_ai_plugin-25.17.0rc1.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
+backend_ai_plugin-25.17.0rc1.dist-info/entry_points.txt,sha256=XxdR8AJRnWYCT-BgkqvFySRw_WjL0r9M43fRAVszaqY,56
+backend_ai_plugin-25.17.0rc1.dist-info/namespace_packages.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+backend_ai_plugin-25.17.0rc1.dist-info/top_level.txt,sha256=TJAp5TUfTUztZSUatbygths7CWRrFfnOMCtZ-DIcw6c,3
+backend_ai_plugin-25.17.0rc1.dist-info/RECORD,,

{backend_ai_plugin-25.6.10.dist-info → backend_ai_plugin-25.17.0rc1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.0.0)
+Generator: setuptools (80.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

backend_ai_plugin-25.6.10.dist-info/RECORD

@@ -1,11 +0,0 @@
-ai/backend/plugin/VERSION,sha256=IdjD0AeESmbRCLn3D9uKMHyJbRqIVbzuxR8wQMPd7TQ,7
-ai/backend/plugin/__init__.py,sha256=HKBIEWtrpEk2KY3fB3Xb72N0xz5zoYUyn2HfZ75bTsc,96
-ai/backend/plugin/cli.py,sha256=oG3KUWot27WlRVwrpG9i2U-OF6XjwuweVdxVmUe4a4c,5443
-ai/backend/plugin/entrypoint.py,sha256=iCTPCsWEXEAn411qqgdpPyrO40eUEOuDhi4f9LmFu7I,9181
-ai/backend/plugin/py.typed,sha256=L3M0nPxGMCVTGcbI38G0aomWrOnRTY4HVjsWWRWRjsI,12
-backend_ai_plugin-25.6.10.dist-info/METADATA,sha256=85OdvRx8zNcZ2Bk-Iq2jXNmaoCgKu58e71grjO3lXrg,1456
-backend_ai_plugin-25.6.10.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
-backend_ai_plugin-25.6.10.dist-info/entry_points.txt,sha256=XxdR8AJRnWYCT-BgkqvFySRw_WjL0r9M43fRAVszaqY,56
-backend_ai_plugin-25.6.10.dist-info/namespace_packages.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-backend_ai_plugin-25.6.10.dist-info/top_level.txt,sha256=TJAp5TUfTUztZSUatbygths7CWRrFfnOMCtZ-DIcw6c,3
-backend_ai_plugin-25.6.10.dist-info/RECORD,,