Przeglądaj źródła

add dotenv file

jihuaqiang 3 tygodni temu
rodzic
commit
c90cd808f2
1 zmienionych plików z 5 dodań i 0 usunięć
  1. 5 0
      create/generate_persona.py

+ 5 - 0
create/generate_persona.py

@@ -6,6 +6,11 @@ import asyncio
 import sys
 from pathlib import Path
 
+from dotenv import load_dotenv
+
+# 加载项目根目录 .env
+load_dotenv(Path(__file__).parent.parent / ".env")
+
 from create.utils import run_agent
 
 DEFAULT_MODEL = "google/gemini-3-flash-preview"