Update prompts/prompts.py

prompts/prompts.py

@@ -1,154 +1,14 @@
 import yaml
-import io, contextlib
 
 def load_prompts():
-    files = ["prompts/base.yml", "prompts/contenteval.yml",  "prompts/planning.yml"]
-    merged = {}
-    for path in files:
-        with open(path, "r", encoding="utf-8") as f:
-            data = yaml.safe_load(f) or {}
-            if isinstance(data, dict):
-                merged.update(data)
-
-    def _to_str(v):
-        if v is None:
-            return ""                         # avoid literal "None"
-        if isinstance(v, str):
-            return v
-        if isinstance(v, dict) and "content" in v:
-            return str(v["content"])          # flatten chat-style entries
-        if isinstance(v, (list, tuple)):
-            return "\n".join(map(str, v))     # readable multiline
-        return str(v)
-
-    merged_prompts =  {str(k): _to_str(v) for k, v in merged.items()}
-    
-    #print(merged_prompts)
-    
-    return {str(k): _to_str(v) for k, v in merged.items()}
-
-
-
-def print_default_prompt_templates(agent):
-    import tempfile, pathlib, yaml, json, inspect
-    
-    from dataclasses import is_dataclass, asdict
-
-    with tempfile.TemporaryDirectory() as td:
-        agent.save(td)
-        for fname in ("prompt.yaml","prompts.yaml","prompt.yml","prompts.yml"):
-            p = pathlib.Path(td)/fname
-            if p.exists():
-                print(f"\n== {fname} ==\n")
-                print(p.read_text(encoding="utf-8"))
-                break
-    
-    pass
-    
-    def safe_get(obj, attr, default=None):
-        try:
-            return getattr(obj, attr)
-        except Exception:
-            return default
-
-    def pick(obj, fields):
-        out = {}
-        for k in fields:
-            v = safe_get(obj, k, None)
-            # keep falsy-but-valid (e.g., ""), just skip None
-            if v is not None:
-                out[k] = v
-        return out
-
-    def normalize_for_yaml(x):
-        # Strings, lists, dicts are safe; other objects → string repr
-        if isinstance(x, (str, int, float, bool)) or x is None:
-            return x
-        if isinstance(x, (list, tuple)):
-            return [normalize_for_yaml(v) for v in x]
-        if isinstance(x, dict):
-            return {str(k): normalize_for_yaml(v) for k, v in x.items()}
-        # Dataclasses emitted upstream should’ve been dict-ified already
-        return str(x)
-
-    print("Trying to print the in-memory PromptTemplates object")
-    print("agent type:", type(agent))
-    print("has prompt_templates:", hasattr(agent, "prompt_templates"))
-
-
-
-
+    system_prompt_override = "prompts/code_agent.yml"
     
     try:
-        pt = safe_get(agent, "prompt_templates", None)
-        print("prompt_templates object:", pt)
-        if pt is None:
-            raise AttributeError("agent.prompt_templates is None or missing")
-
-        # If it’s a dataclass, get a dict in one shot
-        if is_dataclass(pt):
-            out = asdict(pt)
-        else:
-            out = {}
-            sp = safe_get(pt, "system_prompt", None)
-            if sp is not None:
-                out["system_prompt"] = sp
-
-            planning = safe_get(pt, "planning", None)
-            if planning is not None:
-                out["planning"] = pick(planning, [
-                    "plan",
-                    "update_plan_pre_messages",
-                    "update_plan_post_messages",
-                ])
-
-            managed = safe_get(pt, "managed_agent", None)
-            if managed is not None:
-                out["managed_agent"] = pick(managed, ["task", "report"])
-
-            final_answer = safe_get(pt, "final_answer", None)
-            if final_answer is not None:
-                out["final_answer"] = pick(final_answer, ["pre_messages", "post_messages"])
-
-        # Normalize any message objects to YAML-safe structures
-        out = normalize_for_yaml(out)
-
-        # Pretty print quick peeks (safe)
-        for path in [
-            ("system_prompt",),
-            ("planning", "plan"),
-            ("planning", "update_plan_pre_messages"),
-            ("planning", "update_plan_post_messages"),
-            ("final_answer", "pre_messages"),
-            ("final_answer", "post_messages"),
-        ]:
-            cur = out
-            for key in path:
-                if isinstance(cur, dict) and key in cur:
-                    cur = cur[key]
-                else:
-                    cur = None
-                    break
-            print(".".join(path)+":", (cur if cur is not None else "<absent>"))
-
-        print("\n---- prompt templates (YAML) ----")
-        print(yaml.safe_dump(out, sort_keys=False, allow_unicode=True))
-        return
-
+        with open(system_prompt_override, 'r', encoding='utf-8') as file:
+            return file.read()
+    except FileNotFoundError:
+        raise FileNotFoundError(f"Prompt file '{system_prompt_override}' not found.")
     except Exception as e:
-        print(f"[warn] couldn't access agent.prompt_templates directly: {e}")
+        raise Exception(f"Failed to read prompt file: {e}")
+
 
-    # Fallback: export prompt file and print it
-    try:
-        with tempfile.TemporaryDirectory() as td:
-            agent.save(td)  # some builds write prompt.yaml (or prompts.yaml)
-            # try common filenames
-            for fname in ("prompt.yaml", "prompts.yaml", "prompt.yml", "prompts.yml"):
-                p = pathlib.Path(td) / fname
-                if p.exists():
-                    print(f"\n---- exported {fname} ----")
-                    print(p.read_text(encoding="utf-8"))
-                    return
-            print("[warn] agent.save() produced no prompt file I recognize")
-    except Exception as e:
-        print(f"[error] failed to export defaults: {e}")