统一database config,修复.env

This commit is contained in:
NepPure
2025-03-02 10:54:31 +08:00
parent 35589649da
commit 86b76496f6
14 changed files with 46 additions and 118 deletions

View File

@@ -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()