fix: 改进服务可用性检测 - 本地VLLM实际测试连接,智谱rerank降级到LLM方案
All checks were successful
构建并部署 AI Agent 服务 / deploy (push) Successful in 5m52s
All checks were successful
构建并部署 AI Agent 服务 / deploy (push) Successful in 5m52s
This commit is contained in:
@@ -277,22 +277,10 @@ class ZhipuRerankProvider(BaseServiceProvider[BaseRerankService]):
|
||||
def is_available(self) -> bool:
|
||||
"""
|
||||
检查智谱 API 重排服务是否可用
|
||||
注意:zhipuai 库 2.0.1 版本没有 rerank API,直接返回 False,降级到 LLM 方案
|
||||
"""
|
||||
if not ZHIPUAI_API_KEY:
|
||||
logger.warning("ZHIPUAI_API_KEY 未配置")
|
||||
return False
|
||||
|
||||
try:
|
||||
service = ZhipuRerankService(model=self._model)
|
||||
test_scores = service.compute_scores("test query", ["test document"])
|
||||
logger.info(f"智谱重排服务可用")
|
||||
return True
|
||||
except ImportError:
|
||||
logger.warning("zhipuai 库未安装")
|
||||
return False
|
||||
except Exception as e:
|
||||
logger.warning(f"智谱重排服务不可用: {e}")
|
||||
return False
|
||||
logger.warning("智谱 rerank API 在当前 zhipuai 库版本不可用,降级到 LLM 方案")
|
||||
return False
|
||||
|
||||
def get_service(self) -> BaseRerankService:
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user