import { CloudflareAgentsClient } from "@ag-ui/cloudflare-agents";

const agent = new CloudflareAgentsClient({
  url: "wss://your-agent.workers.dev",
});

agent
  .runAgent({
    threadId: "thread-123",
    runId: "run-456",
    messages: [{ role: "user", content: "What's the weather?" }],
  })
  .subscribe({
    next: (event) => {
      switch (event.type) {
        case "TEXT_MESSAGE_CONTENT":
          process.stdout.write(event.delta);
          break;
        case "STATE_SNAPSHOT":
          console.log("State:", event.snapshot);
          break;
        case "TOOL_CALL_START":
          console.log("Calling tool:", event.toolCallName);
          break;
      }
    },
    complete: () => console.log("Done"),
  });

import { AgentsToAGUIAdapter, createSSEResponse } from "@ag-ui/cloudflare-agents";
import { streamText } from "ai";
import { openai } from "@ai-sdk/openai";
import { from } from "rxjs";

const adapter = new AgentsToAGUIAdapter();

export default {
  async fetch(request: Request): Promise<Response> {
    const { messages, threadId, runId } = await request.json();

    const stream = streamText({
      model: openai("gpt-4"),
      messages,
    });

    const events$ = from(
      adapter.adaptStreamToAGUI(stream, threadId, runId, messages)
    );

    return createSSEResponse(events$);
  },
};

import { Agent } from "agents";
import { streamText } from "ai";
import { openai } from "@ai-sdk/openai";
import { AgentsToAGUIAdapter } from "@ag-ui/cloudflare-agents";
import { z } from "zod";

export class MyAgent extends Agent {
  private adapter = new AgentsToAGUIAdapter();

  async onMessage(ws: WebSocket, raw: string | ArrayBuffer) {
    const data = typeof raw === "string" ? raw : new TextDecoder().decode(raw);
    const { messages, threadId } = JSON.parse(data);

    const stream = streamText({
      model: openai("gpt-4"),
      messages,
      tools: {
        getWeather: {
          description: "Get current weather",
          parameters: z.object({ location: z.string() }),
          execute: async ({ location }) => ({ temperature: 72, condition: "sunny" }),
        },
      },
    });

    for await (const event of this.adapter.adaptStreamToAGUI(
      stream, threadId, crypto.randomUUID(), messages
    )) {
      ws.send(JSON.stringify(event));
    }
  }
}