RAG, étape 3 : compléter la consigne

Partie 6 · ★★

Le RAG ne change aucun poids : il complète la consigne. On place les passages retenus avant la question, puis le modèle aligné prédit la suite et peut citer les extraits. Le format est imposé : une première ligne Voici N extraits du protocole., une ligne [1] …, [2] … par passage (numérotés à partir de 1), puis une dernière ligne Question : …. Les lignes sont jointes par un retour à la ligne.

Complète prompt_rag et construis le prompt avec les 2 passages retenus.

def prompt_rag(passages_gardes, question):
    lignes = ["Voici " + str(len(___)) + " extraits du protocole."]
    for i, p in enumerate(passages_gardes, start=___):
        lignes.append("[" + str(i) + "] " + p)
    lignes.append("Question : " + question)
    return "\n".join(lignes)

passages = [
    "Fièvre sans toux : faire un bilan urinaire dans les six heures.",
    "Les visites sont autorisées de 14 h à 19 h.",
    "Si le bilan urinaire est positif, débuter un traitement adapté.",
]
question = "Que faire si le bilan urinaire est positif ?"
top2 = [2, 0]
prompt = prompt_rag([passages[i] for i in top2], question)

Ctrl/⌘+Entrée

Exercice suivant → Classer les façons d'obtenir un modèle utile