统一database config,修复.env

src/plugins/memory_system/draw_memory.py

@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+import os
 import sys
 import jieba
 from llm_module import LLMModel
@@ -153,9 +154,12 @@ class Memory_graph:
 def main():
     # 初始化数据库
     Database.initialize(
-        "127.0.0.1",
-        27017,
-        "MegBot"
+        host= os.getenv("MONGODB_HOST"),
+        port= int(os.getenv("MONGODB_PORT")),
+        db_name=  os.getenv("DATABASE_NAME"),
+        username= os.getenv("MONGODB_USERNAME"),
+        password= os.getenv("MONGODB_PASSWORD"),
+        auth_source=os.getenv("MONGODB_AUTH_SOURCE")
     )
     
     memory_graph = Memory_graph()

src/plugins/memory_system/llm_module.py

@@ -1,12 +1,8 @@
 import os
 import requests
-from dotenv import load_dotenv
 from typing import Tuple, Union
 import time
 
-# 加载环境变量
-load_dotenv()
-
 class LLMModel:
     # def __init__(self, model_name="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", **kwargs):
     def __init__(self, model_name="Pro/deepseek-ai/DeepSeek-V3", **kwargs):

src/plugins/memory_system/llm_module_memory_make.py

@@ -1,23 +1,9 @@
 import os
 import requests
-from dotenv import load_dotenv
 from typing import Tuple, Union
 import time
 from ..chat.config import BotConfig
 
-# 获取当前文件的绝对路径
-current_dir = os.path.dirname(os.path.abspath(__file__))
-root_dir = os.path.abspath(os.path.join(current_dir, '..', '..', '..'))
-env_path = os.path.join(root_dir, 'config', '.env')
-
-# 加载环境变量
-print(f"尝试从 {env_path} 加载环境变量配置")
-if os.path.exists(env_path):
-    load_dotenv(env_path)
-    print("成功加载环境变量配置")
-else:
-    print(f"环境变量配置文件不存在: {env_path}")
-
 class LLMModel:
     # def __init__(self, model_name="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", **kwargs):
     def __init__(self, model_name="Pro/deepseek-ai/DeepSeek-V3", **kwargs):

src/plugins/memory_system/memory_make.py

@@ -9,7 +9,6 @@ import datetime
 import random
 import time
 import os
-from dotenv import load_dotenv
 # from chat.config import global_config
 sys.path.append("C:/GitHub/MaiMBot")  # 添加项目根目录到 Python 路径
 from src.common.database import Database  # 使用正确的导入语法
@@ -219,24 +218,14 @@ def calculate_information_content(text):
 
 
 def main():
-    # 获取当前文件的绝对路径
-    current_dir = os.path.dirname(os.path.abspath(__file__))
-    root_dir = os.path.abspath(os.path.join(current_dir, '..', '..', '..'))
-    env_path = os.path.join(root_dir, 'config', '.env')
-
-    # 加载环境变量
-    print(f"尝试从 {env_path} 加载环境变量配置")
-    if os.path.exists(env_path):
-        load_dotenv(env_path)
-        print("成功加载环境变量配置")
-    else:
-        print(f"环境变量配置文件不存在: {env_path}")
-
     # 初始化数据库
     Database.initialize(
-        "127.0.0.1",
-        27017,
-        "MegBot"
+          host= os.getenv("MONGODB_HOST"),
+          port= int(os.getenv("MONGODB_PORT")),
+          db_name=  os.getenv("DATABASE_NAME"),
+          username= os.getenv("MONGODB_USERNAME"),
+          password= os.getenv("MONGODB_PASSWORD"),
+          auth_source=os.getenv("MONGODB_AUTH_SOURCE")
     )
     
     memory_graph = Memory_graph()