From 9f131ef4d122759c263e5e232f11546c43b047ff Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?The=20Vinh=20LUONG=20=28L=C6=AF=C6=A0NG=20Th=E1=BA=BF=20Vi?= =?UTF-8?q?nh=29?= Date: Sun, 15 Sep 2024 18:06:50 -0700 Subject: [PATCH] rename examples/FinanceBench/rag module --- examples/FinanceBench/Makefile | 2 +- examples/FinanceBench/make.bat | 2 +- examples/FinanceBench/rag-test.py | 2 +- examples/FinanceBench/{rag_default.py => rag.py} | 0 examples/FinanceBench/streamlit-main.py | 2 +- tests/core/resource/test_file_resource.py | 2 +- 6 files changed, 5 insertions(+), 5 deletions(-) rename examples/FinanceBench/{rag_default.py => rag.py} (100%) diff --git a/examples/FinanceBench/Makefile b/examples/FinanceBench/Makefile index 66a41df49..d75b9bec1 100644 --- a/examples/FinanceBench/Makefile +++ b/examples/FinanceBench/Makefile @@ -41,7 +41,7 @@ openai-assist: rag-default-answer: - @poetry run python rag_default.py ${id} + @poetry run python rag.py ${id} rag-test: @poetry run python rag-test.py ${doc} diff --git a/examples/FinanceBench/make.bat b/examples/FinanceBench/make.bat index 67f51a9bb..ed8d2e8e7 100644 --- a/examples/FinanceBench/make.bat +++ b/examples/FinanceBench/make.bat @@ -85,7 +85,7 @@ IF "%TARGET%"=="streamlit-run" GOTO streamlit-run :rag-default-answer - poetry run python rag_default.py %2 + poetry run python rag.py %2 GOTO end :rag-test diff --git a/examples/FinanceBench/rag-test.py b/examples/FinanceBench/rag-test.py index 4f0a40029..20d32bb0d 100644 --- a/examples/FinanceBench/rag-test.py +++ b/examples/FinanceBench/rag-test.py @@ -7,7 +7,7 @@ # pylint: disable=wrong-import-order from data_and_knowledge import DocName, Doc, RAG_GROUND_TRUTHS from eval import get_lm, EVAL_PROMPT_TEMPLATE -from rag_default import get_or_create_file_resource +from rag import get_or_create_file_resource DEFS: dict[str, str] = RAG_GROUND_TRUTHS['defs'] diff --git a/examples/FinanceBench/rag_default.py b/examples/FinanceBench/rag.py similarity index 100% rename from examples/FinanceBench/rag_default.py rename to examples/FinanceBench/rag.py diff --git a/examples/FinanceBench/streamlit-main.py b/examples/FinanceBench/streamlit-main.py index 4dca23d76..7f3111fbb 100644 --- a/examples/FinanceBench/streamlit-main.py +++ b/examples/FinanceBench/streamlit-main.py @@ -6,7 +6,7 @@ from data_and_knowledge import DocName, Doc, DOC_NAMES, ExpertPlanId as TaskId, EXPERT_PROGRAMS from htp_oodar_agent import get_or_create_agent, get_or_create_adaptations -from rag_default import get_or_create_file_resource +from rag import get_or_create_file_resource TASK_IDS: list[TaskId] = list(EXPERT_PROGRAMS) diff --git a/tests/core/resource/test_file_resource.py b/tests/core/resource/test_file_resource.py index 23d418647..cb00a6ddd 100644 --- a/tests/core/resource/test_file_resource.py +++ b/tests/core/resource/test_file_resource.py @@ -7,7 +7,7 @@ sys.path.append(str(FINANCE_BENCH_DIR_PATH)) # pylint: disable=wrong-import-order,wrong-import-position -from rag_default import get_or_create_file_resource # noqa: E402 +from rag import get_or_create_file_resource # noqa: E402 def test_file_resource():