Cloudflare Agents integration

import { createOpenAI } from "@ai-sdk/openai";

export interface Env {
  ORQ_API_KEY: string;
}

export const gateway = ({ ORQ_API_KEY }: Env) =>
  createOpenAI({
    baseURL: "https://my.orq.ai/v3/router",
    apiKey: ORQ_API_KEY,
  });

import { Agent } from "agents";
import { generateText } from "ai";

import { gateway, type Env } from "./gateway";

export class Assistant extends Agent<Env> {
  async onRequest(request: Request): Promise<Response> {
    const prompt = new URL(request.url).searchParams.get("prompt") ?? "Explain quantum computing in simple terms.";
    const orq = gateway(this.env);

    const { text } = await generateText({
      model: orq("openai/gpt-5.6-sol"),
      prompt,
    });

    return Response.json({ text });
  }
}

import { AIChatAgent } from "@cloudflare/ai-chat";
import { convertToModelMessages, streamText } from "ai";

import { gateway } from "./gateway";

export class ChatAgent extends AIChatAgent {
  async onChatMessage() {
    const result = streamText({
      model: gateway(this.env)("openai/gpt-5.6-sol"),
      messages: await convertToModelMessages(this.messages),
    });

    return result.toUIMessageStreamResponse();
  }
}

import { generateText } from "ai";

import { gateway, type Env } from "./gateway";

// Swap the model id for any provider in the catalog:
// anthropic/claude-sonnet-5, google/gemini-2.5-pro, openai/gpt-5.6-luna
export async function ask(env: Env, modelId: string, prompt: string) {
  const orq = gateway(env);

  const { text } = await generateText({
    model: orq(modelId),
    prompt,
  });

  return text;
}

{
  "observability": {
    "traces": {
      "enabled": true,
      "destinations": ["orq-traces"],
      "head_sampling_rate": 1,
      "persist": false
    },
    "logs": {
      "enabled": true,
      "destinations": ["orq-logs"],
      "head_sampling_rate": 1,
      "persist": false
    }
  }
}