The Database for Agentic Workloads

import fetch from "node-fetch";
const BASE = "https://serverless.tidbapi.com/v1beta1";
async function createBranch(clusterId, branchName, publicKey, privateKey) {
  const res = await fetch(`${BASE}/clusters/${clusterId}/branches`, {
    method: "POST",
    headers: {
      "Content-Type": "application/json",
      Authorization:
        "Basic " + Buffer.from(`${publicKey}:${privateKey}`).toString("base64"),
    },
    body: JSON.stringify({ displayName: branchName }),
  });
  if (!res.ok) throw new Error(`Create branch failed: ${res.status}`);
  const { branchId } = await res.json();
  return branchId;
}
// Example usage
createBranch(
  "1234567890",
  "new-feature-branch",
  process.env.TIDB_CLOUD_PUBLIC_KEY,
  process.env.TIDB_CLOUD_PRIVATE_KEY
)
  .then((id) => console.log("Branch created:", id))
  .catch(console.error);

import { TiDBClient } from "tidb-serverless";

const db = new TiDBClient({ url: process.env.TIDB_URL });

async function saveAgentState(agentId, state) {
  await db.execute(
    `INSERT INTO agent_memory (agent_id, state, updated_at)
     VALUES (?, ?, NOW())
     ON DUPLICATE KEY UPDATE state = VALUES(state), updated_at = NOW()`,
    [agentId, JSON.stringify(state)]
  );
}

async function restoreAgentState(agentId) {
  const [row] = await db.execute(
    `SELECT state FROM agent_memory WHERE agent_id = ? LIMIT 1`,
    [agentId]
  );
  return row ? JSON.parse(row.state) : null;
}

// Example usage
const state = await restoreAgentState("agent-42");
await saveAgentState("agent-42", {
  ...state,
  step: "summarize",
  outputs: ["draft-1.md"],
  tokensUsed: 4210,
});

import { TiDBClient } from "tidb-serverless";

const db = new TiDBClient({ url: process.env.TIDB_URL });

async function assignTask(taskId, agentId) {
  await db.execute("BEGIN");
  try {
    const [task] = await db.execute(
      `SELECT status FROM tasks WHERE id = ? FOR UPDATE`,
      [taskId]
    );
    if (task.status !== "pending") throw new Error("Task already claimed");

    await db.execute(
      `UPDATE tasks SET status = "running", agent_id = ? WHERE id = ?`,
      [agentId, taskId]
    );
    await db.execute("COMMIT");
  } catch (err) {
    await db.execute("ROLLBACK");
    throw err;
  }
}

// Two agents racing for the same task — only one wins
await Promise.all([
  assignTask("task-99", "agent-A"),
  assignTask("task-99", "agent-B"),
]);