From d1c4b0d1ff8291b13af5437677021719dac3e893 Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <wenjie.zhang@stu.jiangnan.edu.cn>
Date: Tue, 11 Mar 2025 14:18:48 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E6=97=A5=E5=BF=97=E7=BB=86?=
 =?UTF-8?q?=E8=8A=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/config/__init__.py   | 1 +
 src/core/graphbase.py    | 4 +---
 src/core/startup.py      | 1 +
 src/models/chat_model.py | 1 +
 4 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/config/__init__.py b/src/config/__init__.py
index ab785c52..d0b516c5 100644
--- a/src/config/__init__.py
+++ b/src/config/__init__.py
@@ -95,6 +95,7 @@ class Config(SimpleConfig):
 
         model_provider_info = self.model_names.get(self.model_provider, {})
         self.model_dir = os.environ.get("MODEL_DIR", "")
+        logger.info(f"MODEL_DIR: {self.model_dir}; 如果是在 docker 中运行，会自动挂载 MODEL_DIR 到 /models 目录，请检查 docker compose 文件")
 
         # 检查模型提供商是否存在
         if self.model_provider != "custom":
diff --git a/src/core/graphbase.py b/src/core/graphbase.py
index 65f1d7c5..5ea803cf 100644
--- a/src/core/graphbase.py
+++ b/src/core/graphbase.py
@@ -26,9 +26,7 @@ class GraphDatabase:
         os.makedirs(self.work_dir, exist_ok=True)
 
         # 尝试加载已保存的图数据库信息
-        if self.load_graph_info():
-            logger.info(f"已加载图数据库信息：{self.kgdb_name}")
-        else:
+        if not self.load_graph_info():
             logger.info(f"未找到已保存的图数据库信息，将创建新的配置")
 
         self.start()
diff --git a/src/core/startup.py b/src/core/startup.py
index 69e301a7..f89e263d 100644
--- a/src/core/startup.py
+++ b/src/core/startup.py
@@ -21,6 +21,7 @@ class Startup:
         self.dbm = DataBaseManager(self.config)
         self.retriever = Retriever(self.config, self.dbm, self.model)
 
+        logger.info(f"Loading lite model: {self.config.model_name_lite}")
         self.model_lite = select_model(self.config,
                                        model_provider=self.config.model_provider_lite,
                                        model_name=self.config.model_name_lite)
diff --git a/src/models/chat_model.py b/src/models/chat_model.py
index 548369be..1ed3267b 100644
--- a/src/models/chat_model.py
+++ b/src/models/chat_model.py
@@ -60,6 +60,7 @@ class CustomModel(OpenAIBase):
         model_name = model_info["name"]
         api_key = model_info["api_key"]
         base_url = get_docker_safe_url(model_info["api_base"])
+        logger.info(f"> Custom model: {model_name}, base_url: {base_url}")
 
         super().__init__(api_key=api_key, base_url=base_url, model_name=model_name)