From 952f0dd92d35dc77ba5727abcbd65056182ddc26 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?St=C3=A9phan?= Date: Mon, 19 May 2025 14:17:37 +0200 Subject: [PATCH] Update rag_md.py --- rag_md.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rag_md.py b/rag_md.py index 5b95144..5dcaeb4 100644 --- a/rag_md.py +++ b/rag_md.py @@ -26,7 +26,7 @@ def collect_markdown_files(root_dir): return texts, sources, raw_contents # 2. Initialisation -ROOT_DIR = "mes_fiches" +ROOT_DIR = "Corpus" print("🔍 Chargement des fichiers markdown...") documents, paths, raw_contents = collect_markdown_files(ROOT_DIR) print(f"📄 {len(documents)} fichiers chargés.") @@ -84,7 +84,7 @@ while True: print("\n🧠 Appel au modèle Ollama...\n") res = requests.post( "http://localhost:11434/api/generate", - json={"model": "llama3", "prompt": prompt, "stream": False} + json={"model": "llama3-8b-fast:latest", "prompt": prompt, "stream": False} ) print("📘 Fichiers utilisés :\n", fichiers_utilisés)