FabNum/Code
4
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects 1 Releases Wiki Activity

Update rag_md.py

Browse Source
This commit is contained in:
Stéphan Peccini 2025-05-19 14:17:37 +02:00
parent d5fffdce14
commit 952f0dd92d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
rag_md.py
View File

@ -26,7 +26,7 @@ def collect_markdown_files(root_dir):
return texts, sources, raw_contents return texts, sources, raw_contents
# 2. Initialisation # 2. Initialisation
ROOT_DIR = "mes_fiches" ROOT_DIR = "Corpus"
print("🔍 Chargement des fichiers markdown...") print("🔍 Chargement des fichiers markdown...")
documents, paths, raw_contents = collect_markdown_files(ROOT_DIR) documents, paths, raw_contents = collect_markdown_files(ROOT_DIR)
print(f"📄 {len(documents)} fichiers chargés.") print(f"📄 {len(documents)} fichiers chargés.")
@ -84,7 +84,7 @@ while True:
print("\n🧠 Appel au modèle Ollama...\n") print("\n🧠 Appel au modèle Ollama...\n")
res = requests.post( res = requests.post(
"http://localhost:11434/api/generate", "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) print("📘 Fichiers utilisés :\n", fichiers_utilisés)