Update megumin_agent/bootstrap.py

megumin_agent/bootstrap.py

@@ -30,6 +30,18 @@ def _metadata_filename() -> str:
     return os.getenv("MEGUMIN_FAISS_METADATA_FILENAME", "megumin_questions_meta.json")
 
 
+def _fact_dataset_filename() -> str:
+    return os.getenv("MEGUMIN_HF_FACT_DATASET_FILENAME", "namuwiki_qa.json")
+
+
+def _fact_index_filename() -> str:
+    return os.getenv("MEGUMIN_HF_FACT_INDEX_FILENAME", "namuwiki_questions.faiss")
+
+
+def _fact_metadata_filename() -> str:
+    return os.getenv("MEGUMIN_HF_FACT_METADATA_FILENAME", "namuwiki_questions_meta.json")
+
+
 def bootstrap_environment() -> None:
     load_dotenv(PROJECT_ROOT / ".env", override=True)
     if ADK_SRC.exists():
@@ -48,6 +60,9 @@ def resolve_dataset_dir() -> Path:
             _dataset_filename(),
             _index_filename(),
             _metadata_filename(),
+            _fact_dataset_filename(),
+            _fact_index_filename(),
+            _fact_metadata_filename(),
         )
         for artifact_name in artifact_names:
             try:
@@ -59,7 +74,7 @@ def resolve_dataset_dir() -> Path:
                     local_dir=str(RUNTIME_DATASET_DIR),
                 )
             except Exception:
-                if artifact_name != _dataset_filename():
+                if artifact_name not in {_dataset_filename(), _fact_dataset_filename()}:
                     continue
                 raise
         return RUNTIME_DATASET_DIR