123 lines
3.6 KiB
TypeScript
123 lines
3.6 KiB
TypeScript
import { describe, expect, test } from "vitest";
|
|
import type { UIMessage } from "ai";
|
|
|
|
import { shouldAutomaticallyContinueAssistant } from "./assistant-auto-submit";
|
|
|
|
const messagesWithToolPart = (part: UIMessage["parts"][number]): UIMessage[] => [
|
|
{
|
|
id: "assistant-message",
|
|
role: "assistant",
|
|
parts: [{ type: "step-start" }, part],
|
|
},
|
|
];
|
|
|
|
describe("shouldAutomaticallyContinueAssistant", () => {
|
|
test("continues after a human or frontend tool supplies its result", () => {
|
|
expect(
|
|
shouldAutomaticallyContinueAssistant({
|
|
messages: messagesWithToolPart({
|
|
type: "dynamic-tool",
|
|
toolName: "ask_user",
|
|
toolCallId: "ask-1",
|
|
input: { question: "Continue?" },
|
|
state: "output-available",
|
|
output: { answer: "Yes" },
|
|
}),
|
|
}),
|
|
).toBe(true);
|
|
});
|
|
|
|
test("continues after a mutation approval response", () => {
|
|
expect(
|
|
shouldAutomaticallyContinueAssistant({
|
|
messages: messagesWithToolPart({
|
|
type: "dynamic-tool",
|
|
toolName: "app_create",
|
|
toolCallId: "create-1",
|
|
input: { name: "YouTube" },
|
|
state: "approval-responded",
|
|
approval: { id: "approval-1", approved: true },
|
|
}),
|
|
}),
|
|
).toBe(true);
|
|
});
|
|
|
|
test("continues an approved mutation when the same step also reports a provider-executed server tool", () => {
|
|
expect(
|
|
shouldAutomaticallyContinueAssistant({
|
|
messages: [
|
|
{
|
|
id: "assistant-message",
|
|
role: "assistant",
|
|
parts: [
|
|
{ type: "step-start" },
|
|
{
|
|
type: "dynamic-tool",
|
|
toolName: "openrouter_web_search",
|
|
toolCallId: "search-1",
|
|
input: { query: "PokéAPI" },
|
|
state: "input-available",
|
|
providerExecuted: true,
|
|
},
|
|
{
|
|
type: "dynamic-tool",
|
|
toolName: "customWidget_createFromPreview",
|
|
toolCallId: "create-1",
|
|
input: { previewSessionId: "preview-1" },
|
|
state: "approval-responded",
|
|
approval: { id: "approval-1", approved: true },
|
|
},
|
|
],
|
|
},
|
|
],
|
|
}),
|
|
).toBe(true);
|
|
});
|
|
|
|
test("waits while a human tool has no result", () => {
|
|
expect(
|
|
shouldAutomaticallyContinueAssistant({
|
|
messages: messagesWithToolPart({
|
|
type: "dynamic-tool",
|
|
toolName: "ask_user",
|
|
toolCallId: "ask-1",
|
|
input: { question: "Continue?" },
|
|
state: "input-available",
|
|
}),
|
|
}),
|
|
).toBe(false);
|
|
});
|
|
|
|
test("continues a completed browser-tool step without treating an earlier agent step as user input", () => {
|
|
expect(
|
|
shouldAutomaticallyContinueAssistant({
|
|
messages: [
|
|
{
|
|
id: "assistant-message",
|
|
role: "assistant",
|
|
parts: [
|
|
{ type: "step-start" },
|
|
{
|
|
type: "dynamic-tool",
|
|
toolName: "icon_findIcons",
|
|
toolCallId: "icon-1",
|
|
input: { searchText: "homarr" },
|
|
state: "input-available",
|
|
},
|
|
{ type: "step-start" },
|
|
{
|
|
type: "dynamic-tool",
|
|
toolName: "refresh_current_view",
|
|
toolCallId: "refresh-1",
|
|
input: {},
|
|
state: "output-available",
|
|
output: { success: true },
|
|
},
|
|
],
|
|
},
|
|
],
|
|
}),
|
|
).toBe(true);
|
|
});
|
|
});
|