Quickstart

import { createClient } from "@rivet-dev/agentos/client";
import type { registry } from "./server";

const client = createClient<typeof registry>({ endpoint: "http://localhost:6420" });
const handle = client.vm.getOrCreate("my-agent");

// Subscribe to streaming events. The payload is inferred from the event schema.
const conn = handle.connect();
conn.on("sessionEvent", (data) => {
  console.log(data.event);
});

// Create a session and send a prompt. createSession returns the session ID.
const sessionId = await handle.createSession("pi", {
  env: { ANTHROPIC_API_KEY: process.env.ANTHROPIC_API_KEY! },
});
await handle.sendPrompt(
  sessionId,
  "Write a hello world script to /workspace/hello.js",
);

// Read the file the agent created
const content = await handle.readFile("/workspace/hello.js");
console.log(new TextDecoder().decode(content));

import { createRivetKit } from "@rivet-dev/agentos/react";
import { useState } from "react";
import type { registry } from "./server";

const { useActor } = createRivetKit<typeof registry>("http://localhost:6420");

export function Agent() {
  const [log, setLog] = useState("");
  const agent = useActor({ name: "vm", key: "my-agent" });

  // Stream agent events into component state
  agent.useEvent("sessionEvent", (data) => {
    setLog((prev) => prev + JSON.stringify(data.event) + "\n");
  });

  async function run() {
    // In production, inject credentials on the server (see /docs/llm-credentials)
    // createSession returns the session ID.
    const sessionId = await agent.connection?.createSession("pi", {
      env: { ANTHROPIC_API_KEY: process.env.VITE_ANTHROPIC_API_KEY! },
    });
    if (!sessionId) return;
    await agent.connection?.sendPrompt(
      sessionId,
      "Write a hello world script to /workspace/hello.js",
    );
  }

  return (
    <div>
      <button onClick={run}>Run agent</button>
      <pre>{log}</pre>
    </div>
  );
}