diff options
Diffstat (limited to 'packages/cli/src/nonInteractiveCli.ts')
| -rw-r--r-- | packages/cli/src/nonInteractiveCli.ts | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/packages/cli/src/nonInteractiveCli.ts b/packages/cli/src/nonInteractiveCli.ts index b8b8ac3f..2db28eba 100644 --- a/packages/cli/src/nonInteractiveCli.ts +++ b/packages/cli/src/nonInteractiveCli.ts @@ -63,9 +63,19 @@ export async function runNonInteractive( const chat = await geminiClient.getChat(); const abortController = new AbortController(); let currentMessages: Content[] = [{ role: 'user', parts: [{ text: input }] }]; - + let turnCount = 0; try { while (true) { + turnCount++; + if ( + config.getMaxSessionTurns() > 0 && + turnCount > config.getMaxSessionTurns() + ) { + console.error( + '\n Reached max session turns for this session. Increase the number of turns by specifying maxSessionTurns in settings.json.', + ); + return; + } const functionCalls: FunctionCall[] = []; const responseStream = await chat.sendMessageStream( |
