npm install @better-agent/workers-ai

import { betterAgent, defineAgent } from "@better-agent/core";
import { createWorkersAI } from "@better-agent/workers-ai";

type Env = {
  AI: Ai;
};

export default {
  async fetch(_request: Request, env: Env): Promise<Response> {
    const workersAI = createWorkersAI({
      binding: env.AI,
    });

    const supportAgent = defineAgent({
      name: "support",
      model: workersAI("@cf/moonshotai/kimi-k2.5"),
      instruction: "You help customers.",
    });

    const app = betterAgent({ agents: [supportAgent] });

    return app.handler(_request);
  },
} satisfies ExportedHandler<Env>;

[ai]
binding = "AI"

const workersAI = createWorkersAI({
  accountId: process.env.CLOUDFLARE_ACCOUNT_ID,
  apiKey: process.env.CLOUDFLARE_API_TOKEN,
});

const workersAI = createWorkersAI({
  binding: env.AI,
  gateway: { id: "my-gateway" },
});

const text = workersAI.text("@cf/moonshotai/kimi-k2.5");

const normalize = defineTool({
  name: "normalize_text",
  description: "Normalize customer text before saving it.",
  inputSchema: z.object({
    content: z.string(),
  }),
  execute: async ({ content }) => {
    const result = await text.generate({
      input: `Clean up grammar without changing meaning:\n\n${content}`,
    });

    return { content: result.text };
  },
});

const embedding = workersAI.embedding("@cf/baai/bge-base-en-v1.5");
const image = workersAI.image("@cf/black-forest-labs/flux-1-schnell");
const speech = workersAI.speech("@cf/myshell-ai/melotts");
const transcription = workersAI.transcription("@cf/openai/whisper-large-v3-turbo");

await app.agent("support").run({
  messages,
  providerOptions: {
    "workers-ai": {
      reasoning_effort: "low",
    },
  },
});