Initial commit

reflact/prompts/__init__.py

@@ -0,0 +1,63 @@
+"""Prompt loading utilities for ReflACT.
+
+Prompts are stored as ``.md`` files and loaded at runtime:
+
+- **Generic** prompts live in ``reflact/prompts/*.md``
+- **Env-specific** prompts live in ``reflact/envs/<env>/prompts/*.md``
+
+``load_prompt(name, env)`` tries the env-specific path first, then falls
+back to the generic default.
+"""
+from __future__ import annotations
+
+import os
+
+_PROMPTS_DIR = os.path.dirname(os.path.abspath(__file__))
+_REFLACT_DIR = os.path.dirname(_PROMPTS_DIR)
+
+_cache: dict[str, str] = {}
+
+
+def _read_file(path: str) -> str | None:
+    if path in _cache:
+        return _cache[path]
+    if not os.path.isfile(path):
+        return None
+    with open(path, encoding="utf-8") as f:
+        content = f.read()
+    _cache[path] = content
+    return content
+
+
+def load_prompt(name: str, env: str | None = None) -> str:
+    """Load a prompt by name with env-specific override and generic fallback.
+
+    Lookup order:
+      1. ``reflact/envs/{env}/prompts/{name}.md``  (if *env* given)
+      2. ``reflact/prompts/{name}.md``              (generic default)
+
+    Raises ``FileNotFoundError`` if neither path exists.
+    """
+    if env is not None:
+        env_path = os.path.join(_REFLACT_DIR, "envs", env, "prompts", f"{name}.md")
+        content = _read_file(env_path)
+        if content is not None:
+            return content
+
+    generic_path = os.path.join(_PROMPTS_DIR, f"{name}.md")
+    content = _read_file(generic_path)
+    if content is not None:
+        return content
+
+    searched = []
+    if env is not None:
+        searched.append(os.path.join("reflact/envs", env, "prompts", f"{name}.md"))
+    searched.append(f"reflact/prompts/{name}.md")
+    raise FileNotFoundError(
+        f"Prompt '{name}' not found. Searched: {', '.join(searched)}"
+    )
+
+
+def clear_cache() -> None:
+    """Clear the prompt file cache (useful for testing)."""
+    _cache.clear()