agent-godmode

import osfrom pathlib import Path
# OPENAI_API_KEY is required only for steps that call Chat Completions (LLM + agent loops).# Workspace + direct edit_file work without a key.# Set via OS env or e.g. %env OPENAI_API_KEY sk-... in IPython# Local-only optional override — never commit a real key:# os.environ["OPENAI_API_KEY"] = "sk-..."
from openai import OpenAI
from agent_godmode import (    AgentWorkspace,    OPENAI_TOOL_DEFINITIONS,    SYSTEM_PROMPT_V1,    run_agent_loop,)
HAS_OPENAI_KEY = bool(os.environ.get("OPENAI_API_KEY"))client = OpenAI() if HAS_OPENAI_KEY else NoneMODEL = "gpt-4o-mini"
if not HAS_OPENAI_KEY:    print(        "Note: OPENAI_API_KEY not set — Chat Completions examples will raise until you set it. "        "Workspace + direct edit_file still work."    )

# Fixed workspace — all reads/writes/commands stay under this folderWORK_DIR = Path(r"D:\Avi-assign")WORK_DIR.mkdir(parents=True, exist_ok=True)print("Workspace:", WORK_DIR.resolve())
hello = WORK_DIR / "hello.txt"if not hello.exists():    hello.write_text("Hello from Avi-assign workspace.\n", encoding="utf-8")
ws = AgentWorkspace(WORK_DIR)print(ws.read_file("hello.txt"))print("--- list_files ---")print(ws.list_files(".", recursive=False))

if client is None:    raise ValueError(        "Set OPENAI_API_KEY to run this block (e.g. export OPENAI_API_KEY=... or %env in IPython). "        "Skip if you only want workspace / edit_file demos."    )
# 1) Context from disk (read-only)context = ws.read_file("hello.txt")
# 2) Ask the model to author the entire new file; no static template for the bodyuser_prompt = (    "Here is the current contents of hello.txt in my workspace:\n\n"    f"---\n{context}\n---\n\n"    "Write ONLY the body of a new Markdown file (no preamble, no code fences) "    "with a title line and two bullet points explaining what this greeting is for.")
resp = client.chat.completions.create(    model=MODEL,    messages=[        {            "role": "system",            "content": "You output only the file body the user asked for. No extra commentary.",        },        {"role": "user", "content": user_prompt},    ],)
generated = (resp.choices[0].message.content or "").strip()if not generated:    raise RuntimeError("LLM returned empty content; nothing to write.")
# 3) Persist exactly what the LLM producedout_rel = "llm_generated_notes.md"ws.write_file(out_rel, generated, mode="overwrite")print(f"Wrote {out_rel!r} ({len(generated)} chars from model)\n")print(ws.read_file(out_rel))

# Direct edit_file (no Chat Completions call).# If `ws` is not defined yet (e.g. you skipped §3), the next few lines create it (same WORK_DIR).from pathlib import Path
from agent_godmode import AgentWorkspace
if "ws" not in globals():    WORK_DIR = Path(r"D:\Avi-assign")    WORK_DIR.mkdir(parents=True, exist_ok=True)    ws = AgentWorkspace(WORK_DIR)
demo_edit = "edit_demo.txt"ws.write_file(    demo_edit,    "version: 1\nstatus: draft\nfooter: end\n",    mode="overwrite",)print("--- before ---")print(ws.read_file(demo_edit), end="")print(ws.edit_file(demo_edit, old_string="status: draft", new_string="status: ready"))print("--- after ---")print(ws.read_file(demo_edit), end="")

if client is None:    raise ValueError(        "Set OPENAI_API_KEY to run this block. "        "Skip if you only need workspace or direct edit_file."    )

def complete(messages, tools):    """One Chat Completions turn; return OpenAI-shaped dict for run_agent_loop."""    resp = client.chat.completions.create(        model=MODEL,        messages=messages,        tools=tools,        tool_choice="auto",    )    return resp.model_dump()

answer = run_agent_loop(    complete,    "Use tools only. List the workspace root, read hello.txt, then call write_file on "    "agent_notes.txt. The `content` argument must be your own freshly written summary "    "(several sentences) based only on what you read—do not paste boilerplate.",    ws,    system_prompt=SYSTEM_PROMPT_V1,    max_turns=12,)print("--- final answer ---")print(answer)print("--- agent_notes.txt (if created by tool write_file) ---")p = WORK_DIR / "agent_notes.txt"print(p.read_text(encoding="utf-8") if p.exists() else "(missing)")

from pathlib import Path
from agent_godmode import AgentWorkspace
if "ws" not in globals():    WORK_DIR = Path(r"D:\Avi-assign")    WORK_DIR.mkdir(parents=True, exist_ok=True)    ws = AgentWorkspace(WORK_DIR)if "complete" not in globals():    raise NameError("Define `complete` in §6 (after imports) before running this block.")if client is None:    raise ValueError(        "Set OPENAI_API_KEY to run this block. "        "The direct edit_file example in §5 works without a key."    )
target = "edit_agent_target.txt"ws.write_file(    target,    "# Demo\nThere are three erorrs in this sentance.\n",    mode="overwrite",)edit_answer = run_agent_loop(    complete,    (        f"Use tools only. Read `{target}`. Then use edit_file (not write_file) to fix typos: "        "change erorrs to errors and sentance to sentence. "        "Copy old_string exactly from read_file; use two edit_file calls or replace_all where appropriate."    ),    ws,    system_prompt=SYSTEM_PROMPT_V1,    max_turns=14,)print("--- agent (edit_file) answer ---")print(edit_answer)print("--- file after agent ---")print(ws.read_file(target), end="")

ws = AgentWorkspace(    r"D:\Avi-assign",    allowed_commands=frozenset({"python", "uv"}),    command_timeout_sec=60.0,)

from agent_godmode import WorkspaceTools, OPENAI_TOOL_DEFINITIONS, SYSTEM_PROMPT_V1from agent_godmode.config import config_from_root
tools = WorkspaceTools(config_from_root(r"D:\Avi-assign"))print(tools.read_file("hello.txt"))

final_system = SYSTEM_PROMPT_V1 + "\n\n" + "Your org rules here."