Agents and tools

import { privateKeyToAccount } from "viem/accounts";

const BASE = "https://useantimatter.com";
const wallet = privateKeyToAccount(process.env.WALLET_PRIVATE_KEY);

// 1. The exact sign in message for this wallet.
const { message } = await fetch(`${BASE}/api/auth/nonce?address=${wallet.address}`).then((r) =>
  r.json(),
);

// 2. Sign it and exchange the signature for a session.
const signature = await wallet.signMessage({ message });
const verify = await fetch(`${BASE}/api/auth/verify`, {
  method: "POST",
  headers: { "content-type": "application/json" },
  body: JSON.stringify({ message, signature }),
});
const cookie = verify.headers
  .getSetCookie()
  .map((c) => c.split(";")[0])
  .join("; ");

// 3. Make a key with the session. It is shown once, so save it.
const minted = await fetch(`${BASE}/api/keys`, {
  method: "POST",
  headers: { "content-type": "application/json", cookie },
  body: JSON.stringify({ name: "my-agent" }),
}).then((r) => r.json());
console.log(minted.key); // save it as ANTIMATTER_API_KEY

import { randomBytes } from "node:crypto";
import { privateKeyToAccount } from "viem/accounts";

const BASE = "https://useantimatter.com";
const wallet = privateKeyToAccount(process.env.WALLET_PRIVATE_KEY); // holds USDC
const b64 = (v) => Buffer.from(JSON.stringify(v)).toString("base64");
const unb64 = (s) => JSON.parse(Buffer.from(s, "base64").toString());

const url = `${BASE}/v1/chat/completions`;
const body = JSON.stringify({
  model: "inclusionai/ling-3.0-tiny",
  max_tokens: 200,
  messages: [{ role: "user", content: "Hello" }],
});
const headers = { "content-type": "application/json" };

// 1. Ask without a key. The answer is 402 with the price on each network.
let res = await fetch(url, { method: "POST", headers, body });
if (res.status === 402) {
  const required = unb64(res.headers.get("payment-required"));
  const accepted = required.accepts[0]; // USDC on the first accepted network
  const chainId = Number(accepted.network.split(":")[1]);
  const now = Math.floor(Date.now() / 1000);
  const authorization = {
    from: wallet.address,
    to: accepted.payTo,
    value: accepted.amount,
    validAfter: String(now - 60),
    validBefore: String(now + accepted.maxTimeoutSeconds),
    nonce: `0x${randomBytes(32).toString("hex")}`,
  };

  // 2. Sign an EIP-3009 transfer of exactly that amount. Nothing moves yet.
  const signature = await wallet.signTypedData({
    domain: {
      name: accepted.extra.name,
      version: accepted.extra.version,
      chainId,
      verifyingContract: accepted.asset,
    },
    types: {
      TransferWithAuthorization: [
        { name: "from", type: "address" },
        { name: "to", type: "address" },
        { name: "value", type: "uint256" },
        { name: "validAfter", type: "uint256" },
        { name: "validBefore", type: "uint256" },
        { name: "nonce", type: "bytes32" },
      ],
    },
    primaryType: "TransferWithAuthorization",
    message: {
      ...authorization,
      value: BigInt(authorization.value),
      validAfter: BigInt(authorization.validAfter),
      validBefore: BigInt(authorization.validBefore),
    },
  });

  // 3. Send the same request again with the payment attached.
  const payment = { x402Version: 2, resource: required.resource, accepted, payload: { signature, authorization } };
  res = await fetch(url, {
    method: "POST",
    headers: { ...headers, "payment-signature": b64(payment) },
    body,
  });
}
console.log(res.status, (await res.json()).choices?.[0]?.message?.content);
console.log(unb64(res.headers.get("payment-response") ?? "e30=")); // the settlement

const BASE = "https://useantimatter.com";
const KEY = process.env.ANTIMATTER_API_KEY; // a key with the agents scope
const call = (path, body) =>
  fetch(`${BASE}${path}`, {
    method: body ? "POST" : "GET",
    headers: { authorization: `Bearer ${KEY}`, "content-type": "application/json" },
    body: body ? JSON.stringify(body) : undefined,
  }).then((r) => r.json());

// 1. An agent with $5 moved from this account, a $1 daily cap and auto refill.
const agent = await call("/v1/agents", {
  name: "researcher",
  fund_micros: 5_000_000,
  models: ["inclusionai/ling-3.0-tiny", "antimatter/auto"],
  max_cost_per_request_micros: 50_000,
  daily_cap_micros: 1_000_000,
  auto_refill: { amount_micros: 1_000_000, daily_max_micros: 2_000_000 },
});
console.log(agent.api_key); // the agent's own am_live_ key, shown once

// 2. The agent uses its key like any client; it spends its balance and nothing else.
// 3. See what it spent, then move what is left back.
const seen = await call(`/v1/agents/${agent.id}`);
console.log(seen.balance_micros, seen.spend_by_model);
await call(`/v1/agents/${agent.id}/drain`, {});