admin-valentin/AIClient-2-API
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: remove content deduplication to fix token counting and display

Browse source
This commit is contained in:
Sanyela 2025-12-03 14:07:51 +08:00
parent d7a2332e24
commit 82314ce018
1 changed files with 5 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/claude/claude-kiro.js
View file

@ -1098,7 +1098,6 @@ async initializeAuth(forceRefresh = false) {
const stream = response.data; const stream = response.data;
let buffer = ''; let buffer = '';
const processedPositions = new Set(); // 避免重复处理
for await (const chunk of stream) { for await (const chunk of stream) {
buffer += chunk.toString(); buffer += chunk.toString();
@ -1107,16 +1106,12 @@ async initializeAuth(forceRefresh = false) {
const { events, remaining } = this.parseAwsEventStreamBuffer(buffer); const { events, remaining } = this.parseAwsEventStreamBuffer(buffer);
buffer = remaining; buffer = remaining;
// 只 yield 新的事件 // yield 所有事件（不再去重，因为重复内容是有效的）
for (const event of events) { for (const event of events) {
const eventKey = `${event.type}:${event.data}`; if (event.type === 'content' && event.data) {
if (!processedPositions.has(eventKey)) { yield { type: 'content', content: event.data };
processedPositions.add(eventKey); } else if (event.type === 'toolUse') {
if (event.type === 'content' && event.data) { yield { type: 'toolUse', toolUse: event.data };
yield { type: 'content', content: event.data };
} else if (event.type === 'toolUse') {
yield { type: 'toolUse', toolUse: event.data };
}
} }
} }
} }