From a43dc99210476e6361dd8d769533f90c56c11ff3 Mon Sep 17 00:00:00 2001
From: jiajiu123 <1771663559@qq.com>
Date: Fri, 28 Feb 2025 18:00:15 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BB=8E.env=E4=B8=AD=E8=8E=B7=E5=8F=96?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=BA=93=E5=8F=82=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/plugins/schedule/schedule_generator.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

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: