diff --git a/src/utils/common.js b/src/utils/common.js index 43eacc6..f3cb01f 100644 --- a/src/utils/common.js +++ b/src/utils/common.js @@ -494,14 +494,14 @@ export async function handleStreamRequest(res, service, model, requestBody, from // 如果底层未标记,且不跳过错误计数,则在此处标记 if (!credentialMarkedUnhealthy && !skipErrorCount && providerPoolManager && pooluuid) { // 400 报错码通常是请求参数问题,不记录为提供商错误 - if (status === 400) { + if (error.code === 400) { logger.info(`[Provider Pool] Skipping unhealthy marking for ${toProvider} (${pooluuid}) due to status 400 (client error)`); } else { logger.info(`[Provider Pool] Marking ${toProvider} as unhealthy due to stream error (status: ${status || 'unknown'})`); // 如果是号池模式,并且请求处理失败,则标记当前使用的提供者为不健康 providerPoolManager.markProviderUnhealthy(toProvider, { uuid: pooluuid - }); + }, error.message); credentialMarkedUnhealthy = true; } } else if (credentialMarkedUnhealthy) { @@ -695,14 +695,14 @@ export async function handleUnaryRequest(res, service, model, requestBody, fromP // 如果底层未标记,且不跳过错误计数,则在此处标记 if (!credentialMarkedUnhealthy && !skipErrorCount && providerPoolManager && pooluuid) { // 400 报错码通常是请求参数问题,不记录为提供商错误 - if (status === 400) { + if (error.code === 400) { logger.info(`[Provider Pool] Skipping unhealthy marking for ${toProvider} (${pooluuid}) due to status 400 (client error)`); } else { logger.info(`[Provider Pool] Marking ${toProvider} as unhealthy due to unary error (status: ${status || 'unknown'})`); // 如果是号池模式,并且请求处理失败,则标记当前使用的提供者为不健康 providerPoolManager.markProviderUnhealthy(toProvider, { uuid: pooluuid - }); + }, error.message); credentialMarkedUnhealthy = true; } } else if (credentialMarkedUnhealthy) { @@ -814,7 +814,7 @@ export async function handleModelListRequest(req, res, service, endpointType, CO // 如果是号池模式,并且请求处理失败,则标记当前使用的提供者为不健康 providerPoolManager.markProviderUnhealthy(toProvider, { uuid: pooluuid - }); + }, error.message); } } }