fix: 改进服务可用性检测 - 本地VLLM实际测试连接,智谱rerank降级到LLM方案
All checks were successful
构建并部署 AI Agent 服务 / deploy (push) Successful in 5m52s

This commit is contained in:
2026-05-01 02:15:02 +08:00
parent 704898a8ed
commit 3b16f7dd38
2 changed files with 37 additions and 28 deletions

View File

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