From 3266c4876239fc6e63c0ef2d7b0fd894e5e55e42 Mon Sep 17 00:00:00 2001 From: kyy Date: Tue, 18 Mar 2025 17:28:07 +0900 Subject: [PATCH] Select qarquet files --- main.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/main.py b/main.py index b9021c0..0ce0d77 100644 --- a/main.py +++ b/main.py @@ -9,18 +9,19 @@ from llama_index.llms.ollama import Ollama data_path = './projects/example_01' # 폴더명 확인 OLLAMA_BASE_URL = "autorag-ollama:11434" -autorag.generator_models["ollama"] = autorag.LazyInit(Ollama, base_url=OLLAMA_BASE_URL, model="phi4", request_timeout=300, num_gpus=1) autorag.generator_models["ollama"] = autorag.LazyInit(Ollama, base_url=OLLAMA_BASE_URL, model="gemma3:12b", request_timeout=300, num_gpus=1) -# autorag.generator_models["ollama"] = autorag.LazyInit(Ollama, base_url=OLLAMA_BASE_URL, model="deepseek-r1:14b", request_timeout=300, num_gpus=1) -# autorag.generator_models["ollama"] = autorag.LazyInit(Ollama, base_url=OLLAMA_BASE_URL, model="aya-expanse:8b", request_timeout=300, num_gpus=1) +autorag.generator_models["ollama"] = autorag.LazyInit(Ollama, base_url=OLLAMA_BASE_URL, model="phi4", request_timeout=300, num_gpus=1) +autorag.generator_models["ollama"] = autorag.LazyInit(Ollama, base_url=OLLAMA_BASE_URL, model="deepseek-r1:14b", request_timeout=300, num_gpus=1) +autorag.generator_models["ollama"] = autorag.LazyInit(Ollama, base_url=OLLAMA_BASE_URL, model="aya-expanse:8b", request_timeout=300, num_gpus=1) -# autorag/embedding/base.py 임베딩 모델추가함 +data_index=1 @click.command() @click.option('--config', type=click.Path(exists=True), default=os.path.join(data_path, 'config.yaml')) -@click.option('--qa_data_path', type=click.Path(exists=True), default=os.path.join(data_path, 'qa.parquet')) -@click.option('--corpus_data_path', type=click.Path(exists=True), default=os.path.join(data_path, 'corpus.parquet')) -@click.option('--project_dir', type=click.Path(exists=False), default=os.path.join(data_path, 'benchmark')) +@click.option('--qa_data_path', type=click.Path(exists=True), default=os.path.join(data_path, f'qa_{data_index}.parquet')) +@click.option('--corpus_data_path', type=click.Path(exists=True), default=os.path.join(data_path, f'corpus_{data_index}.parquet')) +@click.option('--project_dir', type=click.Path(exists=False), default=os.path.join(data_path, f'benchmark_ck{data_index}')) + def main(config, qa_data_path, corpus_data_path, project_dir): load_dotenv()