diff --git a/src/plugins/schedule/schedule_generator.py b/src/plugins/schedule/schedule_generator.py index 72d7dd523..415e278c0 100644 --- a/src/plugins/schedule/schedule_generator.py +++ b/src/plugins/schedule/schedule_generator.py @@ -1,6 +1,8 @@ import datetime +import os from typing import List, Dict from .schedule_llm_module import LLMModel +from dotenv import load_dotenv from ...common.database import Database # 使用正确的导入语法 @@ -10,9 +12,9 @@ from ...common.database import Database # 使用正确的导入语法 # from src.common.database import Database # 使用正确的导入语法 Database.initialize( - "127.0.0.1", - 27017, - "MegBot" + os.getenv("MONGODB_HOST"), + int(os.getenv("MONGODB_PORT")), + os.getenv("DATABASE_NAME") ) class ScheduleGenerator: