From faf964691b253315302e21d8ec7613dc01dde713 Mon Sep 17 00:00:00 2001 From: Kit Langton Date: Thu, 16 Jul 2026 22:45:42 -0400 Subject: [PATCH] fix(core): initialize provider state before catalog transforms (#37419) --- packages/core/src/plugin/provider/openai.ts | 2 +- packages/core/src/plugin/provider/opencode.ts | 3 +-- .../core/test/plugin/provider-openai.test.ts | 25 ++----------------- 3 files changed, 4 insertions(+), 26 deletions(-) diff --git a/packages/core/src/plugin/provider/openai.ts b/packages/core/src/plugin/provider/openai.ts index 111f66105a..013388aa55 100644 --- a/packages/core/src/plugin/provider/openai.ts +++ b/packages/core/src/plugin/provider/openai.ts @@ -176,6 +176,7 @@ export const OpenAIPlugin = define({ draft.method.update(browser) draft.method.update(headless) }) + yield* load() yield* ctx.catalog.transform((evt) => { for (const item of evt.provider.list()) { if (!ProviderV2.isAISDK(item.provider.package)) continue @@ -213,7 +214,6 @@ export const OpenAIPlugin = define({ Stream.runForEach(refresh), Effect.forkScoped({ startImmediately: true }), ) - yield* refresh().pipe(Effect.forkScoped) yield* ctx.aisdk.hook( "sdk", Effect.fn(function* (evt) { diff --git a/packages/core/src/plugin/provider/opencode.ts b/packages/core/src/plugin/provider/opencode.ts index 1d0488bdad..1e8c52e6fc 100644 --- a/packages/core/src/plugin/provider/opencode.ts +++ b/packages/core/src/plugin/provider/opencode.ts @@ -114,7 +114,7 @@ export const OpencodePlugin = define { for (const [providerID, item] of Object.entries(providers ?? {})) { catalog.provider.update(providerID, (provider) => { @@ -194,7 +194,6 @@ export const OpencodePlugin = define(value: T | undefined): T { return value } -function eventually( - effect: Effect.Effect, - predicate: (value: A) => boolean, - remaining = 1000, -): Effect.Effect { - return Effect.gen(function* () { - const value = yield* effect - if (predicate(value)) return value - if (remaining === 0) return yield* Effect.fail(new Error("Timed out waiting for value")) - yield* Effect.promise(() => Bun.sleep(1)) - return yield* eventually(effect, predicate, remaining - 1) - }) -} - function fakeSelectorSdk(calls: string[]) { const make = (method: string) => (id: string) => { calls.push(`${method}:${id}`) @@ -219,12 +205,8 @@ describe("OpenAIPlugin", () => { }) yield* addPlugin() - const eligible = required( - yield* eventually( - catalog.model.get(ProviderV2.ID.openai, ModelV2.ID.make("gpt-5.5")), - (model) => model?.cost.length === 0, - ), - ) + const eligible = required(yield* catalog.model.get(ProviderV2.ID.openai, ModelV2.ID.make("gpt-5.5"))) + expect(eligible.cost).toEqual([]) expect(eligible.enabled).toBe(true) expect(required(yield* catalog.model.get(ProviderV2.ID.openai, ModelV2.ID.make("gpt-5.5-pro"))).enabled).toBe( false, @@ -260,9 +242,6 @@ describe("OpenAIPlugin", () => { value: Credential.Key.make({ type: "key", key: "sk-test" }), }) yield* addPlugin() - // The connection refresh is asynchronous; give it time to settle before - // asserting nothing was filtered. - yield* Effect.promise(() => Bun.sleep(25)) expect(required(yield* catalog.model.get(ProviderV2.ID.openai, ModelV2.ID.make("gpt-5.5"))).enabled).toBe(true) expect(required(yield* catalog.model.get(ProviderV2.ID.openai, ModelV2.ID.make("gpt-4.1"))).enabled).toBe(true)