From 3bc9b031ebe95f6fa433dc7fe133739a526409a7 Mon Sep 17 00:00:00 2001 From: AnthonyAxenov Date: Tue, 19 Aug 2025 10:18:40 +0800 Subject: [PATCH] =?UTF-8?q?=D0=A4=D0=B8=D0=BA=D1=81=20=D0=BE=D1=88=D0=B8?= =?UTF-8?q?=D0=B1=D0=BA=D0=B8=20name=20'false'=20is=20not=20defined?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- @rag/3_rag.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/@rag/3_rag.py b/@rag/3_rag.py index 56eff45..19c0f4d 100644 --- a/@rag/3_rag.py +++ b/@rag/3_rag.py @@ -34,7 +34,7 @@ class LocalRAGSystem: db_path: str = "ready_rag", ollama_url: str = "http://localhost:11434", 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.db_path = Path(db_path) @@ -350,7 +350,7 @@ class LocalRAGSystem: json={ "model": self.chat_model, "prompt": prompt, - "stream": false + "stream": False }, timeout=600 ) @@ -443,7 +443,7 @@ def main(): parser.add_argument("--question", type=str, help="Вопрос для поиска") parser.add_argument("--md-folder", default="output_md", help="Папка с markdown файлами") 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="Количество результатов поиска") args = parser.parse_args()