@google/gemini-cli-core 0.0.3-preview.4

package/dist/src/core/turn.d.ts

@@ -0,0 +1,125 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Part, PartListUnion, GenerateContentResponse, FunctionDeclaration, FinishReason } from '@google/genai';
+import type { ToolCallConfirmationDetails, ToolResult, ToolResultDisplay } from '../tools/tools.js';
+import type { ToolErrorType } from '../tools/tool-error.js';
+import type { GeminiChat } from './geminiChat.js';
+export interface ServerTool {
+    name: string;
+    schema: FunctionDeclaration;
+    execute(params: Record<string, unknown>, signal?: AbortSignal): Promise<ToolResult>;
+    shouldConfirmExecute(params: Record<string, unknown>, abortSignal: AbortSignal): Promise<ToolCallConfirmationDetails | false>;
+}
+export declare enum GeminiEventType {
+    Content = "content",
+    ToolCallRequest = "tool_call_request",
+    ToolCallResponse = "tool_call_response",
+    ToolCallConfirmation = "tool_call_confirmation",
+    UserCancelled = "user_cancelled",
+    Error = "error",
+    ChatCompressed = "chat_compressed",
+    Thought = "thought",
+    MaxSessionTurns = "max_session_turns",
+    Finished = "finished",
+    LoopDetected = "loop_detected"
+}
+export interface StructuredError {
+    message: string;
+    status?: number;
+}
+export interface GeminiErrorEventValue {
+    error: StructuredError;
+}
+export interface ToolCallRequestInfo {
+    callId: string;
+    name: string;
+    args: Record<string, unknown>;
+    isClientInitiated: boolean;
+    prompt_id: string;
+}
+export interface ToolCallResponseInfo {
+    callId: string;
+    responseParts: Part[];
+    resultDisplay: ToolResultDisplay | undefined;
+    error: Error | undefined;
+    errorType: ToolErrorType | undefined;
+}
+export interface ServerToolCallConfirmationDetails {
+    request: ToolCallRequestInfo;
+    details: ToolCallConfirmationDetails;
+}
+export type ThoughtSummary = {
+    subject: string;
+    description: string;
+};
+export type ServerGeminiContentEvent = {
+    type: GeminiEventType.Content;
+    value: string;
+};
+export type ServerGeminiThoughtEvent = {
+    type: GeminiEventType.Thought;
+    value: ThoughtSummary;
+};
+export type ServerGeminiToolCallRequestEvent = {
+    type: GeminiEventType.ToolCallRequest;
+    value: ToolCallRequestInfo;
+};
+export type ServerGeminiToolCallResponseEvent = {
+    type: GeminiEventType.ToolCallResponse;
+    value: ToolCallResponseInfo;
+};
+export type ServerGeminiToolCallConfirmationEvent = {
+    type: GeminiEventType.ToolCallConfirmation;
+    value: ServerToolCallConfirmationDetails;
+};
+export type ServerGeminiUserCancelledEvent = {
+    type: GeminiEventType.UserCancelled;
+};
+export type ServerGeminiErrorEvent = {
+    type: GeminiEventType.Error;
+    value: GeminiErrorEventValue;
+};
+export declare enum CompressionStatus {
+    /** The compression was successful */
+    COMPRESSED = 1,
+    /** The compression failed due to the compression inflating the token count */
+    COMPRESSION_FAILED_INFLATED_TOKEN_COUNT = 2,
+    /** The compression failed due to an error counting tokens */
+    COMPRESSION_FAILED_TOKEN_COUNT_ERROR = 3,
+    /** The compression was not necessary and no action was taken */
+    NOOP = 4
+}
+export interface ChatCompressionInfo {
+    originalTokenCount: number;
+    newTokenCount: number;
+    compressionStatus: CompressionStatus;
+}
+export type ServerGeminiChatCompressedEvent = {
+    type: GeminiEventType.ChatCompressed;
+    value: ChatCompressionInfo | null;
+};
+export type ServerGeminiMaxSessionTurnsEvent = {
+    type: GeminiEventType.MaxSessionTurns;
+};
+export type ServerGeminiFinishedEvent = {
+    type: GeminiEventType.Finished;
+    value: FinishReason;
+};
+export type ServerGeminiLoopDetectedEvent = {
+    type: GeminiEventType.LoopDetected;
+};
+export type ServerGeminiStreamEvent = ServerGeminiContentEvent | ServerGeminiToolCallRequestEvent | ServerGeminiToolCallResponseEvent | ServerGeminiToolCallConfirmationEvent | ServerGeminiUserCancelledEvent | ServerGeminiErrorEvent | ServerGeminiChatCompressedEvent | ServerGeminiThoughtEvent | ServerGeminiMaxSessionTurnsEvent | ServerGeminiFinishedEvent | ServerGeminiLoopDetectedEvent;
+export declare class Turn {
+    private readonly chat;
+    private readonly prompt_id;
+    readonly pendingToolCalls: ToolCallRequestInfo[];
+    private debugResponses;
+    finishReason: FinishReason | undefined;
+    constructor(chat: GeminiChat, prompt_id: string);
+    run(req: PartListUnion, signal: AbortSignal): AsyncGenerator<ServerGeminiStreamEvent>;
+    private handlePendingFunctionCall;
+    getDebugResponses(): GenerateContentResponse[];
+}

