@agentunion/kite 1.0.6 → 1.2.0

package/core/launcher/module_scanner.py

@@ -34,9 +34,9 @@ class ModuleInfo:
     module_dir: str = ""                # absolute path to the module directory
     launch: LaunchConfig = field(default_factory=LaunchConfig)
 
-    def is_core(self, project_root: str) -> bool:
-        """Core modules live directly under {project_root}/core/."""
-        core_dir = os.path.join(project_root, "core")
+    def is_core(self) -> bool:
+        """Core modules live directly under {KITE_PROJECT}/core/."""
+        core_dir = os.path.join(os.environ["KITE_PROJECT"], "core")
         return os.path.normcase(self.module_dir).startswith(os.path.normcase(core_dir + os.sep))
 
 
@@ -89,18 +89,19 @@ def _parse_frontmatter(text: str) -> dict:
 class ModuleScanner:
     """Discover modules via configurable sources."""
 
-    def __init__(self, project_root: str, discovery: dict = None, launcher_dir: str = ""):
-        self.project_root = project_root
+    def __init__(self, discovery: dict = None, launcher_dir: str = ""):
         self.discovery = discovery
+        project_root = os.environ["KITE_PROJECT"]
         self.launcher_dir = launcher_dir or os.path.join(project_root, "core", "launcher")
 
     def scan(self) -> dict[str, ModuleInfo]:
         """Return dict of {module_name: ModuleInfo}. Duplicate names are skipped."""
         modules = {}
+        project_root = os.environ["KITE_PROJECT"]
 
         # Built-in: always scan core/ (depth 1) and extensions/ (depth 2)
-        self._scan_dir(os.path.join(self.project_root, "core"), 1, modules)
-        self._scan_dir(os.path.join(self.project_root, "extensions"), 2, modules)
+        self._scan_dir(os.path.join(project_root, "core"), 1, modules)
+        self._scan_dir(os.path.join(project_root, "extensions"), 2, modules)
 
         # Extra sources from discovery config
         if self.discovery:
@@ -118,7 +119,7 @@ class ModuleScanner:
                 if not path:
                     continue
                 if not os.path.isabs(path):
-                    path = os.path.join(self.project_root, path)
+                    path = os.path.join(project_root, path)
 
                 if src_type == "scan_dir":
                     max_depth = int(src.get("max_depth", 2))
@@ -156,7 +157,7 @@ class ModuleScanner:
                     if not line or line.startswith("#"):
                         continue
                     if not os.path.isabs(line):
-                        line = os.path.join(self.project_root, line)
+                        line = os.path.join(os.environ["KITE_PROJECT"], line)
                     self._add_module(line, modules)
         except Exception as e:
             print(f"[launcher] WARNING: failed to read module list {list_path}: {e}")