Client SDK

import { createClient } from "rivetkit/client";
import type { registry } from "./server";

const client = createClient<typeof registry>();
const agent = client.agent.getOrCreate(["support", "customer-123"]);

const root = await agent.harness.root();
const { messages } = await agent.conversation.context(root.id);
console.log(`${messages.length} messages so far`);

const conn = agent.connect();
conn.onStatusChange((status) => console.log(`[connection ${status}]`));

// The text of the current answer printed so far.
let printed = "";

const unsubscribe = conn.on("pi.events", ({ events }) => {
	for (const event of events) {
		switch (event.type) {
			case "message_update":
				for (const change of event.changes) {
					if (change.type === "text_delta") {
						process.stdout.write(change.delta);
						printed += change.delta;
					}
				}
				break;
			case "message_end": {
				// A short answer can arrive whole here, with no text_delta before it.
				const message = event.entry.model?.[0];
				if (message?.role === "assistant") {
					const text = message.content
						.flatMap((block) => (block.type === "text" ? [block.text] : []))
						.join("");
					process.stdout.write(text.slice(printed.length));
				}
				printed = "";
				break;
			}
			case "tool_execution_start":
				console.log(`\n> ${event.toolName}`);
				break;
			case "auto_retry_start":
				console.log(`\nRetrying (attempt ${event.attempt}): ${event.errorMessage}`);
				break;
		}
	}
});
// Sends this conversation's events to this connection.
await conn.conversation.watchEvents(root.id);

const result = await conn.prompt("Clone https://github.com/honojs/hono and summarize its README.", {
	requestId: crypto.randomUUID(),
});
if (result.status === "unanswered") console.error(`\nNo answer: ${result.reason}`);

unsubscribe();
await conn.dispose();

import { createRegistry } from "@earendil-works/pi-durable";
import { CodingTools } from "@earendil-works/pi-durable/tools";
import { pi } from "@rivet-dev/pi";
import { e2bProvider } from "@rivet-dev/sandbox-adapter/e2b";
import { setup } from "rivetkit";

const extensions = createRegistry();
extensions.install(CodingTools);

const agent = pi({
	model: "anthropic/claude-opus-5-5",
	registry: extensions,
	sandbox: e2bProvider(),
});

export const registry = setup({ use: { agent } });

registry.start();

import { createClient } from "rivetkit/client";
import type { registry } from "./server";

async function fetchAgentToken(): Promise<{ agentId: string; token: string }> {
	const response = await fetch("/agent-token", { method: "POST", cache: "no-store" });
	if (!response.ok) throw new Error("Could not get an agent token.");
	return (await response.json()) as { agentId: string; token: string };
}

const { agentId } = await fetchAgentToken();

const client = createClient<typeof registry>({
	endpoint: "https://api.rivet.dev",
	namespace: "production",
	getToken: async () => (await fetchAgentToken()).token,
});

const conn = client.agent.getForId(agentId).connect();
const result = await conn.prompt("Clone https://github.com/honojs/hono and summarize its README.");
console.log(result.status === "done" ? result.text : `No answer: ${result.reason}`);

import { Hono } from "hono";
import { createClient } from "rivetkit/client";
import type { registry } from "./server";

const admin = createClient<typeof registry>();

// Replace with your own auth, such as verifying a session cookie. Never trust a user id the client sends.
async function authenticateUser(request: Request): Promise<string | null> {
	return request.headers.get("x-user-id");
}

const app = new Hono();

app.post("/agent-token", async (c) => {
	const userId = await authenticateUser(c.req.raw);
	if (!userId) return c.json({ error: "unauthorized" }, 401);

	const agent = admin.agent.getOrCreate(["support", userId]);
	const { token } = await agent.issueToken({ subject: userId, expiresIn: 900 });
	return c.json({ agentId: await agent.resolve(), token }, 200, { "Cache-Control": "no-store" });
});

export default app;