package/dist/src/core/turn.js

@@ -0,0 +1,154 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { getResponseText } from '../utils/partUtils.js';
+import { reportError } from '../utils/errorReporting.js';
+import { getErrorMessage, UnauthorizedError, toFriendlyError, } from '../utils/errors.js';
+export var GeminiEventType;
+(function (GeminiEventType) {
+    GeminiEventType["Content"] = "content";
+    GeminiEventType["ToolCallRequest"] = "tool_call_request";
+    GeminiEventType["ToolCallResponse"] = "tool_call_response";
+    GeminiEventType["ToolCallConfirmation"] = "tool_call_confirmation";
+    GeminiEventType["UserCancelled"] = "user_cancelled";
+    GeminiEventType["Error"] = "error";
+    GeminiEventType["ChatCompressed"] = "chat_compressed";
+    GeminiEventType["Thought"] = "thought";
+    GeminiEventType["MaxSessionTurns"] = "max_session_turns";
+    GeminiEventType["Finished"] = "finished";
+    GeminiEventType["LoopDetected"] = "loop_detected";
+})(GeminiEventType || (GeminiEventType = {}));
+export var CompressionStatus;
+(function (CompressionStatus) {
+    /** The compression was successful */
+    CompressionStatus[CompressionStatus["COMPRESSED"] = 1] = "COMPRESSED";
+    /** The compression failed due to the compression inflating the token count */
+    CompressionStatus[CompressionStatus["COMPRESSION_FAILED_INFLATED_TOKEN_COUNT"] = 2] = "COMPRESSION_FAILED_INFLATED_TOKEN_COUNT";
+    /** The compression failed due to an error counting tokens */
+    CompressionStatus[CompressionStatus["COMPRESSION_FAILED_TOKEN_COUNT_ERROR"] = 3] = "COMPRESSION_FAILED_TOKEN_COUNT_ERROR";
+    /** The compression was not necessary and no action was taken */
+    CompressionStatus[CompressionStatus["NOOP"] = 4] = "NOOP";
+})(CompressionStatus || (CompressionStatus = {}));
+// A turn manages the agentic loop turn within the server context.
+export class Turn {
+    chat;
+    prompt_id;
+    pendingToolCalls;
+    debugResponses;
+    finishReason;
+    constructor(chat, prompt_id) {
+        this.chat = chat;
+        this.prompt_id = prompt_id;
+        this.pendingToolCalls = [];
+        this.debugResponses = [];
+        this.finishReason = undefined;
+    }
+    // The run method yields simpler events suitable for server logic
+    async *run(req, signal) {
+        try {
+            const responseStream = await this.chat.sendMessageStream({
+                message: req,
+                config: {
+                    abortSignal: signal,
+                },
+            }, this.prompt_id);
+            for await (const resp of responseStream) {
+                if (signal?.aborted) {
+                    yield { type: GeminiEventType.UserCancelled };
+                    // Do not add resp to debugResponses if aborted before processing
+                    return;
+                }
+                this.debugResponses.push(resp);
+                const thoughtPart = resp.candidates?.[0]?.content?.parts?.[0];
+                if (thoughtPart?.thought) {
+                    // Thought always has a bold "subject" part enclosed in double asterisks
+                    // (e.g., **Subject**). The rest of the string is considered the description.
+                    const rawText = thoughtPart.text ?? '';
+                    const subjectStringMatches = rawText.match(/\*\*(.*?)\*\*/s);
+                    const subject = subjectStringMatches
+                        ? subjectStringMatches[1].trim()
+                        : '';
+                    const description = rawText.replace(/\*\*(.*?)\*\*/s, '').trim();
+                    const thought = {
+                        subject,
+                        description,
+                    };
+                    yield {
+                        type: GeminiEventType.Thought,
+                        value: thought,
+                    };
+                    continue;
+                }
+                const text = getResponseText(resp);
+                if (text) {
+                    yield { type: GeminiEventType.Content, value: text };
+                }
+                // Handle function calls (requesting tool execution)
+                const functionCalls = resp.functionCalls ?? [];
+                for (const fnCall of functionCalls) {
+                    const event = this.handlePendingFunctionCall(fnCall);
+                    if (event) {
+                        yield event;
+                    }
+                }
+                // Check if response was truncated or stopped for various reasons
+                const finishReason = resp.candidates?.[0]?.finishReason;
+                if (finishReason) {
+                    this.finishReason = finishReason;
+                    yield {
+                        type: GeminiEventType.Finished,
+                        value: finishReason,
+                    };
+                }
+            }
+        }
+        catch (e) {
+            if (signal.aborted) {
+                yield { type: GeminiEventType.UserCancelled };
+                // Regular cancellation error, fail gracefully.
+                return;
+            }
+            const error = toFriendlyError(e);
+            if (error instanceof UnauthorizedError) {
+                throw error;
+            }
+            const contextForReport = [...this.chat.getHistory(/*curated*/ true), req];
+            await reportError(error, 'Error when talking to Gemini API', contextForReport, 'Turn.run-sendMessageStream');
+            const status = typeof error === 'object' &&
+                error !== null &&
+                'status' in error &&
+                typeof error.status === 'number'
+                ? error.status
+                : undefined;
+            const structuredError = {
+                message: getErrorMessage(error),
+                status,
+            };
+            await this.chat.maybeIncludeSchemaDepthContext(structuredError);
+            yield { type: GeminiEventType.Error, value: { error: structuredError } };
+            return;
+        }
+    }
+    handlePendingFunctionCall(fnCall) {
+        const callId = fnCall.id ??
+            `${fnCall.name}-${Date.now()}-${Math.random().toString(16).slice(2)}`;
+        const name = fnCall.name || 'undefined_tool_name';
+        const args = (fnCall.args || {});
+        const toolCallRequest = {
+            callId,
+            name,
+            args,
+            isClientInitiated: false,
+            prompt_id: this.prompt_id,
+        };
+        this.pendingToolCalls.push(toolCallRequest);
+        // Yield a request for the tool call, not the pending/confirming status
+        return { type: GeminiEventType.ToolCallRequest, value: toolCallRequest };
+    }
+    getDebugResponses() {
+        return this.debugResponses;
+    }
+}
+//# sourceMappingURL=turn.js.map

