refactor: expose kb helper APIs

Author: eddiman

src/adjutant/capabilities/kb/query.py

@@ -160,10 +160,10 @@ async def kb_query(
     Raises:
         KBQueryError: If KB not found or query fails.
     """
-    from adjutant.capabilities.kb.run import KBNotFoundError, _get_kb
+    from adjutant.capabilities.kb.run import KBNotFoundError, get_kb
 
     try:
-        entry = _get_kb(adj_dir, kb_name)
+        entry = get_kb(adj_dir, kb_name)
     except KBNotFoundError as exc:
         raise KBQueryError(str(exc)) from exc
 
@@ -251,10 +251,10 @@ def kb_write(
     Raises:
         KBQueryError: If KB not found or instruction fails validation.
     """
-    from adjutant.capabilities.kb.run import KBNotFoundError, _get_kb
+    from adjutant.capabilities.kb.run import KBNotFoundError, get_kb
 
     try:
-        entry = _get_kb(adj_dir, kb_name)
+        entry = get_kb(adj_dir, kb_name)
     except KBNotFoundError as exc:
         raise KBQueryError(str(exc)) from exc
 

src/adjutant/capabilities/kb/run.py

@@ -45,7 +45,7 @@ class KBRunError(Exception):
     """Raised when the operation script exits non-zero."""
 
 
-def _get_kb(adj_dir: Path, kb_name: str) -> dict[str, str]:
+def get_kb(adj_dir: Path, kb_name: str) -> dict[str, str]:
     """Return the registry entry for kb_name or raise KBNotFoundError.
 
     Delegates to the canonical registry parser in manage.py and converts
@@ -62,6 +62,11 @@ def _get_kb(adj_dir: Path, kb_name: str) -> dict[str, str]:
     return entry.as_dict()
 
 
+def _get_kb(adj_dir: Path, kb_name: str) -> dict[str, str]:
+    """Backward-compatible alias for older call sites and tests."""
+    return get_kb(adj_dir, kb_name)
+
+
 _VALID_OPERATION = re.compile(r"^[a-z][a-z0-9_-]*$")
 
 
@@ -144,7 +149,7 @@ def get_operation_script(adj_dir: Path, kb_name: str, operation: str) -> Path:
             "Use lowercase letters, digits, hyphens, or underscores."
         )
 
-    entry = _get_kb(adj_dir, kb_name)
+    entry = get_kb(adj_dir, kb_name)
     kb_path_str = entry.get("path", "")
     if not kb_path_str:
         raise KBNotFoundError(f"KB '{kb_name}' has no path in registry.")
@@ -232,7 +237,7 @@ def kb_run(
             "Use lowercase letters, digits, hyphens, or underscores."
         )
 
-    entry = _get_kb(adj_dir, kb_name)
+    entry = get_kb(adj_dir, kb_name)
     kb_path_str = entry.get("path", "")
     if not kb_path_str:
         raise KBNotFoundError(f"KB '{kb_name}' has no path in registry.")