26 lines
603 B
TypeScript
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(),
|
|
};
|
|
}
|