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)