can1357
1e601b9094
- Replaced custom MockAssistantStream helpers with createMockModel streams across agent tests. - Removed manual queueMicrotask stream-event scripting in favor of scripted mock responses. - Consolidated helper fixtures by deleting local aliases and reusing shared user-message/model helpers. - Updated test assertions to use mock.calls and mock.model metadata for call and context validation.
33 lines
729 B
TypeScript
33 lines
729 B
TypeScript
import type { AssistantMessage, Usage, UserMessage } from "@oh-my-pi/pi-ai";
|
|
|
|
export function createUserMessage(text: string): UserMessage {
|
|
return { role: "user", content: text, timestamp: Date.now() };
|
|
}
|
|
|
|
export function createAssistantMessage(
|
|
content: AssistantMessage["content"],
|
|
stopReason: AssistantMessage["stopReason"] = "stop",
|
|
): AssistantMessage {
|
|
return {
|
|
role: "assistant",
|
|
content,
|
|
api: "mock",
|
|
provider: "mock",
|
|
model: "mock-model",
|
|
usage: createUsage(),
|
|
stopReason,
|
|
timestamp: Date.now(),
|
|
};
|
|
}
|
|
|
|
function createUsage(): Usage {
|
|
return {
|
|
input: 0,
|
|
output: 0,
|
|
cacheRead: 0,
|
|
cacheWrite: 0,
|
|
totalTokens: 0,
|
|
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
|
|
};
|
|
}
|