Update prompts/prompts.py

prompts/prompts.py

@@ -25,3 +25,60 @@ def load_prompts():
     print(merged_prompts)
     
     return {str(k): _to_str(v) for k, v in merged.items()}
+
+
+
+def print_default_prompt_templates(
+    
+    base_url,
+    model_id,
+    api_key=None,
+):
+    import os, tempfile, pathlib, yaml
+    from smolagents import CodeAgent, InferenceClientModel
+
+    api_key = api_key or os.getenv("HF_TOKEN")
+
+    model = InferenceClientModel(
+        base_url=base_url,
+        api_key=api_key,              # or token=api_key
+        model_id=model_id,
+        temperature=0.2,
+        max_tokens=1200,
+    )
+    agent = CodeAgent(tools=[], model=model)
+
+    # Try to print from the in-memory PromptTemplates object
+    try:
+        pt = getattr(agent, "prompt_templates", None)
+        if pt is not None:
+            def pick(obj, fields):
+                return {k: getattr(obj, k) for k in fields if getattr(obj, k, None)}
+            out = {}
+            if getattr(pt, "system_prompt", None):
+                out["system_prompt"] = pt.system_prompt
+            if getattr(pt, "planning", None):
+                out["planning"] = pick(pt.planning, ["plan", "update_plan_pre_messages", "update_plan_post_messages"])
+            if getattr(pt, "managed_agent", None):
+                out["managed_agent"] = pick(pt.managed_agent, ["task", "report"])
+            if getattr(pt, "final_answer", None):
+                out["final_answer"] = pick(pt.final_answer, ["pre_messages", "post_messages"])
+            print(yaml.safe_dump(out, sort_keys=False, allow_unicode=True))
+            return
+    except Exception as e:
+        print(f"[warn] couldn't access agent.prompt_templates directly: {e}")
+
+    # Fallback: export prompt.yaml and print it
+    try:
+        with tempfile.TemporaryDirectory() as td:
+            agent.save(td)  # should write prompt.yaml
+            p = pathlib.Path(td) / "prompt.yaml"
+            if p.exists():
+                print(p.read_text(encoding="utf-8"))
+            else:
+                print("[warn] agent.save() produced no prompt.yaml")
+    except Exception as e:
+        print(f"[error] failed to export defaults: {e}")
+
+
+