rename examples/FinanceBench/rag module

aitomatic · Sep 16, 2024 · 9f131ef · 9f131ef
1 parent f95a87f
commit 9f131ef
Show file tree

Hide file tree

Showing 6 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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 → examples/FinanceBench/rag.py b/examples/FinanceBench/rag_default.py → examples/FinanceBench/rag.py
diff --git 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
@@ -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():