Offline-Capable Email Agents with Jan

from openai import OpenAI
import requests
import json

"cm"># Point OpenAI client at Jan's local server
client = OpenAI(
    base_url="http://localhost:1337/v1",
    api_key="jan"  "cm"># Jan doesn't require a real key
)

MULTIMAIL_API = "https://api.multimail.dev/v1"
MM_HEADERS = {"Authorization": "Bearer mm_live_your_api_key"}

email_tools = [
    {
        "type": "function",
        "function": {
            "name": "send_email",
            "description": "Send an email through MultiMail. In gated_send mode, queues for human approval.",
            "parameters": {
                "type": "object",
                "properties": {
                    "mailbox_id": {"type": "string", "description": "Mailbox to send from"},
                    "to": {"type": "string", "description": "Recipient email"},
                    "subject": {"type": "string", "description": "Subject line"},
                    "body": {"type": "string", "description": "Email body"}
                },
                "required": ["mailbox_id", "to", "subject", "body"]
            }
        }
    },
    {
        "type": "function",
        "function": {
            "name": "check_inbox",
            "description": "Check inbox for recent messages.",
            "parameters": {
                "type": "object",
                "properties": {
                    "mailbox_id": {"type": "string", "description": "Mailbox to check"},
                    "limit": {"type": "integer", "description": "Max messages"}
                },
                "required": ["mailbox_id"]
            }
        }
    }
]

def execute_tool(name, args):
    if name == "send_email":
        resp = requests.post(f"{MULTIMAIL_API}/send", headers=MM_HEADERS, json=args)
    elif name == "check_inbox":
        resp = requests.get(
            f"{MULTIMAIL_API}/mailboxes/{args[&"cm">#039;mailbox_id']}/inbox",
            headers=MM_HEADERS, params={"limit": args.get("limit", 10)}
        )
    elif name == "reply_email":
        resp = requests.post(f"{MULTIMAIL_API}/reply", headers=MM_HEADERS, json=args)
    else:
        return {"error": f"Unknown tool: {name}"}
    return resp.json()

def run_email_agent(user_message, mailbox_id):
    messages = [
        {"role": "system", "content": f"You are an email assistant for mailbox {mailbox_id}. "
         f"Emails use gated_send mode and queue for human approval."},
        {"role": "user", "content": user_message}
    ]
    while True:
        response = client.chat.completions.create(
            model="llama-3.3-70b",  # Use your loaded model name
            messages=messages,
            tools=email_tools
        )
        msg = response.choices[0].message
        if msg.tool_calls:
            messages.append(msg)
            for tc in msg.tool_calls:
                result = execute_tool(
                    tc.function.name,
                    json.loads(tc.function.arguments)
                )
                messages.append({
                    "role": "tool", "tool_call_id": tc.id,
                    "content": json.dumps(result)
                })
        else:
            return msg.content

print(run_email_agent("Check my inbox", "mbx_abc123"))

"cm">// Jan extension structure for MultiMail integration
"cm">// Place in ~/.jan/extensions/multimail/

const MULTIMAIL_API = 'https://api.multimail.dev/v1';

interface EmailTool {
  name: string;
  execute: (args: Record<string, unknown>) => Promise<unknown>;
}

const emailTools: EmailTool[] = [
  {
    name: 'send_email',
    execute: async (args) => {
      const resp = await fetch(`${MULTIMAIL_API}/send`, {
        method: 'POST',
        headers: {
          'Authorization': `Bearer ${getApiKey()}`,
          'Content-Type': 'application/json'
        },
        body: JSON.stringify(args)
      });
      return resp.json();
    }
  },
  {
    name: 'check_inbox',
    execute: async (args) => {
      const mailboxId = args.mailbox_id as string;
      const resp = await fetch(
        `${MULTIMAIL_API}/mailboxes/${mailboxId}/inbox`,
        { headers: { 'Authorization': `Bearer ${getApiKey()}` } }
      );
      return resp.json();
    }
  }
];

function getApiKey(): string {
  "cm">// Read from Jan's secure settings
  return process.env.MULTIMAIL_API_KEY || '';
}