Introduction

Avec les progrès actuels de l'IA, il est facile de configurer un modèle d'IA génératif sur votre ordinateur pour créer un chatbot.

Dans cet article, nous verrons comment configurer un chatbot sur votre système à l'aide d'Ollama et Next.js

Configurer Ollama

Commençons par configurer Ollama sur notre système. Visitez ollama.com et téléchargez-le pour votre système d'exploitation. Cela nous permettra d'utiliser la commande ollama dans le terminal/invite de commande.

Vérifiez la version d'Ollama en utilisant la commande ollama -v
Consultez la liste des modèles sur la page de la bibliothèque Ollama.

Télécharger et exécuter un modèle

Pour télécharger et exécuter un modèle, exécutez la commande ollama run
Exemple : ollama exécute lama3.1 ou ollama exécute gemma2

Vous pourrez discuter avec le modèle directement dans le terminal.

Configurer l'application Web

Configuration de base pour Next.js

• Téléchargez et installez la dernière version de Node.js
• Accédez au dossier souhaité et exécutez npx create-next-app@latest pour générer le projet Next.js.
• Il posera quelques questions pour générer du code passe-partout. Pour ce tutoriel, nous conserverons tout par défaut.
• Ouvrez le projet nouvellement créé dans l'éditeur de code de votre choix. Nous allons utiliser VS Code.

Installation des dépendances

Il existe quelques packages npm qui doivent être installés pour utiliser l'ollama.

1. ai de vercel.
2. ollama La bibliothèque JavaScript Ollama offre le moyen le plus simple d'intégrer votre projet JavaScript à Ollama.
3. ollama-ai-provider aide à connecter ai et ollama ensemble.
4. Les résultats du chat réagissent-markdown seront formatés dans le style markdown, pour analyser le démarque, nous allons utiliser le package réagir-markdown.

Pour installer ces dépendances, exécutez npm i ai ollama ollama-ai-provider.

Créer une page de discussion

Sous app/src se trouve un fichier nommé page.tsx.

Supprimons tout ce qu'il contient et commençons par le composant fonctionnel de base :

src/app/page.tsx

export default function Home() {
  return (
    

        {/* Code here... */}
    
  );
}

Commençons par importer le hook useChat depuis ai/react et react-markdown

"use client";
import { useChat } from "ai/react";
import Markdown from "react-markdown";

Parce que nous utilisons un hook, nous devons convertir cette page en composant client.

Astuce : Vous pouvez créer un composant distinct pour le chat et l'appeler dans le page.tsx pour limiter l'utilisation des composants clients.

Dans le composant, obtenez les messages, l'entrée, handleInputChange et handleSubmit à partir du hook useChat.

    const { messages, input, handleInputChange, handleSubmit } = useChat();

Dans JSX, créez un formulaire de saisie pour obtenir la saisie de l'utilisateur afin d'initier une conversation.

La bonne idée à ce sujet est que nous n'avons pas besoin de corriger le gestionnaire ou de maintenir un état pour la valeur d'entrée, le hook useChat nous le fournit.

Nous pouvons afficher les messages en parcourant le tableau des messages.

    messages.map((m, i) => (
{m}
)

La version stylisée basée sur le rôle de l'expéditeur ressemble à ceci :

  

    

      {messages.length ? (
        messages.map((m, i) => {
          return m.role === "user" ? (
            

              You
              

                {m.content}
            
          ) : (
            

              AI
              

                {m.content}
            
          );
        })
      ) : (
        

          
Local AI Chat
        
      )}
    
  

Jetons un œil à l'ensemble du fichier
src/app/page.tsx

"use client";
import { useChat } from "ai/react";
import Markdown from "react-markdown";

export default function Home() {
    const { messages, input, handleInputChange, handleSubmit } = useChat();
    return (
        

              

                
            
        
    );
}

Avec cela, la partie frontend est terminée. Passons maintenant à l'API.

API de gestion

Commençons par créer route.ts dans app/api/chat.
Basé sur la convention de dénomination Next.js, cela nous permettra de gérer les requêtes sur le point de terminaison localhost:3000/api/chat.

src/app/api/chat/route.ts

import { createOllama } from "ollama-ai-provider";
import { streamText } from "ai";

const ollama = createOllama();

export async function POST(req: Request) {
  const { messages } = await req.json();

  const result = await streamText({
    model: ollama("llama3.1"),
    messages,
  });
  return result.toDataStreamResponse();
}

Le code ci-dessus utilise essentiellement ollama et vercel ai pour diffuser les données en réponse.

• createOllama crée une instance de l'ollama qui communiquera avec le modèle installé sur le système.
• La fonction POST est le gestionnaire de route sur le point de terminaison /api/chat avec la méthode post.
• Le corps de la requête contient la liste de tous les messages précédents. C'est donc une bonne idée de le limiter, sinon les performances se dégraderont avec le temps. Dans cet exemple, la fonction ollama prend "llama3.1" comme modèle pour générer la réponse basée sur le tableau de messages.

IA générative sur votre système

Exécutez npm run dev pour démarrer le serveur en mode développement.
Ouvrez le navigateur et accédez à localhost:3000 pour voir les résultats.
Si tout est correctement configuré, vous pourrez parler à votre propre chatbot.

Vous pouvez trouver le code source ici : https://github.com/parasbansal/ai-chat

Faites-moi savoir si vous avez des questions dans les commentaires, j'essaierai d'y répondre.