Local Axiom AI API Documentation

curl -X POST https://local-axiom.com/api/chat \\
  -H "Content-Type: application/json" \\
  -d '{
  "model": "qwen3_4b",
  "system_prompt": "You are a helpful programming assistant",
  "messages": [
    {
      "role": "user",
      "content": "Write a Python web server"
    }
  ]
}'

import requests, json, re, urllib3, os, time

# ----------------------------------------------------
# Configuration
# ----------------------------------------------------
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
API_URL = "https://local-axiom.com/api/chat"
MODEL   = "qwen3_4b"

# ----------------------------------------------------
# System Prompts
# ----------------------------------------------------
DEV_PROMPT = """
You are a Developer Agent. Your goal is to write code and create files to complete a task.
Respond ONLY with a JSON object:
{"action": "create_file", "path": "file.py", "content": "..."}
{"action": "complete", "message": "finished"}
  
If you receive 'CRITIC FEEDBACK', use it to fix your code.
"""

CRITIC_PROMPT = """
You are a Senior Code Reviewer.
Review the code provided by the developer.
1. If the code is perfect and meets all requirements, respond ONLY with: {"status": "PASS"}
2. If there are errors, bugs, or missing features, respond with: {"status": "FAIL", "feedback": "Detailed explanation of what to fix"}
"""

def extract_json(text):
    try:
        match = re.search(r'(\{.*\})', text, re.DOTALL)
        return json.loads(match.group(1)) if match else None
    except:
        return None

def call_axiom(system_prompt, messages, session_id=None):
    payload = {
        "model": MODEL,
        "system_prompt": system_prompt,
        "messages": messages
    }
    headers = {"Content-Type": "application/json"}
    if session_id:
        headers["X-Session-Id"] = session_id
    try:
        response = requests.post(API_URL, json=payload, headers=headers,
                                 verify=False, timeout=60)
        resp_json = response.json()
        # Grab session ID from header if present
        new_session = response.headers.get("X-Session-Id")
        return resp_json.get("message", ""), new_session
    except Exception as e:
        return f"Error: {str(e)}", None

def run_task(task_description):
    dev_messages = [{"role": "user", "content": f"Task: {task_description}"}]
    session_id = None
    print(f"\n?? Starting: {task_description}")

    for step in range(5):  # limit to 5 iterations
        raw_res, session_id = call_axiom(DEV_PROMPT, dev_messages, session_id)
        data = extract_json(raw_res)

        if not data:
            dev_messages.append({"role": "user", "content": "Please only output JSON."})
            continue

        if data.get("action") == "create_file":
            path, content = data.get("path"), data.get("content")
            with open(path, "w") as f:
                f.write(content)
            print(f"???  Dev: Created {path}")

            # Critic review
            print(f"?? Critic: Reviewing {path}...")
            critic_input = [{"role": "user",
                             "content": f"Task was: {task_description}\nCode written:\n{content}"}]
            critic_res, _ = call_axiom(CRITIC_PROMPT, critic_input, session_id)
            review = extract_json(critic_res)

            if review and review.get("status") == "PASS":
                print("? Critic: Code looks good! Task Complete.")
                return
            elif review and review.get("status") == "FAIL":
                feedback = review.get("feedback")
                print(f"? Critic Feedback: {feedback}")
                dev_messages.append({"role": "assistant", "content": raw_res})
                dev_messages.append({"role": "user",
                                     "content": f"CRITIC FEEDBACK: {feedback}. Please fix the code and recreate the file."})
            else:
                print("?? Critic gave invalid response, assuming pass.")
                return
        elif data.get("action") == "complete":
            print(f"? Dev marked task as complete.")
            break

if __name__ == "__main__":
    task = input("Enter task: ")
    if task:
        run_task(task)