feat: project agent messages for assistant ui

2026-07-10 06:22:21 +07:00
parent 3e4c660ac1
commit 143f11f74c
2 changed files with 239 additions and 0 deletions
@@ -0,0 +1,144 @@
+import { describe, expect, it } from "vitest";
+import type { AgentMessage } from "../../demo/agent/events.js";
+import { projectAgentMessagesForAssistant } from "./assistantRuntimeProjection.js";
+
+describe("assistantRuntimeProjection", () => {
+  it("projects text and tool call parts into assistant-ui content parts", () => {
+    const messages: ReadonlyArray<AgentMessage> = [
+      {
+        id: "assistant-1",
+        role: "assistant",
+        parts: [
+          { type: "text", text: "I will inspect the run." },
+          {
+            type: "tool-call",
+            call: { id: "call-1", name: "readRunTrace", input: { run_id: "run_1" } },
+          },
+        ],
+      },
+    ];
+
+    const projected = projectAgentMessagesForAssistant(messages);
+
+    expect(projected).toHaveLength(1);
+    expect(projected[0]).toMatchObject({
+      id: "assistant-1",
+      role: "assistant",
+    });
+    expect(projected[0]?.content).toEqual([
+      { type: "text", text: "I will inspect the run." },
+      {
+        type: "tool-call",
+        toolCallId: "call-1",
+        toolName: "readRunTrace",
+        args: { run_id: "run_1" },
+      },
+    ]);
+  });
+
+  it("projects tool results as tool-role messages", () => {
+    const messages: ReadonlyArray<AgentMessage> = [
+      {
+        id: "tool-result-message",
+        role: "assistant",
+        parts: [
+          {
+            type: "tool-result",
+            result: {
+              callId: "call-1",
+              name: "readRunTrace",
+              status: "success",
+              output: { frames: 3 },
+            },
+          },
+        ],
+      },
+    ];
+
+    const projected = projectAgentMessagesForAssistant(messages);
+
+    expect(projected[0]).toMatchObject({
+      id: "tool-result-message",
+      role: "tool",
+      content: [
+        {
+          type: "tool-result",
+          toolCallId: "call-1",
+          toolName: "readRunTrace",
+          result: { frames: 3 },
+          isError: false,
+        },
+      ],
+    });
+  });
+
+  it("projects approval requests as human tool calls with contract metadata", () => {
+    const messages: ReadonlyArray<AgentMessage> = [
+      {
+        id: "approval",
+        role: "assistant",
+        parts: [
+          {
+            type: "approval-request",
+            callId: "call-approval",
+            name: "resumeIssueReview",
+            prompt: "Submit resume request?",
+            contract: {
+              kind: "issue_review",
+              outcomes: ["submitted", "cancelled"],
+              resumeSchema: { type: "object" },
+              resumePayloadPreview: { selected_issue_ids: ["risk-1"] },
+              runId: "run_recorded_lda_report",
+            },
+          },
+        ],
+      },
+    ];
+
+    const projected = projectAgentMessagesForAssistant(messages);
+
+    expect(projected[0]?.content).toEqual([
+      { type: "text", text: "Submit resume request?" },
+      {
+        type: "tool-call",
+        toolCallId: "call-approval",
+        toolName: "resumeIssueReview",
+        args: {
+          prompt: "Submit resume request?",
+          contract: {
+            kind: "issue_review",
+            outcomes: ["submitted", "cancelled"],
+            resumeSchema: { type: "object" },
+            resumePayloadPreview: { selected_issue_ids: ["risk-1"] },
+            runId: "run_recorded_lda_report",
+          },
+        },
+      },
+    ]);
+  });
+
+  it("keeps presentation actions and errors visible as text/tool evidence", () => {
+    const messages: ReadonlyArray<AgentMessage> = [
+      {
+        id: "mixed",
+        role: "assistant",
+        parts: [
+          { type: "presentation-action", action: { type: "selectWorkflowNode", nodeId: "review_issues" } },
+          { type: "error", message: "provider failed" },
+        ],
+      },
+    ];
+
+    const projected = projectAgentMessagesForAssistant(messages);
+
+    expect(projected[0]?.content).toEqual([
+      {
+        type: "tool-call",
+        toolCallId: "presentation-selectWorkflowNode",
+        toolName: "presentation.selectWorkflowNode",
+        args: { type: "selectWorkflowNode", nodeId: "review_issues" },
+      },
+      { type: "text", text: "provider failed" },
+    ]);
+  });
+});
@@ -0,0 +1,95 @@
+import type { AgentMessage, AgentMessagePart } from "../../demo/agent/events.js";
+
+export type AssistantToolRenderPayload = {
+  readonly toolCallId: string;
+  readonly toolName: string;
+  readonly args?: unknown;
+  readonly result?: unknown;
+  readonly isError?: boolean;
+};
+
+type AssistantContentPart =
+  | { readonly type: "text"; readonly text: string }
+  | {
+      readonly type: "tool-call";
+      readonly toolCallId: string;
+      readonly toolName: string;
+      readonly args: unknown;
+    }
+  | {
+      readonly type: "tool-result";
+      readonly toolCallId: string;
+      readonly toolName: string;
+      readonly result: unknown;
+      readonly isError: boolean;
+    };
+
+export type AssistantProjectedMessage = {
+  readonly id: string;
+  readonly role: "user" | "assistant" | "tool";
+  readonly content: ReadonlyArray<AssistantContentPart>;
+  readonly metadata?: { readonly unstable_state?: string | undefined };
+};
+
+const projectPart = (part: AgentMessagePart): AssistantContentPart[] => {
+  switch (part.type) {
+    case "text":
+      return [{ type: "text", text: part.text }];
+    case "tool-call":
+      return [{
+        type: "tool-call",
+        toolCallId: part.call.id,
+        toolName: part.call.name,
+        args: part.call.input,
+      }];
+    case "tool-result":
+      return [{
+        type: "tool-result",
+        toolCallId: part.result.callId,
+        toolName: part.result.name,
+        result: part.result.output,
+        isError: part.result.status === "failure",
+      }];
+    case "presentation-action":
+      return [{
+        type: "tool-call",
+        toolCallId: `presentation-${part.action.type}`,
+        toolName: `presentation.${part.action.type}`,
+        args: part.action,
+      }];
+    case "approval-request":
+      return [
+        { type: "text", text: part.prompt },
+        {
+          type: "tool-call",
+          toolCallId: part.callId,
+          toolName: part.name,
+          args: {
+            prompt: part.prompt,
+            contract: part.contract,
+          },
+        },
+      ];
+    case "error":
+      return [{ type: "text", text: part.message }];
+  }
+};
+
+const messageRoleFor = (message: AgentMessage): AssistantProjectedMessage["role"] => {
+  const onlyToolResults = message.parts.length > 0
+    && message.parts.every((part) => part.type === "tool-result");
+  if (onlyToolResults) return "tool";
+  return message.role === "user" ? "user" : "assistant";
+};
+
+export const projectAgentMessagesForAssistant = (
+  messages: ReadonlyArray<AgentMessage>,
+): AssistantProjectedMessage[] =>
+  messages.map((message) => {
+    const content = message.parts.flatMap(projectPart);
+    return {
+      id: message.id,
+      role: messageRoleFor(message),
+      content,
+    };
+  });