summaryrefslogtreecommitdiff
path: root/packages/cli/src/gemini.tsx
diff options
context:
space:
mode:
Diffstat (limited to 'packages/cli/src/gemini.tsx')
-rw-r--r--packages/cli/src/gemini.tsx8
1 files changed, 5 insertions, 3 deletions
diff --git a/packages/cli/src/gemini.tsx b/packages/cli/src/gemini.tsx
index 11875593..9cfaef37 100644
--- a/packages/cli/src/gemini.tsx
+++ b/packages/cli/src/gemini.tsx
@@ -95,9 +95,11 @@ async function main() {
const geminiClient = new GeminiClient(config);
const chat = await geminiClient.startChat();
try {
- for await (const event of geminiClient.sendMessageStream(chat, [
- { text: input },
- ])) {
+ for await (const event of geminiClient.sendMessageStream(
+ chat,
+ [{ text: input }],
+ new AbortController().signal,
+ )) {
if (event.type === 'content') {
process.stdout.write(event.value);
}