@office-ai/aioncli-core 0.2.3 → 0.18.4

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

@@ -0,0 +1,170 @@
+/**
+ * @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', async () => {
+            const errorHandler = vi.fn();
+            messageBus.on('error', errorHandler);
+            // @ts-expect-error - Testing invalid message
+            await messageBus.publish({ invalid: 'message' });
+            expect(errorHandler).toHaveBeenCalledWith(expect.objectContaining({
+                message: expect.stringContaining('Invalid message structure'),
+            }));
+        });
+        it('should validate tool confirmation requests have correlationId', async () => {
+            const errorHandler = vi.fn();
+            messageBus.on('error', errorHandler);
+            // @ts-expect-error - Testing missing correlationId
+            await messageBus.publish({
+                type: MessageBusType.TOOL_CONFIRMATION_REQUEST,
+                toolCall: { name: 'test' },
+            });
+            expect(errorHandler).toHaveBeenCalled();
+        });
+        it('should emit confirmation response when policy allows', async () => {
+            vi.spyOn(policyEngine, 'check').mockResolvedValue({
+                decision: 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',
+            };
+            await 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', async () => {
+            vi.spyOn(policyEngine, 'check').mockResolvedValue({
+                decision: 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',
+            };
+            await 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', async () => {
+            vi.spyOn(policyEngine, 'check').mockResolvedValue({
+                decision: 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',
+            };
+            await messageBus.publish(request);
+            expect(requestHandler).toHaveBeenCalledWith(request);
+        });
+        it('should emit other message types directly', async () => {
+            const successHandler = vi.fn();
+            messageBus.subscribe(MessageBusType.TOOL_EXECUTION_SUCCESS, successHandler);
+            const message = {
+                type: MessageBusType.TOOL_EXECUTION_SUCCESS,
+                toolCall: { name: 'test-tool' },
+                result: 'success',
+            };
+            await messageBus.publish(message);
+            expect(successHandler).toHaveBeenCalledWith(message);
+        });
+    });
+    describe('subscribe/unsubscribe', () => {
+        it('should allow subscribing to specific message types', async () => {
+            const handler = vi.fn();
+            messageBus.subscribe(MessageBusType.TOOL_EXECUTION_SUCCESS, handler);
+            const message = {
+                type: MessageBusType.TOOL_EXECUTION_SUCCESS,
+                toolCall: { name: 'test' },
+                result: 'test',
+            };
+            await messageBus.publish(message);
+            expect(handler).toHaveBeenCalledWith(message);
+        });
+        it('should allow unsubscribing from message types', async () => {
+            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',
+            };
+            await messageBus.publish(message);
+            expect(handler).not.toHaveBeenCalled();
+        });
+        it('should support multiple subscribers for the same message type', async () => {
+            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',
+            };
+            await messageBus.publish(message);
+            expect(handler1).toHaveBeenCalledWith(message);
+            expect(handler2).toHaveBeenCalledWith(message);
+        });
+    });
+    describe('error handling', () => {
+        it('should not crash on errors during message processing', async () => {
+            const errorHandler = vi.fn();
+            messageBus.on('error', errorHandler);
+            // Mock policyEngine to throw an error
+            vi.spyOn(policyEngine, 'check').mockImplementation(async () => {
+                throw new Error('Policy check failed');
+            });
+            const request = {
+                type: MessageBusType.TOOL_CONFIRMATION_REQUEST,
+                toolCall: { name: 'test-tool' },
+                correlationId: '123',
+            };
+            // Should not throw
+            await expect(messageBus.publish(request)).resolves.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,KAAK,IAAI,EAAE;YACrD,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7B,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;YAErC,6CAA6C;YAC7C,MAAM,UAAU,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;YAEjD,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,KAAK,IAAI,EAAE;YAC7E,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7B,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;YAErC,mDAAmD;YACnD,MAAM,UAAU,CAAC,OAAO,CAAC;gBACvB,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,KAAK,IAAI,EAAE;YACpE,EAAE,CAAC,KAAK,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAAC;gBAChD,QAAQ,EAAE,cAAc,CAAC,KAAK;aAC/B,CAAC,CAAC;YAEH,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,MAAM,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAElC,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,KAAK,IAAI,EAAE;YACrE,EAAE,CAAC,KAAK,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAAC;gBAChD,QAAQ,EAAE,cAAc,CAAC,IAAI;aAC9B,CAAC,CAAC;YAEH,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,MAAM,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAElC,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,KAAK,IAAI,EAAE;YACnE,EAAE,CAAC,KAAK,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAAC;gBAChD,QAAQ,EAAE,cAAc,CAAC,QAAQ;aAClC,CAAC,CAAC;YAEH,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,MAAM,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAElC,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,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,MAAM,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAElC,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,KAAK,IAAI,EAAE;YAClE,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,MAAM,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAElC,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,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,MAAM,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAElC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,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,MAAM,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAElC,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,KAAK,IAAI,EAAE;YACpE,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,KAAK,IAAI,EAAE;gBAC5D,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,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;YAEjE,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,49 @@
+/**
+ * @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",
+    UPDATE_POLICY = "update-policy"
+}
+export interface ToolConfirmationRequest {
+    type: MessageBusType.TOOL_CONFIRMATION_REQUEST;
+    toolCall: FunctionCall;
+    correlationId: string;
+    serverName?: string;
+}
+export interface ToolConfirmationResponse {
+    type: MessageBusType.TOOL_CONFIRMATION_RESPONSE;
+    correlationId: string;
+    confirmed: boolean;
+    /**
+     * When true, indicates that policy decision was ASK_USER and the tool should
+     * show its legacy confirmation UI instead of auto-proceeding.
+     */
+    requiresUserConfirmation?: boolean;
+}
+export interface UpdatePolicy {
+    type: MessageBusType.UPDATE_POLICY;
+    toolName: string;
+}
+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 | UpdatePolicy;

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

