Fix AgentInterface to requestUpdate on all message lifecycle events

Was only handling message_update, now also handles message_start,
message_end, turn_start, turn_end, agent_start, agent_end.
This commit is contained in:
Mario Zechner 2025-12-28 11:22:56 +01:00
parent 977e4ea6ef
commit 347d4cf729
2 changed files with 74 additions and 64 deletions

View file

@ -131,13 +131,23 @@ export class AgentInterface extends LitElement {
}
if (!this.session) return;
this._unsubscribeSession = this.session.subscribe(async (ev: AgentEvent) => {
if (ev.type === "message_update") {
if (this._streamingContainer) {
const isStreaming = this.session?.state.isStreaming || false;
this._streamingContainer.isStreaming = isStreaming;
this._streamingContainer.setMessage(ev.message, !isStreaming);
}
this.requestUpdate();
switch (ev.type) {
case "message_start":
case "message_end":
case "turn_start":
case "turn_end":
case "agent_start":
case "agent_end":
this.requestUpdate();
break;
case "message_update":
if (this._streamingContainer) {
const isStreaming = this.session?.state.isStreaming || false;
this._streamingContainer.isStreaming = isStreaming;
this._streamingContainer.setMessage(ev.message, !isStreaming);
}
this.requestUpdate();
break;
}
});
}