import { CloudflareAGUIAdapter } from 'ag-ui-cloudflare';

const adapter = new CloudflareAGUIAdapter({
  accountId: 'your-account-id',
  apiToken: 'your-api-token',
  model: '@cf/meta/llama-3.1-8b-instruct'
});

// Execute with AG-UI events
const messages = [
  { role: 'user', content: 'Hello, how are you?' }
];

for await (const event of adapter.execute(messages)) {
  console.log(event.type, event.data);
}

import { CopilotRuntime } from '@copilotkit/runtime';
import { CloudflareAGUIAdapter } from 'ag-ui-cloudflare';

const adapter = new CloudflareAGUIAdapter({
  accountId: process.env.CLOUDFLARE_ACCOUNT_ID!,
  apiToken: process.env.CLOUDFLARE_API_TOKEN!,
});

const runtime = new CopilotRuntime({
  adapter,
});

export async function POST(req: Request) {
  return runtime.handle(req);
}

import { CloudflareProviders } from 'ag-ui-cloudflare';

// Auto-select best model based on capabilities
const adapter = CloudflareProviders.auto({
  accountId: 'your-account-id',
  apiToken: 'your-api-token',
  tools: [...], // Will use Llama 3.3 70B if tools provided
});

// Or choose specific model providers
const fastAdapter = CloudflareProviders.llama3_8b(config);
const powerfulAdapter = CloudflareProviders.llama3_70b(config);
const functionAdapter = CloudflareProviders.llama3_3_70b(config); // Supports tools

import { CloudflareProviders } from 'ag-ui-cloudflare';

const adapter = CloudflareProviders.createWithGateway(
  accountId,
  apiToken,
  gatewayId,
  '@cf/meta/llama-3.1-70b-instruct'
);

// Benefits: Response caching, rate limiting, analytics

const tools = [{
  type: 'function',
  function: {
    name: 'get_weather',
    description: 'Get current weather',
    parameters: {
      type: 'object',
      properties: {
        location: { type: 'string' }
      }
    }
  }
}];

const adapter = new CloudflareAGUIAdapter({
  accountId: 'your-account-id',
  apiToken: 'your-api-token',
  model: '@cf/meta/llama-3.3-70b-instruct', // Only model with function calling
  tools
});

for await (const event of adapter.execute(messages)) {
  if (event.type === 'TOOL_CALL_START') {
    console.log('Calling tool:', event.data.toolName);
  }
}

const stages = [
  { name: 'Research', instruction: 'Research the topic' },
  { name: 'Outline', instruction: 'Create an outline' },
  { name: 'Write', instruction: 'Write the full article' }
];

for await (const event of adapter.progressiveGeneration('Write about AI', stages)) {
  if (event.type === 'PROGRESS') {
    console.log(`Progress: ${event.data.progress}% - ${event.data.message}`);
  }
}

class CloudflareAGUIAdapter {
  constructor(options: CloudflareAGUIAdapterOptions)

  execute(messages: CloudflareMessage[], context?: Record<string, any>): AsyncGenerator<AGUIEvent>

  executeWithTools(messages: CloudflareMessage[], tools: Tool[], context?: Record<string, any>): AsyncGenerator<AGUIEvent>

  progressiveGeneration(prompt: string, stages: Stage[]): AsyncGenerator<AGUIEvent>

  setModel(model: CloudflareModel): void

  getCapabilities(): ModelCapabilities

  listAvailableModels(): Promise<string[]>
}

interface CloudflareAGUIAdapterOptions {
  accountId: string;        // Cloudflare account ID
  apiToken: string;         // API token with Workers AI permissions
  model?: CloudflareModel;  // Model to use (defaults to llama-3.1-8b)
  baseURL?: string;         // Custom API endpoint
  gatewayId?: string;       // AI Gateway ID for caching/analytics
  systemPrompt?: string;    // System prompt for all requests
  tools?: Tool[];           // Available tools for function calling
  streamingEnabled?: boolean; // Enable streaming (default: true)
}