从.env中获取数据库参数

pull/8/head
jiajiu123 2025-02-28 18:00:15 +08:00
parent 78d3747128
commit a43dc99210
1 changed files with 5 additions and 3 deletions

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: