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

package/dist/src/confirmation-bus/message-bus.test.js

@@ -0,0 +1,164 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { MessageBus } from './message-bus.js';
+import { PolicyEngine } from '../policy/policy-engine.js';
+import { PolicyDecision } from '../policy/types.js';
+import { MessageBusType, } from './types.js';
+describe('MessageBus', () => {
+    let messageBus;
+    let policyEngine;
+    beforeEach(() => {
+        policyEngine = new PolicyEngine();
+        messageBus = new MessageBus(policyEngine);
+    });
+    describe('publish', () => {
+        it('should emit error for invalid message', () => {
+            const errorHandler = vi.fn();
+            messageBus.on('error', errorHandler);
+            // @ts-expect-error - Testing invalid message
+            messageBus.publish({ invalid: 'message' });
+            expect(errorHandler).toHaveBeenCalledWith(expect.objectContaining({
+                message: expect.stringContaining('Invalid message structure'),
+            }));
+        });
+        it('should validate tool confirmation requests have correlationId', () => {
+            const errorHandler = vi.fn();
+            messageBus.on('error', errorHandler);
+            // @ts-expect-error - Testing missing correlationId
+            messageBus.publish({
+                type: MessageBusType.TOOL_CONFIRMATION_REQUEST,
+                toolCall: { name: 'test' },
+            });
+            expect(errorHandler).toHaveBeenCalled();
+        });
+        it('should emit confirmation response when policy allows', () => {
+            vi.spyOn(policyEngine, 'check').mockReturnValue(PolicyDecision.ALLOW);
+            const responseHandler = vi.fn();
+            messageBus.subscribe(MessageBusType.TOOL_CONFIRMATION_RESPONSE, responseHandler);
+            const request = {
+                type: MessageBusType.TOOL_CONFIRMATION_REQUEST,
+                toolCall: { name: 'test-tool', args: {} },
+                correlationId: '123',
+            };
+            messageBus.publish(request);
+            const expectedResponse = {
+                type: MessageBusType.TOOL_CONFIRMATION_RESPONSE,
+                correlationId: '123',
+                confirmed: true,
+            };
+            expect(responseHandler).toHaveBeenCalledWith(expectedResponse);
+        });
+        it('should emit rejection and response when policy denies', () => {
+            vi.spyOn(policyEngine, 'check').mockReturnValue(PolicyDecision.DENY);
+            const responseHandler = vi.fn();
+            const rejectionHandler = vi.fn();
+            messageBus.subscribe(MessageBusType.TOOL_CONFIRMATION_RESPONSE, responseHandler);
+            messageBus.subscribe(MessageBusType.TOOL_POLICY_REJECTION, rejectionHandler);
+            const request = {
+                type: MessageBusType.TOOL_CONFIRMATION_REQUEST,
+                toolCall: { name: 'test-tool', args: {} },
+                correlationId: '123',
+            };
+            messageBus.publish(request);
+            const expectedRejection = {
+                type: MessageBusType.TOOL_POLICY_REJECTION,
+                toolCall: { name: 'test-tool', args: {} },
+            };
+            expect(rejectionHandler).toHaveBeenCalledWith(expectedRejection);
+            const expectedResponse = {
+                type: MessageBusType.TOOL_CONFIRMATION_RESPONSE,
+                correlationId: '123',
+                confirmed: false,
+            };
+            expect(responseHandler).toHaveBeenCalledWith(expectedResponse);
+        });
+        it('should pass through to UI when policy says ASK_USER', () => {
+            vi.spyOn(policyEngine, 'check').mockReturnValue(PolicyDecision.ASK_USER);
+            const requestHandler = vi.fn();
+            messageBus.subscribe(MessageBusType.TOOL_CONFIRMATION_REQUEST, requestHandler);
+            const request = {
+                type: MessageBusType.TOOL_CONFIRMATION_REQUEST,
+                toolCall: { name: 'test-tool', args: {} },
+                correlationId: '123',
+            };
+            messageBus.publish(request);
+            expect(requestHandler).toHaveBeenCalledWith(request);
+        });
+        it('should emit other message types directly', () => {
+            const successHandler = vi.fn();
+            messageBus.subscribe(MessageBusType.TOOL_EXECUTION_SUCCESS, successHandler);
+            const message = {
+                type: MessageBusType.TOOL_EXECUTION_SUCCESS,
+                toolCall: { name: 'test-tool' },
+                result: 'success',
+            };
+            messageBus.publish(message);
+            expect(successHandler).toHaveBeenCalledWith(message);
+        });
+    });
+    describe('subscribe/unsubscribe', () => {
+        it('should allow subscribing to specific message types', () => {
+            const handler = vi.fn();
+            messageBus.subscribe(MessageBusType.TOOL_EXECUTION_SUCCESS, handler);
+            const message = {
+                type: MessageBusType.TOOL_EXECUTION_SUCCESS,
+                toolCall: { name: 'test' },
+                result: 'test',
+            };
+            messageBus.publish(message);
+            expect(handler).toHaveBeenCalledWith(message);
+        });
+        it('should allow unsubscribing from message types', () => {
+            const handler = vi.fn();
+            messageBus.subscribe(MessageBusType.TOOL_EXECUTION_SUCCESS, handler);
+            messageBus.unsubscribe(MessageBusType.TOOL_EXECUTION_SUCCESS, handler);
+            const message = {
+                type: MessageBusType.TOOL_EXECUTION_SUCCESS,
+                toolCall: { name: 'test' },
+                result: 'test',
+            };
+            messageBus.publish(message);
+            expect(handler).not.toHaveBeenCalled();
+        });
+        it('should support multiple subscribers for the same message type', () => {
+            const handler1 = vi.fn();
+            const handler2 = vi.fn();
+            messageBus.subscribe(MessageBusType.TOOL_EXECUTION_SUCCESS, handler1);
+            messageBus.subscribe(MessageBusType.TOOL_EXECUTION_SUCCESS, handler2);
+            const message = {
+                type: MessageBusType.TOOL_EXECUTION_SUCCESS,
+                toolCall: { name: 'test' },
+                result: 'test',
+            };
+            messageBus.publish(message);
+            expect(handler1).toHaveBeenCalledWith(message);
+            expect(handler2).toHaveBeenCalledWith(message);
+        });
+    });
+    describe('error handling', () => {
+        it('should not crash on errors during message processing', () => {
+            const errorHandler = vi.fn();
+            messageBus.on('error', errorHandler);
+            // Mock policyEngine to throw an error
+            vi.spyOn(policyEngine, 'check').mockImplementation(() => {
+                throw new Error('Policy check failed');
+            });
+            const request = {
+                type: MessageBusType.TOOL_CONFIRMATION_REQUEST,
+                toolCall: { name: 'test-tool' },
+                correlationId: '123',
+            };
+            // Should not throw
+            expect(() => messageBus.publish(request)).not.toThrow();
+            // Should emit error
+            expect(errorHandler).toHaveBeenCalledWith(expect.objectContaining({
+                message: 'Policy check failed',
+            }));
+        });
+    });
+});
+//# sourceMappingURL=message-bus.test.js.map

