fix: restore SSE streaming via StreamConsumer for text messages

This commit is contained in:
admin
2026-05-05 15:16:01 +00:00
Unverified
parent 5312694f6d
commit e49e379096

View File

@@ -675,24 +675,31 @@ export async function initBot(config, api, tools, skills, agents) {
await queueRequest(key, text, async () => {
await ctx.api.sendChatAction(ctx.chat.id, 'typing');
// Use self-correction + direct send (streaming disabled for reliability)
// Create stream consumer for real-time edit-in-place
const consumer = new StreamConsumer(ctx, { editInterval: 1000 });
const runPromise = consumer.run();
// Wrap chatWithAI with self-correction + streaming
const chatWithCorrection = withSelfCorrection(async (msgs) => {
return await chatWithAI(msgs);
return await chatWithAI(msgs, { onDelta: (token) => consumer.onDelta(token) });
});
const result = await chatWithCorrection([
{ role: 'system', content: buildSystemPrompt(svc) },
{ role: 'user', content: text },
]);
// Send formatted response
if (result) {
logger.info(`📤 Sending response (${result.length} chars)`);
// Signal completion and wait for final edit
consumer.finish();
await runPromise;
// If streaming failed to deliver (no message sent), fallback to plain send
if (!consumer.alreadySent && result) {
logger.info(`📤 Streaming failed, fallback send (${result.length} chars)`);
await sendFormatted(ctx, result);
logger.info('✅ Response delivered');
} else {
logger.warn('Empty response from AI — nothing to send');
logger.info('✅ Stream delivered');
}
// ── Self-learning: extract patterns from this interaction ──