fix: implement disabled skills functionality in skill resolution

This commit is contained in:
Muhammad Noor Misyuari 2026-01-26 14:21:28 +07:00
parent 976ffaeb0d
commit 2236a940f8
No known key found for this signature in database
GPG Key ID: C78CF07E782BB4B2
8 changed files with 151 additions and 31 deletions

View File

@ -41,7 +41,7 @@ describe("createBuiltinAgents with model overrides", () => {
} }
// #when // #when
const agents = await createBuiltinAgents([], overrides, undefined, TEST_DEFAULT_MODEL) const agents = await createBuiltinAgents([], overrides, undefined, TEST_DEFAULT_MODEL, undefined, undefined, [], undefined, undefined)
// #then // #then
expect(agents.sisyphus.model).toBe("github-copilot/gpt-5.2") expect(agents.sisyphus.model).toBe("github-copilot/gpt-5.2")
@ -103,7 +103,7 @@ describe("createBuiltinAgents with model overrides", () => {
const cacheSpy = spyOn(connectedProvidersCache, "readConnectedProvidersCache").mockReturnValue(["openai"]) const cacheSpy = spyOn(connectedProvidersCache, "readConnectedProvidersCache").mockReturnValue(["openai"])
// #when // #when
const agents = await createBuiltinAgents([], {}, undefined, TEST_DEFAULT_MODEL) const agents = await createBuiltinAgents([], {}, undefined, TEST_DEFAULT_MODEL, undefined, undefined, [], undefined, undefined)
// #then - oracle resolves via connected cache fallback to openai/gpt-5.2 (not system default) // #then - oracle resolves via connected cache fallback to openai/gpt-5.2 (not system default)
expect(agents.oracle.model).toBe("openai/gpt-5.2") expect(agents.oracle.model).toBe("openai/gpt-5.2")
@ -132,7 +132,7 @@ describe("createBuiltinAgents with model overrides", () => {
} }
// #when // #when
const agents = await createBuiltinAgents([], overrides, undefined, TEST_DEFAULT_MODEL) const agents = await createBuiltinAgents([], overrides, undefined, TEST_DEFAULT_MODEL, undefined, undefined, [], undefined, undefined)
// #then // #then
expect(agents.oracle.model).toBe("openai/gpt-5.2") expect(agents.oracle.model).toBe("openai/gpt-5.2")
@ -148,7 +148,7 @@ describe("createBuiltinAgents with model overrides", () => {
} }
// #when // #when
const agents = await createBuiltinAgents([], overrides, undefined, TEST_DEFAULT_MODEL) const agents = await createBuiltinAgents([], overrides, undefined, TEST_DEFAULT_MODEL, undefined, undefined, [], undefined, undefined)
// #then // #then
expect(agents.oracle.model).toBe("anthropic/claude-sonnet-4") expect(agents.oracle.model).toBe("anthropic/claude-sonnet-4")
@ -164,12 +164,25 @@ describe("createBuiltinAgents with model overrides", () => {
} }
// #when // #when
const agents = await createBuiltinAgents([], overrides, undefined, TEST_DEFAULT_MODEL) const agents = await createBuiltinAgents([], overrides, undefined, TEST_DEFAULT_MODEL, undefined, undefined, [], undefined, undefined)
// #then // #then
expect(agents.sisyphus.model).toBe("github-copilot/gpt-5.2") expect(agents.sisyphus.model).toBe("github-copilot/gpt-5.2")
expect(agents.sisyphus.temperature).toBe(0.5) expect(agents.sisyphus.temperature).toBe(0.5)
}) })
test("createBuiltinAgents excludes disabled skills from availableSkills", async () => {
// #given
const disabledSkills = new Set(["playwright"])
// #when
const agents = await createBuiltinAgents([], {}, undefined, TEST_DEFAULT_MODEL, undefined, undefined, [], undefined, undefined, undefined, disabledSkills)
// #then
expect(agents.sisyphus.prompt).not.toContain("playwright")
expect(agents.sisyphus.prompt).toContain("frontend-ui-ux")
expect(agents.sisyphus.prompt).toContain("git-master")
})
}) })
describe("createBuiltinAgents without systemDefaultModel", () => { describe("createBuiltinAgents without systemDefaultModel", () => {

View File

@ -233,7 +233,8 @@ export async function createBuiltinAgents(
discoveredSkills: LoadedSkill[] = [], discoveredSkills: LoadedSkill[] = [],
client?: any, client?: any,
browserProvider?: BrowserAutomationProvider, browserProvider?: BrowserAutomationProvider,
uiSelectedModel?: string uiSelectedModel?: string,
disabledSkills?: Set<string>
): Promise<Record<string, AgentConfig>> { ): Promise<Record<string, AgentConfig>> {
const connectedProviders = readConnectedProvidersCache() const connectedProviders = readConnectedProvidersCache()
// IMPORTANT: Do NOT pass client to fetchAvailableModels during plugin initialization. // IMPORTANT: Do NOT pass client to fetchAvailableModels during plugin initialization.
@ -257,7 +258,7 @@ export async function createBuiltinAgents(
description: categories?.[name]?.description ?? CATEGORY_DESCRIPTIONS[name] ?? "General tasks", description: categories?.[name]?.description ?? CATEGORY_DESCRIPTIONS[name] ?? "General tasks",
})) }))
const builtinSkills = createBuiltinSkills({ browserProvider }) const builtinSkills = createBuiltinSkills({ browserProvider, disabledSkills })
const builtinSkillNames = new Set(builtinSkills.map(s => s.name)) const builtinSkillNames = new Set(builtinSkills.map(s => s.name))
const builtinAvailable: AvailableSkill[] = builtinSkills.map((skill) => ({ const builtinAvailable: AvailableSkill[] = builtinSkills.map((skill) => ({
@ -290,16 +291,16 @@ export async function createBuiltinAgents(
const override = agentOverrides[agentName] const override = agentOverrides[agentName]
?? Object.entries(agentOverrides).find(([key]) => key.toLowerCase() === agentName.toLowerCase())?.[1] ?? Object.entries(agentOverrides).find(([key]) => key.toLowerCase() === agentName.toLowerCase())?.[1]
const requirement = AGENT_MODEL_REQUIREMENTS[agentName] const requirement = AGENT_MODEL_REQUIREMENTS[agentName]
// Check if agent requires a specific model // Check if agent requires a specific model
if (requirement?.requiresModel && availableModels) { if (requirement?.requiresModel && availableModels) {
if (!isModelAvailable(requirement.requiresModel, availableModels)) { if (!isModelAvailable(requirement.requiresModel, availableModels)) {
continue continue
} }
} }
const isPrimaryAgent = isFactory(source) && source.mode === "primary" const isPrimaryAgent = isFactory(source) && source.mode === "primary"
const resolution = applyModelResolution({ const resolution = applyModelResolution({
uiSelectedModel: isPrimaryAgent ? uiSelectedModel : undefined, uiSelectedModel: isPrimaryAgent ? uiSelectedModel : undefined,
userModel: override?.model, userModel: override?.model,
@ -374,7 +375,7 @@ export async function createBuiltinAgents(
availableSkills, availableSkills,
availableCategories availableCategories
) )
if (sisyphusResolvedVariant) { if (sisyphusResolvedVariant) {
sisyphusConfig = { ...sisyphusConfig, variant: sisyphusResolvedVariant } sisyphusConfig = { ...sisyphusConfig, variant: sisyphusResolvedVariant }
} }
@ -419,7 +420,7 @@ export async function createBuiltinAgents(
availableSkills, availableSkills,
availableCategories availableCategories
) )
hephaestusConfig = { ...hephaestusConfig, variant: hephaestusResolvedVariant ?? "medium" } hephaestusConfig = { ...hephaestusConfig, variant: hephaestusResolvedVariant ?? "medium" }
const hepOverrideCategory = (hephaestusOverride as Record<string, unknown> | undefined)?.category as string | undefined const hepOverrideCategory = (hephaestusOverride as Record<string, unknown> | undefined)?.category as string | undefined
@ -467,7 +468,7 @@ export async function createBuiltinAgents(
availableSkills, availableSkills,
userCategories: categories, userCategories: categories,
}) })
if (atlasResolvedVariant) { if (atlasResolvedVariant) {
orchestratorConfig = { ...orchestratorConfig, variant: atlasResolvedVariant } orchestratorConfig = { ...orchestratorConfig, variant: atlasResolvedVariant }
} }

View File

@ -86,4 +86,58 @@ describe("createBuiltinSkills", () => {
expect(defaultSkills).toHaveLength(4) expect(defaultSkills).toHaveLength(4)
expect(agentBrowserSkills).toHaveLength(4) expect(agentBrowserSkills).toHaveLength(4)
}) })
test("should exclude playwright when it is in disabledSkills", () => {
// #given
const options = { disabledSkills: new Set(["playwright"]) }
// #when
const skills = createBuiltinSkills(options)
// #then
expect(skills.map((s) => s.name)).not.toContain("playwright")
expect(skills.map((s) => s.name)).toContain("frontend-ui-ux")
expect(skills.map((s) => s.name)).toContain("git-master")
expect(skills.map((s) => s.name)).toContain("dev-browser")
expect(skills.length).toBe(3)
})
test("should exclude multiple skills when they are in disabledSkills", () => {
// #given
const options = { disabledSkills: new Set(["playwright", "git-master"]) }
// #when
const skills = createBuiltinSkills(options)
// #then
expect(skills.map((s) => s.name)).not.toContain("playwright")
expect(skills.map((s) => s.name)).not.toContain("git-master")
expect(skills.map((s) => s.name)).toContain("frontend-ui-ux")
expect(skills.map((s) => s.name)).toContain("dev-browser")
expect(skills.length).toBe(2)
})
test("should return an empty array when all skills are disabled", () => {
// #given
const options = {
disabledSkills: new Set(["playwright", "frontend-ui-ux", "git-master", "dev-browser"]),
}
// #when
const skills = createBuiltinSkills(options)
// #then
expect(skills.length).toBe(0)
})
test("should return all skills when disabledSkills set is empty", () => {
// #given
const options = { disabledSkills: new Set<string>() }
// #when
const skills = createBuiltinSkills(options)
// #then
expect(skills.length).toBe(4)
})
}) })

View File

@ -11,12 +11,19 @@ import {
export interface CreateBuiltinSkillsOptions { export interface CreateBuiltinSkillsOptions {
browserProvider?: BrowserAutomationProvider browserProvider?: BrowserAutomationProvider
disabledSkills?: Set<string>
} }
export function createBuiltinSkills(options: CreateBuiltinSkillsOptions = {}): BuiltinSkill[] { export function createBuiltinSkills(options: CreateBuiltinSkillsOptions = {}): BuiltinSkill[] {
const { browserProvider = "playwright" } = options const { browserProvider = "playwright", disabledSkills } = options
const browserSkill = browserProvider === "agent-browser" ? agentBrowserSkill : playwrightSkill const browserSkill = browserProvider === "agent-browser" ? agentBrowserSkill : playwrightSkill
return [browserSkill, frontendUiUxSkill, gitMasterSkill, devBrowserSkill] const skills = [browserSkill, frontendUiUxSkill, gitMasterSkill, devBrowserSkill]
if (!disabledSkills) {
return skills
}
return skills.filter((skill) => !disabledSkills.has(skill.name))
} }

View File

@ -33,10 +33,12 @@ describe("resolveSkillContent", () => {
expect(result).toBeNull() expect(result).toBeNull()
}) })
it("should return null for empty string", () => { it("should return null for disabled skill", () => {
// given: builtin skills // given: frontend-ui-ux skill disabled
// when: resolving content for empty string const options = { disabledSkills: new Set(["frontend-ui-ux"]) }
const result = resolveSkillContent("")
// when: resolving content for disabled skill
const result = resolveSkillContent("frontend-ui-ux", options)
// then: returns null // then: returns null
expect(result).toBeNull() expect(result).toBeNull()
@ -96,6 +98,20 @@ describe("resolveMultipleSkills", () => {
expect(result.notFound).toEqual(["skill-one", "skill-two", "skill-three"]) expect(result.notFound).toEqual(["skill-one", "skill-two", "skill-three"])
}) })
it("should treat disabled skills as not found", () => {
// #given: frontend-ui-ux disabled, playwright not disabled
const skillNames = ["frontend-ui-ux", "playwright"]
const options = { disabledSkills: new Set(["frontend-ui-ux"]) }
// #when: resolving multiple skills with disabled one
const result = resolveMultipleSkills(skillNames, options)
// #then: frontend-ui-ux in notFound, playwright resolved
expect(result.resolved.size).toBe(1)
expect(result.resolved.has("playwright")).toBe(true)
expect(result.notFound).toEqual(["frontend-ui-ux"])
})
it("should preserve skill order in resolved map", () => { it("should preserve skill order in resolved map", () => {
// given: list of skill names in specific order // given: list of skill names in specific order
const skillNames = ["playwright", "frontend-ui-ux"] const skillNames = ["playwright", "frontend-ui-ux"]
@ -122,10 +138,12 @@ describe("resolveSkillContentAsync", () => {
expect(result).toContain("Role: Designer-Turned-Developer") expect(result).toContain("Role: Designer-Turned-Developer")
}) })
it("should return null for non-existent skill", async () => { it("should return null for disabled skill", async () => {
// given: non-existent skill name // given: frontend-ui-ux disabled
// when: resolving content async const options = { disabledSkills: new Set(["frontend-ui-ux"]) }
const result = await resolveSkillContentAsync("definitely-not-a-skill-12345")
// when: resolving content async for disabled skill
const result = await resolveSkillContentAsync("frontend-ui-ux", options)
// then: returns null // then: returns null
expect(result).toBeNull() expect(result).toBeNull()
@ -160,6 +178,20 @@ describe("resolveMultipleSkillsAsync", () => {
expect(result.resolved.get("playwright")).toContain("Playwright Browser Automation") expect(result.resolved.get("playwright")).toContain("Playwright Browser Automation")
}) })
it("should treat disabled skills as not found async", async () => {
// #given: frontend-ui-ux disabled
const skillNames = ["frontend-ui-ux", "playwright"]
const options = { disabledSkills: new Set(["frontend-ui-ux"]) }
// #when: resolving multiple skills async with disabled one
const result = await resolveMultipleSkillsAsync(skillNames, options)
// #then: frontend-ui-ux in notFound, playwright resolved
expect(result.resolved.size).toBe(1)
expect(result.resolved.has("playwright")).toBe(true)
expect(result.notFound).toEqual(["frontend-ui-ux"])
})
it("should NOT inject watermark when both options are disabled", async () => { it("should NOT inject watermark when both options are disabled", async () => {
// given: git-master skill with watermark disabled // given: git-master skill with watermark disabled
const skillNames = ["git-master"] const skillNames = ["git-master"]

View File

@ -8,6 +8,7 @@ import type { GitMasterConfig, BrowserAutomationProvider } from "../../config/sc
export interface SkillResolutionOptions { export interface SkillResolutionOptions {
gitMasterConfig?: GitMasterConfig gitMasterConfig?: GitMasterConfig
browserProvider?: BrowserAutomationProvider browserProvider?: BrowserAutomationProvider
disabledSkills?: Set<string>
} }
const cachedSkillsByProvider = new Map<string, LoadedSkill[]>() const cachedSkillsByProvider = new Map<string, LoadedSkill[]>()
@ -23,7 +24,12 @@ async function getAllSkills(options?: SkillResolutionOptions): Promise<LoadedSki
const [discoveredSkills, builtinSkillDefs] = await Promise.all([ const [discoveredSkills, builtinSkillDefs] = await Promise.all([
discoverSkills({ includeClaudeCodePaths: true }), discoverSkills({ includeClaudeCodePaths: true }),
Promise.resolve(createBuiltinSkills({ browserProvider: options?.browserProvider })), Promise.resolve(
createBuiltinSkills({
browserProvider: options?.browserProvider,
disabledSkills: options?.disabledSkills,
})
),
]) ])
const builtinSkillsAsLoaded: LoadedSkill[] = builtinSkillDefs.map((skill) => ({ const builtinSkillsAsLoaded: LoadedSkill[] = builtinSkillDefs.map((skill) => ({
@ -122,7 +128,10 @@ export function injectGitMasterConfig(template: string, config?: GitMasterConfig
} }
export function resolveSkillContent(skillName: string, options?: SkillResolutionOptions): string | null { export function resolveSkillContent(skillName: string, options?: SkillResolutionOptions): string | null {
const skills = createBuiltinSkills({ browserProvider: options?.browserProvider }) const skills = createBuiltinSkills({
browserProvider: options?.browserProvider,
disabledSkills: options?.disabledSkills,
})
const skill = skills.find((s) => s.name === skillName) const skill = skills.find((s) => s.name === skillName)
if (!skill) return null if (!skill) return null
@ -137,7 +146,10 @@ export function resolveMultipleSkills(skillNames: string[], options?: SkillResol
resolved: Map<string, string> resolved: Map<string, string>
notFound: string[] notFound: string[]
} { } {
const skills = createBuiltinSkills({ browserProvider: options?.browserProvider }) const skills = createBuiltinSkills({
browserProvider: options?.browserProvider,
disabledSkills: options?.disabledSkills,
})
const skillMap = new Map(skills.map((s) => [s.name, s.template])) const skillMap = new Map(skills.map((s) => [s.name, s.template]))
const resolved = new Map<string, string>() const resolved = new Map<string, string>()

View File

@ -414,9 +414,7 @@ const OhMyOpenCodePlugin: Plugin = async (ctx) => {
}); });
const disabledSkills = new Set(pluginConfig.disabled_skills ?? []); const disabledSkills = new Set(pluginConfig.disabled_skills ?? []);
const systemMcpNames = getSystemMcpServerNames(); const systemMcpNames = getSystemMcpServerNames();
const builtinSkills = createBuiltinSkills({ browserProvider }).filter( const builtinSkills = createBuiltinSkills({ browserProvider, disabledSkills }).filter((skill) => {
(skill) => {
if (disabledSkills.has(skill.name as never)) return false;
if (skill.mcpConfig) { if (skill.mcpConfig) {
for (const mcpName of Object.keys(skill.mcpConfig)) { for (const mcpName of Object.keys(skill.mcpConfig)) {
if (systemMcpNames.has(mcpName)) return false; if (systemMcpNames.has(mcpName)) return false;

View File

@ -157,6 +157,7 @@ export function createConfigHandler(deps: ConfigHandlerDeps) {
// config.model represents the currently active model in OpenCode (including UI selection) // config.model represents the currently active model in OpenCode (including UI selection)
// Pass it as uiSelectedModel so it takes highest priority in model resolution // Pass it as uiSelectedModel so it takes highest priority in model resolution
const currentModel = config.model as string | undefined; const currentModel = config.model as string | undefined;
const disabledSkills = new Set<string>(pluginConfig.disabled_skills ?? []);
const builtinAgents = await createBuiltinAgents( const builtinAgents = await createBuiltinAgents(
migratedDisabledAgents, migratedDisabledAgents,
pluginConfig.agents, pluginConfig.agents,
@ -167,7 +168,8 @@ export function createConfigHandler(deps: ConfigHandlerDeps) {
allDiscoveredSkills, allDiscoveredSkills,
ctx.client, ctx.client,
browserProvider, browserProvider,
currentModel // uiSelectedModel - takes highest priority currentModel, // uiSelectedModel - takes highest priority
disabledSkills
); );
// Claude Code agents: Do NOT apply permission migration // Claude Code agents: Do NOT apply permission migration
@ -358,7 +360,8 @@ export function createConfigHandler(deps: ConfigHandlerDeps) {
: {}; : {};
const planDemoteConfig = shouldDemotePlan const planDemoteConfig = shouldDemotePlan
? { mode: "subagent" as const } ? { mode: "subagent" as const
}
: undefined; : undefined;
config.agent = { config.agent = {