Add GENERATION_ENTERED event

ba52ceb979fcda143e069bed0f5828042be2c5e6

Cohee <18619528+Cohee1207@users.noreply.github.com>

Parents: e814d8c019232a8ef79f81ed914b0c3f56bbae5d
1 files changed, +5 -0Showing whitespace changes
public/script.js+5 -0
@@ -412,6 +412,7 @@ export const event_types = {
412 MESSAGE_FILE_EMBEDDED: 'message_file_embedded',412 MESSAGE_FILE_EMBEDDED: 'message_file_embedded',
413 IMPERSONATE_READY: 'impersonate_ready',413 IMPERSONATE_READY: 'impersonate_ready',
414 CHAT_CHANGED: 'chat_id_changed',414 CHAT_CHANGED: 'chat_id_changed',
415 GENERATION_ENTERED: 'generation_entered',
415 GENERATION_STARTED: 'generation_started',416 GENERATION_STARTED: 'generation_started',
416 GENERATION_STOPPED: 'generation_stopped',417 GENERATION_STOPPED: 'generation_stopped',
417 GENERATION_ENDED: 'generation_ended',418 GENERATION_ENDED: 'generation_ended',
@@ -3339,6 +3340,9 @@ export async function Generate(type, { automatic_trigger, force_name2, quiet_pro
3339 setGenerationProgress(0);3340 setGenerationProgress(0);
3340 generation_started = new Date();3341 generation_started = new Date();
33413342
3343 // Occurs every time, even if the generation is aborted due to slash commands execution
3344 await eventSource.emit(event_types.GENERATION_ENTERED, type, { automatic_trigger, force_name2, quiet_prompt, quietToLoud, skipWIAN, force_chid, signal, quietImage }, dryRun);
3345
3342 // Don't recreate abort controller if signal is passed3346 // Don't recreate abort controller if signal is passed
3343 if (!(abortController && signal)) {3347 if (!(abortController && signal)) {
3344 abortController = new AbortController();3348 abortController = new AbortController();
@@ -3358,6 +3362,7 @@ export async function Generate(type, { automatic_trigger, force_name2, quiet_pro
3358 }3362 }
3359 }3363 }
33603364
3365 // Occurs only if the generation is not aborted due to slash commands execution
3361 await eventSource.emit(event_types.GENERATION_STARTED, type, { automatic_trigger, force_name2, quiet_prompt, quietToLoud, skipWIAN, force_chid, signal, quietImage }, dryRun);3366 await eventSource.emit(event_types.GENERATION_STARTED, type, { automatic_trigger, force_name2, quiet_prompt, quietToLoud, skipWIAN, force_chid, signal, quietImage }, dryRun);
33623367
3363 if (main_api == 'kobold' && kai_settings.streaming_kobold && !kai_flags.can_use_streaming) {3368 if (main_api == 'kobold' && kai_settings.streaming_kobold && !kai_flags.can_use_streaming) {