Skip to content

Commit 8db03e4

Browse files
committed
Update worker_events.py
1 parent 3e6b12d commit 8db03e4

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

tldw_chatbook/Event_Handlers/worker_events.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ async def handle_api_call_worker_state_changed(app: 'TldwCli', event: Worker.Sta
7777
error_msg_text = Text.from_markup(
7878
f"[bold red]Error:[/]\nAI response for worker '{worker_name}' received, but its display widget was missing.")
7979
# Use plain text for ChatMessage content if it doesn't support Text directly
80-
app.notify(f"Error: AI response for worker '{worker_name}' received, but its display widget was missing.", severity="error", timeout=10)
80+
#app.notify(f"Error: AI response for worker '{worker_name}' received, but its display widget was missing.", severity="error", timeout=3)
8181
except QueryError:
8282
logger.error(f"Fallback: Could not find chat container #{prefix}-log.")
8383
app.current_ai_message_widget = None
@@ -220,7 +220,16 @@ async def handle_api_call_worker_state_changed(app: 'TldwCli', event: Worker.Sta
220220

221221
static_text_widget_in_ai_msg.update(Text.from_markup(f"[bold red]{escaped_error_for_display}[/]"))
222222
ai_message_widget.mark_generation_complete()
223-
app.current_ai_message_widget = None
223+
224+
if app.current_chat_is_streaming:
225+
logger.info(f"Worker '{worker_name}' failed during an active stream. Posting StreamDone with error.")
226+
# Post StreamDone event with the error details
227+
app.post_message(StreamDone(full_text=ai_message_widget.message_text, error=str(error_from_worker)))
228+
# DO NOT set app.current_ai_message_widget = None here.
229+
# The handle_stream_done handler will be responsible for this.
230+
else:
231+
logger.info(f"Worker '{worker_name}' failed (non-streaming). Clearing current_ai_message_widget.")
232+
app.current_ai_message_widget = None
224233

225234
if chat_container.is_mounted:
226235
chat_container.scroll_end(animate=True)

0 commit comments

Comments
 (0)