优化日志细节
This commit is contained in:
parent
5f532f3aea
commit
d1c4b0d1ff
@ -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":
|
||||
|
||||
@ -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()
|
||||
|
||||
@ -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)
|
||||
|
||||
@ -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)
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user