package/dist/src/confirmation-bus/message-bus.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"message-bus.test.js","sourceRoot":"","sources":["../../../src/confirmation-bus/message-bus.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EACL,cAAc,GAKf,MAAM,YAAY,CAAC;AAEpB,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,IAAI,UAAsB,CAAC;IAC3B,IAAI,YAA0B,CAAC;IAE/B,UAAU,CAAC,GAAG,EAAE;QACd,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC;QAClC,UAAU,GAAG,IAAI,UAAU,CAAC,YAAY,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7B,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;YAErC,6CAA6C;YAC7C,UAAU,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;YAE3C,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC,2BAA2B,CAAC;aAC9D,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7B,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;YAErC,mDAAmD;YACnD,UAAU,CAAC,OAAO,CAAC;gBACjB,IAAI,EAAE,cAAc,CAAC,yBAAyB;gBAC9C,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;aAC3B,CAAC,CAAC;YAEH,MAAM,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,EAAE,CAAC,KAAK,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YAEtE,MAAM,eAAe,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,UAAU,CAAC,SAAS,CAClB,cAAc,CAAC,0BAA0B,EACzC,eAAe,CAChB,CAAC;YAEF,MAAM,OAAO,GAA4B;gBACvC,IAAI,EAAE,cAAc,CAAC,yBAAyB;gBAC9C,QAAQ,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE;gBACzC,aAAa,EAAE,KAAK;aACrB,CAAC;YAEF,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAE5B,MAAM,gBAAgB,GAA6B;gBACjD,IAAI,EAAE,cAAc,CAAC,0BAA0B;gBAC/C,aAAa,EAAE,KAAK;gBACpB,SAAS,EAAE,IAAI;aAChB,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,EAAE,CAAC,KAAK,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;YAErE,MAAM,eAAe,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACjC,UAAU,CAAC,SAAS,CAClB,cAAc,CAAC,0BAA0B,EACzC,eAAe,CAChB,CAAC;YACF,UAAU,CAAC,SAAS,CAClB,cAAc,CAAC,qBAAqB,EACpC,gBAAgB,CACjB,CAAC;YAEF,MAAM,OAAO,GAA4B;gBACvC,IAAI,EAAE,cAAc,CAAC,yBAAyB;gBAC9C,QAAQ,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE;gBACzC,aAAa,EAAE,KAAK;aACrB,CAAC;YAEF,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAE5B,MAAM,iBAAiB,GAAwB;gBAC7C,IAAI,EAAE,cAAc,CAAC,qBAAqB;gBAC1C,QAAQ,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE;aAC1C,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;YAEjE,MAAM,gBAAgB,GAA6B;gBACjD,IAAI,EAAE,cAAc,CAAC,0BAA0B;gBAC/C,aAAa,EAAE,KAAK;gBACpB,SAAS,EAAE,KAAK;aACjB,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,EAAE,CAAC,KAAK,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;YAEzE,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/B,UAAU,CAAC,SAAS,CAClB,cAAc,CAAC,yBAAyB,EACxC,cAAc,CACf,CAAC;YAEF,MAAM,OAAO,GAA4B;gBACvC,IAAI,EAAE,cAAc,CAAC,yBAAyB;gBAC9C,QAAQ,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE;gBACzC,aAAa,EAAE,KAAK;aACrB,CAAC;YAEF,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAE5B,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/B,UAAU,CAAC,SAAS,CAClB,cAAc,CAAC,sBAAsB,EACrC,cAAc,CACf,CAAC;YAEF,MAAM,OAAO,GAAiC;gBAC5C,IAAI,EAAE,cAAc,CAAC,sBAA+B;gBACpD,QAAQ,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;gBAC/B,MAAM,EAAE,SAAS;aAClB,CAAC;YAEF,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAE5B,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACxB,UAAU,CAAC,SAAS,CAAC,cAAc,CAAC,sBAAsB,EAAE,OAAO,CAAC,CAAC;YAErE,MAAM,OAAO,GAAiC;gBAC5C,IAAI,EAAE,cAAc,CAAC,sBAA+B;gBACpD,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;gBAC1B,MAAM,EAAE,MAAM;aACf,CAAC;YAEF,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAE5B,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACxB,UAAU,CAAC,SAAS,CAAC,cAAc,CAAC,sBAAsB,EAAE,OAAO,CAAC,CAAC;YACrE,UAAU,CAAC,WAAW,CAAC,cAAc,CAAC,sBAAsB,EAAE,OAAO,CAAC,CAAC;YAEvE,MAAM,OAAO,GAAiC;gBAC5C,IAAI,EAAE,cAAc,CAAC,sBAA+B;gBACpD,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;gBAC1B,MAAM,EAAE,MAAM;aACf,CAAC;YAEF,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAE5B,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAEzB,UAAU,CAAC,SAAS,CAAC,cAAc,CAAC,sBAAsB,EAAE,QAAQ,CAAC,CAAC;YACtE,UAAU,CAAC,SAAS,CAAC,cAAc,CAAC,sBAAsB,EAAE,QAAQ,CAAC,CAAC;YAEtE,MAAM,OAAO,GAAiC;gBAC5C,IAAI,EAAE,cAAc,CAAC,sBAA+B;gBACpD,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;gBAC1B,MAAM,EAAE,MAAM;aACf,CAAC;YAEF,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAE5B,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;YAC/C,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7B,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;YAErC,sCAAsC;YACtC,EAAE,CAAC,KAAK,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBACtD,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACzC,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAA4B;gBACvC,IAAI,EAAE,cAAc,CAAC,yBAAyB;gBAC9C,QAAQ,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;gBAC/B,aAAa,EAAE,KAAK;aACrB,CAAC;YAEF,mBAAmB;YACnB,MAAM,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;YAExD,oBAAoB;YACpB,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,qBAAqB;aAC/B,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/confirmation-bus/types.d.ts

@@ -0,0 +1,38 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type FunctionCall } from '@google/genai';
+export declare enum MessageBusType {
+    TOOL_CONFIRMATION_REQUEST = "tool-confirmation-request",
+    TOOL_CONFIRMATION_RESPONSE = "tool-confirmation-response",
+    TOOL_POLICY_REJECTION = "tool-policy-rejection",
+    TOOL_EXECUTION_SUCCESS = "tool-execution-success",
+    TOOL_EXECUTION_FAILURE = "tool-execution-failure"
+}
+export interface ToolConfirmationRequest {
+    type: MessageBusType.TOOL_CONFIRMATION_REQUEST;
+    toolCall: FunctionCall;
+    correlationId: string;
+}
+export interface ToolConfirmationResponse {
+    type: MessageBusType.TOOL_CONFIRMATION_RESPONSE;
+    correlationId: string;
+    confirmed: boolean;
+}
+export interface ToolPolicyRejection {
+    type: MessageBusType.TOOL_POLICY_REJECTION;
+    toolCall: FunctionCall;
+}
+export interface ToolExecutionSuccess<T = unknown> {
+    type: MessageBusType.TOOL_EXECUTION_SUCCESS;
+    toolCall: FunctionCall;
+    result: T;
+}
+export interface ToolExecutionFailure<E = Error> {
+    type: MessageBusType.TOOL_EXECUTION_FAILURE;
+    toolCall: FunctionCall;
+    error: E;
+}
+export type Message = ToolConfirmationRequest | ToolConfirmationResponse | ToolPolicyRejection | ToolExecutionSuccess | ToolExecutionFailure;

package/dist/src/confirmation-bus/types.js

@@ -0,0 +1,15 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import {} from '@google/genai';
+export var MessageBusType;
+(function (MessageBusType) {
+    MessageBusType["TOOL_CONFIRMATION_REQUEST"] = "tool-confirmation-request";
+    MessageBusType["TOOL_CONFIRMATION_RESPONSE"] = "tool-confirmation-response";
+    MessageBusType["TOOL_POLICY_REJECTION"] = "tool-policy-rejection";
+    MessageBusType["TOOL_EXECUTION_SUCCESS"] = "tool-execution-success";
+    MessageBusType["TOOL_EXECUTION_FAILURE"] = "tool-execution-failure";
+})(MessageBusType || (MessageBusType = {}));
+//# sourceMappingURL=types.js.map

package/dist/src/confirmation-bus/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/confirmation-bus/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAqB,MAAM,eAAe,CAAC;AAElD,MAAM,CAAN,IAAY,cAMX;AAND,WAAY,cAAc;IACxB,yEAAuD,CAAA;IACvD,2EAAyD,CAAA;IACzD,iEAA+C,CAAA;IAC/C,mEAAiD,CAAA;IACjD,mEAAiD,CAAA;AACnD,CAAC,EANW,cAAc,KAAd,cAAc,QAMzB"}

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

