diff --git a/@rag/3_rag.py b/@rag/3_rag.py index 19c0f4d..47304ae 100644 --- a/@rag/3_rag.py +++ b/@rag/3_rag.py @@ -96,7 +96,7 @@ class LocalRAGSystem: return None - def chunk_text(self, text: str, chunk_size: int = 1000, overlap: int = 200) -> List[str]: + def chunk_text(self, text: str, chunk_size: int = 500, overlap: int = 100) -> List[str]: """Разбиваем текст на чанки с перекрытием""" chunks = [] start = 0 @@ -444,7 +444,7 @@ def main(): 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="phi4-mini:3.8b", help="Модель для чата") - parser.add_argument("--results", type=int, default=5, help="Количество результатов поиска") + parser.add_argument("--results", type=int, default=10, help="Количество результатов поиска") args = parser.parse_args()