From 8b68c6024992f0b42153392788226f024fba1116 Mon Sep 17 00:00:00 2001 From: leehk Date: Wed, 12 Mar 2025 15:31:04 +0800 Subject: [PATCH] rename src folder from adaptive_rag_evaluation to rag_adaptive_evaluation --- app/llmops/main.py | 6 +++--- .../MLproject | 2 +- .../python_env.yml | 0 .../run.py | 0 app/llmops/src/rag_cot_evaluation/run.py | 8 ++++++++ 5 files changed, 12 insertions(+), 4 deletions(-) rename app/llmops/src/{adaptive_rag_evaluation => rag_adaptive_evaluation}/MLproject (96%) rename app/llmops/src/{adaptive_rag_evaluation => rag_adaptive_evaluation}/python_env.yml (100%) rename app/llmops/src/{adaptive_rag_evaluation => rag_adaptive_evaluation}/run.py (100%) 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):