fix: apply model_redirects to /v1/messages/count_tokens endpoint

This commit is contained in:
1e0n
2025-10-14 02:35:59 +08:00
parent 302c5e6657
commit 62cdc11cec
2 changed files with 7 additions and 4 deletions

View File

@@ -514,7 +514,7 @@ async function handleCountTokens(req, res) {
try {
const anthropicRequest = req.body;
const modelId = anthropicRequest.model;
const modelId = getRedirectedModelId(anthropicRequest.model);
if (!modelId) {
return res.status(400).json({ error: 'model is required' });
@@ -559,13 +559,16 @@ async function handleCountTokens(req, res) {
// 构建 count_tokens 端点 URL
const countTokensUrl = endpoint.base_url.replace('/v1/messages', '/v1/messages/count_tokens');
// 更新请求体中的模型ID为重定向后的ID
const modifiedRequest = { ...anthropicRequest, model: modelId };
logInfo(`Forwarding to count_tokens endpoint: ${countTokensUrl}`);
logRequest('POST', countTokensUrl, headers, anthropicRequest);
logRequest('POST', countTokensUrl, headers, modifiedRequest);
const response = await fetch(countTokensUrl, {
method: 'POST',
headers,
body: JSON.stringify(anthropicRequest)
body: JSON.stringify(modifiedRequest)
});
logInfo(`Response status: ${response.status}`);