File size: 1,978 Bytes
25d91a4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | import { createApiRegistry, createLlmRuntime } from "@openclaw/ai";
import type { AssistantMessage, Model } from "@openclaw/llm-core";
import { describe, expect, it, vi } from "vitest";
import { createZeroUsageFixture } from "../agents/test-helpers/usage-fixtures.js";
import { bindModelLlmRuntime } from "./model-runtime-binding.js";
import { streamSimple } from "./stream.js";
import { createAssistantMessageEventStream } from "./utils/event-stream.js";
describe("LLM stream facade", () => {
it("routes a prepared model through its lifecycle runtime", async () => {
const registry = createApiRegistry();
const runtime = createLlmRuntime(registry);
const expected = createAssistantMessageEventStream();
const message = {
role: "assistant",
content: [{ type: "text", text: "complete" }],
api: "test-lifecycle-api",
provider: "test-provider",
model: "test-model",
usage: createZeroUsageFixture(),
stopReason: "stop",
timestamp: Date.now(),
} satisfies AssistantMessage;
const providerStream = vi.fn(() => expected);
registry.registerApiProvider({
api: "test-lifecycle-api",
stream: providerStream,
streamSimple: providerStream,
});
const model = bindModelLlmRuntime(
{
api: "test-lifecycle-api",
provider: "test-provider",
id: "test-model",
name: "Test Model",
baseUrl: "https://example.test",
reasoning: false,
input: ["text"],
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
contextWindow: 1024,
maxTokens: 512,
} satisfies Model,
runtime,
);
const output = streamSimple(model, { messages: [] });
expect(providerStream).not.toHaveBeenCalled();
expected.push({ type: "done", reason: "stop", message });
expected.end();
await expect(output.result()).resolves.toEqual(message);
expect(providerStream).toHaveBeenCalledOnce();
});
});
|