@assistant-ui/core 0.3.11 → 0.3.13

package/src/utils/normalizePartStatus.ts

@@ -2,7 +2,9 @@ import type {
   MessagePartStatus,
   MessagePartStreamStatus,
   ThreadAssistantMessagePart,
+  ThreadMessage,
   ThreadUserMessagePart,
+  ToolCallMessagePartStatus,
 } from "../types/message";
 
 export const COMPLETE_STATUS: MessagePartStatus = Object.freeze({
@@ -53,3 +55,28 @@ export const normalizePartStatus = (
 
   return INCOMPLETE_STATUSES[normalizedReason];
 };
+
+export const toMessagePartStatus = (
+  message: ThreadMessage,
+  partIndex: number,
+  part: ThreadUserMessagePart | ThreadAssistantMessagePart,
+): ToolCallMessagePartStatus => {
+  if (message.role !== "assistant") return COMPLETE_STATUS;
+
+  if (part.type === "tool-call") {
+    if (part.result === undefined) {
+      return message.status as ToolCallMessagePartStatus;
+    } else {
+      return COMPLETE_STATUS;
+    }
+  }
+
+  if (message.status.type === "running") {
+    const status = normalizePartStatus(part);
+    if (status) return status;
+  }
+
+  const isLastPart = partIndex === Math.max(0, message.content.length - 1);
+  if (message.status.type === "requires-action") return COMPLETE_STATUS;
+  return isLastPart ? (message.status as MessagePartStatus) : COMPLETE_STATUS;
+};

package/src/tests/default-thread-composer-runtime-core.test.ts

@@ -1,82 +0,0 @@
-import { describe, expect, it, vi } from "vitest";
-import { DefaultThreadComposerRuntimeCore } from "../runtime/base/default-thread-composer-runtime-core";
-import type { ThreadRuntimeCore } from "../runtime/interfaces/thread-runtime-core";
-import type { AppendMessage, ThreadMessage } from "../types/message";
-
-const makeRuntime = (options?: {
-  messages?: ThreadMessage[];
-  composerMetadata?: Record<string, unknown>;
-}) => {
-  const append = vi.fn();
-  const runtime = {
-    append,
-    getModelContext: () => ({
-      ...(options?.composerMetadata
-        ? { unstable_composerMetadata: options.composerMetadata }
-        : {}),
-    }),
-    subscribe: () => () => {},
-    capabilities: { cancel: false },
-    messages: options?.messages ?? [],
-    isSendDisabled: false,
-    composer: { runConfig: {} },
-  } as unknown as Omit<ThreadRuntimeCore, "composer">;
-  return { runtime, append };
-};
-
-const snapshotMessage = (id: string, state: unknown): ThreadMessage =>
-  ({
-    id,
-    role: "user",
-    createdAt: new Date(),
-    content: [{ type: "text", text: "old" }],
-    attachments: [],
-    metadata: {
-      custom: { interactables: [{ id: "n1", name: "note", state }] },
-    },
-  }) as ThreadMessage;
-
-const live = (state: unknown) => ({
-  interactables: [{ id: "n1", name: "note", state }],
-});
-
-const send = async (runtime: Omit<ThreadRuntimeCore, "composer">) => {
-  const composer = new DefaultThreadComposerRuntimeCore(runtime);
-  composer.setText("hello");
-  await composer.send();
-};
-
-describe("DefaultThreadComposerRuntimeCore interactables snapshot gating", () => {
-  it("stamps a baseline on the first send of the thread", async () => {
-    const { runtime, append } = makeRuntime({
-      composerMetadata: live({ v: 1 }),
-    });
-    await send(runtime);
-    const appended = append.mock.calls[0]![0] as AppendMessage;
-    expect(appended.metadata?.custom?.interactables).toEqual([
-      { id: "n1", name: "note", state: { v: 1 } },
-    ]);
-  });
-
-  it("stamps nothing when the history already carries the live state", async () => {
-    const { runtime, append } = makeRuntime({
-      messages: [snapshotMessage("msg-1", { v: 1 })],
-      composerMetadata: live({ v: 1 }),
-    });
-    await send(runtime);
-    const appended = append.mock.calls[0]![0] as AppendMessage;
-    expect(appended.metadata?.custom?.interactables).toBeUndefined();
-  });
-
-  it("stamps the new state when it diverged from the latest snapshot", async () => {
-    const { runtime, append } = makeRuntime({
-      messages: [snapshotMessage("msg-1", { v: 1 })],
-      composerMetadata: live({ v: 2 }),
-    });
-    await send(runtime);
-    const appended = append.mock.calls[0]![0] as AppendMessage;
-    expect(appended.metadata?.custom?.interactables).toEqual([
-      { id: "n1", name: "note", state: { v: 2 } },
-    ]);
-  });
-});