Update megumin_agent/bootstrap.py

megumin_agent/bootstrap.py

@@ -26,6 +26,10 @@ def _index_filename() -> str:
     return os.getenv("MEGUMIN_FAISS_INDEX_FILENAME", "megumin_questions.faiss")
 
 
+def _qa_index_filename() -> str:
+    return os.getenv("MEGUMIN_FAISS_QA_INDEX_FILENAME", "megumin_question_answer.faiss")
+
+
 def _metadata_filename() -> str:
     return os.getenv("MEGUMIN_FAISS_METADATA_FILENAME", "megumin_questions_meta.json")
 
@@ -38,6 +42,10 @@ def _fact_index_filename() -> str:
     return os.getenv("MEGUMIN_HF_FACT_INDEX_FILENAME", "namuwiki_questions.faiss")
 
 
+def _fact_qa_index_filename() -> str:
+    return os.getenv("MEGUMIN_HF_FACT_QA_INDEX_FILENAME", "namuwiki_question_answer.faiss")
+
+
 def _fact_metadata_filename() -> str:
     return os.getenv("MEGUMIN_HF_FACT_METADATA_FILENAME", "namuwiki_questions_meta.json")
 
@@ -59,9 +67,11 @@ def resolve_dataset_dir() -> Path:
         artifact_names = (
             _dataset_filename(),
             _index_filename(),
+            _qa_index_filename(),
             _metadata_filename(),
             _fact_dataset_filename(),
             _fact_index_filename(),
+            _fact_qa_index_filename(),
             _fact_metadata_filename(),
         )
         for artifact_name in artifact_names: