mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-17 20:05:05 +00:00
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:
parent
977e4ea6ef
commit
347d4cf729
2 changed files with 74 additions and 64 deletions
|
|
@ -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;
|
||||
}
|
||||
});
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue