ghost-reader 0.1.0__py3-none-any.whl

ghost_reader/.release-version

@@ -0,0 +1 @@
+0.1.0

ghost_reader/__init__.py

@@ -0,0 +1,3 @@
+"""Ghost Reader local-first CLI package."""
+
+__version__ = "0.2.3"

ghost_reader/agent_loader.py

@@ -0,0 +1,64 @@
+"""Load agent YAML definitions for Ghost Reader personas."""
+
+from pathlib import Path
+from typing import Optional
+
+import yaml
+
+
+def discover_agent_personas(home: Path) -> list[str]:
+    """Scan agents/ directory for *.agent.yaml files and extract persona IDs."""
+    agents_dir = home / "agents"
+    persona_ids: list[str] = []
+    if agents_dir.exists():
+        for f in sorted(agents_dir.glob("*.agent.yaml")):
+            try:
+                raw = f.read_text(encoding="utf-8")
+                agent = yaml.safe_load(raw)
+                pid = (agent.get("metadata") or {}).get("persona_id")
+                if pid:
+                    persona_ids.append(pid)
+            except (yaml.YAMLError, OSError, AttributeError):
+                continue
+    return persona_ids or _legacy_personas()
+
+
+def _legacy_personas() -> list[str]:
+    """Fallback to hardcoded persona list when no agent YAML files found."""
+    return ["mara", "dex", "pip", "elena", "rook"]
+
+
+def load_agent_yaml(home: Path, persona_id: str) -> Optional[dict]:
+    """Load an agent YAML for a specific persona ID. Returns parsed dict or None."""
+    agents_dir = home / "agents"
+    if not agents_dir.exists():
+        return None
+    for f in sorted(agents_dir.glob("*.agent.yaml")):
+        try:
+            raw = f.read_text(encoding="utf-8")
+            agent = yaml.safe_load(raw)
+            pid = (agent.get("metadata") or {}).get("persona_id")
+            if pid == persona_id:
+                return dict(agent)
+        except (yaml.YAMLError, OSError, AttributeError):
+            continue
+    return None
+
+
+def load_persona_from_agent(home: Path, persona_id: str) -> Optional[dict]:
+    """Load an agent YAML and extract persona-compatible data.
+    
+    Returns a dict with id, source, model, and system_prompt fields,
+    suitable for use by read_persona() and review generation.
+    """
+    agent = load_agent_yaml(home, persona_id)
+    if not agent:
+        return None
+    return {
+        "id": persona_id,
+        "source": "agent_yaml",
+        "model": agent.get("model"),
+        "system_prompt": agent.get("system"),
+        "tools": agent.get("tools", []),
+        "metadata": agent.get("metadata", {}),
+    }