Update megumin_agent/agent.py

megumin_agent/agent.py

@@ -3,8 +3,8 @@ from __future__ import annotations
 import os
 from typing import Any
 
-from .bootstrap import PROJECT_ROOT
 from .bootstrap import bootstrap_environment
+from .bootstrap import resolve_dataset_dir
 
 bootstrap_environment()
 
@@ -14,7 +14,7 @@ from google.adk.tools.tool_context import ToolContext
 
 from .retrieval import JsonQaRetriever
 
-DATASET_DIR = PROJECT_ROOT / "data" / "processed"
+DATASET_DIR = resolve_dataset_dir()
 MODEL_NAME = os.getenv("MEGUMIN_AGENT_MODEL", "gemini-3.1-flash-lite-preview")