package/dist/src/core/turn.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"turn.js","sourceRoot":"","sources":["../../../src/core/turn.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAgBH,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EACL,eAAe,EACf,iBAAiB,EACjB,eAAe,GAChB,MAAM,oBAAoB,CAAC;AAkB5B,MAAM,CAAN,IAAY,eAYX;AAZD,WAAY,eAAe;IACzB,sCAAmB,CAAA;IACnB,wDAAqC,CAAA;IACrC,0DAAuC,CAAA;IACvC,kEAA+C,CAAA;IAC/C,mDAAgC,CAAA;IAChC,kCAAe,CAAA;IACf,qDAAkC,CAAA;IAClC,sCAAmB,CAAA;IACnB,wDAAqC,CAAA;IACrC,wCAAqB,CAAA;IACrB,iDAA8B,CAAA;AAChC,CAAC,EAZW,eAAe,KAAf,eAAe,QAY1B;AAuED,MAAM,CAAN,IAAY,iBAYX;AAZD,WAAY,iBAAiB;IAC3B,qCAAqC;IACrC,qEAAc,CAAA;IAEd,8EAA8E;IAC9E,+HAAuC,CAAA;IAEvC,6DAA6D;IAC7D,yHAAoC,CAAA;IAEpC,gEAAgE;IAChE,yDAAI,CAAA;AACN,CAAC,EAZW,iBAAiB,KAAjB,iBAAiB,QAY5B;AAwCD,kEAAkE;AAClE,MAAM,OAAO,IAAI;IAMI;IACA;IANV,gBAAgB,CAAwB;IACzC,cAAc,CAA4B;IAClD,YAAY,CAA2B;IAEvC,YACmB,IAAgB,EAChB,SAAiB;QADjB,SAAI,GAAJ,IAAI,CAAY;QAChB,cAAS,GAAT,SAAS,CAAQ;QAElC,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAC;QAC3B,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;IAChC,CAAC;IACD,iEAAiE;IACjE,KAAK,CAAC,CAAC,GAAG,CACR,GAAkB,EAClB,MAAmB;QAEnB,IAAI,CAAC;YACH,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,iBAAiB,CACtD;gBACE,OAAO,EAAE,GAAG;gBACZ,MAAM,EAAE;oBACN,WAAW,EAAE,MAAM;iBACpB;aACF,EACD,IAAI,CAAC,SAAS,CACf,CAAC;YAEF,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;gBACxC,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;oBACpB,MAAM,EAAE,IAAI,EAAE,eAAe,CAAC,aAAa,EAAE,CAAC;oBAC9C,iEAAiE;oBACjE,OAAO;gBACT,CAAC;gBACD,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAE/B,MAAM,WAAW,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC9D,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;oBACzB,wEAAwE;oBACxE,6EAA6E;oBAC7E,MAAM,OAAO,GAAG,WAAW,CAAC,IAAI,IAAI,EAAE,CAAC;oBACvC,MAAM,oBAAoB,GAAG,OAAO,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;oBAC7D,MAAM,OAAO,GAAG,oBAAoB;wBAClC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE;wBAChC,CAAC,CAAC,EAAE,CAAC;oBACP,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;oBACjE,MAAM,OAAO,GAAmB;wBAC9B,OAAO;wBACP,WAAW;qBACZ,CAAC;oBAEF,MAAM;wBACJ,IAAI,EAAE,eAAe,CAAC,OAAO;wBAC7B,KAAK,EAAE,OAAO;qBACf,CAAC;oBACF,SAAS;gBACX,CAAC;gBAED,MAAM,IAAI,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;gBACnC,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM,EAAE,IAAI,EAAE,eAAe,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;gBACvD,CAAC;gBAED,oDAAoD;gBACpD,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC;gBAC/C,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;oBACnC,MAAM,KAAK,GAAG,IAAI,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC;oBACrD,IAAI,KAAK,EAAE,CAAC;wBACV,MAAM,KAAK,CAAC;oBACd,CAAC;gBACH,CAAC;gBAED,iEAAiE;gBACjE,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;gBAExD,IAAI,YAAY,EAAE,CAAC;oBACjB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;oBACjC,MAAM;wBACJ,IAAI,EAAE,eAAe,CAAC,QAAQ;wBAC9B,KAAK,EAAE,YAA4B;qBACpC,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,MAAM,EAAE,IAAI,EAAE,eAAe,CAAC,aAAa,EAAE,CAAC;gBAC9C,+CAA+C;gBAC/C,OAAO;YACT,CAAC;YAED,MAAM,KAAK,GAAG,eAAe,CAAC,CAAC,CAAC,CAAC;YACjC,IAAI,KAAK,YAAY,iBAAiB,EAAE,CAAC;gBACvC,MAAM,KAAK,CAAC;YACd,CAAC;YAED,MAAM,gBAAgB,GAAG,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC;YAC1E,MAAM,WAAW,CACf,KAAK,EACL,kCAAkC,EAClC,gBAAgB,EAChB,4BAA4B,CAC7B,CAAC;YACF,MAAM,MAAM,GACV,OAAO,KAAK,KAAK,QAAQ;gBACzB,KAAK,KAAK,IAAI;gBACd,QAAQ,IAAI,KAAK;gBACjB,OAAQ,KAA6B,CAAC,MAAM,KAAK,QAAQ;gBACvD,CAAC,CAAE,KAA4B,CAAC,MAAM;gBACtC,CAAC,CAAC,SAAS,CAAC;YAChB,MAAM,eAAe,GAAoB;gBACvC,OAAO,EAAE,eAAe,CAAC,KAAK,CAAC;gBAC/B,MAAM;aACP,CAAC;YACF,MAAM,IAAI,CAAC,IAAI,CAAC,8BAA8B,CAAC,eAAe,CAAC,CAAC;YAChE,MAAM,EAAE,IAAI,EAAE,eAAe,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,EAAE,CAAC;YACzE,OAAO;QACT,CAAC;IACH,CAAC;IAEO,yBAAyB,CAC/B,MAAoB;QAEpB,MAAM,MAAM,GACV,MAAM,CAAC,EAAE;YACT,GAAG,MAAM,CAAC,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QACxE,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,qBAAqB,CAAC;QAClD,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAA4B,CAAC;QAE5D,MAAM,eAAe,GAAwB;YAC3C,MAAM;YACN,IAAI;YACJ,IAAI;YACJ,iBAAiB,EAAE,KAAK;YACxB,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC;QAEF,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAE5C,uEAAuE;QACvE,OAAO,EAAE,IAAI,EAAE,eAAe,CAAC,eAAe,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC;IAC3E,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;CACF"}

