mirror of
https://github.com/Wei-Shaw/claude-relay-service.git
synced 2026-01-23 00:53:33 +00:00
feat: 重构和更新 Claude Code 验证器组件
- 重构 src/utils/contents.js,添加完整的 prompt 定义和相似度检测功能 - 删除重复的 src/utils/text-similarity.js 文件 - 更新 claudeCodeValidator.js 中的验证逻辑和错误处理 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -1,12 +1,6 @@
|
||||
const logger = require('../../utils/logger')
|
||||
const { CLIENT_DEFINITIONS } = require('../clientDefinitions')
|
||||
const {
|
||||
haikuSystemPrompt,
|
||||
claudeOtherSystemPrompt1,
|
||||
claudeOtherSystemPrompt2,
|
||||
claudeOtherSystemPromptCompact
|
||||
} = require('../../utils/contents')
|
||||
const { simple: similaritySimple } = require('../../utils/text-similarity')
|
||||
const { bestSimilarityByTemplates, SYSTEM_PROMPT_THRESHOLD } = require('../../utils/contents')
|
||||
|
||||
/**
|
||||
* Claude Code CLI 验证器
|
||||
@@ -56,35 +50,17 @@ class ClaudeCodeValidator {
|
||||
return false
|
||||
}
|
||||
|
||||
if (model.startsWith('claude-3-5-haiku')) {
|
||||
return true
|
||||
}
|
||||
|
||||
const systemEntries = Array.isArray(body.system) ? body.system : []
|
||||
const system0Text =
|
||||
systemEntries.length > 0 && typeof systemEntries[0]?.text === 'string'
|
||||
? systemEntries[0].text
|
||||
: null
|
||||
const system1Text =
|
||||
systemEntries.length > 1 && typeof systemEntries[1]?.text === 'string'
|
||||
? systemEntries[1].text
|
||||
: null
|
||||
|
||||
if (!system0Text || !system1Text) {
|
||||
return false
|
||||
for (const entry of systemEntries) {
|
||||
const rawText = typeof entry?.text === 'string' ? entry.text : ''
|
||||
const { bestScore } = bestSimilarityByTemplates(rawText)
|
||||
if (bestScore < SYSTEM_PROMPT_THRESHOLD) {
|
||||
logger.error(
|
||||
`Claude system prompt similarity below threshold: score=${bestScore.toFixed(4)}, threshold=${SYSTEM_PROMPT_THRESHOLD}, prompt=${rawText}`
|
||||
)
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
const sys0 = similaritySimple(system0Text, claudeOtherSystemPrompt1, 0.9)
|
||||
if (!sys0.passed) {
|
||||
return false
|
||||
}
|
||||
|
||||
const sys1 = similaritySimple(system1Text, claudeOtherSystemPrompt2, 0.5)
|
||||
const sysCompact = similaritySimple(system1Text, claudeOtherSystemPromptCompact, 0.9)
|
||||
if (!sys1.passed && !sysCompact.passed) {
|
||||
return false
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
@@ -101,7 +77,8 @@ class ClaudeCodeValidator {
|
||||
// 1. 先检查是否是 Claude Code 的 User-Agent
|
||||
// 格式: claude-cli/1.0.86 (external, cli) sdk-cli sdk-py
|
||||
|
||||
const claudeCodePattern = /^claude-cli\/[\d.]+(?:[-\w]*)?\s+\(external,\s*(?:cli|sdk-[a-z]+)\)$/i
|
||||
const claudeCodePattern =
|
||||
/^claude-cli\/[\d.]+(?:[-\w]*)?\s+\(external,\s*(?:cli|sdk-[a-z]+)\)$/i
|
||||
|
||||
if (!claudeCodePattern.test(userAgent)) {
|
||||
// 不是 Claude Code 的请求,此验证器不处理
|
||||
|
||||
Reference in New Issue
Block a user