fix(default-mode,multimodal-looker,delegate-task): preserve user-expected behavior
default-mode (system-transform): -e5463e2dbintroduced auto-activation of ultrawork+ralph-loop, anddc2e082acthen skipped the ultrawork system prompt whenever ralph_loop was also enabled. Net effect: the keyword-detector still showed 'Default ultrawork mode enabled' to the user, but the first turn had none of the ultrawork behavior. Loop continuation kept the ultrawork prefix, so the contract was honored only on later iterations. - Drop the skip so the initial turn matches what the toast advertises. New matrix test pins all four (ultrawork, ralph_loop) combinations. multimodal-looker: - Prompt claimed 'read' and 'call_omo_agent' were available, but the look_at invocation runtime explicitly disables both via READ_ENABLED and createAgentToolAllowlist([]). Small VL models trusted the prompt and looped on rejected tool calls (#4116). - Rewrite the agent prompt to describe direct-attachment analysis and forbid tool/agent calls. Add a consistency test that extracts the prompt's 'available tools' claim and compares it against the configured allowlist. delegate-task (skill-resolver): -088693697filtered per-agent restricted skills at the skill tool and builtin agent prompt layers, but delegate-task itself happily injected whatever skill name a caller passed. A target agent could be force-fed a skill marked agent: oracle just by listing it in load_skills. - Thread the target agent through resolveSkills and silently filter skills whose definition.agent does not include it. Public skills with no restriction are unaffected. Regression test pins the bypass.
This commit is contained in:
@@ -1,17 +1,67 @@
|
||||
import { describe, test, expect } from "bun:test"
|
||||
import { createAgentToolAllowlist } from "../shared/permission-compat"
|
||||
import { READ_ENABLED } from "../tools/look-at/look-at-prompt"
|
||||
import { createMultimodalLookerAgent } from "./multimodal-looker"
|
||||
|
||||
function extractAvailableToolClaims(prompt: string): readonly string[] {
|
||||
const availableToolsLine = prompt
|
||||
.split("\n")
|
||||
.find((line) => line.toLowerCase().includes("available tools"))
|
||||
if (availableToolsLine === undefined) {
|
||||
return []
|
||||
}
|
||||
|
||||
const tools: string[] = []
|
||||
for (const match of availableToolsLine.matchAll(/['`]([^'`]+)['`]/g)) {
|
||||
const toolName = match[1]
|
||||
if (toolName !== undefined) {
|
||||
tools.push(toolName)
|
||||
}
|
||||
}
|
||||
|
||||
return [...new Set(tools)].sort()
|
||||
}
|
||||
|
||||
function allowedToolNames(
|
||||
toolAllowlist: ReturnType<typeof createAgentToolAllowlist>
|
||||
): readonly string[] {
|
||||
return Object.entries(toolAllowlist.permission)
|
||||
.filter(([toolName, permission]) => toolName !== "*" && permission === "allow")
|
||||
.map(([toolName]) => toolName)
|
||||
.sort()
|
||||
}
|
||||
|
||||
function createLookAtRuntimeToolAllowlist(): ReturnType<typeof createAgentToolAllowlist> {
|
||||
return createAgentToolAllowlist(READ_ENABLED ? ["read"] : [])
|
||||
}
|
||||
|
||||
describe("createMultimodalLookerAgent", () => {
|
||||
test("prompt explicitly enumerates the agent's available tools to prevent death loop on small VL models", () => {
|
||||
test("prompt available tool claims match the look_at runtime allowlist", () => {
|
||||
// given
|
||||
const agent = createMultimodalLookerAgent("openai/gpt-5-nano")
|
||||
const runtimeToolAllowlist = createLookAtRuntimeToolAllowlist()
|
||||
|
||||
// when
|
||||
const prompt = typeof agent.prompt === "string" ? agent.prompt : ""
|
||||
const promptToolClaims = extractAvailableToolClaims(prompt)
|
||||
const runtimeToolNames = allowedToolNames(runtimeToolAllowlist)
|
||||
|
||||
// then
|
||||
expect(promptToolClaims).toEqual(runtimeToolNames)
|
||||
})
|
||||
|
||||
test("prompt denies tool use to prevent death loop on small VL models", () => {
|
||||
// given
|
||||
const agent = createMultimodalLookerAgent("openai/gpt-5-nano")
|
||||
|
||||
// when
|
||||
const prompt = typeof agent.prompt === "string" ? agent.prompt : ""
|
||||
const normalizedPrompt = prompt.toLowerCase()
|
||||
|
||||
// then
|
||||
expect(prompt).toMatch(/available tools/i)
|
||||
expect(prompt).toContain("read")
|
||||
expect(normalizedPrompt).toContain("never")
|
||||
expect(normalizedPrompt).toContain("tools")
|
||||
expect(extractAvailableToolClaims(prompt)).toEqual([])
|
||||
})
|
||||
|
||||
test("prompt instructs the agent never to call other tools", () => {
|
||||
|
||||
@@ -23,28 +23,28 @@ export function createMultimodalLookerAgent(model: string): AgentConfig {
|
||||
...restrictions,
|
||||
prompt: `You interpret media files that cannot be read as plain text.
|
||||
|
||||
Your only available tools are 'read' and 'call_omo_agent'. Always use 'read' to load the file first, then analyze the returned content. Never attempt to call any other tool.
|
||||
During look_at invocations, the file or image is already attached to the message. Analyze the attachment directly. Never call tools, never spawn other agents, and never try to load the file by path.
|
||||
|
||||
Your job: examine the attached file and extract ONLY what was requested.
|
||||
|
||||
When to use you:
|
||||
- Media files the Read tool cannot interpret
|
||||
- Media files that need visual or document interpretation
|
||||
- Extracting specific information or summaries from documents
|
||||
- Describing visual content in images or diagrams
|
||||
- When analyzed/extracted data is needed, not raw file contents
|
||||
|
||||
When NOT to use you:
|
||||
- Source code or plain text files needing exact contents (use Read)
|
||||
- Files that need editing afterward (need literal content from Read)
|
||||
- Source code or plain text files needing exact contents
|
||||
- Files that need editing afterward
|
||||
- Simple file reading where no interpretation is needed
|
||||
|
||||
How you work:
|
||||
1. Receive a file path and a goal describing what to extract
|
||||
2. Read and analyze the file deeply
|
||||
1. Receive an attached file or image and a goal describing what to extract
|
||||
2. Analyze the attachment deeply
|
||||
3. Return ONLY the relevant extracted information
|
||||
4. The main agent never processes the raw file - you save context tokens
|
||||
|
||||
For PDFs and documents: Use the Read tool to load the file content first, then extract text, structure, tables, data from specific sections
|
||||
For PDFs and documents: extract text, structure, tables, and data from specific sections
|
||||
For images: describe layouts, UI elements, text, diagrams, charts
|
||||
For diagrams: explain relationships, flows, architecture depicted
|
||||
|
||||
|
||||
@@ -0,0 +1,231 @@
|
||||
import { afterEach, beforeEach, describe, expect, test } from "bun:test"
|
||||
import type { OhMyOpenCodeConfig } from "../config"
|
||||
import type { DefaultModeConfig } from "../config/schema/default-mode"
|
||||
import type { CreatedHooks } from "../create-hooks"
|
||||
import { _resetForTesting, setMainSession } from "../features/claude-code-session-state"
|
||||
import { createKeywordDetectorHook } from "../hooks/keyword-detector"
|
||||
import { unsafeTestValue } from "../../test-support/unsafe-test-value"
|
||||
import { createChatMessageHandler, type ChatMessageHandlerOutput } from "./chat-message"
|
||||
import { createSystemTransformHandler } from "./system-transform"
|
||||
import type { PluginContext } from "./types"
|
||||
|
||||
const ULTRAWORK_INSTRUCTION_MARKER = "<ultrawork-mode>matrix ultrawork instructions"
|
||||
const FIRST_TURN_PROMPT = "ship the default-mode priority behavior"
|
||||
const DEFAULT_ULTRAWORK_TOAST = "Default ultrawork mode enabled. All agents at your disposal."
|
||||
|
||||
type ToastCall = {
|
||||
readonly body: {
|
||||
readonly title: string
|
||||
readonly message: string
|
||||
readonly variant: string
|
||||
readonly duration: number
|
||||
}
|
||||
}
|
||||
|
||||
type RalphLoopCall = {
|
||||
readonly sessionID: string
|
||||
readonly prompt: string
|
||||
readonly options: Record<string, unknown>
|
||||
}
|
||||
|
||||
type MatrixCase = {
|
||||
readonly name: string
|
||||
readonly ultrawork: boolean
|
||||
readonly ralphLoop: boolean
|
||||
readonly expectUltraworkSystem: boolean
|
||||
readonly expectToast: boolean
|
||||
readonly expectRalphLoop: boolean
|
||||
}
|
||||
|
||||
const DEFAULT_MODE_CASES = [
|
||||
{
|
||||
name: "neither default mode enabled",
|
||||
ultrawork: false,
|
||||
ralphLoop: false,
|
||||
expectUltraworkSystem: false,
|
||||
expectToast: false,
|
||||
expectRalphLoop: false,
|
||||
},
|
||||
{
|
||||
name: "ultrawork default mode only",
|
||||
ultrawork: true,
|
||||
ralphLoop: false,
|
||||
expectUltraworkSystem: true,
|
||||
expectToast: true,
|
||||
expectRalphLoop: false,
|
||||
},
|
||||
{
|
||||
name: "ralph loop default mode only",
|
||||
ultrawork: false,
|
||||
ralphLoop: true,
|
||||
expectUltraworkSystem: false,
|
||||
expectToast: false,
|
||||
expectRalphLoop: true,
|
||||
},
|
||||
{
|
||||
name: "ultrawork and ralph loop default modes together",
|
||||
ultrawork: true,
|
||||
ralphLoop: true,
|
||||
expectUltraworkSystem: true,
|
||||
expectToast: true,
|
||||
expectRalphLoop: true,
|
||||
},
|
||||
] satisfies readonly MatrixCase[]
|
||||
|
||||
function createDefaultMode(testCase: MatrixCase): DefaultModeConfig {
|
||||
return {
|
||||
ultrawork: testCase.ultrawork,
|
||||
ralph_loop: testCase.ralphLoop,
|
||||
}
|
||||
}
|
||||
|
||||
function createPluginContext(toasts: ToastCall[]): PluginContext {
|
||||
return unsafeTestValue<PluginContext>({
|
||||
client: {
|
||||
tui: {
|
||||
showToast: async (toast: ToastCall): Promise<void> => {
|
||||
toasts.push(toast)
|
||||
},
|
||||
},
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
function createPluginConfig(defaultMode: DefaultModeConfig): OhMyOpenCodeConfig {
|
||||
return unsafeTestValue<OhMyOpenCodeConfig>({
|
||||
default_mode: defaultMode,
|
||||
})
|
||||
}
|
||||
|
||||
function createFirstMessageVariantGate() {
|
||||
let isFirstMessage = true
|
||||
return {
|
||||
shouldOverride: (): boolean => isFirstMessage,
|
||||
markApplied: (): void => {
|
||||
isFirstMessage = false
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
function createHooks(startLoopCalls: RalphLoopCall[]): CreatedHooks {
|
||||
return unsafeTestValue<CreatedHooks>({
|
||||
ralphLoop: {
|
||||
startLoop: (sessionID: string, prompt: string, options?: Record<string, unknown>): boolean => {
|
||||
startLoopCalls.push({ sessionID, prompt, options: options ?? {} })
|
||||
return true
|
||||
},
|
||||
cancelLoop: (): boolean => true,
|
||||
getState: () => null,
|
||||
event: async (): Promise<void> => {},
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
async function renderSystemPrompt(defaultMode: DefaultModeConfig): Promise<string> {
|
||||
const handler = createSystemTransformHandler(
|
||||
defaultMode,
|
||||
() => ULTRAWORK_INSTRUCTION_MARKER,
|
||||
)
|
||||
const output = { system: ["base system prompt"] }
|
||||
|
||||
await handler(
|
||||
{
|
||||
sessionID: "system-transform-session",
|
||||
model: { id: "gpt-5.5", providerID: "openai" },
|
||||
},
|
||||
output,
|
||||
)
|
||||
|
||||
return output.system.join("\n")
|
||||
}
|
||||
|
||||
async function collectDefaultModeToasts(
|
||||
defaultMode: DefaultModeConfig,
|
||||
sessionID: string,
|
||||
): Promise<readonly ToastCall[]> {
|
||||
const toasts: ToastCall[] = []
|
||||
const hook = createKeywordDetectorHook(
|
||||
createPluginContext(toasts),
|
||||
undefined,
|
||||
undefined,
|
||||
undefined,
|
||||
defaultMode,
|
||||
)
|
||||
|
||||
await hook["chat.message"](
|
||||
{ sessionID, agent: "sisyphus" },
|
||||
{
|
||||
message: {},
|
||||
parts: [{ type: "text", text: FIRST_TURN_PROMPT }],
|
||||
},
|
||||
)
|
||||
|
||||
return toasts
|
||||
}
|
||||
|
||||
async function collectRalphLoopCalls(
|
||||
defaultMode: DefaultModeConfig,
|
||||
sessionID: string,
|
||||
): Promise<readonly RalphLoopCall[]> {
|
||||
const startLoopCalls: RalphLoopCall[] = []
|
||||
const handler = createChatMessageHandler({
|
||||
ctx: createPluginContext([]),
|
||||
pluginConfig: createPluginConfig(defaultMode),
|
||||
firstMessageVariantGate: createFirstMessageVariantGate(),
|
||||
hooks: createHooks(startLoopCalls),
|
||||
})
|
||||
const output: ChatMessageHandlerOutput = {
|
||||
message: {},
|
||||
parts: [{ type: "text", text: FIRST_TURN_PROMPT }],
|
||||
}
|
||||
|
||||
await handler(
|
||||
{
|
||||
sessionID,
|
||||
agent: "sisyphus",
|
||||
model: { providerID: "openai", modelID: "gpt-5.5" },
|
||||
},
|
||||
output,
|
||||
)
|
||||
|
||||
return startLoopCalls
|
||||
}
|
||||
|
||||
describe("default-mode priority matrix", () => {
|
||||
beforeEach(() => {
|
||||
_resetForTesting()
|
||||
})
|
||||
|
||||
afterEach(() => {
|
||||
_resetForTesting()
|
||||
})
|
||||
|
||||
for (const testCase of DEFAULT_MODE_CASES) {
|
||||
test(`#given ${testCase.name} #when first user turn runs #then prompt toast and loop state match config`, async () => {
|
||||
// given
|
||||
const defaultMode = createDefaultMode(testCase)
|
||||
const sessionID = `default-mode-${testCase.ultrawork}-${testCase.ralphLoop}`
|
||||
setMainSession(sessionID)
|
||||
|
||||
// when
|
||||
const systemPrompt = await renderSystemPrompt(defaultMode)
|
||||
const toasts = await collectDefaultModeToasts(defaultMode, sessionID)
|
||||
const startLoopCalls = await collectRalphLoopCalls(defaultMode, sessionID)
|
||||
|
||||
// then
|
||||
expect(systemPrompt.includes(ULTRAWORK_INSTRUCTION_MARKER)).toBe(
|
||||
testCase.expectUltraworkSystem,
|
||||
)
|
||||
expect(toasts.map((toast) => toast.body.message)).toEqual(
|
||||
testCase.expectToast ? [DEFAULT_ULTRAWORK_TOAST] : [],
|
||||
)
|
||||
expect(startLoopCalls.length > 0).toBe(testCase.expectRalphLoop)
|
||||
if (testCase.expectRalphLoop) {
|
||||
expect(startLoopCalls).toHaveLength(1)
|
||||
expect(startLoopCalls[0]?.sessionID).toBe(sessionID)
|
||||
expect(startLoopCalls[0]?.prompt).toBe(FIRST_TURN_PROMPT)
|
||||
expect(startLoopCalls[0]?.options["ultrawork"]).toBe(testCase.ultrawork)
|
||||
}
|
||||
})
|
||||
}
|
||||
})
|
||||
@@ -12,13 +12,6 @@ export function createSystemTransformHandler(
|
||||
return async (input, output): Promise<void> => {
|
||||
if (!defaultMode?.ultrawork || !getUltraworkMessage) return
|
||||
|
||||
// When ralph_loop is also enabled, skip system prompt injection: the loop's
|
||||
// own continuation mechanism handles ultrawork re-injection on each iteration.
|
||||
// Injecting both would be redundant — the continuation prompt prepends
|
||||
// "ultrawork" and re-triggers keyword detection, defeating invisibility.
|
||||
// The `ultrawork` flag still controls loop behavior (500 iters + Oracle gate).
|
||||
if (defaultMode?.ralph_loop) return
|
||||
|
||||
// Avoid re-injecting if the ultrawork prompt is already in the system prompt
|
||||
// (e.g. after compaction the system prompt is rebuilt and this hook fires again)
|
||||
if (output.system.some((part) => part.includes(ULTRAWORK_MODE_TAG))) return
|
||||
|
||||
@@ -173,6 +173,35 @@ describe("resolveSkillContent — nativeSkills integration", () => {
|
||||
expect(result.content).toContain("SHORT_NAME_BODY")
|
||||
})
|
||||
|
||||
it("#given an agent-restricted OMO skill #when another target agent requests it #then filters the restricted skill but keeps public skills", async () => {
|
||||
// given
|
||||
const oracleSkillDir = join(TEST_DIR, ".opencode", "skills", "oracle-only-skill")
|
||||
mkdirSync(oracleSkillDir, { recursive: true })
|
||||
writeFileSync(
|
||||
join(oracleSkillDir, "SKILL.md"),
|
||||
"---\nname: oracle-only-skill\ndescription: Oracle only\nagent: oracle\n---\nORACLE_ONLY_BODY",
|
||||
)
|
||||
|
||||
const publicSkillDir = join(TEST_DIR, ".opencode", "skills", "public-skill")
|
||||
mkdirSync(publicSkillDir, { recursive: true })
|
||||
writeFileSync(
|
||||
join(publicSkillDir, "SKILL.md"),
|
||||
"---\nname: public-skill\ndescription: Public skill\n---\nPUBLIC_BODY",
|
||||
)
|
||||
|
||||
// when
|
||||
const result = await resolveSkillContent(["oracle-only-skill", "public-skill"], {
|
||||
directory: TEST_DIR,
|
||||
targetAgent: "explore",
|
||||
})
|
||||
|
||||
// then
|
||||
expect(result.error).toBeNull()
|
||||
expect(result.content).not.toContain("ORACLE_ONLY_BODY")
|
||||
expect(result.content).toContain("PUBLIC_BODY")
|
||||
expect(result.contents).toHaveLength(1)
|
||||
})
|
||||
|
||||
it("#given no nativeSkills passed #when resolved #then behaves like pre-fix (no native discovery)", async () => {
|
||||
// when
|
||||
const result = await resolveSkillContent(["does-not-exist"], {
|
||||
|
||||
@@ -6,6 +6,7 @@ import {
|
||||
injectGitMasterConfig,
|
||||
} from "../../features/opencode-skill-loader/skill-content"
|
||||
import type { LoadedSkill } from "../../features/opencode-skill-loader/types"
|
||||
import { getAgentConfigKey } from "../../shared/agent-display-names"
|
||||
import { log } from "../../shared/logger"
|
||||
import { mergeNativeSkills } from "../skill/native-skills"
|
||||
import type { NativeSkillEntry } from "../skill/native-skills"
|
||||
@@ -18,10 +19,18 @@ type ResolveSkillContentOptions = {
|
||||
disabledSkills?: Set<string>
|
||||
teamModeEnabled?: boolean
|
||||
directory?: string
|
||||
targetAgent?: string
|
||||
nativeSkills?: DelegateTaskToolOptions["nativeSkills"]
|
||||
nativeSkillEntries?: NativeSkillEntry[]
|
||||
}
|
||||
|
||||
function isSkillAllowedForTargetAgent(skill: LoadedSkill, targetAgent: string | undefined): boolean {
|
||||
const restrictedAgent = skill.definition.agent
|
||||
if (!restrictedAgent) return true
|
||||
if (!targetAgent) return false
|
||||
return getAgentConfigKey(restrictedAgent) === getAgentConfigKey(targetAgent)
|
||||
}
|
||||
|
||||
async function loadNativeSkillEntries(
|
||||
nativeSkills: DelegateTaskToolOptions["nativeSkills"] | undefined,
|
||||
nativeSkillEntries: NativeSkillEntry[] | undefined,
|
||||
@@ -55,13 +64,34 @@ export async function resolveSkillContent(
|
||||
|
||||
const resolved = new Map<string, string>()
|
||||
const notFound: string[] = []
|
||||
let unfilteredDiscoveredSkills: LoadedSkill[] | undefined
|
||||
|
||||
const getUnfilteredDiscoveredSkills = async (): Promise<LoadedSkill[]> => {
|
||||
if (unfilteredDiscoveredSkills) return unfilteredDiscoveredSkills
|
||||
unfilteredDiscoveredSkills = await discoverSkills({
|
||||
includeClaudeCodePaths: true,
|
||||
directory: options.directory,
|
||||
})
|
||||
return unfilteredDiscoveredSkills
|
||||
}
|
||||
|
||||
for (const name of skills) {
|
||||
const skill = matchSkillByName(baseSkills, name)
|
||||
let skill = matchSkillByName(baseSkills, name)
|
||||
if (!skill && options.browserProvider === undefined && !options.disabledSkills?.has(name)) {
|
||||
skill = matchSkillByName(await getUnfilteredDiscoveredSkills(), name)
|
||||
}
|
||||
if (!skill) {
|
||||
notFound.push(name)
|
||||
continue
|
||||
}
|
||||
if (!isSkillAllowedForTargetAgent(skill, options.targetAgent)) {
|
||||
log("[skill-resolver] filtered agent-restricted skill for delegate target", {
|
||||
skill: skill.name,
|
||||
restricted_agent: skill.definition.agent,
|
||||
target_agent: options.targetAgent ?? "(unknown)",
|
||||
})
|
||||
continue
|
||||
}
|
||||
const template = extractSkillTemplate(skill)
|
||||
if (name === "git-master") {
|
||||
resolved.set(name, injectGitMasterConfig(template, options.gitMasterConfig))
|
||||
|
||||
@@ -74,6 +74,7 @@ export function createDelegateTask(options: DelegateTaskToolOptions): ToolDefini
|
||||
disabledSkills: options.disabledSkills,
|
||||
teamModeEnabled: options.teamModeEnabled,
|
||||
directory: options.directory,
|
||||
targetAgent: delegateTaskArgs.subagent_type,
|
||||
nativeSkills: options.nativeSkills,
|
||||
nativeSkillEntries,
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user