LLM/moltbot
1
0
Fork 0
mirror of https://github.com/moltbot/moltbot.git synced 2026-05-13 15:47:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: tighten deepinfra model discovery assertions

Browse Source
This commit is contained in:
Peter Steinberger
2026-05-09 09:17:27 +01:00
parent 2c4ec93245
commit 3b0dafae1e
1 changed files with 15 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
extensions/deepinfra/provider-models.test.ts
View File

@@ -77,14 +77,17 @@ describe("discoverDeepInfraModels", () => {
await withFetchPathTest(mockFetch, async () => {
const models = await discoverDeepInfraModels();
expect(mockFetch).toHaveBeenCalledWith(
DEEPINFRA_MODELS_URL,
expect.objectContaining({
headers: { Accept: "application/json" },
}),
);
expect(mockFetch).toHaveBeenCalledOnce();
const [fetchUrl, fetchInit] = mockFetch.mock.calls[0] ?? [];
const fetchSignal = Reflect.get(fetchInit ?? {}, "signal");
expect(fetchUrl).toBe(DEEPINFRA_MODELS_URL);
expect(fetchSignal).toBeInstanceOf(AbortSignal);
expect(fetchInit).toEqual({
headers: { Accept: "application/json" },
signal: fetchSignal,
});
expect(models).toEqual([
expect.objectContaining({
{
id: "openai/gpt-oss-120b",
name: "openai/gpt-oss-120b",
reasoning: true,
@@ -92,8 +95,8 @@ describe("discoverDeepInfraModels", () => {
contextWindow: 131072,
maxTokens: 65536,
cost: { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 0 },
compat: expect.objectContaining({ supportsUsageInStreaming: true }),
}),
compat: { supportsUsageInStreaming: true },
},
]);
});
});
@@ -133,13 +136,15 @@ describe("discoverDeepInfraModels", () => {
await withFetchPathTest(mockFetch, async () => {
const [model] = await discoverDeepInfraModels();
expect(model).toMatchObject({
expect(model).toEqual({
id: "some/model",
name: "some/model",
reasoning: false,
input: ["text"],
contextWindow: 128000,
maxTokens: 8192,
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
compat: { supportsUsageInStreaming: true },
});
});
});