bilalb/Philosophy-RAG-demo
1
0
Fork 0
forked from AI_team/Philosophy-RAG-demo
Code Pull Requests Activity

✨ Change config file location using argument parser

Browse Source
This commit is contained in:
Ruben Lucas 2025-04-17 13:06:27 +02:00
parent a58ef2f365
commit 9935d9e8d3
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
generic_rag/app.py
View File

@ -1,3 +1,4 @@
import argparse
import json import json
import logging import logging
import os import os
@ -17,6 +18,8 @@ from generic_rag.parsers.parser import add_pdf_files, add_urls
logger = logging.getLogger("sogeti-rag") logger = logging.getLogger("sogeti-rag")
logger.setLevel(logging.DEBUG) logger.setLevel(logging.DEBUG)
PROJECT_ROOT = Path(__file__).resolve().parent.parent
system_prompt = ( system_prompt = (
"You are an assistant for question-answering tasks. " "You are an assistant for question-answering tasks. "
"If the question is in Dutch, answer in Dutch. If the question is in English, answer in English." "If the question is in Dutch, answer in Dutch. If the question is in English, answer in English."
@ -24,13 +27,21 @@ system_prompt = (
"If you don't know the answer, say that you don't know." "If you don't know the answer, say that you don't know."
) )
PROJECT_ROOT = Path(__file__).resolve().parent.parent parser = argparse.ArgumentParser(description="A Sogeti Netherlands Generic RAG demo.")
CONFIG_FILE_PATH = PROJECT_ROOT / "config.yaml" parser.add_argument(
"-c",
"--config",
type=Path,
default=PROJECT_ROOT / "config.yaml",
help="Path to configuration file (YAML format). Defaults to 'config.yaml' in project root.",
)
args = parser.parse_args()
try: try:
settings: AppSettings = load_settings(CONFIG_FILE_PATH) settings: AppSettings = load_settings(args.config)
except (FileNotFoundError, Exception) as e: except (FileNotFoundError, Exception) as e:
logger.error(f"Failed to load configuration from {CONFIG_FILE_PATH}. Exiting.") logger.error(f"Failed to load configuration from {args.config}. Exiting.")
sys.exit(1) sys.exit(1)
embedding_function = get_embedding_model(settings) embedding_function = get_embedding_model(settings)