From 0528586ad2ef4c30f99de7d493b3ca2ba4652650 Mon Sep 17 00:00:00 2001 From: NepPure Date: Sun, 2 Mar 2025 11:25:48 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E9=81=97=E6=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/gui/reasoning_gui.py | 9 ++++++--- src/plugins/memory_system/memory.py | 10 +++++++--- 2 files changed, 13 insertions(+), 6 deletions(-) diff --git a/src/gui/reasoning_gui.py b/src/gui/reasoning_gui.py index 356be3bd..f12b5979 100644 --- a/src/gui/reasoning_gui.py +++ b/src/gui/reasoning_gui.py @@ -331,9 +331,12 @@ class ReasoningGUI: 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") ) app = ReasoningGUI() diff --git a/src/plugins/memory_system/memory.py b/src/plugins/memory_system/memory.py index af6aab39..4deb28d6 100644 --- a/src/plugins/memory_system/memory.py +++ b/src/plugins/memory_system/memory.py @@ -1,4 +1,5 @@ # -*- coding: utf-8 -*- +import os import jieba from .llm_module import LLMModel import networkx as nx @@ -263,9 +264,12 @@ def topic_what(text, topic): start_time = time.time() Database.initialize( - global_config.MONGODB_HOST, - global_config.MONGODB_PORT, - global_config.DATABASE_NAME + 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()