@@ -0,0 +1,46 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Content, GenerateContentConfig, Part } from '@google/genai';
+import type { Config } from '../config/config.js';
+import type { ContentGenerator } from './contentGenerator.js';
+/**
+ * Options for the generateJson utility function.
+ */
+export interface GenerateJsonOptions {
+    /** The input prompt or history. */
+    contents: Content[];
+    /** The required JSON schema for the output. */
+    schema: Record<string, unknown>;
+    /** The specific model to use for this task. */
+    model: string;
+    /**
+     * Task-specific system instructions.
+     * If omitted, no system instruction is sent.
+     */
+    systemInstruction?: string | Part | Part[] | Content;
+    /**
+     * Overrides for generation configuration (e.g., temperature).
+     */
+    config?: Omit<GenerateContentConfig, 'systemInstruction' | 'responseJsonSchema' | 'responseMimeType' | 'tools' | 'abortSignal'>;
+    /** Signal for cancellation. */
+    abortSignal: AbortSignal;
+    /**
+     * A unique ID for the prompt, used for logging/telemetry correlation.
+     */
+    promptId: string;
+}
+/**
+ * A client dedicated to stateless, utility-focused LLM calls.
+ */
+export declare class BaseLlmClient {
+    private readonly contentGenerator;
+    private readonly config;
+    private readonly defaultUtilityConfig;
+    constructor(contentGenerator: ContentGenerator, config: Config);
+    generateJson(options: GenerateJsonOptions): Promise<Record<string, unknown>>;
+    generateEmbedding(texts: string[]): Promise<number[][]>;
+    private cleanJsonResponse;
+}

package/dist/src/core/baseLlmClient.js

