anthony/ollama
1
0
Fork 0
Code Activity

Фикс ошибки name 'false' is not defined

Browse Source
This commit is contained in:
Anthony Axenov
2025-08-19 10:18:40 +08:00
parent 64a63f048a
commit 3bc9b031eb
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
@rag/3_rag.py
View File

@@ -34,7 +34,7 @@ class LocalRAGSystem:
db_path: str = "ready_rag", db_path: str = "ready_rag",
ollama_url: str = "http://localhost:11434", ollama_url: str = "http://localhost:11434",
embed_model: str = "nomic-embed-text", embed_model: str = "nomic-embed-text",
chat_model: str = "qwen2.5:7b"): chat_model: str = "phi4-mini:3.8b"):
self.md_folder = Path(md_folder) self.md_folder = Path(md_folder)
self.db_path = Path(db_path) self.db_path = Path(db_path)
@@ -350,7 +350,7 @@ class LocalRAGSystem:
json={ json={
"model": self.chat_model, "model": self.chat_model,
"prompt": prompt, "prompt": prompt,
"stream": false "stream": False
}, },
timeout=600 timeout=600
) )
@@ -443,7 +443,7 @@ def main():
parser.add_argument("--question", type=str, help="Вопрос для поиска") parser.add_argument("--question", type=str, help="Вопрос для поиска")
parser.add_argument("--md-folder", default="output_md", help="Папка с markdown файлами") parser.add_argument("--md-folder", default="output_md", help="Папка с markdown файлами")
parser.add_argument("--embed-model", default="nomic-embed-text", help="Модель для эмбеддингов") parser.add_argument("--embed-model", default="nomic-embed-text", help="Модель для эмбеддингов")
parser.add_argument("--chat-model", default="llama3.2:3b", help="Модель для чата") parser.add_argument("--chat-model", default="phi4-mini:3.8b", help="Модель для чата")
parser.add_argument("--results", type=int, default=5, help="Количество результатов поиска") parser.add_argument("--results", type=int, default=5, help="Количество результатов поиска")
args = parser.parse_args() args = parser.parse_args()