优化日志细节

This commit is contained in:
Wenjie Zhang 2025-03-11 14:18:48 +08:00
parent 5f532f3aea
commit d1c4b0d1ff
4 changed files with 4 additions and 3 deletions

View File

@ -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":

View File

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

View File

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

View File

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