github/claude-code-best
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(agents): propagate allowedAgentTypes from agent tool resolution to query engine

Browse Source
This commit is contained in:
yhangf 2026-05-09 15:15:10 +08:00
parent 1efba6b17a
commit 396e4cdbe3
2 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/QueryEngine.ts
View File

@ -140,6 +140,7 @@ export type QueryEngineConfig = {
commands: Command[]
mcpClients: MCPServerConnection[]
agents: AgentDefinition[]
allowedAgentTypes?: string[]
canUseTool: CanUseToolFn
getAppState: () => AppState
setAppState: (f: (prev: AppState) => AppState) => void
@ -240,6 +241,7 @@ export class QueryEngine {
replayUserMessages = false,
includePartialMessages = false,
agents = [],
allowedAgentTypes,
setSDKStatus,
orphanedPermission,
} = this.config
@ -383,7 +385,7 @@ export class QueryEngine {
isNonInteractiveSession: true,
customSystemPrompt,
appendSystemPrompt,
agentDefinitions: { activeAgents: agents, allAgents: [] },
agentDefinitions: { activeAgents: agents, allAgents: [], ...(allowedAgentTypes ? { allowedAgentTypes } : {}) },
theme: resolveThemeSetting(getGlobalConfig().theme),
maxBudgetUsd,
},
@ -533,7 +535,7 @@ export class QueryEngine {
customSystemPrompt,
appendSystemPrompt,
theme: resolveThemeSetting(getGlobalConfig().theme),
agentDefinitions: { activeAgents: agents, allAgents: [] },
agentDefinitions: { activeAgents: agents, allAgents: [], ...(allowedAgentTypes ? { allowedAgentTypes } : {}) },
maxBudgetUsd,
},
getAppState,
@ -1273,6 +1275,7 @@ export async function* ask({
includePartialMessages = false,
handleElicitation,
agents = [],
allowedAgentTypes,
setSDKStatus,
orphanedPermission,
}: {
@ -1305,6 +1308,7 @@ export async function* ask({
includePartialMessages?: boolean
handleElicitation?: ToolUseContext['handleElicitation']
agents?: AgentDefinition[]
allowedAgentTypes?: string[]
setSDKStatus?: (status: SDKStatus) => void
orphanedPermission?: OrphanedPermission
}): AsyncGenerator<SDKMessage, void, unknown> {
@ -1314,6 +1318,7 @@ export async function* ask({
commands,
mcpClients,
agents: agents ?? [],
allowedAgentTypes,
canUseTool,
getAppState,
setAppState,

9
src/cli/print.ts
View File

@ -2032,9 +2032,11 @@ function runHeadlessStreaming(
const mainThreadAgentDef = currentAgents.find(
a => a.agentType === getMainThreadAgentType(),
)
const allTools = mainThreadAgentDef
? resolveAgentTools(mainThreadAgentDef, rawTools, false, true).resolvedTools
: rawTools
const resolvedAgentResult = mainThreadAgentDef
? resolveAgentTools(mainThreadAgentDef, rawTools, false, true)
: null
const allTools = resolvedAgentResult ? resolvedAgentResult.resolvedTools : rawTools
const mainThreadAllowedAgentTypes = resolvedAgentResult?.allowedAgentTypes
for (const uuid of batchUuids) {
notifyCommandLifecycle(uuid, 'started')
@ -2245,6 +2247,7 @@ function runHeadlessStreaming(
: undefined,
),
agents: currentAgents,
allowedAgentTypes: mainThreadAllowedAgentTypes,
orphanedPermission: cmd.orphanedPermission,
setSDKStatus: status => {
output.enqueue({