从.env中获取数据库参数

This commit is contained in:
jiajiu123
2025-02-28 18:00:15 +08:00
parent 78d3747128
commit a43dc99210

View File

@@ -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: