chat Premier chatbot Python en ligne de commande

mkdir chatbot-ollama
cd chatbot-ollama

# Créer et activer le venv
python -m venv venv                  # Windows
python3 -m venv venv                 # macOS / Linux

venv\Scripts\activate                # Windows
source venv/bin/activate             # macOS / Linux

# Installer la bibliothèque officielle Ollama pour Python
pip install ollama

import ollama  # on importe la bibliothèque qu'on vient d'installer

# La fonction chat() prend un modèle et une liste de messages.
# Chaque message est un dictionnaire avec deux clés :
#   "role"    : "user" (l'utilisateur) ou "assistant" (le modèle)
#   "content" : le texte du message

reponse = ollama.chat(
    model="gemma4:e2b",
    messages=[
        {"role": "user", "content": "Bonjour ! Qui es-tu ?"}
    ]
)

# La réponse est un objet. Le texte généré est dans :
# reponse.message.content
print(reponse.message.content)

python chatbot.py

import ollama

# L'historique démarre vide. On peut aussi y mettre un "system prompt"
# pour donner des instructions globales au modèle.
historique = [
    {
        "role": "system",
        "content": (
            "Tu es un assistant en français, sympathique et concis. "
            "Tu réponds toujours en français, même si on te parle dans une autre langue."
        )
    }
]

print("=== Chatbot Ollama (gemma4:e2b) ===")
print("Tape 'quitter' pour terminer.\n")

while True:  # une boucle infinie — on continue jusqu'à ce que l'utilisateur quitte
    
    # 1. Demander un message à l'utilisateur
    message_utilisateur = input("Toi : ").strip()
    
    # 2. Permettre de quitter proprement
    if message_utilisateur.lower() in ["quitter", "exit", "bye", "q"]:
        print("Au revoir !")
        break  # sort de la boucle while
    
    # 3. Ne rien faire si l'utilisateur n'a rien tapé
    if not message_utilisateur:
        continue  # passe à l'itération suivante de la boucle
    
    # 4. Ajouter le message de l'utilisateur à l'historique
    historique.append({
        "role": "user",
        "content": message_utilisateur
    })
    
    # 5. Envoyer TOUT l'historique au modèle
    reponse = ollama.chat(
        model="gemma4:e2b",
        messages=historique
    )
    
    # 6. Extraire le texte de la réponse
    texte_reponse = reponse.message.content
    
    # 7. Ajouter la réponse de l'assistant à l'historique
    historique.append({
        "role": "assistant",
        "content": texte_reponse
    })
    
    # 8. Afficher la réponse
    print(f"\nAssistant : {texte_reponse}\n")

import ollama
import sys  # pour flush le buffer d'affichage

historique = [
    {
        "role": "system",
        "content": (
            "Tu es un assistant en français, sympathique et concis. "
            "Tu réponds toujours en français."
        )
    }
]

print("=== Chatbot Ollama (gemma4:e2b) ===")
print("Tape 'quitter' pour terminer.\n")

while True:
    message_utilisateur = input("Toi : ").strip()

    if message_utilisateur.lower() in ["quitter", "exit", "bye", "q"]:
        print("Au revoir !")
        break

    if not message_utilisateur:
        continue

    historique.append({"role": "user", "content": message_utilisateur})

    print("Assistant : ", end="", flush=True)  # afficher sans saut de ligne

    texte_complet = ""  # on accumule le texte pour l'historique

    # stream=True active le streaming
    for chunk in ollama.chat(model="gemma4:e2b", messages=historique, stream=True):
        morceau = chunk.message.content
        print(morceau, end="", flush=True)  # afficher sans saut de ligne
        texte_complet += morceau           # accumuler

    print("\n")  # saut de ligne à la fin de la réponse

    historique.append({"role": "assistant", "content": texte_complet})

    # Remplace le bloc "for chunk in ollama.chat..." par ceci :
    try:
        print("Assistant : ", end="", flush=True)
        texte_complet = ""

        for chunk in ollama.chat(model="gemma4:e2b", messages=historique, stream=True):
            morceau = chunk.message.content
            print(morceau, end="", flush=True)
            texte_complet += morceau

        print("\n")
        historique.append({"role": "assistant", "content": texte_complet})

    except Exception as erreur:
        # Si quelque chose se passe mal, on affiche un message clair
        print(f"\n[Erreur] Impossible de contacter Ollama : {erreur}")
        print("Vérifie qu'Ollama est bien lancé sur ta machine.\n")
        # On retire le dernier message utilisateur de l'historique
        # car la réponse n'a pas été générée
        historique.pop()