diff --git a/app/llmops/main.py b/app/llmops/main.py index 809b2f1..4c04a63 100644 --- a/app/llmops/main.py +++ b/app/llmops/main.py @@ -9,7 +9,7 @@ _steps = [ "etl_chromadb_pdf", "etl_chromadb_scanned_pdf", # the performance for scanned pdf may not be good "rag_cot_evaluation", - "adaptive_rag_evaluation", + "rag_adaptive_evaluation", "test_rag_cot" ] @@ -132,7 +132,7 @@ def go(config: DictConfig): }, ) - if "adaptive_rag_evaluation" in active_steps: + if "rag_adaptive_evaluation" in active_steps: if config["prompt_engineering"]["run_id_chromadb"] == "None": # Look for run_id that has artifact logged as documents @@ -152,7 +152,7 @@ def go(config: DictConfig): run_id = config["prompt_engineering"]["run_id_chromadb"] _ = mlflow.run( - os.path.join(hydra.utils.get_original_cwd(), "src", "adaptive_rag_evaluation"), + os.path.join(hydra.utils.get_original_cwd(), "src", "rag_adaptive_evaluation"), "main", parameters={ "query": config["prompt_engineering"]["query"], diff --git a/app/llmops/src/adaptive_rag_evaluation/MLproject b/app/llmops/src/rag_adaptive_evaluation/MLproject similarity index 96% rename from app/llmops/src/adaptive_rag_evaluation/MLproject rename to app/llmops/src/rag_adaptive_evaluation/MLproject index 521e783..48c1dad 100644 --- a/app/llmops/src/adaptive_rag_evaluation/MLproject +++ b/app/llmops/src/rag_adaptive_evaluation/MLproject @@ -1,4 +1,4 @@ -name: adaptive_rag_evaluation +name: rag_adaptive_evaluation python_env: python_env.yml entry_points: diff --git a/app/llmops/src/adaptive_rag_evaluation/python_env.yml b/app/llmops/src/rag_adaptive_evaluation/python_env.yml similarity index 100% rename from app/llmops/src/adaptive_rag_evaluation/python_env.yml rename to app/llmops/src/rag_adaptive_evaluation/python_env.yml diff --git a/app/llmops/src/adaptive_rag_evaluation/run.py b/app/llmops/src/rag_adaptive_evaluation/run.py similarity index 100% rename from app/llmops/src/adaptive_rag_evaluation/run.py rename to app/llmops/src/rag_adaptive_evaluation/run.py diff --git a/app/llmops/src/rag_cot_evaluation/run.py b/app/llmops/src/rag_cot_evaluation/run.py index 054d11b..aa773b1 100644 --- a/app/llmops/src/rag_cot_evaluation/run.py +++ b/app/llmops/src/rag_cot_evaluation/run.py @@ -18,6 +18,14 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false" GEMINI_API_KEY = config("GOOGLE_API_KEY", cast=str) DEEKSEEK_API_KEY = config("DEEKSEEK_API_KEY", cast=str) MOONSHOT_API_KEY = config("MOONSHOT_API_KEY", cast=str) +LANGSMITH_API_KEY = config("LANGSMITH_API_KEY", cast=str) +LANGSMITH_TRACING = config("LANGSMITH_TRACING", cast=str) +LANGSMITH_PROJECT = config("LANGSMITH_PROJECT", cast=str) +os.environ["TOKENIZERS_PARALLELISM"] = "false" +os.environ["LANGSMITH_API_KEY"] = LANGSMITH_API_KEY +os.environ["LANGSMITH_TRACING"] = LANGSMITH_TRACING +os.environ["LANGSMITH_ENDPOINT"] = "https://api.smith.langchain.com" +os.environ["LANGSMITH_PROJECT"] = LANGSMITH_PROJECT def go(args):