let5see/droid2api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
package.json
View File

@@ -1,6 +1,6 @@
{ {
"name": "droid2api", "name": "droid2api",
"version": "1.3.3", "version": "1.3.4",
"description": "OpenAI Compatible API Proxy", "description": "OpenAI Compatible API Proxy",
"main": "server.js", "main": "server.js",
"type": "module", "type": "module",

9
routes.js
View File

@@ -514,7 +514,7 @@ async function handleCountTokens(req, res) {
try { try {
const anthropicRequest = req.body; const anthropicRequest = req.body;
const modelId = anthropicRequest.model; const modelId = getRedirectedModelId(anthropicRequest.model);
if (!modelId) { if (!modelId) {
return res.status(400).json({ error: 'model is required' }); return res.status(400).json({ error: 'model is required' });
@@ -559,13 +559,16 @@ async function handleCountTokens(req, res) {
// 构建 count_tokens 端点 URL // 构建 count_tokens 端点 URL
const countTokensUrl = endpoint.base_url.replace('/v1/messages', '/v1/messages/count_tokens'); 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}`); logInfo(`Forwarding to count_tokens endpoint: ${countTokensUrl}`);
logRequest('POST', countTokensUrl, headers, anthropicRequest); logRequest('POST', countTokensUrl, headers, modifiedRequest);
const response = await fetch(countTokensUrl, { const response = await fetch(countTokensUrl, {
method: 'POST', method: 'POST',
headers, headers,
body: JSON.stringify(anthropicRequest) body: JSON.stringify(modifiedRequest)
}); });
logInfo(`Response status: ${response.status}`); logInfo(`Response status: ${response.status}`);