Assistant IA

Installer Wrangler CLI

Installez Wrangler, l'outil de ligne de commande pour Cloudflare Workers :

npm install -g wrangler
wrangler login

Créer un nouveau projet Workers

Créez un nouveau projet Workers pour votre chatbot :

wrangler init ai-chatbot
cd ai-chatbot

Configurer le projet pour Workers AI

Modifiez le fichier wrangler.toml pour activer Workers AI :

name = "ai-chatbot"
main = "src/index.js"
compatibility_date = "2023-10-30"

[ai]
binding = "AI"

Créer le code du chatbot

Modifiez le fichier src/index.js pour implémenter le chatbot :

export default {
  async fetch(request, env) {
    // Gérer les requêtes CORS pour les requêtes cross-origin
    if (request.method === 'OPTIONS') {
      return handleCORS();
    }

    // Vérifier si c'est une requête POST
    if (request.method !== 'POST') {
      return new Response('Méthode non autorisée', { status: 405 });
    }

    try {
      // Récupérer le corps de la requête
      const { messages } = await request.json();

      // Vérifier si les messages sont valides
      if (!messages || !Array.isArray(messages) || messages.length === 0) {
        return new Response('Format de message invalide', { status: 400 });
      }

      // Appeler le modèle d'IA
      const response = await env.AI.run('@cf/meta/llama-2-7b-chat-int8', {
        messages: messages.map(msg => ({
          role: msg.role,
          content: msg.content
        })),
        max_tokens: 800
      });

      // Retourner la réponse avec les en-têtes CORS
      return corsify(new Response(JSON.stringify({ response }), {
        headers: {
          'Content-Type': 'application/json'
        }
      }));
    } catch (error) {
      console.error('Erreur:', error);
      return corsify(new Response(JSON.stringify({ error: error.message }), {
        status: 500,
        headers: {
          'Content-Type': 'application/json'
        }
      }));
    }
  }
};

// Fonction pour gérer les requêtes CORS
function handleCORS() {
  return new Response(null, {
    headers: {
      'Access-Control-Allow-Origin': '*',
      'Access-Control-Allow-Methods': 'POST, OPTIONS',
      'Access-Control-Allow-Headers': 'Content-Type',
      'Access-Control-Max-Age': '86400'
    }
  });
}

// Fonction pour ajouter les en-têtes CORS à une réponse
function corsify(response) {
  const headers = new Headers(response.headers);
  headers.set('Access-Control-Allow-Origin', '*');
  
  return new Response(response.body, {
    status: response.status,
    statusText: response.statusText,
    headers
  });
}

Déployer le Worker

Déployez votre Worker sur Cloudflare :

wrangler deploy

Notez l'URL de votre Worker, elle ressemblera à https://ai-chatbot.your-username.workers.dev