UNPKG

@assistant-ui/react

Version:

React components for AI chat.

192 lines 6.26 kB
// src/api/MessageRuntime.ts import { getThreadMessageText } from "../utils/getThreadMessageText.mjs"; import { MessageAttachmentRuntimeImpl } from "./AttachmentRuntime.mjs"; import { EditComposerRuntimeImpl } from "./ComposerRuntime.mjs"; import { ContentPartRuntimeImpl } from "./ContentPartRuntime.mjs"; import { NestedSubscriptionSubject } from "./subscribable/NestedSubscriptionSubject.mjs"; import { SKIP_UPDATE } from "./subscribable/SKIP_UPDATE.mjs"; import { ShallowMemoizeSubject } from "./subscribable/ShallowMemoizeSubject.mjs"; var COMPLETE_STATUS = { type: "complete" }; var toContentPartStatus = (message, partIndex, part) => { if (message.role !== "assistant") return COMPLETE_STATUS; if (part.type === "tool-call") { if (!part.result) { return message.status; } else { return COMPLETE_STATUS; } } const isLastPart = partIndex === Math.max(0, message.content.length - 1); if (message.status.type === "requires-action") return COMPLETE_STATUS; return isLastPart ? message.status : COMPLETE_STATUS; }; var EMPTY_CONTENT = Object.freeze({ type: "text", text: "" }); var getContentPartState = (message, partIndex) => { let part = message.content[partIndex]; if (!part) { if (message.content.length === 0 && partIndex === 0) { part = EMPTY_CONTENT; } else { return SKIP_UPDATE; } } else if (message.content.length === 1 && part.type === "text" && part.text.length === 0) { part = EMPTY_CONTENT; } const status = toContentPartStatus(message, partIndex, part); return Object.freeze({ ...part, part, status }); }; var MessageRuntimeImpl = class { constructor(_core, _threadBinding) { this._core = _core; this._threadBinding = _threadBinding; this.composer = new EditComposerRuntimeImpl( new NestedSubscriptionSubject({ path: { ...this.path, ref: this.path.ref + `${this.path.ref}.composer`, composerSource: "edit" }, getState: () => this._threadBinding.getState().getEditComposer(this._core.getState().id), subscribe: (callback) => this._threadBinding.subscribe(callback) }), () => this._threadBinding.getState().beginEdit(this._core.getState().id) ); } get path() { return this._core.path; } composer; getState() { return this._core.getState(); } reload() { const state = this._core.getState(); if (state.role !== "assistant") throw new Error("Can only reload assistant messages"); this._threadBinding.getState().startRun(state.parentId); } speak() { const state = this._core.getState(); return this._threadBinding.getState().speak(state.id); } stopSpeaking() { const state = this._core.getState(); const thread = this._threadBinding.getState(); if (thread.speech?.messageId === state.id) { this._threadBinding.getState().stopSpeaking(); } else { throw new Error("Message is not being spoken"); } } submitFeedback({ type }) { const state = this._core.getState(); this._threadBinding.getState().submitFeedback({ messageId: state.id, type }); } switchToBranch({ position, branchId }) { const state = this._core.getState(); if (branchId && position) { throw new Error("May not specify both branchId and position"); } else if (!branchId && !position) { throw new Error("Must specify either branchId or position"); } const thread = this._threadBinding.getState(); const branches = thread.getBranches(state.id); let targetBranch = branchId; if (position === "previous") { targetBranch = branches[state.branchNumber - 2]; } else if (position === "next") { targetBranch = branches[state.branchNumber]; } if (!targetBranch) throw new Error("Branch not found"); this._threadBinding.getState().switchToBranch(targetBranch); } unstable_getCopyText() { return getThreadMessageText(this.getState()); } subscribe(callback) { return this._core.subscribe(callback); } getContentPartByIndex(idx) { if (idx < 0) throw new Error("Content part index must be >= 0"); return new ContentPartRuntimeImpl( new ShallowMemoizeSubject({ path: { ...this.path, ref: this.path.ref + `${this.path.ref}.content[${idx}]`, contentPartSelector: { type: "index", index: idx } }, getState: () => { return getContentPartState(this.getState(), idx); }, subscribe: (callback) => this._core.subscribe(callback) }), this._core, this._threadBinding ); } getContentPartByToolCallId(toolCallId) { return new ContentPartRuntimeImpl( new ShallowMemoizeSubject({ path: { ...this.path, ref: this.path.ref + `${this.path.ref}.content[toolCallId=${JSON.stringify(toolCallId)}]`, contentPartSelector: { type: "toolCallId", toolCallId } }, getState: () => { const state = this._core.getState(); const idx = state.content.findIndex( (part) => part.type === "tool-call" && part.toolCallId === toolCallId ); if (idx === -1) return SKIP_UPDATE; return getContentPartState(state, idx); }, subscribe: (callback) => this._core.subscribe(callback) }), this._core, this._threadBinding ); } getAttachmentByIndex(idx) { return new MessageAttachmentRuntimeImpl( new ShallowMemoizeSubject({ path: { ...this.path, ref: this.path.ref + `${this.path.ref}.attachments[${idx}]`, attachmentSource: "message", attachmentSelector: { type: "index", index: idx } }, getState: () => { const attachments = this.getState().attachments; const attachment = attachments?.[idx]; if (!attachment) return SKIP_UPDATE; return { ...attachment, attachment, source: "message" }; }, subscribe: (callback) => this._core.subscribe(callback) }) ); } }; export { EMPTY_CONTENT, MessageRuntimeImpl, toContentPartStatus }; //# sourceMappingURL=MessageRuntime.mjs.map