@@ -0,0 +1,16 @@
+/**
+ * @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["UPDATE_POLICY"] = "update-policy";
+})(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,cAOX;AAPD,WAAY,cAAc;IACxB,yEAAuD,CAAA;IACvD,2EAAyD,CAAA;IACzD,iEAA+C,CAAA;IAC/C,mEAAiD,CAAA;IACjD,mEAAiD,CAAA;IACjD,iDAA+B,CAAA;AACjC,CAAC,EAPW,cAAc,KAAd,cAAc,QAOzB"}

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

@@ -0,0 +1,17 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Load cached API key
+ */
+export declare function loadApiKey(): Promise<string | null>;
+/**
+ * Save API key
+ */
+export declare function saveApiKey(apiKey: string | null | undefined): Promise<void>;
+/**
+ * Clear cached API key
+ */
+export declare function clearApiKey(): Promise<void>;

package/dist/src/core/apiKeyCredentialStorage.js

@@ -0,0 +1,64 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { HybridTokenStorage } from '../mcp/token-storage/hybrid-token-storage.js';
+import { debugLogger } from '../utils/debugLogger.js';
+const KEYCHAIN_SERVICE_NAME = 'gemini-cli-api-key';
+const DEFAULT_API_KEY_ENTRY = 'default-api-key';
+const storage = new HybridTokenStorage(KEYCHAIN_SERVICE_NAME);
+/**
+ * Load cached API key
+ */
+export async function loadApiKey() {
+    try {
+        const credentials = await storage.getCredentials(DEFAULT_API_KEY_ENTRY);
+        if (credentials?.token?.accessToken) {
+            return credentials.token.accessToken;
+        }
+        return null;
+    }
+    catch (error) {
+        // Log other errors but don't crash, just return null so user can re-enter key
+        debugLogger.error('Failed to load API key from storage:', error);
+        return null;
+    }
+}
+/**
+ * Save API key
+ */
+export async function saveApiKey(apiKey) {
+    if (!apiKey || apiKey.trim() === '') {
+        try {
+            await storage.deleteCredentials(DEFAULT_API_KEY_ENTRY);
+        }
+        catch (error) {
+            // Ignore errors when deleting, as it might not exist
+            debugLogger.warn('Failed to delete API key from storage:', error);
+        }
+        return;
+    }
+    // Wrap API key in OAuthCredentials format as required by HybridTokenStorage
+    const credentials = {
+        serverName: DEFAULT_API_KEY_ENTRY,
+        token: {
+            accessToken: apiKey,
+            tokenType: 'ApiKey',
+        },
+        updatedAt: Date.now(),
+    };
+    await storage.setCredentials(credentials);
+}
+/**
+ * Clear cached API key
+ */
+export async function clearApiKey() {
+    try {
+        await storage.deleteCredentials(DEFAULT_API_KEY_ENTRY);
+    }
+    catch (error) {
+        debugLogger.error('Failed to clear API key from storage:', error);
+    }
+}
+//# sourceMappingURL=apiKeyCredentialStorage.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"apiKeyCredentialStorage.js","sourceRoot":"","sources":["../../../src/core/apiKeyCredentialStorage.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,kBAAkB,EAAE,MAAM,8CAA8C,CAAC;AAElF,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAEtD,MAAM,qBAAqB,GAAG,oBAAoB,CAAC;AACnD,MAAM,qBAAqB,GAAG,iBAAiB,CAAC;AAEhD,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,qBAAqB,CAAC,CAAC;AAE9D;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,UAAU;IAC9B,IAAI,CAAC;QACH,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,qBAAqB,CAAC,CAAC;QAExE,IAAI,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;YACpC,OAAO,WAAW,CAAC,KAAK,CAAC,WAAW,CAAC;QACvC,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,8EAA8E;QAC9E,WAAW,CAAC,KAAK,CAAC,sCAAsC,EAAE,KAAK,CAAC,CAAC;QACjE,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,MAAiC;IAEjC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,MAAM,OAAO,CAAC,iBAAiB,CAAC,qBAAqB,CAAC,CAAC;QACzD,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,qDAAqD;YACrD,WAAW,CAAC,IAAI,CAAC,wCAAwC,EAAE,KAAK,CAAC,CAAC;QACpE,CAAC;QACD,OAAO;IACT,CAAC;IAED,4EAA4E;IAC5E,MAAM,WAAW,GAAqB;QACpC,UAAU,EAAE,qBAAqB;QACjC,KAAK,EAAE;YACL,WAAW,EAAE,MAAM;YACnB,SAAS,EAAE,QAAQ;SACpB;QACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACtB,CAAC;IAEF,MAAM,OAAO,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;AAC5C,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW;IAC/B,IAAI,CAAC;QACH,MAAM,OAAO,CAAC,iBAAiB,CAAC,qBAAqB,CAAC,CAAC;IACzD,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,WAAW,CAAC,KAAK,CAAC,uCAAuC,EAAE,KAAK,CAAC,CAAC;IACpE,CAAC;AACH,CAAC"}

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

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

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

