mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-05-13 05:12:55 -07:00
Gaurav
78 lines
2.5 KiB
TypeScript
78 lines
2.5 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2026 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import type { Config } from '../config/config.js';
|
|
import { Scheduler } from '../scheduler/scheduler.js';
|
|
import type {
|
|
ToolCallRequestInfo,
|
|
CompletedToolCall,
|
|
} from '../scheduler/types.js';
|
|
import type { ToolRegistry } from '../tools/tool-registry.js';
|
|
import type { EditorType } from '../utils/editor.js';
|
|
|
|
/**
|
|
* Options for scheduling agent tools.
|
|
*/
|
|
export interface AgentSchedulingOptions {
|
|
/** The unique ID for this agent's scheduler. */
|
|
schedulerId: string;
|
|
/** The ID of the tool call that invoked this agent. */
|
|
parentCallId?: string;
|
|
/** The tool registry specific to this agent. */
|
|
toolRegistry: ToolRegistry;
|
|
/** AbortSignal for cancellation. */
|
|
signal: AbortSignal;
|
|
/** Optional function to get the preferred editor for tool modifications. */
|
|
getPreferredEditor?: () => EditorType | undefined;
|
|
/** Optional function to be notified when the scheduler is waiting for user confirmation. */
|
|
onWaitingForConfirmation?: (waiting: boolean) => void;
|
|
}
|
|
|
|
/**
|
|
* Schedules a batch of tool calls for an agent using the new event-driven Scheduler.
|
|
*
|
|
* @param config The global runtime configuration.
|
|
* @param requests The list of tool call requests from the agent.
|
|
* @param options Scheduling options including registry and IDs.
|
|
* @returns A promise that resolves to the completed tool calls.
|
|
*/
|
|
export async function scheduleAgentTools(
|
|
config: Config,
|
|
requests: ToolCallRequestInfo[],
|
|
options: AgentSchedulingOptions,
|
|
): Promise<CompletedToolCall[]> {
|
|
const {
|
|
schedulerId,
|
|
parentCallId,
|
|
toolRegistry,
|
|
signal,
|
|
getPreferredEditor,
|
|
onWaitingForConfirmation,
|
|
} = options;
|
|
|
|
// Create a proxy/override of the config to provide the agent-specific tool registry.
|
|
// eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
|
|
const agentConfig: Config = Object.create(config);
|
|
agentConfig.getToolRegistry = () => toolRegistry;
|
|
agentConfig.getMessageBus = () => toolRegistry.getMessageBus();
|
|
// Override toolRegistry property so AgentLoopContext reads the agent-specific registry.
|
|
Object.defineProperty(agentConfig, 'toolRegistry', {
|
|
get: () => toolRegistry,
|
|
configurable: true,
|
|
});
|
|
|
|
const scheduler = new Scheduler({
|
|
config: agentConfig,
|
|
messageBus: toolRegistry.getMessageBus(),
|
|
getPreferredEditor: getPreferredEditor ?? (() => undefined),
|
|
schedulerId,
|
|
parentCallId,
|
|
onWaitingForConfirmation,
|
|
});
|
|
|
|
return scheduler.schedule(requests, signal);
|
|
}
|