Files
2026-07-13 12:22:59 +08:00

26 lines
603 B
TypeScript

import type { AssistantMessage } from "@oh-my-pi/pi-ai";
/**
* Shared factory for building a minimal mock `AssistantMessage`
* used by AgentSession test suites.
*/
export function createAssistantMessage(text: string): AssistantMessage {
return {
role: "assistant",
content: [{ type: "text", text }],
api: "anthropic-messages",
provider: "anthropic",
model: "mock",
usage: {
input: 0,
output: 0,
cacheRead: 0,
cacheWrite: 0,
totalTokens: 0,
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
},
stopReason: "stop",
timestamp: Date.now(),
};
}