aes-cli 0.4.3__tar.gz → 0.6.0__tar.gz

{aes_cli-0.4.3 → aes_cli-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aes-cli
-Version: 0.4.3
+Version: 0.6.0
 Summary: CLI tool for the Agentic Engineering Standard
 Author: Hiro
 License: Apache-2.0

{aes_cli-0.4.3 → aes_cli-0.6.0}/aes/__init__.py

@@ -2,4 +2,4 @@
 
 from __future__ import annotations
 
-__version__ = "0.4.3"
+__version__ = "0.6.0"

{aes_cli-0.4.3 → aes_cli-0.6.0}/aes/__main__.py

@@ -16,6 +16,8 @@ from aes.commands.install import install_cmd
 from aes.commands.search import search_cmd
 from aes.commands.status import status_cmd
 from aes.commands.sync import sync_cmd
+from aes.commands.upgrade import upgrade_cmd
+from aes.commands.bom import bom_cmd
 
 
 def _prompt_language() -> None:
@@ -77,6 +79,8 @@ cli.add_command(install_cmd, "install")
 cli.add_command(sync_cmd, "sync")
 cli.add_command(status_cmd, "status")
 cli.add_command(search_cmd, "search")
+cli.add_command(upgrade_cmd, "upgrade")
+cli.add_command(bom_cmd, "bom")
 
 
 if __name__ == "__main__":

aes_cli-0.6.0/aes/commands/bom.py

@@ -0,0 +1,131 @@
+"""aes bom — Display the Agent Bill of Materials."""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+import click
+import yaml
+from rich.console import Console
+from rich.table import Table
+
+from aes.config import AGENT_DIR, BOM_FILE
+from aes.i18n import t
+
+console = Console()
+
+
+def _load_bom(agent_dir: Path) -> dict:
+    """Load and return bom.yaml, or empty dict."""
+    bom_path = agent_dir / BOM_FILE
+    if not bom_path.exists():
+        return {}
+    with open(bom_path) as f:
+        data = yaml.safe_load(f)
+    return data if isinstance(data, dict) else {}
+
+
+@click.command("bom")
+@click.argument("path", default=".", type=click.Path(exists=True))
+def bom_cmd(path: str) -> None:
+    """Display the Agent Bill of Materials (AI-BOM).
+
+    Shows all models, frameworks, tools, and data sources
+    declared in .agent/bom.yaml.
+
+    \b
+    Examples:
+      aes bom                    # current project
+      aes bom ./my-project       # specific project
+    """
+    project_root = Path(path).resolve()
+    agent_dir = project_root / AGENT_DIR
+
+    if not agent_dir.exists():
+        console.print(f"[red]{t('common.error')}:[/] {t('common.no_agent_dir', agent_dir=AGENT_DIR, path=project_root)}")
+        raise SystemExit(1)
+
+    bom = _load_bom(agent_dir)
+    if not bom:
+        console.print(f"\n  [dim]{t('bom.not_found')}[/]\n")
+        return
+
+    console.print(f"\n[bold]{t('bom.title')}[/]  [dim](AES {bom.get('aes_bom', '?')})[/]\n")
+
+    # Models
+    models = bom.get("models", [])
+    if models:
+        table = Table(title=t("bom.models_table"), show_header=True, header_style="bold")
+        table.add_column(t("bom.col_name"), style="cyan")
+        table.add_column(t("bom.col_provider"))
+        table.add_column(t("bom.col_purpose"))
+        table.add_column(t("bom.col_license"))
+        for m in models:
+            table.add_row(
+                m.get("name", "?"),
+                m.get("provider", "?"),
+                m.get("purpose", "-"),
+                m.get("license", "-"),
+            )
+        console.print(table)
+        console.print()
+
+    # Frameworks
+    frameworks = bom.get("frameworks", [])
+    if frameworks:
+        table = Table(title=t("bom.frameworks_table"), show_header=True, header_style="bold")
+        table.add_column(t("bom.col_name"), style="cyan")
+        table.add_column(t("bom.col_version"))
+        table.add_column(t("bom.col_license"))
+        for fw in frameworks:
+            table.add_row(
+                fw.get("name", "?"),
+                fw.get("version", "-"),
+                fw.get("license", "-"),
+            )
+        console.print(table)
+        console.print()
+
+    # Tools
+    tools = bom.get("tools", [])
+    if tools:
+        table = Table(title=t("bom.tools_table"), show_header=True, header_style="bold")
+        table.add_column(t("bom.col_name"), style="cyan")
+        table.add_column(t("bom.col_type"))
+        table.add_column(t("bom.col_version"))
+        table.add_column(t("bom.col_source"))
+        for tool in tools:
+            table.add_row(
+                tool.get("name", "?"),
+                tool.get("type", "?"),
+                tool.get("version", "-"),
+                tool.get("source", "-"),
+            )
+        console.print(table)
+        console.print()
+
+    # Data Sources
+    data_sources = bom.get("data_sources", [])
+    if data_sources:
+        table = Table(title=t("bom.data_sources_table"), show_header=True, header_style="bold")
+        table.add_column(t("bom.col_name"), style="cyan")
+        table.add_column(t("bom.col_type"))
+        table.add_column(t("bom.col_uri"))
+        table.add_column(t("bom.col_license"))
+        for ds in data_sources:
+            table.add_row(
+                ds.get("name", "?"),
+                ds.get("type", "?"),
+                ds.get("uri", "-"),
+                ds.get("license", "-"),
+            )
+        console.print(table)
+        console.print()
+
+    # Summary
+    console.print(f"[bold]{t('bom.summary')}[/]")
+    console.print(f"  {t('bom.models_count', count=len(models))}")
+    console.print(f"  {t('bom.frameworks_count', count=len(frameworks))}")
+    console.print(f"  {t('bom.tools_count', count=len(tools))}")
+    console.print(f"  {t('bom.data_sources_count', count=len(data_sources))}")
+    console.print()

{aes_cli-0.4.3 → aes_cli-0.6.0}/aes/commands/init.py

@@ -20,6 +20,8 @@ from rich.tree import Tree
 from aes.config import (
     AGENT_DIR,
     AGENTIGNORE_FILE,
+    BOM_FILE,
+    DECISIONS_DIR,
     LOCAL_EXAMPLE_FILE,
     LOCAL_FILE,
     SCAFFOLD_DIR,
@@ -427,6 +429,7 @@ def _print_post_init_summary(
 
     cmd_branch = tree.add("commands/")
     cmd_branch.add("setup.md")
+    cmd_branch.add("memory.md [dim]/memory[/]")
     if isinstance(domain_config, DomainConfig) and domain_config.workflow_commands:
         for cmd_def in domain_config.workflow_commands:
             cmd_branch.add(f"{cmd_def.id}.md [dim]{cmd_def.trigger}[/]")
@@ -619,6 +622,7 @@ def init_cmd(
     agent_dir.mkdir(exist_ok=True)
     (agent_dir / MEMORY_DIR).mkdir(exist_ok=True)
     (agent_dir / MEMORY_DIR / "sessions").mkdir(exist_ok=True)
+    (agent_dir / DECISIONS_DIR).mkdir(parents=True, exist_ok=True)
     (agent_dir / OVERRIDES_DIR).mkdir(exist_ok=True)
 
     if skills:
@@ -663,6 +667,10 @@ def init_cmd(
     content = _render_template(env, "permissions.yaml.jinja", context)
     (agent_dir / "permissions.yaml").write_text(content)
 
+    # bom.yaml
+    content = _render_template(env, "bom.yaml.jinja", context)
+    (agent_dir / BOM_FILE).write_text(content)
+
     # .agentignore
     agentignore_path = project_root / AGENTIGNORE_FILE
     if not agentignore_path.exists():
@@ -702,10 +710,12 @@ def init_cmd(
     memory_content = f"# {name} — Agent Memory\n\n## Project Overview\n\n## Architecture\n\n## Status\n\n## Key Patterns\n"
     (agent_dir / MEMORY_DIR / "project.md").write_text(memory_content)
 
-    # Commands directory + /setup runbook
+    # Commands directory + /setup and /memory runbooks
     (agent_dir / COMMANDS_DIR).mkdir(exist_ok=True)
     content = _render_template(env, "setup.md.jinja", context)
     (agent_dir / COMMANDS_DIR / "setup.md").write_text(content)
+    content = _render_template(env, "memory_command.md.jinja", context)
+    (agent_dir / COMMANDS_DIR / "memory.md").write_text(content)
 
     # Workflow command runbooks
     if domain_config and domain_config.workflow_commands:

{aes_cli-0.4.3 → aes_cli-0.6.0}/aes/commands/inspect.py

@@ -12,7 +12,7 @@ import yaml
 from rich.console import Console
 from rich.table import Table
 
-from aes.config import AGENT_DIR
+from aes.config import AGENT_DIR, BOM_FILE, DECISIONS_DIR
 from aes.i18n import t
 from aes.registry import (
     fetch_index,
@@ -205,6 +205,47 @@ def _inspect_local(path: str) -> None:
         console.print(table)
         console.print()
 
+    # Models
+    models = manifest.get("models", [])
+    if models:
+        console.print(f"[bold]{t('inspect.models_section')}[/]")
+        for m in models:
+            purpose = m.get("purpose", "")
+            purpose_str = f" [dim]({purpose})[/]" if purpose else ""
+            console.print(f"  {m.get('name', '?')} — {m.get('provider', '?')}{purpose_str}")
+        console.print()
+
+    # Provenance
+    provenance = manifest.get("provenance", {})
+    if provenance:
+        console.print(f"[bold]{t('inspect.provenance_section')}[/]")
+        if provenance.get("created_by"):
+            console.print(f"  {t('inspect.provenance_created_by', value=provenance['created_by'])}")
+        if provenance.get("source"):
+            console.print(f"  {t('inspect.provenance_source', value=provenance['source'])}")
+        console.print()
+
+    # BOM summary
+    bom_path = agent_dir / BOM_FILE
+    if bom_path.exists():
+        bom = _load_yaml(bom_path)
+        n_models = len(bom.get("models", []))
+        n_frameworks = len(bom.get("frameworks", []))
+        n_tools = len(bom.get("tools", []))
+        n_data = len(bom.get("data_sources", []))
+        console.print(f"[bold]{t('inspect.bom_section')}[/]")
+        console.print(f"  {t('inspect.bom_summary', models=n_models, frameworks=n_frameworks, tools=n_tools, data=n_data)}")
+        console.print()
+
+    # Decision records count
+    decisions_dir = agent_dir / DECISIONS_DIR
+    if decisions_dir.exists() and decisions_dir.is_dir():
+        dr_count = len(list(decisions_dir.glob("*.yaml")))
+        if dr_count > 0:
+            console.print(f"[bold]{t('inspect.decisions_section')}[/]")
+            console.print(f"  {t('inspect.decisions_count', count=dr_count)}")
+            console.print()
+
     # Summary
     console.print(f"[bold]{t('inspect.summary')}[/]")
     console.print(f"  {t('inspect.skills_count', count=len(skills))}")

{aes_cli-0.4.3 → aes_cli-0.6.0}/aes/commands/publish.py

@@ -3,6 +3,8 @@
 from __future__ import annotations
 
 import fnmatch
+import hashlib
+import json
 import shutil
 import sys
 import tarfile
@@ -14,7 +16,7 @@ import click
 import yaml
 from rich.console import Console
 
-from aes.config import AGENT_DIR, MANIFEST_FILE
+from aes.config import AGENT_DIR, BOM_FILE, MANIFEST_FILE
 from aes.i18n import t
 
 console = Console()
@@ -171,6 +173,53 @@ def _validate_before_publish(project_root: Path) -> bool:
     return True
 
 
+def _media_type_for_path(path: str) -> str:
+    """Return a media type string for a file path."""
+    if path.endswith(".yaml") or path.endswith(".yml"):
+        return "application/vnd.aes.agent-config.v1+yaml"
+    if path.endswith(".md"):
+        return "text/markdown"
+    if path.endswith(".json"):
+        return "application/json"
+    return "application/octet-stream"
+
+
+def _build_package_manifest(
+    name: str,
+    version: str,
+    aes_version: str,
+    pkg_type: str,
+    files: List[tuple],
+) -> dict:
+    """Build an aes-manifest.json dict.
+
+    *files* is a list of ``(arcname, file_path)`` tuples.
+    """
+    layers = []
+    for arcname, file_path in files:
+        data = file_path.read_bytes()
+        digest = hashlib.sha256(data).hexdigest()
+        layers.append({
+            "mediaType": _media_type_for_path(arcname),
+            "digest": f"sha256:{digest}",
+            "size": len(data),
+            "path": arcname,
+        })
+
+    return {
+        "schemaVersion": 1,
+        "mediaType": "application/vnd.aes.package.v1+tar+gzip",
+        "config": {
+            "name": name,
+            "version": version,
+            "type": pkg_type,
+            "aes": aes_version,
+        },
+        "layers": layers,
+        "signature": None,
+    }
+
+
 def _publish_template_dir(
     project_root: Path,
     output_dir: Path,
@@ -194,6 +243,7 @@ def _publish_template_dir(
 
     name = manifest.get("name", "unknown")
     version = manifest.get("version", "0.0.0")
+    aes_version = manifest.get("aes", "1.0")
 
     # Build exclusion list
     if include_all:
@@ -208,17 +258,35 @@ def _publish_template_dir(
     tarball_name = f"{name}-{version}.tar.gz"
     tarball_path = output_dir / tarball_name
 
+    # Collect files and build package manifest
+    included_files: List[tuple] = []  # (rel_path_in_agent, absolute_path)
+
+    for file_path in sorted(agent_dir.rglob("*")):
+        if not file_path.is_file():
+            continue
+        rel = file_path.relative_to(agent_dir)
+        rel_str = str(rel)
+        if _is_excluded(rel_str, excludes):
+            continue
+        included_files.append((f"{AGENT_DIR}/{rel_str}", file_path))
+
+    # Generate aes-manifest.json
+    pkg_manifest = _build_package_manifest(
+        name, version, aes_version, "template", included_files,
+    )
+
     with tarfile.open(tarball_path, "w:gz") as tar:
-        for file_path in sorted(agent_dir.rglob("*")):
-            if not file_path.is_file():
-                continue
-            rel = file_path.relative_to(agent_dir)
-            rel_str = str(rel)
-            if _is_excluded(rel_str, excludes):
-                continue
-            arcname = f"{name}/{AGENT_DIR}/{rel_str}"
+        for rel_path, file_path in included_files:
+            arcname = f"{name}/{rel_path}"
             tar.add(file_path, arcname=arcname)
 
+        # Add aes-manifest.json at the package root
+        manifest_json = json.dumps(pkg_manifest, indent=2).encode()
+        import io
+        info = tarfile.TarInfo(name=f"{name}/aes-manifest.json")
+        info.size = len(manifest_json)
+        tar.addfile(info, io.BytesIO(manifest_json))
+
     return tarball_path
 
 

aes_cli-0.6.0/aes/commands/upgrade.py

@@ -0,0 +1,282 @@
+"""aes upgrade — Upgrade .agent/ to the current AES spec version."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from pathlib import Path
+from typing import List, Optional
+
+import click
+import yaml
+from jinja2 import ChoiceLoader, Environment, FileSystemLoader
+from rich.console import Console
+
+from aes.config import AGENT_DIR, COMMANDS_DIR, MANIFEST_FILE, SCAFFOLD_DIR
+from aes.i18n import t
+from aes.migrations import (
+    CURRENT_SPEC_VERSION,
+    Migration,
+    MigrationFile,
+    applicable_migrations,
+)
+
+console = Console()
+
+
+# ---------------------------------------------------------------------------
+# Upgrade plan data structures
+# ---------------------------------------------------------------------------
+
+
+@dataclass
+class PlannedFile:
+    """A file to create during upgrade."""
+
+    migration_file: MigrationFile
+    create_file: bool  # whether the file needs creating
+    add_manifest_entry: bool  # whether the manifest entry needs adding
+
+
+@dataclass
+class UpgradePlan:
+    """The full upgrade plan."""
+
+    current_version: str
+    target_version: str
+    migrations: List[Migration] = field(default_factory=list)
+    planned_files: List[PlannedFile] = field(default_factory=list)
+
+    @property
+    def has_changes(self) -> bool:
+        return bool(self.planned_files) or self.current_version != self.target_version
+
+
+# ---------------------------------------------------------------------------
+# Plan computation
+# ---------------------------------------------------------------------------
+
+
+def _compute_plan(agent_dir: Path, manifest: dict) -> UpgradePlan:
+    """Compare current .agent/ against migrations to determine needed changes."""
+    current_version = manifest.get("aes", "1.0")
+    migrations = applicable_migrations(current_version)
+
+    plan = UpgradePlan(
+        current_version=current_version,
+        target_version=CURRENT_SPEC_VERSION,
+        migrations=migrations,
+    )
+
+    for migration in migrations:
+        for mf in migration.files:
+            file_path = agent_dir / mf.relative_path
+            file_exists = file_path.exists()
+
+            entry_exists = False
+            if mf.manifest_entry and mf.manifest_section:
+                entry_id = mf.manifest_entry.get("id", "")
+                section = manifest.get(mf.manifest_section, [])
+                entry_exists = any(
+                    item.get("id") == entry_id for item in section
+                )
+
+            needs_file = not file_exists
+            needs_entry = bool(mf.manifest_entry) and not entry_exists
+
+            if needs_file or needs_entry:
+                plan.planned_files.append(
+                    PlannedFile(
+                        migration_file=mf,
+                        create_file=needs_file,
+                        add_manifest_entry=needs_entry,
+                    )
+                )
+
+    return plan
+
+
+# ---------------------------------------------------------------------------
+# Plan display
+# ---------------------------------------------------------------------------
+
+
+def _display_plan(plan: UpgradePlan) -> None:
+    """Display the upgrade plan without applying it."""
+    console.print()
+    console.print(f"[bold]{t('upgrade.title')}[/]")
+    console.print()
+    console.print(
+        f"  {t('upgrade.current_version', version=plan.current_version)}"
+    )
+    console.print(
+        f"  {t('upgrade.target_version', version=plan.target_version)}"
+    )
+    console.print()
+
+    for migration in plan.migrations:
+        console.print(
+            f"  [bold cyan]{t('upgrade.migration_header', from_v=migration.from_version, to_v=migration.to_version, description=migration.description)}[/]"
+        )
+
+    for pf in plan.planned_files:
+        mf = pf.migration_file
+        if pf.create_file:
+            console.print(
+                f"    [green]+[/] {mf.relative_path}  [dim]({t('upgrade.new_file')})[/]"
+            )
+        if pf.add_manifest_entry and mf.manifest_entry:
+            entry_id = mf.manifest_entry.get("id", "?")
+            console.print(
+                f"    [green]+[/] agent.yaml: {mf.manifest_section}[]  [dim]({t('upgrade.add_entry', id=entry_id)})[/]"
+            )
+
+    if plan.current_version != plan.target_version:
+        console.print(
+            f"    [yellow]~[/] agent.yaml: aes  [dim]({t('upgrade.bump_version', old=plan.current_version, new=plan.target_version)})[/]"
+        )
+
+    console.print()
+    console.print(f"  [dim]{t('upgrade.run_apply')}[/]")
+
+
+# ---------------------------------------------------------------------------
+# Apply logic
+# ---------------------------------------------------------------------------
+
+
+def _apply_plan(
+    agent_dir: Path,
+    manifest: dict,
+    plan: UpgradePlan,
+    env: Environment,
+    context: dict,
+) -> int:
+    """Apply the upgrade plan. Returns number of files created."""
+    created = 0
+
+    for pf in plan.planned_files:
+        mf = pf.migration_file
+
+        # Create missing file from template
+        if pf.create_file:
+            tmpl = env.get_template(mf.template_name)
+            content = tmpl.render(**context)
+            file_path = agent_dir / mf.relative_path
+            file_path.parent.mkdir(parents=True, exist_ok=True)
+            file_path.write_text(content)
+            console.print(
+                f"  [green]+[/] {t('upgrade.created_file', path=mf.relative_path)}"
+            )
+            created += 1
+
+        # Add manifest entry
+        if pf.add_manifest_entry and mf.manifest_entry:
+            section = manifest.setdefault(mf.manifest_section, [])
+            section.append(dict(mf.manifest_entry))
+
+    # Bump aes version
+    if plan.current_version != plan.target_version:
+        manifest["aes"] = plan.target_version
+
+    # Write updated agent.yaml
+    yaml_path = agent_dir / MANIFEST_FILE
+    with open(yaml_path, "w") as f:
+        yaml.safe_dump(
+            manifest,
+            f,
+            default_flow_style=False,
+            sort_keys=False,
+            allow_unicode=True,
+        )
+    console.print(
+        f"  [yellow]~[/] {t('upgrade.updated_manifest', old=plan.current_version, new=plan.target_version)}"
+    )
+
+    return created
+
+
+# ---------------------------------------------------------------------------
+# CLI command
+# ---------------------------------------------------------------------------
+
+
+@click.command("upgrade")
+@click.argument("path", default=".", type=click.Path(exists=True))
+@click.option("--apply", "do_apply", is_flag=True, help="Apply the upgrade (default: dry-run).")
+def upgrade_cmd(path: str, do_apply: bool) -> None:
+    """Upgrade .agent/ to the current AES spec version.
+
+    Adds missing files and manifest entries introduced in newer versions
+    without touching existing customizations.
+
+    Without --apply, shows what would change (dry-run).
+
+    \b
+    Examples:
+      aes upgrade                    # show upgrade plan
+      aes upgrade --apply            # apply and auto-sync
+      aes upgrade /path/to/project   # target a specific project
+    """
+    project_root = Path(path).resolve()
+    agent_dir = project_root / AGENT_DIR
+
+    if not agent_dir.exists() or not (agent_dir / MANIFEST_FILE).exists():
+        console.print(
+            f"[red]{t('common.error')}:[/] {t('upgrade.no_agent_dir')}"
+        )
+        console.print(f"[dim]{t('common.run_init_hint')}[/]")
+        raise SystemExit(1)
+
+    # Load manifest
+    with open(agent_dir / MANIFEST_FILE) as f:
+        manifest = yaml.safe_load(f) or {}
+
+    # Compute plan
+    plan = _compute_plan(agent_dir, manifest)
+
+    if not plan.has_changes:
+        console.print(
+            f"\n  {t('upgrade.up_to_date', version=plan.target_version)}\n"
+        )
+        return
+
+    if not do_apply:
+        _display_plan(plan)
+        return
+
+    # Build Jinja2 environment (locale-aware, same as init.py)
+    from aes.i18n import get_current_locale
+
+    locale = get_current_locale()
+    loaders = []
+    if locale != "en":
+        locale_dir = SCAFFOLD_DIR / locale
+        if locale_dir.exists():
+            loaders.append(FileSystemLoader(str(locale_dir)))
+    loaders.append(FileSystemLoader(str(SCAFFOLD_DIR)))
+
+    env = Environment(
+        loader=ChoiceLoader(loaders),
+        keep_trailing_newline=True,
+    )
+
+    # Template context — minimal, just what the templates need
+    context = {
+        "name": manifest.get("name", "project"),
+        "domain": manifest.get("domain", "other"),
+        "language": manifest.get("runtime", {}).get("language", "other"),
+    }
+
+    console.print()
+    created = _apply_plan(agent_dir, manifest, plan, env, context)
+
+    # Auto-sync
+    from aes.commands.sync import run_sync
+
+    synced = run_sync(project_root, force=True, quiet=True)
+    if synced > 0:
+        console.print(f"  [green]{t('upgrade.synced', count=synced)}[/]")
+
+    console.print(
+        f"\n  [bold green]{t('upgrade.applied', count=created)}[/]\n"
+    )

{aes_cli-0.4.3 → aes_cli-0.6.0}/aes/config.py

@@ -29,9 +29,11 @@ REGISTRY_DIR = "registry"
 WORKFLOWS_DIR = "workflows"
 COMMANDS_DIR = "commands"
 MEMORY_DIR = "memory"
+DECISIONS_DIR = "memory/decisions"
 OVERRIDES_DIR = "overrides"
 LOCAL_FILE = "local.yaml"
 LOCAL_EXAMPLE_FILE = "local.example.yaml"
+BOM_FILE = "bom.yaml"
 
 # Schema file mapping
 SCHEMA_MAP = {
@@ -40,4 +42,6 @@ SCHEMA_MAP = {
     "workflow": "workflow.schema.json",
     "registry": "registry.schema.json",
     "permissions": "permissions.schema.json",
+    "bom": "bom.schema.json",
+    "decision-record": "decision-record.schema.json",
 }