@office-ai/aioncli-core 0.2.3 → 0.8.1

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,54 @@
+/**
+ * @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;
+    /**
+     * 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;
+    private readonly defaultUtilityConfig;
+    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,190 @@
+/**
+ * @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;
+    // 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, 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 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 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 { contents, schema, model, abortSignal, systemInstruction, promptId, maxAttempts, } = options;
+        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,
+                ...this.defaultUtilityConfig,
+                ...options.config,
+                ...(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

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;AAcH,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AACjD,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;AACrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,8CAA8C,CAAC;AAEhF,MAAM,oBAAoB,GAAG,CAAC,CAAC;AAwC/B;;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,EACR,WAAW,GACZ,GAAG,OAAO,CAAC;QAEZ,iCAAiC;QACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,yBAAyB,EAAE,EAAE,QAAQ,CAAC;QACnE,MAAM,QAAQ,GAAG,QAAQ,KAAK,QAAQ,CAAC,UAAU,CAAC;QAElD,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;QAC7C,CAAC;QAED,iCAAiC;QACjC,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,oBAAoB,GAAG,CAAC,QAAiC,EAAE,EAAE;gBACjE,MAAM,IAAI,GAAG,eAAe,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,CAAC;gBAC/C,IAAI,CAAC,IAAI,EAAE,CAAC;oBACV,OAAO,IAAI,CAAC,CAAC,0BAA0B;gBACzC,CAAC;gBACD,IAAI,CAAC;oBACH,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;oBAChD,OAAO,KAAK,CAAC;gBACf,CAAC;gBAAC,OAAO,EAAE,EAAE,CAAC;oBACZ,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,CAAC,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,OAAO,EAAE;gBAC7C,oBAAoB;gBACpB,WAAW,EAAE,WAAW,IAAI,oBAAoB;aACjD,CAAC,CAAC;YAEH,iEAAiE;YACjE,OAAO,IAAI,CAAC,KAAK,CACf,IAAI,CAAC,iBAAiB,CAAC,eAAe,CAAC,MAAM,CAAE,CAAC,IAAI,EAAE,EAAE,KAAK,CAAC,CAC/D,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;gBACxB,MAAM,KAAK,CAAC;YACd,CAAC;YAED,yEAAyE;YACzE,IACE,KAAK,YAAY,KAAK;gBACtB,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,0BAA0B,CAAC,EAClD,CAAC;gBACD,MAAM,WAAW,CACf,KAAK,EACL,4EAA4E,EAC5E,QAAQ,EACR,8BAA8B,CAC/B,CAAC;YACJ,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;;OAEG;IACK,KAAK,CAAC,qBAAqB,CACjC,OAA4B;QAE5B,MAAM,EACJ,QAAQ,EACR,MAAM,EACN,KAAK,EACL,WAAW,EACX,iBAAiB,EACjB,QAAQ,EACR,WAAW,GACZ,GAAG,OAAO,CAAC;QAEZ,IAAI,CAAC;YACH,wDAAwD;YACxD,MAAM,mBAAmB,GAAwB;gBAC/C,IAAI,EAAE,mBAAmB;gBACzB,WAAW,EAAE,yCAAyC;gBACtD,UAAU,EAAE,MAAgB;aAC7B,CAAC;YAEF,MAAM,KAAK,GAAW;gBACpB;oBACE,oBAAoB,EAAE,CAAC,mBAAmB,CAAC;iBAC5C;aACF,CAAC;YAEF,MAAM,aAAa,GAA0B;gBAC3C,WAAW;gBACX,GAAG,IAAI,CAAC,oBAAoB;gBAC5B,GAAG,OAAO,CAAC,MAAM;gBACjB,GAAG,CAAC,iBAAiB,IAAI,EAAE,iBAAiB,EAAE,CAAC;gBAC/C,KAAK;aACN,CAAC;YAEF,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,EAAE;gBAC7C,WAAW,EAAE,WAAW,IAAI,oBAAoB;aACjD,CAAC,CAAC;YAEH,yCAAyC;YACzC,MAAM,aAAa,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAE/C,IAAI,aAAa,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9C,MAAM,YAAY,GAAG,aAAa,CAAC,IAAI,CACrC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,mBAAmB,CAC5C,CAAC;gBACF,IAAI,YAAY,IAAI,YAAY,CAAC,IAAI,EAAE,CAAC;oBACtC,OAAO,YAAY,CAAC,IAAI,CAAC;gBAC3B,CAAC;YACH,CAAC;YAED,+CAA+C;YAC/C,IAAI,IAAI,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;YAEnC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,MAAM,KAAK,GAAG,IAAI,KAAK,CACrB,2DAA2D,CAC5D,CAAC;gBACF,MAAM,WAAW,CACf,KAAK,EACL,iEAAiE,EACjE,QAAQ,EACR,oCAAoC,CACrC,CAAC;gBACF,MAAM,KAAK,CAAC;YACd,CAAC;YAED,+BAA+B;YAC/B,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,WAAW,CACf,UAAU,EACV,2DAA2D,EAC3D;oBACE,yBAAyB,EAAE,IAAI;oBAC/B,uBAAuB,EAAE,QAAQ;iBAClC,EACD,2BAA2B,CAC5B,CAAC;gBACF,MAAM,IAAI,KAAK,CACb,yCAAyC,eAAe,CACtD,UAAU,CACX,EAAE,CACJ,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;gBACxB,MAAM,KAAK,CAAC;YACd,CAAC;YAED,MAAM,WAAW,CACf,KAAK,EACL,+CAA+C,EAC/C,QAAQ,EACR,yBAAyB,CAC1B,CAAC;YACF,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 {};