chore: rename toModelMessage -> toModelMessages

This commit is contained in:
Aiden Cline
2026-01-20 16:16:23 -06:00
parent 5c3e9cfa2c
commit 0c4ffec857
4 changed files with 16 additions and 16 deletions

View File

@@ -149,7 +149,7 @@ export namespace SessionCompaction {
tools: {}, tools: {},
system: [], system: [],
messages: [ messages: [
...MessageV2.toModelMessage(input.messages), ...MessageV2.toModelMessages(input.messages),
{ {
role: "user", role: "user",
content: [ content: [

View File

@@ -432,7 +432,7 @@ export namespace MessageV2 {
}) })
export type WithParts = z.infer<typeof WithParts> export type WithParts = z.infer<typeof WithParts>
export function toModelMessage(input: WithParts[]): ModelMessage[] { export function toModelMessages(input: WithParts[]): ModelMessage[] {
const result: UIMessage[] = [] const result: UIMessage[] = []
for (const msg of input) { for (const msg of input) {

View File

@@ -598,7 +598,7 @@ export namespace SessionPrompt {
sessionID, sessionID,
system: [...(await SystemPrompt.environment()), ...(await SystemPrompt.custom())], system: [...(await SystemPrompt.environment()), ...(await SystemPrompt.custom())],
messages: [ messages: [
...MessageV2.toModelMessage(sessionMessages), ...MessageV2.toModelMessages(sessionMessages),
...(isLastStep ...(isLastStep
? [ ? [
{ {
@@ -1800,7 +1800,7 @@ NOTE: At any point in time through this workflow you should feel free to ask the
}, },
...(hasOnlySubtaskParts ...(hasOnlySubtaskParts
? [{ role: "user" as const, content: subtaskParts.map((p) => p.prompt).join("\n") }] ? [{ role: "user" as const, content: subtaskParts.map((p) => p.prompt).join("\n") }]
: MessageV2.toModelMessage(contextMessages)), : MessageV2.toModelMessages(contextMessages)),
], ],
}) })
const text = await result.text.catch((err) => log.error("failed to generate title", { error: err })) const text = await result.text.catch((err) => log.error("failed to generate title", { error: err }))

View File

@@ -66,7 +66,7 @@ describe("session.message-v2.toModelMessage", () => {
}, },
] ]
expect(MessageV2.toModelMessage(input)).toStrictEqual([ expect(MessageV2.toModelMessages(input)).toStrictEqual([
{ {
role: "user", role: "user",
content: [{ type: "text", text: "hello" }], content: [{ type: "text", text: "hello" }],
@@ -91,7 +91,7 @@ describe("session.message-v2.toModelMessage", () => {
}, },
] ]
expect(MessageV2.toModelMessage(input)).toStrictEqual([]) expect(MessageV2.toModelMessages(input)).toStrictEqual([])
}) })
test("includes synthetic text parts", () => { test("includes synthetic text parts", () => {
@@ -122,7 +122,7 @@ describe("session.message-v2.toModelMessage", () => {
}, },
] ]
expect(MessageV2.toModelMessage(input)).toStrictEqual([ expect(MessageV2.toModelMessages(input)).toStrictEqual([
{ {
role: "user", role: "user",
content: [{ type: "text", text: "hello" }], content: [{ type: "text", text: "hello" }],
@@ -189,7 +189,7 @@ describe("session.message-v2.toModelMessage", () => {
}, },
] ]
expect(MessageV2.toModelMessage(input)).toStrictEqual([ expect(MessageV2.toModelMessages(input)).toStrictEqual([
{ {
role: "user", role: "user",
content: [ content: [
@@ -259,7 +259,7 @@ describe("session.message-v2.toModelMessage", () => {
}, },
] ]
expect(MessageV2.toModelMessage(input)).toStrictEqual([ expect(MessageV2.toModelMessages(input)).toStrictEqual([
{ {
role: "user", role: "user",
content: [{ type: "text", text: "run tool" }], content: [{ type: "text", text: "run tool" }],
@@ -341,7 +341,7 @@ describe("session.message-v2.toModelMessage", () => {
}, },
] ]
expect(MessageV2.toModelMessage(input)).toStrictEqual([ expect(MessageV2.toModelMessages(input)).toStrictEqual([
{ {
role: "user", role: "user",
content: [{ type: "text", text: "run tool" }], content: [{ type: "text", text: "run tool" }],
@@ -408,7 +408,7 @@ describe("session.message-v2.toModelMessage", () => {
}, },
] ]
expect(MessageV2.toModelMessage(input)).toStrictEqual([ expect(MessageV2.toModelMessages(input)).toStrictEqual([
{ {
role: "user", role: "user",
content: [{ type: "text", text: "run tool" }], content: [{ type: "text", text: "run tool" }],
@@ -461,7 +461,7 @@ describe("session.message-v2.toModelMessage", () => {
}, },
] ]
expect(MessageV2.toModelMessage(input)).toStrictEqual([]) expect(MessageV2.toModelMessages(input)).toStrictEqual([])
}) })
test("includes aborted assistant messages only when they have non-step-start/reasoning content", () => { test("includes aborted assistant messages only when they have non-step-start/reasoning content", () => {
@@ -504,7 +504,7 @@ describe("session.message-v2.toModelMessage", () => {
}, },
] ]
expect(MessageV2.toModelMessage(input)).toStrictEqual([ expect(MessageV2.toModelMessages(input)).toStrictEqual([
{ {
role: "assistant", role: "assistant",
content: [ content: [
@@ -540,7 +540,7 @@ describe("session.message-v2.toModelMessage", () => {
}, },
] ]
expect(MessageV2.toModelMessage(input)).toStrictEqual([ expect(MessageV2.toModelMessages(input)).toStrictEqual([
{ {
role: "assistant", role: "assistant",
content: [{ type: "text", text: "first" }], content: [{ type: "text", text: "first" }],
@@ -567,7 +567,7 @@ describe("session.message-v2.toModelMessage", () => {
}, },
] ]
expect(MessageV2.toModelMessage(input)).toStrictEqual([]) expect(MessageV2.toModelMessages(input)).toStrictEqual([])
}) })
test("converts pending/running tool calls to error results to prevent dangling tool_use", () => { test("converts pending/running tool calls to error results to prevent dangling tool_use", () => {
@@ -614,7 +614,7 @@ describe("session.message-v2.toModelMessage", () => {
}, },
] ]
const result = MessageV2.toModelMessage(input) const result = MessageV2.toModelMessages(input)
expect(result).toStrictEqual([ expect(result).toStrictEqual([
{ {