Skip to content

Commit 8f27742

Browse files
committed
fix: repair stream end
1 parent e2797c3 commit 8f27742

File tree

3 files changed

+6
-0
lines changed

3 files changed

+6
-0
lines changed

src/backend/src/modules/puterai/ClaudeService.js

+1
Original file line numberDiff line numberDiff line change
@@ -189,6 +189,7 @@ class ClaudeService extends BaseService {
189189
}
190190
}
191191
}
192+
chatStream.end();
192193
usage_promise.resolve(counts);
193194
};
194195

src/backend/src/modules/puterai/OpenAICompletionService.js

+1
Original file line numberDiff line numberDiff line change
@@ -365,6 +365,7 @@ class OpenAICompletionService extends BaseService {
365365
if ( mode === 'text' ) textblock.end();
366366
if ( mode === 'tool' ) toolblock.end();
367367
message.end();
368+
chatStream.end();
368369
};
369370

370371
return new TypedValue({ $: 'ai-chat-intermediate' }, {

src/backend/src/modules/puterai/lib/Streaming.js

+4
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,10 @@ class AIChatStream {
7474
this.stream = stream;
7575
}
7676

77+
end () {
78+
this.stream.end();
79+
}
80+
7781
message () {
7882
return new AIChatMessageStream(this);
7983
}

0 commit comments

Comments
 (0)