mirror of
https://github.com/Wei-Shaw/claude-relay-service.git
synced 2026-03-30 02:31:33 +00:00
style(gemini): apply linter fixes for code quality check
This commit is contained in:
@@ -651,10 +651,11 @@ async function handleMessages(req, res) {
|
|||||||
|
|
||||||
// 尝试从 SSE 流中提取 usage 数据
|
// 尝试从 SSE 流中提取 usage 数据
|
||||||
streamBuffer += chunkStr
|
streamBuffer += chunkStr
|
||||||
|
|
||||||
// 如果 buffer 过大,进行保护性清理(防止内存泄漏)
|
// 如果 buffer 过大,进行保护性清理(防止内存泄漏)
|
||||||
if (streamBuffer.length > 1024 * 1024) { // 1MB
|
if (streamBuffer.length > 1024 * 1024) {
|
||||||
streamBuffer = streamBuffer.slice(-1024 * 64) // 只保留最后 64KB
|
// 1MB
|
||||||
|
streamBuffer = streamBuffer.slice(-1024 * 64) // 只保留最后 64KB
|
||||||
}
|
}
|
||||||
|
|
||||||
const lines = streamBuffer.split('\n')
|
const lines = streamBuffer.split('\n')
|
||||||
@@ -1990,50 +1991,51 @@ async function handleStreamGenerateContent(req, res) {
|
|||||||
|
|
||||||
// 提取 usage 数据
|
// 提取 usage 数据
|
||||||
try {
|
try {
|
||||||
const chunkStr = chunk.toString()
|
const chunkStr = chunk.toString()
|
||||||
streamBuffer += chunkStr
|
streamBuffer += chunkStr
|
||||||
|
|
||||||
// 如果 buffer 过大,进行保护性清理(防止内存泄漏)
|
|
||||||
if (streamBuffer.length > 1024 * 1024) { // 1MB
|
|
||||||
streamBuffer = streamBuffer.slice(-1024 * 64) // 只保留最后 64KB
|
|
||||||
}
|
|
||||||
|
|
||||||
const lines = streamBuffer.split('\n')
|
// 如果 buffer 过大,进行保护性清理(防止内存泄漏)
|
||||||
// 保留最后一行(可能不完整)
|
if (streamBuffer.length > 1024 * 1024) {
|
||||||
streamBuffer = lines.pop() || ''
|
// 1MB
|
||||||
|
streamBuffer = streamBuffer.slice(-1024 * 64) // 只保留最后 64KB
|
||||||
for (const line of lines) {
|
|
||||||
// 只处理可能包含数据的行
|
|
||||||
if (!line.trim() || !line.startsWith('data:')) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
// <20><>试解析 SSE 行
|
|
||||||
const parsed = parseSSELine(line)
|
|
||||||
|
|
||||||
// 检查各种可能的 usage 位置
|
|
||||||
let extractedUsage = null
|
|
||||||
|
|
||||||
if (parsed.type === 'data') {
|
|
||||||
if (parsed.data.response?.usageMetadata) {
|
|
||||||
extractedUsage = parsed.data.response.usageMetadata
|
|
||||||
} else if (parsed.data.usageMetadata) {
|
|
||||||
extractedUsage = parsed.data.usageMetadata
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (extractedUsage) {
|
|
||||||
totalUsage = extractedUsage
|
|
||||||
logger.debug('📊 Captured Gemini usage data:', totalUsage)
|
|
||||||
}
|
|
||||||
} catch (parseError) {
|
|
||||||
// 解析失败忽略,可能是非 JSON 数据
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} catch (error) {
|
|
||||||
logger.warn('⚠️ Error extracting usage data:', error.message)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const lines = streamBuffer.split('\n')
|
||||||
|
// 保留最后一行(可能不完整)
|
||||||
|
streamBuffer = lines.pop() || ''
|
||||||
|
|
||||||
|
for (const line of lines) {
|
||||||
|
// 只处理可能包含数据的行
|
||||||
|
if (!line.trim() || !line.startsWith('data:')) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
// <20><>试解析 SSE 行
|
||||||
|
const parsed = parseSSELine(line)
|
||||||
|
|
||||||
|
// 检查各种可能的 usage 位置
|
||||||
|
let extractedUsage = null
|
||||||
|
|
||||||
|
if (parsed.type === 'data') {
|
||||||
|
if (parsed.data.response?.usageMetadata) {
|
||||||
|
extractedUsage = parsed.data.response.usageMetadata
|
||||||
|
} else if (parsed.data.usageMetadata) {
|
||||||
|
extractedUsage = parsed.data.usageMetadata
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (extractedUsage) {
|
||||||
|
totalUsage = extractedUsage
|
||||||
|
logger.debug('📊 Captured Gemini usage data:', totalUsage)
|
||||||
|
}
|
||||||
|
} catch (parseError) {
|
||||||
|
// 解析失败忽略,可能是非 JSON 数据
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
logger.warn('⚠️ Error extracting usage data:', error.message)
|
||||||
|
}
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
logger.error('Error processing stream chunk:', error)
|
logger.error('Error processing stream chunk:', error)
|
||||||
}
|
}
|
||||||
@@ -2789,24 +2791,24 @@ async function handleStandardStreamGenerateContent(req, res) {
|
|||||||
res.write(outputChunk)
|
res.write(outputChunk)
|
||||||
}
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const usageSource =
|
const usageSource =
|
||||||
processedPayload && processedPayload !== '[DONE]' ? processedPayload : dataPayload
|
processedPayload && processedPayload !== '[DONE]' ? processedPayload : dataPayload
|
||||||
|
|
||||||
if (!usageSource || !usageSource.includes('usageMetadata')) {
|
if (!usageSource || !usageSource.includes('usageMetadata')) {
|
||||||
return
|
return
|
||||||
}
|
|
||||||
|
|
||||||
const usageObj = JSON.parse(usageSource)
|
|
||||||
const usage = usageObj.usageMetadata || usageObj.response?.usageMetadata || usageObj.usage
|
|
||||||
|
|
||||||
if (usage && typeof usage === 'object') {
|
|
||||||
totalUsage = usage
|
|
||||||
logger.debug('📊 Captured Gemini usage data (async):', totalUsage)
|
|
||||||
}
|
|
||||||
} catch (error) {
|
|
||||||
// 提取用量失败时忽略
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const usageObj = JSON.parse(usageSource)
|
||||||
|
const usage = usageObj.usageMetadata || usageObj.response?.usageMetadata || usageObj.usage
|
||||||
|
|
||||||
|
if (usage && typeof usage === 'object') {
|
||||||
|
totalUsage = usage
|
||||||
|
logger.debug('📊 Captured Gemini usage data (async):', totalUsage)
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
// 提取用量失败时忽略
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
streamResponse.on('data', (chunk) => {
|
streamResponse.on('data', (chunk) => {
|
||||||
|
|||||||
Reference in New Issue
Block a user