@@ -0,0 +1,112 @@
+/**
+ * @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 } from '../utils/errors.js';
+import { logMalformedJsonResponse } from '../telemetry/loggers.js';
+import { MalformedJsonResponseEvent } from '../telemetry/types.js';
+import { retryWithBackoff } from '../utils/retry.js';
+/**
+ * A client dedicated to stateless, utility-focused LLM calls.
+ */
+export class BaseLlmClient {
+    contentGenerator;
+    config;
+    // Default configuration for utility tasks
+    defaultUtilityConfig = {
+        temperature: 0,
+        topP: 1,
+    };
+    constructor(contentGenerator, config) {
+        this.contentGenerator = contentGenerator;
+        this.config = config;
+    }
+    async generateJson(options) {
+        const { contents, schema, model, abortSignal, systemInstruction, promptId, } = options;
+        const requestConfig = {
+            abortSignal,
+            ...this.defaultUtilityConfig,
+            ...options.config,
+            ...(systemInstruction && { systemInstruction }),
+            responseJsonSchema: schema,
+            responseMimeType: 'application/json',
+        };
+        try {
+            const apiCall = () => this.contentGenerator.generateContent({
+                model,
+                config: requestConfig,
+                contents,
+            }, promptId);
+            const result = await retryWithBackoff(apiCall);
+            let text = getResponseText(result)?.trim();
+            if (!text) {
+                const error = new Error('API returned an empty response for generateJson.');
+                await reportError(error, 'Error in generateJson: API returned an empty response.', contents, 'generateJson-empty-response');
+                throw error;
+            }
+            text = this.cleanJsonResponse(text, model);
+            try {
+                return JSON.parse(text);
+            }
+            catch (parseError) {
+                const error = new Error(`Failed to parse API response as JSON: ${getErrorMessage(parseError)}`);
+                await reportError(parseError, 'Failed to parse JSON response from generateJson.', {
+                    responseTextFailedToParse: text,
+                    originalRequestContents: contents,
+                }, 'generateJson-parse');
+                throw error;
+            }
+        }
+        catch (error) {
+            if (abortSignal.aborted) {
+                throw error;
+            }
+            if (error instanceof Error &&
+                (error.message === 'API returned an empty response for generateJson.' ||
+                    error.message.startsWith('Failed to parse API response as JSON:'))) {
+                // We perform this check so that we don't report these again.
+            }
+            else {
+                await reportError(error, 'Error generating JSON content via API.', contents, 'generateJson-api');
+            }
+            throw new Error(`Failed to generate JSON content: ${getErrorMessage(error)}`);
+        }
+    }
+    async generateEmbedding(texts) {
+        if (!texts || texts.length === 0) {
+            return [];
+        }
+        const embedModelParams = {
+            model: this.config.getEmbeddingModel(),
+            contents: texts,
+        };
+        const embedContentResponse = await this.contentGenerator.embedContent(embedModelParams);
+        if (!embedContentResponse.embeddings ||
+            embedContentResponse.embeddings.length === 0) {
+            throw new Error('No embeddings found in API response.');
+        }
+        if (embedContentResponse.embeddings.length !== texts.length) {
+            throw new Error(`API returned a mismatched number of embeddings. Expected ${texts.length}, got ${embedContentResponse.embeddings.length}.`);
+        }
+        return embedContentResponse.embeddings.map((embedding, index) => {
+            const values = embedding.values;
+            if (!values || values.length === 0) {
+                throw new Error(`API returned an empty embedding for input text at index ${index}: "${texts[index]}"`);
+            }
+            return values;
+        });
+    }
+    cleanJsonResponse(text, model) {
+        const prefix = '```json';
+        const suffix = '```';
+        if (text.startsWith(prefix) && text.endsWith(suffix)) {
+            logMalformedJsonResponse(this.config, new MalformedJsonResponseEvent(model));
+            return text.substring(prefix.length, text.length - suffix.length).trim();
+        }
+        return text;
+    }
+}
+//# sourceMappingURL=baseLlmClient.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"baseLlmClient.js","sourceRoot":"","sources":["../../../src/core/baseLlmClient.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAUH,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,0BAA0B,EAAE,MAAM,uBAAuB,CAAC;AACnE,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAoCrD;;GAEG;AACH,MAAM,OAAO,aAAa;IAQL;IACA;IARnB,0CAA0C;IACzB,oBAAoB,GAA0B;QAC7D,WAAW,EAAE,CAAC;QACd,IAAI,EAAE,CAAC;KACR,CAAC;IAEF,YACmB,gBAAkC,EAClC,MAAc;QADd,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,WAAM,GAAN,MAAM,CAAQ;IAC9B,CAAC;IAEJ,KAAK,CAAC,YAAY,CAChB,OAA4B;QAE5B,MAAM,EACJ,QAAQ,EACR,MAAM,EACN,KAAK,EACL,WAAW,EACX,iBAAiB,EACjB,QAAQ,GACT,GAAG,OAAO,CAAC;QAEZ,MAAM,aAAa,GAA0B;YAC3C,WAAW;YACX,GAAG,IAAI,CAAC,oBAAoB;YAC5B,GAAG,OAAO,CAAC,MAAM;YACjB,GAAG,CAAC,iBAAiB,IAAI,EAAE,iBAAiB,EAAE,CAAC;YAC/C,kBAAkB,EAAE,MAAM;YAC1B,gBAAgB,EAAE,kBAAkB;SACrC,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,GAAG,EAAE,CACnB,IAAI,CAAC,gBAAgB,CAAC,eAAe,CACnC;gBACE,KAAK;gBACL,MAAM,EAAE,aAAa;gBACrB,QAAQ;aACT,EACD,QAAQ,CACT,CAAC;YAEJ,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,OAAO,CAAC,CAAC;YAE/C,IAAI,IAAI,GAAG,eAAe,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC;YAC3C,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,MAAM,KAAK,GAAG,IAAI,KAAK,CACrB,kDAAkD,CACnD,CAAC;gBACF,MAAM,WAAW,CACf,KAAK,EACL,wDAAwD,EACxD,QAAQ,EACR,6BAA6B,CAC9B,CAAC;gBACF,MAAM,KAAK,CAAC;YACd,CAAC;YAED,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAE3C,IAAI,CAAC;gBACH,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC1B,CAAC;YAAC,OAAO,UAAU,EAAE,CAAC;gBACpB,MAAM,KAAK,GAAG,IAAI,KAAK,CACrB,yCAAyC,eAAe,CAAC,UAAU,CAAC,EAAE,CACvE,CAAC;gBACF,MAAM,WAAW,CACf,UAAU,EACV,kDAAkD,EAClD;oBACE,yBAAyB,EAAE,IAAI;oBAC/B,uBAAuB,EAAE,QAAQ;iBAClC,EACD,oBAAoB,CACrB,CAAC;gBACF,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;gBACxB,MAAM,KAAK,CAAC;YACd,CAAC;YAED,IACE,KAAK,YAAY,KAAK;gBACtB,CAAC,KAAK,CAAC,OAAO,KAAK,kDAAkD;oBACnE,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,uCAAuC,CAAC,CAAC,EACpE,CAAC;gBACD,6DAA6D;YAC/D,CAAC;iBAAM,CAAC;gBACN,MAAM,WAAW,CACf,KAAK,EACL,wCAAwC,EACxC,QAAQ,EACR,kBAAkB,CACnB,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,KAAK,CACb,oCAAoC,eAAe,CAAC,KAAK,CAAC,EAAE,CAC7D,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,KAAe;QACrC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,gBAAgB,GAA2B;YAC/C,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE;YACtC,QAAQ,EAAE,KAAK;SAChB,CAAC;QAEF,MAAM,oBAAoB,GACxB,MAAM,IAAI,CAAC,gBAAgB,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;QAC7D,IACE,CAAC,oBAAoB,CAAC,UAAU;YAChC,oBAAoB,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAC5C,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1D,CAAC;QAED,IAAI,oBAAoB,CAAC,UAAU,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,EAAE,CAAC;YAC5D,MAAM,IAAI,KAAK,CACb,4DAA4D,KAAK,CAAC,MAAM,SAAS,oBAAoB,CAAC,UAAU,CAAC,MAAM,GAAG,CAC3H,CAAC;QACJ,CAAC;QAED,OAAO,oBAAoB,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE;YAC9D,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC;YAChC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACnC,MAAM,IAAI,KAAK,CACb,2DAA2D,KAAK,MAAM,KAAK,CAAC,KAAK,CAAC,GAAG,CACtF,CAAC;YACJ,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,iBAAiB,CAAC,IAAY,EAAE,KAAa;QACnD,MAAM,MAAM,GAAG,SAAS,CAAC;QACzB,MAAM,MAAM,GAAG,KAAK,CAAC;QACrB,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YACrD,wBAAwB,CACtB,IAAI,CAAC,MAAM,EACX,IAAI,0BAA0B,CAAC,KAAK,CAAC,CACtC,CAAC;YACF,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;QAC3E,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}

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

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

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

