style(gemini): apply linter fixes for code quality check

This commit is contained in:
Anonymous Contributor
2026-02-14 15:54:27 +08:00
parent 3f0dabc5fa
commit 3f81afabce

View File

@@ -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) => {