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()