@@ -0,0 +1,71 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { loadApiKey, saveApiKey, clearApiKey, } from './apiKeyCredentialStorage.js';
+const getCredentialsMock = vi.hoisted(() => vi.fn());
+const setCredentialsMock = vi.hoisted(() => vi.fn());
+const deleteCredentialsMock = vi.hoisted(() => vi.fn());
+vi.mock('../mcp/token-storage/hybrid-token-storage.js', () => ({
+    HybridTokenStorage: vi.fn().mockImplementation(() => ({
+        getCredentials: getCredentialsMock,
+        setCredentials: setCredentialsMock,
+        deleteCredentials: deleteCredentialsMock,
+    })),
+}));
+describe('ApiKeyCredentialStorage', () => {
+    beforeEach(() => {
+        vi.clearAllMocks();
+    });
+    it('should load an API key', async () => {
+        getCredentialsMock.mockResolvedValue({
+            serverName: 'default-api-key',
+            token: {
+                accessToken: 'test-key',
+                tokenType: 'ApiKey',
+            },
+            updatedAt: Date.now(),
+        });
+        const apiKey = await loadApiKey();
+        expect(apiKey).toBe('test-key');
+        expect(getCredentialsMock).toHaveBeenCalledWith('default-api-key');
+    });
+    it('should return null if no API key is stored', async () => {
+        getCredentialsMock.mockResolvedValue(null);
+        const apiKey = await loadApiKey();
+        expect(apiKey).toBeNull();
+        expect(getCredentialsMock).toHaveBeenCalledWith('default-api-key');
+    });
+    it('should save an API key', async () => {
+        await saveApiKey('new-key');
+        expect(setCredentialsMock).toHaveBeenCalledWith(expect.objectContaining({
+            serverName: 'default-api-key',
+            token: expect.objectContaining({
+                accessToken: 'new-key',
+                tokenType: 'ApiKey',
+            }),
+        }));
+    });
+    it('should clear an API key when saving empty key', async () => {
+        await saveApiKey('');
+        expect(deleteCredentialsMock).toHaveBeenCalledWith('default-api-key');
+        expect(setCredentialsMock).not.toHaveBeenCalled();
+    });
+    it('should clear an API key when saving null key', async () => {
+        await saveApiKey(null);
+        expect(deleteCredentialsMock).toHaveBeenCalledWith('default-api-key');
+        expect(setCredentialsMock).not.toHaveBeenCalled();
+    });
+    it('should clear an API key', async () => {
+        await clearApiKey();
+        expect(deleteCredentialsMock).toHaveBeenCalledWith('default-api-key');
+    });
+    it('should not throw when clearing an API key fails', async () => {
+        deleteCredentialsMock.mockRejectedValueOnce(new Error('Failed to delete'));
+        await expect(saveApiKey('')).resolves.not.toThrow();
+        expect(deleteCredentialsMock).toHaveBeenCalledWith('default-api-key');
+    });
+});
+//# sourceMappingURL=apiKeyCredentialStorage.test.js.map