@@ -0,0 +1,253 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach, } from 'vitest';
+import { BaseLlmClient } from './baseLlmClient.js';
+import { AuthType } from './contentGenerator.js';
+import { reportError } from '../utils/errorReporting.js';
+import { logMalformedJsonResponse } from '../telemetry/loggers.js';
+import { retryWithBackoff } from '../utils/retry.js';
+import { MalformedJsonResponseEvent } from '../telemetry/types.js';
+import { getErrorMessage } from '../utils/errors.js';
+vi.mock('../utils/errorReporting.js');
+vi.mock('../telemetry/loggers.js');
+vi.mock('../utils/errors.js', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        getErrorMessage: vi.fn((e) => (e instanceof Error ? e.message : String(e))),
+    };
+});
+vi.mock('../utils/retry.js', () => ({
+    retryWithBackoff: vi.fn(async (fn) => await fn()),
+}));
+const mockGenerateContent = vi.fn();
+const mockEmbedContent = vi.fn();
+const mockContentGenerator = {
+    generateContent: mockGenerateContent,
+    embedContent: mockEmbedContent,
+};
+const mockConfig = {
+    getSessionId: vi.fn().mockReturnValue('test-session-id'),
+    getContentGeneratorConfig: vi
+        .fn()
+        .mockReturnValue({ authType: AuthType.USE_GEMINI }),
+    getEmbeddingModel: vi.fn().mockReturnValue('test-embedding-model'),
+};
+// Helper to create a mock GenerateContentResponse
+const createMockResponse = (text) => ({
+    candidates: [{ content: { role: 'model', parts: [{ text }] }, index: 0 }],
+});
+describe('BaseLlmClient', () => {
+    let client;
+    let abortController;
+    let defaultOptions;
+    beforeEach(() => {
+        vi.clearAllMocks();
+        // Reset the mocked implementation for getErrorMessage for accurate error message assertions
+        vi.mocked(getErrorMessage).mockImplementation((e) => e instanceof Error ? e.message : String(e));
+        client = new BaseLlmClient(mockContentGenerator, mockConfig);
+        abortController = new AbortController();
+        defaultOptions = {
+            contents: [{ role: 'user', parts: [{ text: 'Give me a color.' }] }],
+            schema: { type: 'object', properties: { color: { type: 'string' } } },
+            model: 'test-model',
+            abortSignal: abortController.signal,
+            promptId: 'test-prompt-id',
+        };
+    });
+    afterEach(() => {
+        abortController.abort();
+    });
+    describe('generateJson - Success Scenarios', () => {
+        it('should call generateContent with correct parameters, defaults, and utilize retry mechanism', async () => {
+            const mockResponse = createMockResponse('{"color": "blue"}');
+            mockGenerateContent.mockResolvedValue(mockResponse);
+            const result = await client.generateJson(defaultOptions);
+            expect(result).toEqual({ color: 'blue' });
+            // Ensure the retry mechanism was engaged
+            expect(retryWithBackoff).toHaveBeenCalledTimes(1);
+            // Validate the parameters passed to the underlying generator
+            expect(mockGenerateContent).toHaveBeenCalledTimes(1);
+            expect(mockGenerateContent).toHaveBeenCalledWith({
+                model: 'test-model',
+                contents: defaultOptions.contents,
+                config: {
+                    abortSignal: defaultOptions.abortSignal,
+                    temperature: 0,
+                    topP: 1,
+                    responseJsonSchema: defaultOptions.schema,
+                    responseMimeType: 'application/json',
+                    // Crucial: systemInstruction should NOT be in the config object if not provided
+                },
+            }, 'test-prompt-id');
+        });
+        it('should respect configuration overrides', async () => {
+            const mockResponse = createMockResponse('{"color": "red"}');
+            mockGenerateContent.mockResolvedValue(mockResponse);
+            const options = {
+                ...defaultOptions,
+                config: { temperature: 0.8, topK: 10 },
+            };
+            await client.generateJson(options);
+            expect(mockGenerateContent).toHaveBeenCalledWith(expect.objectContaining({
+                config: expect.objectContaining({
+                    temperature: 0.8,
+                    topP: 1, // Default should remain if not overridden
+                    topK: 10,
+                }),
+            }), expect.any(String));
+        });
+        it('should include system instructions when provided', async () => {
+            const mockResponse = createMockResponse('{"color": "green"}');
+            mockGenerateContent.mockResolvedValue(mockResponse);
+            const systemInstruction = 'You are a helpful assistant.';
+            const options = {
+                ...defaultOptions,
+                systemInstruction,
+            };
+            await client.generateJson(options);
+            expect(mockGenerateContent).toHaveBeenCalledWith(expect.objectContaining({
+                config: expect.objectContaining({
+                    systemInstruction,
+                }),
+            }), expect.any(String));
+        });
+        it('should use the provided promptId', async () => {
+            const mockResponse = createMockResponse('{"color": "yellow"}');
+            mockGenerateContent.mockResolvedValue(mockResponse);
+            const customPromptId = 'custom-id-123';
+            const options = {
+                ...defaultOptions,
+                promptId: customPromptId,
+            };
+            await client.generateJson(options);
+            expect(mockGenerateContent).toHaveBeenCalledWith(expect.any(Object), customPromptId);
+        });
+    });
+    describe('generateJson - Response Cleaning', () => {
+        it('should clean JSON wrapped in markdown backticks and log telemetry', async () => {
+            const malformedResponse = '```json\n{"color": "purple"}\n```';
+            mockGenerateContent.mockResolvedValue(createMockResponse(malformedResponse));
+            const result = await client.generateJson(defaultOptions);
+            expect(result).toEqual({ color: 'purple' });
+            expect(logMalformedJsonResponse).toHaveBeenCalledTimes(1);
+            expect(logMalformedJsonResponse).toHaveBeenCalledWith(mockConfig, expect.any(MalformedJsonResponseEvent));
+            // Validate the telemetry event content
+            const event = vi.mocked(logMalformedJsonResponse).mock
+                .calls[0][1];
+            expect(event.model).toBe('test-model');
+        });
+        it('should handle extra whitespace correctly without logging malformed telemetry', async () => {
+            const responseWithWhitespace = '  \n  {"color": "orange"}  \n';
+            mockGenerateContent.mockResolvedValue(createMockResponse(responseWithWhitespace));
+            const result = await client.generateJson(defaultOptions);
+            expect(result).toEqual({ color: 'orange' });
+            expect(logMalformedJsonResponse).not.toHaveBeenCalled();
+        });
+    });
+    describe('generateJson - Error Handling', () => {
+        it('should throw and report error for empty response', async () => {
+            mockGenerateContent.mockResolvedValue(createMockResponse(''));
+            // The final error message includes the prefix added by the client's outer catch block.
+            await expect(client.generateJson(defaultOptions)).rejects.toThrow('Failed to generate JSON content: API returned an empty response for generateJson.');
+            // Verify error reporting details
+            expect(reportError).toHaveBeenCalledTimes(1);
+            expect(reportError).toHaveBeenCalledWith(expect.any(Error), 'Error in generateJson: API returned an empty response.', defaultOptions.contents, 'generateJson-empty-response');
+        });
+        it('should throw and report error for invalid JSON syntax', async () => {
+            const invalidJson = '{"color": "blue"'; // missing closing brace
+            mockGenerateContent.mockResolvedValue(createMockResponse(invalidJson));
+            await expect(client.generateJson(defaultOptions)).rejects.toThrow(/^Failed to generate JSON content: Failed to parse API response as JSON:/);
+            expect(reportError).toHaveBeenCalledTimes(1);
+            expect(reportError).toHaveBeenCalledWith(expect.any(Error), 'Failed to parse JSON response from generateJson.', expect.objectContaining({ responseTextFailedToParse: invalidJson }), 'generateJson-parse');
+        });
+        it('should throw and report generic API errors', async () => {
+            const apiError = new Error('Service Unavailable (503)');
+            // Simulate the generator failing
+            mockGenerateContent.mockRejectedValue(apiError);
+            await expect(client.generateJson(defaultOptions)).rejects.toThrow('Failed to generate JSON content: Service Unavailable (503)');
+            // Verify generic error reporting
+            expect(reportError).toHaveBeenCalledTimes(1);
+            expect(reportError).toHaveBeenCalledWith(apiError, 'Error generating JSON content via API.', defaultOptions.contents, 'generateJson-api');
+        });
+        it('should throw immediately without reporting if aborted', async () => {
+            const abortError = new DOMException('Aborted', 'AbortError');
+            // Simulate abortion happening during the API call
+            mockGenerateContent.mockImplementation(() => {
+                abortController.abort(); // Ensure the signal is aborted when the service checks
+                throw abortError;
+            });
+            const options = {
+                ...defaultOptions,
+                abortSignal: abortController.signal,
+            };
+            await expect(client.generateJson(options)).rejects.toThrow(abortError);
+            // Crucially, it should not report a cancellation as an application error
+            expect(reportError).not.toHaveBeenCalled();
+        });
+    });
+    describe('generateEmbedding', () => {
+        const texts = ['hello world', 'goodbye world'];
+        const testEmbeddingModel = 'test-embedding-model';
+        it('should call embedContent with correct parameters and return embeddings', async () => {
+            const mockEmbeddings = [
+                [0.1, 0.2, 0.3],
+                [0.4, 0.5, 0.6],
+            ];
+            mockEmbedContent.mockResolvedValue({
+                embeddings: [
+                    { values: mockEmbeddings[0] },
+                    { values: mockEmbeddings[1] },
+                ],
+            });
+            const result = await client.generateEmbedding(texts);
+            expect(mockEmbedContent).toHaveBeenCalledTimes(1);
+            expect(mockEmbedContent).toHaveBeenCalledWith({
+                model: testEmbeddingModel,
+                contents: texts,
+            });
+            expect(result).toEqual(mockEmbeddings);
+        });
+        it('should return an empty array if an empty array is passed', async () => {
+            const result = await client.generateEmbedding([]);
+            expect(result).toEqual([]);
+            expect(mockEmbedContent).not.toHaveBeenCalled();
+        });
+        it('should throw an error if API response has no embeddings array', async () => {
+            mockEmbedContent.mockResolvedValue({});
+            await expect(client.generateEmbedding(texts)).rejects.toThrow('No embeddings found in API response.');
+        });
+        it('should throw an error if API response has an empty embeddings array', async () => {
+            mockEmbedContent.mockResolvedValue({
+                embeddings: [],
+            });
+            await expect(client.generateEmbedding(texts)).rejects.toThrow('No embeddings found in API response.');
+        });
+        it('should throw an error if API returns a mismatched number of embeddings', async () => {
+            mockEmbedContent.mockResolvedValue({
+                embeddings: [{ values: [1, 2, 3] }], // Only one for two texts
+            });
+            await expect(client.generateEmbedding(texts)).rejects.toThrow('API returned a mismatched number of embeddings. Expected 2, got 1.');
+        });
+        it('should throw an error if any embedding has nullish values', async () => {
+            mockEmbedContent.mockResolvedValue({
+                embeddings: [{ values: [1, 2, 3] }, { values: undefined }], // Second one is bad
+            });
+            await expect(client.generateEmbedding(texts)).rejects.toThrow('API returned an empty embedding for input text at index 1: "goodbye world"');
+        });
+        it('should throw an error if any embedding has an empty values array', async () => {
+            mockEmbedContent.mockResolvedValue({
+                embeddings: [{ values: [] }, { values: [1, 2, 3] }], // First one is bad
+            });
+            await expect(client.generateEmbedding(texts)).rejects.toThrow('API returned an empty embedding for input text at index 0: "hello world"');
+        });
+        it('should propagate errors from the API call', async () => {
+            mockEmbedContent.mockRejectedValue(new Error('API Failure'));
+            await expect(client.generateEmbedding(texts)).rejects.toThrow('API Failure');
+        });
+    });
+});
+//# sourceMappingURL=baseLlmClient.test.js.map