vrr/openclaw
2
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-13 13:18:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 19

test(openai): prefer exact live registry models

Browse source
This commit is contained in:
Peter Steinberger 2026-04-27 07:22:19 +01:00
parent aac83e00cf
commit cf04fa24d8
No known key found for this signature in database
2 changed files with 36 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

24
extensions/openai/openai-provider.live.test.ts
View file

@ -1,4 +1,6 @@
import { getModel, type Api, type Model } from "@mariozechner/pi-ai";
import OpenAI from "openai";
import type { ProviderRuntimeModel } from "openclaw/plugin-sdk/plugin-entry";
import { describe, expect, it } from "vitest";
import { buildOpenAIProvider } from "./openai-provider.js";
@ -16,6 +18,10 @@ type LiveModelCase = {
maxTokens: number;
};
function findOpenAIModel(modelId: string): Model<Api> | null {
return (getModel("openai", modelId as never) as Model<Api> | undefined) ?? null;
}
function resolveLiveModelCase(modelId: string): LiveModelCase {
switch (modelId) {
case "gpt-5.5":
@ -96,6 +102,10 @@ describeLive("buildOpenAIProvider live", () => {
if (providerId !== "openai") {
return null;
}
const exactModel = findOpenAIModel(id);
if (exactModel) {
return exactModel;
}
if (id === liveCase.templateId) {
return {
id: liveCase.templateId,
@ -108,17 +118,19 @@ describeLive("buildOpenAIProvider live", () => {
cost: liveCase.cost,
contextWindow: liveCase.contextWindow,
maxTokens: liveCase.maxTokens,
};
} satisfies ProviderRuntimeModel;
}
return null;
},
};
const resolved = provider.resolveDynamicModel?.({
provider: "openai",
modelId: liveCase.modelId,
modelRegistry: registry as never,
});
const resolved =
registry.find("openai", liveCase.modelId) ??
provider.resolveDynamicModel?.({
provider: "openai",
modelId: liveCase.modelId,
modelRegistry: registry as never,
});
if (!resolved) {
throw new Error(`openai provider did not resolve ${liveCase.modelId}`);
}

26
extensions/openai/openai.live.test.ts
View file

@ -1,7 +1,7 @@
import fs from "node:fs/promises";
import os from "node:os";
import path from "node:path";
import { getModel } from "@mariozechner/pi-ai";
import { getModel, type Api, type Model } from "@mariozechner/pi-ai";
import { AuthStorage, ModelRegistry } from "@mariozechner/pi-coding-agent";
import OpenAI from "openai";
import type { ResolvedTtsConfig } from "openclaw/plugin-sdk/agent-runtime";
@ -27,6 +27,10 @@ const ModelRegistryCtor = ModelRegistry as unknown as {
new (authStorage: AuthStorage, modelsJsonPath?: string): ModelRegistry;
};
function findOpenAIModel(modelId: string): Model<Api> | null {
return (getModel("openai", modelId as never) as Model<Api> | undefined) ?? null;
}
function resolveTemplateModelId(modelId: string) {
switch (modelId) {
case "gpt-5.5":
@ -42,9 +46,12 @@ function resolveTemplateModelId(modelId: string) {
}
}
function createTemplateModelRegistry(modelId: string): ModelRegistry {
function createLiveModelRegistry(modelId: string): ModelRegistry {
const registry = new ModelRegistryCtor(AuthStorage.inMemory());
const template = getModel("openai", resolveTemplateModelId(modelId));
const template = findOpenAIModel(modelId) ?? findOpenAIModel(resolveTemplateModelId(modelId));
if (!template) {
throw new Error(`Unsupported live OpenAI plugin model: ${modelId}`);
}
registry.registerProvider("openai", {
apiKey: "test",
baseUrl: template.baseUrl,
@ -183,12 +190,15 @@ describeLive("openai plugin live", () => {
it("registers an OpenAI provider that can complete a live request", async () => {
const { providers } = await registerOpenAIPlugin();
const provider = requireRegisteredProvider(providers, "openai");
const modelRegistry = createLiveModelRegistry(LIVE_MODEL_ID);
const resolved = provider.resolveDynamicModel?.({
provider: "openai",
modelId: LIVE_MODEL_ID,
modelRegistry: createTemplateModelRegistry(LIVE_MODEL_ID),
});
const resolved =
modelRegistry.find("openai", LIVE_MODEL_ID) ??
provider.resolveDynamicModel?.({
provider: "openai",
modelId: LIVE_MODEL_ID,
modelRegistry,
});
if (!resolved) {
throw new Error("openai provider did not resolve the live model");