package/dist/src/core/apiKeyCredentialStorage.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"apiKeyCredentialStorage.test.js","sourceRoot":"","sources":["../../../src/core/apiKeyCredentialStorage.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EACL,UAAU,EACV,UAAU,EACV,WAAW,GACZ,MAAM,8BAA8B,CAAC;AAEtC,MAAM,kBAAkB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACrD,MAAM,kBAAkB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACrD,MAAM,qBAAqB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AAExD,EAAE,CAAC,IAAI,CAAC,8CAA8C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7D,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QACpD,cAAc,EAAE,kBAAkB;QAClC,cAAc,EAAE,kBAAkB;QAClC,iBAAiB,EAAE,qBAAqB;KACzC,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,kBAAkB,CAAC,iBAAiB,CAAC;YACnC,UAAU,EAAE,iBAAiB;YAC7B,KAAK,EAAE;gBACL,WAAW,EAAE,UAAU;gBACvB,SAAS,EAAE,QAAQ;aACpB;YACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAChC,MAAM,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,kBAAkB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC1B,MAAM,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,UAAU,CAAC,SAAS,CAAC,CAAC;QAC5B,MAAM,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAC7C,MAAM,CAAC,gBAAgB,CAAC;YACtB,UAAU,EAAE,iBAAiB;YAC7B,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAC7B,WAAW,EAAE,SAAS;gBACtB,SAAS,EAAE,QAAQ;aACpB,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,UAAU,CAAC,EAAE,CAAC,CAAC;QACrB,MAAM,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;QACtE,MAAM,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,UAAU,CAAC,IAAI,CAAC,CAAC;QACvB,MAAM,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;QACtE,MAAM,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,WAAW,EAAE,CAAC;QACpB,MAAM,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,qBAAqB,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC,CAAC;QAC3E,MAAM,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACpD,MAAM,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,50 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Content, Part } from '@google/genai';
+import type { Config } from '../config/config.js';
+import type { ContentGenerator } from './contentGenerator.js';
+import type { ModelConfigKey } from '../services/modelConfigService.js';
+/**
+ * Options for the generateJson utility function.
+ */
+export interface GenerateJsonOptions {
+    /** The desired model config. */
+    modelConfigKey: ModelConfigKey;
+    /** The input prompt or history. */
+    contents: Content[];
+    /** The required JSON schema for the output. */
+    schema: Record<string, unknown>;
+    /**
+     * Task-specific system instructions.
+     * If omitted, no system instruction is sent.
+     */
+    systemInstruction?: string | Part | Part[] | Content;
+    /** Signal for cancellation. */
+    abortSignal: AbortSignal;
+    /**
+     * A unique ID for the prompt, used for logging/telemetry correlation.
+     */
+    promptId: string;
+    /**
+     * The maximum number of attempts for the request.
+     */
+    maxAttempts?: number;
+}
+/**
+ * A client dedicated to stateless, utility-focused LLM calls.
+ */
+export declare class BaseLlmClient {
+    private readonly contentGenerator;
+    private readonly config;
+    constructor(contentGenerator: ContentGenerator, config: Config);
+    generateJson(options: GenerateJsonOptions): Promise<Record<string, unknown>>;
+    /**
+     * OpenAI-specific implementation using function calling instead of responseJsonSchema
+     */
+    private generateJsonForOpenAI;
+    generateEmbedding(texts: string[]): Promise<number[][]>;
+    private cleanJsonResponse;
+}

package/dist/src/core/baseLlmClient.js

@@ -0,0 +1,185 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AuthType } from './contentGenerator.js';
+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';
+import { getFunctionCalls } from '../utils/generateContentResponseUtilities.js';
+const DEFAULT_MAX_ATTEMPTS = 5;
+/**
+ * A client dedicated to stateless, utility-focused LLM calls.
+ */
+export class BaseLlmClient {
+    contentGenerator;
+    config;
+    constructor(contentGenerator, config) {
+        this.contentGenerator = contentGenerator;
+        this.config = config;
+    }
+    async generateJson(options) {
+        const { modelConfigKey, contents, schema, abortSignal, systemInstruction, promptId, maxAttempts, } = options;
+        // Check if using OpenAI authType
+        const authType = this.config.getContentGeneratorConfig()?.authType;
+        const isOpenAI = authType === AuthType.USE_OPENAI;
+        if (isOpenAI) {
+            return this.generateJsonForOpenAI(options);
+        }
+        // Standard Gemini implementation
+        const { model, generateContentConfig } = this.config.modelConfigService.getResolvedConfig(modelConfigKey);
+        const requestConfig = {
+            abortSignal,
+            ...generateContentConfig,
+            ...(systemInstruction && { systemInstruction }),
+            responseJsonSchema: schema,
+            responseMimeType: 'application/json',
+        };
+        try {
+            const apiCall = () => this.contentGenerator.generateContent({
+                model,
+                config: requestConfig,
+                contents,
+            }, promptId);
+            const shouldRetryOnContent = (response) => {
+                const text = getResponseText(response)?.trim();
+                if (!text) {
+                    return true; // Retry on empty response
+                }
+                try {
+                    JSON.parse(this.cleanJsonResponse(text, model));
+                    return false;
+                }
+                catch (_e) {
+                    return true;
+                }
+            };
+            const result = await retryWithBackoff(apiCall, {
+                shouldRetryOnContent,
+                maxAttempts: maxAttempts ?? DEFAULT_MAX_ATTEMPTS,
+            });
+            // If we are here, the content is valid (not empty and parsable).
+            return JSON.parse(this.cleanJsonResponse(getResponseText(result).trim(), model));
+        }
+        catch (error) {
+            if (abortSignal.aborted) {
+                throw error;
+            }
+            // Check if the error is from exhausting retries, and report accordingly.
+            if (error instanceof Error &&
+                error.message.includes('Retry attempts exhausted')) {
+                await reportError(error, 'API returned invalid content (empty or unparsable JSON) after all retries.', contents, 'generateJson-invalid-content');
+            }
+            else {
+                await reportError(error, 'Error generating JSON content via API.', contents, 'generateJson-api');
+            }
+            throw new Error(`Failed to generate JSON content: ${getErrorMessage(error)}`);
+        }
+    }
+    /**
+     * OpenAI-specific implementation using function calling instead of responseJsonSchema
+     */
+    async generateJsonForOpenAI(options) {
+        const { modelConfigKey, contents, schema, abortSignal, systemInstruction, promptId, maxAttempts, } = options;
+        const { model, generateContentConfig } = this.config.modelConfigService.getResolvedConfig(modelConfigKey);
+        try {
+            // Create function declaration for OpenAI-style response
+            const functionDeclaration = {
+                name: 'respond_in_schema',
+                description: 'Provide the response in provided schema',
+                parameters: schema,
+            };
+            const tools = [
+                {
+                    functionDeclarations: [functionDeclaration],
+                },
+            ];
+            const requestConfig = {
+                abortSignal,
+                ...generateContentConfig,
+                ...(systemInstruction && { systemInstruction }),
+                tools,
+            };
+            const apiCall = () => this.contentGenerator.generateContent({
+                model,
+                config: requestConfig,
+                contents,
+            }, promptId);
+            const result = await retryWithBackoff(apiCall, {
+                maxAttempts: maxAttempts ?? DEFAULT_MAX_ATTEMPTS,
+            });
+            // Try to extract function call arguments
+            const functionCalls = getFunctionCalls(result);
+            if (functionCalls && functionCalls.length > 0) {
+                const functionCall = functionCalls.find((call) => call.name === 'respond_in_schema');
+                if (functionCall && functionCall.args) {
+                    return functionCall.args;
+                }
+            }
+            // Fallback: try to parse response text as JSON
+            let text = getResponseText(result);
+            if (!text) {
+                const error = new Error('API returned an empty response for generateJson (OpenAI).');
+                await reportError(error, 'Error in generateJson (OpenAI): API returned an empty response.', contents, 'generateJson-openai-empty-response');
+                throw error;
+            }
+            // Handle markdown wrapped JSON
+            text = this.cleanJsonResponse(text, model);
+            try {
+                return JSON.parse(text);
+            }
+            catch (parseError) {
+                await reportError(parseError, 'Failed to parse JSON response from generateJson (OpenAI).', {
+                    responseTextFailedToParse: text,
+                    originalRequestContents: contents,
+                }, 'generateJson-openai-parse');
+                throw new Error(`Failed to parse API response as JSON: ${getErrorMessage(parseError)}`);
+            }
+        }
+        catch (error) {
+            if (abortSignal.aborted) {
+                throw error;
+            }
+            await reportError(error, 'Error generating JSON content via OpenAI API.', contents, 'generateJson-openai-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