mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-05-14 05:42:54 -07:00
ANDI FAUZAN HEDIANTORO
87 lines
2.5 KiB
TypeScript
87 lines
2.5 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2026 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import {
|
|
type AgentLoopContext,
|
|
ShellExecutionService,
|
|
ShellTool,
|
|
type Config as CoreConfig,
|
|
} from '@google/gemini-cli-core';
|
|
import type {
|
|
AgentShell,
|
|
AgentShellResult,
|
|
AgentShellOptions,
|
|
} from './types.js';
|
|
|
|
/**
|
|
* SDK implementation of {@link AgentShell} that executes commands via the
|
|
* core ShellExecutionService, subject to the agent's security policies.
|
|
*
|
|
* Commands that require interactive confirmation will be rejected since
|
|
* no interactive session is available in headless SDK mode.
|
|
*
|
|
* @remarks In this implementation, stderr is combined into stdout by the
|
|
* underlying ShellExecutionService. As a result, the stderr field of the
|
|
* returned {@link AgentShellResult} will be empty, and both output and
|
|
* stdout will contain the combined output.
|
|
*/
|
|
export class SdkAgentShell implements AgentShell {
|
|
constructor(private readonly config: CoreConfig) {}
|
|
|
|
async exec(
|
|
command: string,
|
|
options?: AgentShellOptions,
|
|
): Promise<AgentShellResult> {
|
|
const cwd = options?.cwd || this.config.getWorkingDir();
|
|
const abortController = new AbortController();
|
|
|
|
// Use ShellTool to check policy
|
|
const loopContext: AgentLoopContext = this.config;
|
|
const shellTool = new ShellTool(this.config, loopContext.messageBus);
|
|
try {
|
|
const invocation = shellTool.build({
|
|
command,
|
|
dir_path: cwd,
|
|
});
|
|
|
|
const confirmation = await invocation.shouldConfirmExecute(
|
|
abortController.signal,
|
|
);
|
|
if (confirmation) {
|
|
throw new Error(
|
|
'Command execution requires confirmation but no interactive session is available.',
|
|
);
|
|
}
|
|
} catch (error) {
|
|
return {
|
|
output: '',
|
|
stdout: '',
|
|
stderr: '',
|
|
exitCode: 1,
|
|
error: error instanceof Error ? error : new Error(String(error)),
|
|
};
|
|
}
|
|
|
|
const handle = await ShellExecutionService.execute(
|
|
command,
|
|
cwd,
|
|
() => {}, // No-op output event handler for now
|
|
abortController.signal,
|
|
false, // shouldUseNodePty: false for headless execution
|
|
this.config.getShellExecutionConfig(),
|
|
);
|
|
|
|
const result = await handle.result;
|
|
|
|
return {
|
|
output: result.output,
|
|
stdout: result.output, // ShellExecutionService combines stdout/stderr usually
|
|
stderr: '', // ShellExecutionService currently combines, so stderr is empty or mixed
|
|
exitCode: result.exitCode,
|
|
};
|
|
}
|
|
}
|