rename src folder from adaptive_rag_evaluation to rag_adaptive_evaluation

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"],

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:

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):