117 lines
4.8 KiB
JavaScript
117 lines
4.8 KiB
JavaScript
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
|
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
};
|
|
import { html, LitElement } from "lit";
|
|
import { property, state } from "lit/decorators.js";
|
|
export class StreamingMessageContainer extends LitElement {
|
|
constructor() {
|
|
super(...arguments);
|
|
this.tools = [];
|
|
this.isStreaming = false;
|
|
this._message = null;
|
|
this._pendingMessage = null;
|
|
this._updateScheduled = false;
|
|
this._immediateUpdate = false;
|
|
}
|
|
createRenderRoot() {
|
|
return this;
|
|
}
|
|
connectedCallback() {
|
|
super.connectedCallback();
|
|
this.style.display = "block";
|
|
}
|
|
// Public method to update the message with batching for performance
|
|
setMessage(message, immediate = false) {
|
|
// Store the latest message
|
|
this._pendingMessage = message;
|
|
// If this is an immediate update (like clearing), apply it right away
|
|
if (immediate || message === null) {
|
|
this._immediateUpdate = true;
|
|
this._message = message;
|
|
this.requestUpdate();
|
|
// Cancel any pending updates since we're clearing
|
|
this._pendingMessage = null;
|
|
this._updateScheduled = false;
|
|
return;
|
|
}
|
|
// Otherwise batch updates for performance during streaming
|
|
if (!this._updateScheduled) {
|
|
this._updateScheduled = true;
|
|
requestAnimationFrame(async () => {
|
|
// Only apply the update if we haven't been cleared
|
|
if (!this._immediateUpdate && this._pendingMessage !== null) {
|
|
// Deep clone the message to ensure Lit detects changes in nested properties
|
|
// (like toolCall.arguments being mutated during streaming)
|
|
this._message = JSON.parse(JSON.stringify(this._pendingMessage));
|
|
this.requestUpdate();
|
|
}
|
|
// Reset for next batch
|
|
this._pendingMessage = null;
|
|
this._updateScheduled = false;
|
|
this._immediateUpdate = false;
|
|
});
|
|
}
|
|
}
|
|
render() {
|
|
// Show loading indicator if loading but no message yet
|
|
if (!this._message) {
|
|
if (this.isStreaming)
|
|
return html `<div class="flex flex-col gap-3 mb-3">
|
|
<span class="mx-4 inline-block w-2 h-4 bg-muted-foreground animate-pulse"></span>
|
|
</div>`;
|
|
return html ``; // Empty until a message is set
|
|
}
|
|
const msg = this._message;
|
|
if (msg.role === "toolResult") {
|
|
// Skip standalone tool result in streaming; the stable list will render paired tool-message
|
|
return html ``;
|
|
}
|
|
else if (msg.role === "user") {
|
|
// Skip standalone tool result in streaming; the stable list will render it immediiately
|
|
return html ``;
|
|
}
|
|
else if (msg.role === "assistant") {
|
|
// Assistant message - render inline tool messages during streaming
|
|
return html `
|
|
<div class="flex flex-col gap-3 mb-3">
|
|
<assistant-message
|
|
.message=${msg}
|
|
.tools=${this.tools}
|
|
.isStreaming=${this.isStreaming}
|
|
.pendingToolCalls=${this.pendingToolCalls}
|
|
.toolResultsById=${this.toolResultsById}
|
|
.hideToolCalls=${false}
|
|
.onCostClick=${this.onCostClick}
|
|
></assistant-message>
|
|
${this.isStreaming ? html `<span class="mx-4 inline-block w-2 h-4 bg-muted-foreground animate-pulse"></span>` : ""}
|
|
</div>
|
|
`;
|
|
}
|
|
}
|
|
}
|
|
__decorate([
|
|
property({ type: Array })
|
|
], StreamingMessageContainer.prototype, "tools", void 0);
|
|
__decorate([
|
|
property({ type: Boolean })
|
|
], StreamingMessageContainer.prototype, "isStreaming", void 0);
|
|
__decorate([
|
|
property({ type: Object })
|
|
], StreamingMessageContainer.prototype, "pendingToolCalls", void 0);
|
|
__decorate([
|
|
property({ type: Object })
|
|
], StreamingMessageContainer.prototype, "toolResultsById", void 0);
|
|
__decorate([
|
|
property({ attribute: false })
|
|
], StreamingMessageContainer.prototype, "onCostClick", void 0);
|
|
__decorate([
|
|
state()
|
|
], StreamingMessageContainer.prototype, "_message", void 0);
|
|
// Register custom element
|
|
if (!customElements.get("streaming-message-container")) {
|
|
customElements.define("streaming-message-container", StreamingMessageContainer);
|
|
}
|
|
//# sourceMappingURL=StreamingMessageContainer.js.map
|