Fix Dockerfile build issue
This commit is contained in:
35
main.py
Normal file
35
main.py
Normal file
@@ -0,0 +1,35 @@
|
||||
import os
|
||||
import autorag
|
||||
import click
|
||||
from autorag.evaluator import Evaluator
|
||||
from dotenv import load_dotenv
|
||||
|
||||
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/embedding/base.py 임베딩 모델추가함
|
||||
|
||||
@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'))
|
||||
|
||||
def main(config, qa_data_path, corpus_data_path, project_dir):
|
||||
load_dotenv()
|
||||
if os.getenv('OPENAI_API_KEY') is None:
|
||||
raise ValueError('OPENAI_API_KEY environment variable is not set')
|
||||
if not os.path.exists(project_dir):
|
||||
os.makedirs(project_dir)
|
||||
evaluator = Evaluator(qa_data_path, corpus_data_path, project_dir=project_dir)
|
||||
evaluator.start_trial(config)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user