Cloudflare Workers AI Edge Functions

export interface Env {
  AI: any;
}

export default {
  async fetch(request: Request, env: Env): Promise<Response> {
    if (request.method !== 'POST') {
      return new Response('Method not allowed', { status: 405 });
    }

    const { messages } = await request.json<{ messages: any[] }>();

    const response = await env.AI.run('@cf/meta/llama-2-7b-chat-int8', {
      messages: [
        { role: 'system', content: 'You are a helpful assistant.' },
        ...messages,
      ],
      stream: true,
    });

    return new Response(response, {
      headers: {
        'content-type': 'text/event-stream',
        'cache-control': 'no-cache',
      },
    });
  },
};

export interface Env {
  AI: any;
  IMAGES: R2Bucket;
}

export default {
  async fetch(request: Request, env: Env): Promise<Response> {
    const { prompt } = await request.json<{ prompt: string }>();

    // Generate image with Stable Diffusion
    const response = await env.AI.run(
      '@cf/stabilityai/stable-diffusion-xl-base-1.0',
      { prompt }
    );

    // Upload to R2
    const imageKey = `${crypto.randomUUID()}.png`;
    await env.IMAGES.put(imageKey, response, {
      httpMetadata: { contentType: 'image/png' },
    });

    const imageUrl = `https://images.example.com/${imageKey}`;

    return Response.json({
      success: true,
      imageUrl,
      prompt,
    });
  },
};

export interface Env {
  AI: any;
  TRANSLATIONS: KVNamespace;
}

export default {
  async fetch(request: Request, env: Env): Promise<Response> {
    const { text, targetLang = 'en' } = await request.json<{
      text: string;
      targetLang: string;
    }>();

    // Check cache first
    const cacheKey = `${text}:${targetLang}`;
    const cached = await env.TRANSLATIONS.get(cacheKey);
    if (cached) {
      return Response.json({ translation: cached, cached: true });
    }

    // Translate with Workers AI
    const response = await env.AI.run('@cf/meta/m2m100-1.2b', {
      text,
      target_lang: targetLang,
    });

    const translation = response.translated_text;

    // Cache for 24 hours
    await env.TRANSLATIONS.put(cacheKey, translation, {
      expirationTtl: 86400,
    });

    return Response.json({ translation, cached: false });
  },
};

name = "ai-worker"
main = "src/index.ts"
compatibility_date = "2025-10-16"

[ai]
binding = "AI"

[[r2_buckets]]
binding = "IMAGES"
bucket_name = "my-images"

[[kv_namespaces]]
binding = "TRANSLATIONS"
id = "your-kv-namespace-id"

[[d1_databases]]
binding = "DB"
database_name = "my-database"
database_id = "your-database-id"

[observability]
enabled = true