package/dist/src/core/turn.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/core/turn.test.js

@@ -0,0 +1,388 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { Turn, GeminiEventType } from './turn.js';
+import { reportError } from '../utils/errorReporting.js';
+const mockSendMessageStream = vi.fn();
+const mockGetHistory = vi.fn();
+const mockMaybeIncludeSchemaDepthContext = vi.fn();
+vi.mock('@google/genai', async (importOriginal) => {
+    const actual = await importOriginal();
+    const MockChat = vi.fn().mockImplementation(() => ({
+        sendMessageStream: mockSendMessageStream,
+        getHistory: mockGetHistory,
+        maybeIncludeSchemaDepthContext: mockMaybeIncludeSchemaDepthContext,
+    }));
+    return {
+        ...actual,
+        Chat: MockChat,
+    };
+});
+vi.mock('../utils/errorReporting', () => ({
+    reportError: vi.fn(),
+}));
+vi.mock('../utils/generateContentResponseUtilities', () => ({
+    getResponseText: (resp) => resp.candidates?.[0]?.content?.parts?.map((part) => part.text).join('') ||
+        undefined,
+}));
+describe('Turn', () => {
+    let turn;
+    let mockChatInstance;
+    beforeEach(() => {
+        vi.resetAllMocks();
+        mockChatInstance = {
+            sendMessageStream: mockSendMessageStream,
+            getHistory: mockGetHistory,
+            maybeIncludeSchemaDepthContext: mockMaybeIncludeSchemaDepthContext,
+        };
+        turn = new Turn(mockChatInstance, 'prompt-id-1');
+        mockGetHistory.mockReturnValue([]);
+        mockSendMessageStream.mockResolvedValue((async function* () { })());
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    describe('constructor', () => {
+        it('should initialize pendingToolCalls and debugResponses', () => {
+            expect(turn.pendingToolCalls).toEqual([]);
+            expect(turn.getDebugResponses()).toEqual([]);
+        });
+    });
+    describe('run', () => {
+        it('should yield content events for text parts', async () => {
+            const mockResponseStream = (async function* () {
+                yield {
+                    candidates: [{ content: { parts: [{ text: 'Hello' }] } }],
+                };
+                yield {
+                    candidates: [{ content: { parts: [{ text: ' world' }] } }],
+                };
+            })();
+            mockSendMessageStream.mockResolvedValue(mockResponseStream);
+            const events = [];
+            const reqParts = [{ text: 'Hi' }];
+            for await (const event of turn.run(reqParts, new AbortController().signal)) {
+                events.push(event);
+            }
+            expect(mockSendMessageStream).toHaveBeenCalledWith({
+                message: reqParts,
+                config: { abortSignal: expect.any(AbortSignal) },
+            }, 'prompt-id-1');
+            expect(events).toEqual([
+                { type: GeminiEventType.Content, value: 'Hello' },
+                { type: GeminiEventType.Content, value: ' world' },
+            ]);
+            expect(turn.getDebugResponses().length).toBe(2);
+        });
+        it('should yield tool_call_request events for function calls', async () => {
+            const mockResponseStream = (async function* () {
+                yield {
+                    functionCalls: [
+                        {
+                            id: 'fc1',
+                            name: 'tool1',
+                            args: { arg1: 'val1' },
+                            isClientInitiated: false,
+                        },
+                        { name: 'tool2', args: { arg2: 'val2' }, isClientInitiated: false }, // No ID
+                    ],
+                };
+            })();
+            mockSendMessageStream.mockResolvedValue(mockResponseStream);
+            const events = [];
+            const reqParts = [{ text: 'Use tools' }];
+            for await (const event of turn.run(reqParts, new AbortController().signal)) {
+                events.push(event);
+            }
+            expect(events.length).toBe(2);
+            const event1 = events[0];
+            expect(event1.type).toBe(GeminiEventType.ToolCallRequest);
+            expect(event1.value).toEqual(expect.objectContaining({
+                callId: 'fc1',
+                name: 'tool1',
+                args: { arg1: 'val1' },
+                isClientInitiated: false,
+            }));
+            expect(turn.pendingToolCalls[0]).toEqual(event1.value);
+            const event2 = events[1];
+            expect(event2.type).toBe(GeminiEventType.ToolCallRequest);
+            expect(event2.value).toEqual(expect.objectContaining({
+                name: 'tool2',
+                args: { arg2: 'val2' },
+                isClientInitiated: false,
+            }));
+            expect(event2.value.callId).toEqual(expect.stringMatching(/^tool2-\d{13}-\w{10,}$/));
+            expect(turn.pendingToolCalls[1]).toEqual(event2.value);
+            expect(turn.getDebugResponses().length).toBe(1);
+        });
+        it('should yield UserCancelled event if signal is aborted', async () => {
+            const abortController = new AbortController();
+            const mockResponseStream = (async function* () {
+                yield {
+                    candidates: [{ content: { parts: [{ text: 'First part' }] } }],
+                };
+                abortController.abort();
+                yield {
+                    candidates: [
+                        {
+                            content: {
+                                parts: [{ text: 'Second part - should not be processed' }],
+                            },
+                        },
+                    ],
+                };
+            })();
+            mockSendMessageStream.mockResolvedValue(mockResponseStream);
+            const events = [];
+            const reqParts = [{ text: 'Test abort' }];
+            for await (const event of turn.run(reqParts, abortController.signal)) {
+                events.push(event);
+            }
+            expect(events).toEqual([
+                { type: GeminiEventType.Content, value: 'First part' },
+                { type: GeminiEventType.UserCancelled },
+            ]);
+            expect(turn.getDebugResponses().length).toBe(1);
+        });
+        it('should yield Error event and report if sendMessageStream throws', async () => {
+            const error = new Error('API Error');
+            mockSendMessageStream.mockRejectedValue(error);
+            const reqParts = [{ text: 'Trigger error' }];
+            const historyContent = [
+                { role: 'model', parts: [{ text: 'Previous history' }] },
+            ];
+            mockGetHistory.mockReturnValue(historyContent);
+            mockMaybeIncludeSchemaDepthContext.mockResolvedValue(undefined);
+            const events = [];
+            for await (const event of turn.run(reqParts, new AbortController().signal)) {
+                events.push(event);
+            }
+            expect(events.length).toBe(1);
+            const errorEvent = events[0];
+            expect(errorEvent.type).toBe(GeminiEventType.Error);
+            expect(errorEvent.value).toEqual({
+                error: { message: 'API Error', status: undefined },
+            });
+            expect(turn.getDebugResponses().length).toBe(0);
+            expect(reportError).toHaveBeenCalledWith(error, 'Error when talking to Gemini API', [...historyContent, reqParts], 'Turn.run-sendMessageStream');
+        });
+        it('should handle function calls with undefined name or args', async () => {
+            const mockResponseStream = (async function* () {
+                yield {
+                    functionCalls: [
+                        { id: 'fc1', name: undefined, args: { arg1: 'val1' } },
+                        { id: 'fc2', name: 'tool2', args: undefined },
+                        { id: 'fc3', name: undefined, args: undefined },
+                    ],
+                };
+            })();
+            mockSendMessageStream.mockResolvedValue(mockResponseStream);
+            const events = [];
+            const reqParts = [{ text: 'Test undefined tool parts' }];
+            for await (const event of turn.run(reqParts, new AbortController().signal)) {
+                events.push(event);
+            }
+            expect(events.length).toBe(3);
+            const event1 = events[0];
+            expect(event1.type).toBe(GeminiEventType.ToolCallRequest);
+            expect(event1.value).toEqual(expect.objectContaining({
+                callId: 'fc1',
+                name: 'undefined_tool_name',
+                args: { arg1: 'val1' },
+                isClientInitiated: false,
+            }));
+            expect(turn.pendingToolCalls[0]).toEqual(event1.value);
+            const event2 = events[1];
+            expect(event2.type).toBe(GeminiEventType.ToolCallRequest);
+            expect(event2.value).toEqual(expect.objectContaining({
+                callId: 'fc2',
+                name: 'tool2',
+                args: {},
+                isClientInitiated: false,
+            }));
+            expect(turn.pendingToolCalls[1]).toEqual(event2.value);
+            const event3 = events[2];
+            expect(event3.type).toBe(GeminiEventType.ToolCallRequest);
+            expect(event3.value).toEqual(expect.objectContaining({
+                callId: 'fc3',
+                name: 'undefined_tool_name',
+                args: {},
+                isClientInitiated: false,
+            }));
+            expect(turn.pendingToolCalls[2]).toEqual(event3.value);
+            expect(turn.getDebugResponses().length).toBe(1);
+        });
+        it('should yield finished event when response has finish reason', async () => {
+            const mockResponseStream = (async function* () {
+                yield {
+                    candidates: [
+                        {
+                            content: { parts: [{ text: 'Partial response' }] },
+                            finishReason: 'STOP',
+                        },
+                    ],
+                };
+            })();
+            mockSendMessageStream.mockResolvedValue(mockResponseStream);
+            const events = [];
+            const reqParts = [{ text: 'Test finish reason' }];
+            for await (const event of turn.run(reqParts, new AbortController().signal)) {
+                events.push(event);
+            }
+            expect(events).toEqual([
+                { type: GeminiEventType.Content, value: 'Partial response' },
+                { type: GeminiEventType.Finished, value: 'STOP' },
+            ]);
+        });
+        it('should yield finished event for MAX_TOKENS finish reason', async () => {
+            const mockResponseStream = (async function* () {
+                yield {
+                    candidates: [
+                        {
+                            content: {
+                                parts: [
+                                    { text: 'This is a long response that was cut off...' },
+                                ],
+                            },
+                            finishReason: 'MAX_TOKENS',
+                        },
+                    ],
+                };
+            })();
+            mockSendMessageStream.mockResolvedValue(mockResponseStream);
+            const events = [];
+            const reqParts = [{ text: 'Generate long text' }];
+            for await (const event of turn.run(reqParts, new AbortController().signal)) {
+                events.push(event);
+            }
+            expect(events).toEqual([
+                {
+                    type: GeminiEventType.Content,
+                    value: 'This is a long response that was cut off...',
+                },
+                { type: GeminiEventType.Finished, value: 'MAX_TOKENS' },
+            ]);
+        });
+        it('should yield finished event for SAFETY finish reason', async () => {
+            const mockResponseStream = (async function* () {
+                yield {
+                    candidates: [
+                        {
+                            content: { parts: [{ text: 'Content blocked' }] },
+                            finishReason: 'SAFETY',
+                        },
+                    ],
+                };
+            })();
+            mockSendMessageStream.mockResolvedValue(mockResponseStream);
+            const events = [];
+            const reqParts = [{ text: 'Test safety' }];
+            for await (const event of turn.run(reqParts, new AbortController().signal)) {
+                events.push(event);
+            }
+            expect(events).toEqual([
+                { type: GeminiEventType.Content, value: 'Content blocked' },
+                { type: GeminiEventType.Finished, value: 'SAFETY' },
+            ]);
+        });
+        it('should not yield finished event when there is no finish reason', async () => {
+            const mockResponseStream = (async function* () {
+                yield {
+                    candidates: [
+                        {
+                            content: { parts: [{ text: 'Response without finish reason' }] },
+                            // No finishReason property
+                        },
+                    ],
+                };
+            })();
+            mockSendMessageStream.mockResolvedValue(mockResponseStream);
+            const events = [];
+            const reqParts = [{ text: 'Test no finish reason' }];
+            for await (const event of turn.run(reqParts, new AbortController().signal)) {
+                events.push(event);
+            }
+            expect(events).toEqual([
+                {
+                    type: GeminiEventType.Content,
+                    value: 'Response without finish reason',
+                },
+            ]);
+            // No Finished event should be emitted
+        });
+        it('should handle multiple responses with different finish reasons', async () => {
+            const mockResponseStream = (async function* () {
+                yield {
+                    candidates: [
+                        {
+                            content: { parts: [{ text: 'First part' }] },
+                            // No finish reason on first response
+                        },
+                    ],
+                };
+                yield {
+                    candidates: [
+                        {
+                            content: { parts: [{ text: 'Second part' }] },
+                            finishReason: 'OTHER',
+                        },
+                    ],
+                };
+            })();
+            mockSendMessageStream.mockResolvedValue(mockResponseStream);
+            const events = [];
+            const reqParts = [{ text: 'Test multiple responses' }];
+            for await (const event of turn.run(reqParts, new AbortController().signal)) {
+                events.push(event);
+            }
+            expect(events).toEqual([
+                { type: GeminiEventType.Content, value: 'First part' },
+                { type: GeminiEventType.Content, value: 'Second part' },
+                { type: GeminiEventType.Finished, value: 'OTHER' },
+            ]);
+        });
+        it('should not crash when cancelled request has malformed error', async () => {
+            const abortController = new AbortController();
+            const errorToThrow = {
+                response: {
+                    data: undefined, // Malformed error data
+                },
+            };
+            mockSendMessageStream.mockImplementation(async () => {
+                abortController.abort();
+                throw errorToThrow;
+            });
+            const events = [];
+            const reqParts = [{ text: 'Test malformed error handling' }];
+            for await (const event of turn.run(reqParts, abortController.signal)) {
+                events.push(event);
+            }
+            expect(events).toEqual([{ type: GeminiEventType.UserCancelled }]);
+            expect(reportError).not.toHaveBeenCalled();
+        });
+    });
+    describe('getDebugResponses', () => {
+        it('should return collected debug responses', async () => {
+            const resp1 = {
+                candidates: [{ content: { parts: [{ text: 'Debug 1' }] } }],
+            };
+            const resp2 = {
+                functionCalls: [{ name: 'debugTool' }],
+            };
+            const mockResponseStream = (async function* () {
+                yield resp1;
+                yield resp2;
+            })();
+            mockSendMessageStream.mockResolvedValue(mockResponseStream);
+            const reqParts = [{ text: 'Hi' }];
+            for await (const _ of turn.run(reqParts, new AbortController().signal)) {
+                // consume stream
+            }
+            expect(turn.getDebugResponses()).toEqual([resp1, resp2]);
+        });
+    });
+});
+//# sourceMappingURL=turn.test.js.map