Give Cerebras its own inbox

import json, os, requests
from cerebras.cloud.sdk import Cerebras

cerebras = Cerebras(api_key=os.environ["CEREBRAS_API_KEY"])
GORK = "https://api.gork.email/v1"
HEADERS = {"Authorization": f"Bearer {os.environ['GORK_API_KEY']}"}

def run_tool(name, args):
    if name == "gork_create_inbox":
        r = requests.post(f"{GORK}/inboxes", headers=HEADERS,
                          json={"username": args["username"]})
        return r.json()["data"]
    if name == "gork_send_email":
        r = requests.post(f"{GORK}/messages/send", headers=HEADERS, json={
            "inboxId": args["inbox_id"], "to": [args["to"]],
            "subject": args["subject"], "text": args["text"]})
        return r.json()["data"]
    if name == "gork_read_emails":
        r = requests.get(f"{GORK}/messages", headers=HEADERS,
                         params={"q": args["query"]})
        return r.json()["data"]
    raise ValueError(f"Unknown tool: {name}")

messages = [
    {"role": "system", "content": "You are an email agent with Gork Mail tools."},
    {"role": "user", "content": "Give the support agent an email address."},
]

while True:
    resp = cerebras.chat.completions.create(
        model="gpt-oss-120b", messages=messages, tools=EMAIL_TOOLS)
    msg = resp.choices[0].message
    if not msg.tool_calls:
        print("Agent:", msg.content)
        break
    messages.append(msg.model_dump())
    for call in msg.tool_calls:
        output = run_tool(call.function.name, json.loads(call.function.arguments))
        messages.append({"role": "tool", "tool_call_id": call.id,
                         "content": json.dumps(output)})