refactor: 使用 claudeCodeValidator 统一验证逻辑

替换 _hasClaudeCodeSystemPrompt 方法,改用 claudeCodeValidator

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
shaw
2025-09-30 11:25:35 +08:00
parent 66b9b391d9
commit 4ab91f233f

View File

@@ -10,6 +10,7 @@ const logger = require('../utils/logger')
const config = require('../../config/config') const config = require('../../config/config')
const claudeCodeHeadersService = require('./claudeCodeHeadersService') const claudeCodeHeadersService = require('./claudeCodeHeadersService')
const redis = require('../models/redis') const redis = require('../models/redis')
const ClaudeCodeValidator = require('../validators/clients/claudeCodeValidator')
class ClaudeRelayService { class ClaudeRelayService {
constructor() { constructor() {
@@ -22,42 +23,16 @@ class ClaudeRelayService {
// 🔍 判断是否是真实的 Claude Code 请求 // 🔍 判断是否是真实的 Claude Code 请求
isRealClaudeCodeRequest(requestBody, clientHeaders) { isRealClaudeCodeRequest(requestBody, clientHeaders) {
// 检查 user-agent 是否匹配 Claude Code 格式 // 使用 claudeCodeValidator 来进行完整的验证
const userAgent = clientHeaders?.['user-agent'] || clientHeaders?.['User-Agent'] || '' // 注意claudeCodeValidator.validate() 需要一个完整的 req 对象
const isClaudeCodeUserAgent = /^claude-cli\/[\d.]+\s+\(/i.test(userAgent) // 我们需要构造一个最小化的 req 对象来满足验证器的需求
const mockReq = {
// 检查系统提示词是否包含 Claude Code 标识(支持两种类型的提示词) headers: clientHeaders || {},
const hasClaudeCodeSystemPrompt = this._hasClaudeCodeSystemPrompt(requestBody) body: requestBody,
path: '/api/v1/messages'
// 只有当 user-agent 匹配且系统提示词正确时,才认为是真实的 Claude Code 请求
return isClaudeCodeUserAgent && hasClaudeCodeSystemPrompt
} }
// 🔍 检查请求中是否包含 Claude Code 系统提示词 return ClaudeCodeValidator.validate(mockReq)
_hasClaudeCodeSystemPrompt(requestBody) {
if (!requestBody || !requestBody.system) {
return false
}
// 如果是字符串格式,一定不是真实的 Claude Code 请求
if (typeof requestBody.system === 'string') {
return false
}
// 处理数组格式 - 检查第一个元素
if (Array.isArray(requestBody.system) && requestBody.system.length > 0) {
const firstItem = requestBody.system[0]
// 检查第一个元素是否包含 Claude Code 相关的提示词
if (firstItem && firstItem.type === 'text' && firstItem.text) {
// Claude Code 的两种典型提示词开头
return (
firstItem.text.startsWith("You are Claude Code, Anthropic's official CLI for Claude.") ||
firstItem.text.startsWith('Analyze if this message indicates a new conversation topic')
)
}
}
return false
} }
// 🚀 转发请求到Claude API // 🚀 转发请求到Claude API