mirror of
https://github.com/Wei-Shaw/claude-relay-service.git
synced 2026-01-22 16:43:35 +00:00
fix: droid去掉count_tokens端点
This commit is contained in:
@@ -61,8 +61,7 @@ const resolveConcurrencyConfig = () => {
|
|||||||
const TOKEN_COUNT_PATHS = new Set([
|
const TOKEN_COUNT_PATHS = new Set([
|
||||||
'/v1/messages/count_tokens',
|
'/v1/messages/count_tokens',
|
||||||
'/api/v1/messages/count_tokens',
|
'/api/v1/messages/count_tokens',
|
||||||
'/claude/v1/messages/count_tokens',
|
'/claude/v1/messages/count_tokens'
|
||||||
'/droid/claude/v1/messages/count_tokens'
|
|
||||||
])
|
])
|
||||||
|
|
||||||
function extractApiKey(req) {
|
function extractApiKey(req) {
|
||||||
|
|||||||
@@ -60,49 +60,6 @@ router.post('/claude/v1/messages', authenticateApiKey, async (req, res) => {
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
router.post('/claude/v1/messages/count_tokens', authenticateApiKey, async (req, res) => {
|
|
||||||
try {
|
|
||||||
const requestBody = { ...req.body }
|
|
||||||
if ('stream' in requestBody) {
|
|
||||||
delete requestBody.stream
|
|
||||||
}
|
|
||||||
const sessionHash = sessionHelper.generateSessionHash(requestBody)
|
|
||||||
|
|
||||||
if (!hasDroidPermission(req.apiKey)) {
|
|
||||||
logger.security(
|
|
||||||
`🚫 API Key ${req.apiKey?.id || 'unknown'} 缺少 Droid 权限,拒绝访问 ${req.originalUrl}`
|
|
||||||
)
|
|
||||||
return res.status(403).json({
|
|
||||||
error: 'permission_denied',
|
|
||||||
message: '此 API Key 未启用 Droid 权限'
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
const result = await droidRelayService.relayRequest(
|
|
||||||
requestBody,
|
|
||||||
req.apiKey,
|
|
||||||
req,
|
|
||||||
res,
|
|
||||||
req.headers,
|
|
||||||
{
|
|
||||||
endpointType: 'anthropic',
|
|
||||||
sessionHash,
|
|
||||||
customPath: '/a/v1/messages/count_tokens',
|
|
||||||
skipUsageRecord: true,
|
|
||||||
disableStreaming: true
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
res.status(result.statusCode).set(result.headers).send(result.body)
|
|
||||||
} catch (error) {
|
|
||||||
logger.error('Droid Claude count_tokens relay error:', error)
|
|
||||||
res.status(500).json({
|
|
||||||
error: 'internal_server_error',
|
|
||||||
message: error.message
|
|
||||||
})
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
// OpenAI 端点 - /v1/responses
|
// OpenAI 端点 - /v1/responses
|
||||||
router.post(['/openai/v1/responses', '/openai/responses'], authenticateApiKey, async (req, res) => {
|
router.post(['/openai/v1/responses', '/openai/responses'], authenticateApiKey, async (req, res) => {
|
||||||
try {
|
try {
|
||||||
|
|||||||
Reference in New Issue
Block a user