fix(plugin-handlers): cache vision-capable provider models

Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-opencode)

Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
This commit is contained in:
YeonGyu-Kim
2026-03-08 02:20:34 +09:00
parent f80181199b
commit dd680357ae
2 changed files with 120 additions and 2 deletions
@@ -0,0 +1,84 @@
/// <reference types="bun-types" />
import { describe, expect, test } from "bun:test"
import { applyProviderConfig } from "./provider-config-handler"
import { createModelCacheState } from "../plugin-state"
import { clearVisionCapableModelsCache, readVisionCapableModelsCache } from "../shared/vision-capable-models-cache"
describe("applyProviderConfig", () => {
test("caches vision-capable models from modalities and capabilities", () => {
// given
const modelCacheState = createModelCacheState()
const visionCapableModelsCache = modelCacheState.visionCapableModelsCache
if (!visionCapableModelsCache) {
throw new Error("visionCapableModelsCache should be initialized")
}
const config = {
provider: {
rundao: {
models: {
"public/qwen3.5-397b": {
modalities: {
input: ["text", "image"],
},
},
"public/text-only": {
modalities: {
input: ["text"],
},
},
},
},
google: {
models: {
"gemini-3-flash": {
capabilities: {
input: {
image: true,
},
},
},
},
},
},
} satisfies Record<string, unknown>
// when
applyProviderConfig({ config, modelCacheState })
// then
expect(Array.from(visionCapableModelsCache.keys())).toEqual([
"rundao/public/qwen3.5-397b",
"google/gemini-3-flash",
])
expect(readVisionCapableModelsCache()).toEqual([
{ providerID: "rundao", modelID: "public/qwen3.5-397b" },
{ providerID: "google", modelID: "gemini-3-flash" },
])
})
test("clears stale vision-capable models when provider config changes", () => {
// given
const modelCacheState = createModelCacheState()
const visionCapableModelsCache = modelCacheState.visionCapableModelsCache
if (!visionCapableModelsCache) {
throw new Error("visionCapableModelsCache should be initialized")
}
visionCapableModelsCache.set("stale/old-model", {
providerID: "stale",
modelID: "old-model",
})
// when
applyProviderConfig({
config: { provider: {} },
modelCacheState,
})
// then
expect(visionCapableModelsCache.size).toBe(0)
expect(readVisionCapableModelsCache()).toEqual([])
})
})
clearVisionCapableModelsCache()
+36 -2
View File
@@ -1,10 +1,31 @@
import type { ModelCacheState } from "../plugin-state";
import type { ModelCacheState, VisionCapableModel } from "../plugin-state";
import { setVisionCapableModelsCache } from "../shared/vision-capable-models-cache"
type ProviderConfig = {
options?: { headers?: Record<string, string> };
models?: Record<string, { limit?: { context?: number } }>;
models?: Record<string, ProviderModelConfig>;
};
type ProviderModelConfig = {
limit?: { context?: number };
modalities?: {
input?: string[];
};
capabilities?: {
input?: {
image?: boolean;
};
};
}
function supportsImageInput(modelConfig: ProviderModelConfig | undefined): boolean {
if (modelConfig?.modalities?.input?.includes("image")) {
return true
}
return modelConfig?.capabilities?.input?.image === true
}
export function applyProviderConfig(params: {
config: Record<string, unknown>;
modelCacheState: ModelCacheState;
@@ -17,6 +38,12 @@ export function applyProviderConfig(params: {
params.modelCacheState.anthropicContext1MEnabled =
anthropicBeta?.includes("context-1m") ?? false;
const visionCapableModelsCache = params.modelCacheState.visionCapableModelsCache
?? new Map<string, VisionCapableModel>()
params.modelCacheState.visionCapableModelsCache = visionCapableModelsCache
visionCapableModelsCache.clear()
setVisionCapableModelsCache(visionCapableModelsCache)
if (!providers) return;
for (const [providerID, providerConfig] of Object.entries(providers)) {
@@ -24,6 +51,13 @@ export function applyProviderConfig(params: {
if (!models) continue;
for (const [modelID, modelConfig] of Object.entries(models)) {
if (supportsImageInput(modelConfig)) {
visionCapableModelsCache.set(
`${providerID}/${modelID}`,
{ providerID, modelID },
)
}
const contextLimit = modelConfig?.limit?.context;
if (!contextLimit) continue;