RYDE-WORK/aimingmed-ai
1
0
Fork 0
mirror of https://github.com/aimingmed/aimingmed-ai.git synced 2026-02-09 00:43:33 +08:00
Code Issues 6 Packages Projects Releases Wiki Activity

rename src folder from adaptive_rag_evaluation to rag_adaptive_evaluation

Browse Source
This commit is contained in:
leehk 2025-03-12 15:31:04 +08:00
parent e8a8fbf826
commit 8b68c60249
5 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/llmops/main.py
View File

@ -9,7 +9,7 @@ _steps = [
"etl_chromadb_pdf", "etl_chromadb_pdf",
"etl_chromadb_scanned_pdf", # the performance for scanned pdf may not be good "etl_chromadb_scanned_pdf", # the performance for scanned pdf may not be good
"rag_cot_evaluation", "rag_cot_evaluation",
"adaptive_rag_evaluation", "rag_adaptive_evaluation",
"test_rag_cot" "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": if config["prompt_engineering"]["run_id_chromadb"] == "None":
# Look for run_id that has artifact logged as documents # 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"] run_id = config["prompt_engineering"]["run_id_chromadb"]
_ = mlflow.run( _ = 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", "main",
parameters={ parameters={
"query": config["prompt_engineering"]["query"], "query": config["prompt_engineering"]["query"],

2
app/llmops/src/adaptive_rag_evaluation/MLproject → app/llmops/src/rag_adaptive_evaluation/MLproject
View File

@ -1,4 +1,4 @@
name: adaptive_rag_evaluation name: rag_adaptive_evaluation
python_env: python_env.yml python_env: python_env.yml
entry_points: entry_points:

0
app/llmops/src/adaptive_rag_evaluation/python_env.yml → app/llmops/src/rag_adaptive_evaluation/python_env.yml
View File

0
app/llmops/src/adaptive_rag_evaluation/run.py → app/llmops/src/rag_adaptive_evaluation/run.py
View File

8
app/llmops/src/rag_cot_evaluation/run.py
View File

@ -18,6 +18,14 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
GEMINI_API_KEY = config("GOOGLE_API_KEY", cast=str) GEMINI_API_KEY = config("GOOGLE_API_KEY", cast=str)
DEEKSEEK_API_KEY = config("DEEKSEEK_API_KEY", cast=str) DEEKSEEK_API_KEY = config("DEEKSEEK_API_KEY", cast=str)
MOONSHOT_API_KEY = config("MOONSHOT_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): def go(args):