@hibanacloud/core 0.3.15

package/dist/src/output/json-formatter.test.js

@@ -0,0 +1,266 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { expect, describe, it } from 'vitest';
+import { JsonFormatter } from './json-formatter.js';
+describe('JsonFormatter', () => {
+    it('should format the response as JSON', () => {
+        const formatter = new JsonFormatter();
+        const response = 'This is a test response.';
+        const formatted = formatter.format(response);
+        const expected = {
+            response,
+        };
+        expect(JSON.parse(formatted)).toEqual(expected);
+    });
+    it('should strip ANSI escape sequences from response text', () => {
+        const formatter = new JsonFormatter();
+        const responseWithAnsi = '\x1B[31mRed text\x1B[0m and \x1B[32mGreen text\x1B[0m';
+        const formatted = formatter.format(responseWithAnsi);
+        const parsed = JSON.parse(formatted);
+        expect(parsed.response).toBe('Red text and Green text');
+    });
+    it('should strip control characters from response text', () => {
+        const formatter = new JsonFormatter();
+        const responseWithControlChars = 'Text with\x07 bell\x08 and\x0B vertical tab';
+        const formatted = formatter.format(responseWithControlChars);
+        const parsed = JSON.parse(formatted);
+        // Only ANSI codes are stripped, other control chars are preserved
+        expect(parsed.response).toBe('Text with\x07 bell\x08 and\x0B vertical tab');
+    });
+    it('should preserve newlines and tabs in response text', () => {
+        const formatter = new JsonFormatter();
+        const responseWithWhitespace = 'Line 1\nLine 2\r\nLine 3\twith tab';
+        const formatted = formatter.format(responseWithWhitespace);
+        const parsed = JSON.parse(formatted);
+        expect(parsed.response).toBe('Line 1\nLine 2\r\nLine 3\twith tab');
+    });
+    it('should format the response as JSON with stats', () => {
+        const formatter = new JsonFormatter();
+        const response = 'This is a test response.';
+        const stats = {
+            models: {
+                'gemini-2.5-pro': {
+                    api: {
+                        totalRequests: 2,
+                        totalErrors: 0,
+                        totalLatencyMs: 5672,
+                    },
+                    tokens: {
+                        prompt: 24401,
+                        candidates: 215,
+                        total: 24719,
+                        cached: 10656,
+                        thoughts: 103,
+                        tool: 0,
+                    },
+                },
+                'gemini-2.5-flash': {
+                    api: {
+                        totalRequests: 2,
+                        totalErrors: 0,
+                        totalLatencyMs: 5914,
+                    },
+                    tokens: {
+                        prompt: 20803,
+                        candidates: 716,
+                        total: 21657,
+                        cached: 0,
+                        thoughts: 138,
+                        tool: 0,
+                    },
+                },
+            },
+            tools: {
+                totalCalls: 1,
+                totalSuccess: 1,
+                totalFail: 0,
+                totalDurationMs: 4582,
+                totalDecisions: {
+                    accept: 0,
+                    reject: 0,
+                    modify: 0,
+                    auto_accept: 1,
+                },
+                byName: {
+                    google_web_search: {
+                        count: 1,
+                        success: 1,
+                        fail: 0,
+                        durationMs: 4582,
+                        decisions: {
+                            accept: 0,
+                            reject: 0,
+                            modify: 0,
+                            auto_accept: 1,
+                        },
+                    },
+                },
+            },
+            files: {
+                totalLinesAdded: 0,
+                totalLinesRemoved: 0,
+            },
+        };
+        const formatted = formatter.format(response, stats);
+        const expected = {
+            response,
+            stats,
+        };
+        expect(JSON.parse(formatted)).toEqual(expected);
+    });
+    it('should format error as JSON', () => {
+        const formatter = new JsonFormatter();
+        const error = {
+            type: 'ValidationError',
+            message: 'Invalid input provided',
+            code: 400,
+        };
+        const formatted = formatter.format(undefined, undefined, error);
+        const expected = {
+            error,
+        };
+        expect(JSON.parse(formatted)).toEqual(expected);
+    });
+    it('should format response with error as JSON', () => {
+        const formatter = new JsonFormatter();
+        const response = 'Partial response';
+        const error = {
+            type: 'TimeoutError',
+            message: 'Request timed out',
+            code: 'TIMEOUT',
+        };
+        const formatted = formatter.format(response, undefined, error);
+        const expected = {
+            response,
+            error,
+        };
+        expect(JSON.parse(formatted)).toEqual(expected);
+    });
+    it('should format error using formatError method', () => {
+        const formatter = new JsonFormatter();
+        const error = new Error('Something went wrong');
+        const formatted = formatter.formatError(error, 500);
+        const parsed = JSON.parse(formatted);
+        expect(parsed).toEqual({
+            error: {
+                type: 'Error',
+                message: 'Something went wrong',
+                code: 500,
+            },
+        });
+    });
+    it('should format custom error using formatError method', () => {
+        class CustomError extends Error {
+            constructor(message) {
+                super(message);
+                this.name = 'CustomError';
+            }
+        }
+        const formatter = new JsonFormatter();
+        const error = new CustomError('Custom error occurred');
+        const formatted = formatter.formatError(error);
+        const parsed = JSON.parse(formatted);
+        expect(parsed).toEqual({
+            error: {
+                type: 'CustomError',
+                message: 'Custom error occurred',
+            },
+        });
+    });
+    it('should format complete JSON output with response, stats, and error', () => {
+        const formatter = new JsonFormatter();
+        const response = 'Partial response before error';
+        const stats = {
+            models: {},
+            tools: {
+                totalCalls: 0,
+                totalSuccess: 0,
+                totalFail: 1,
+                totalDurationMs: 0,
+                totalDecisions: {
+                    accept: 0,
+                    reject: 0,
+                    modify: 0,
+                    auto_accept: 0,
+                },
+                byName: {},
+            },
+            files: {
+                totalLinesAdded: 0,
+                totalLinesRemoved: 0,
+            },
+        };
+        const error = {
+            type: 'ApiError',
+            message: 'Rate limit exceeded',
+            code: 429,
+        };
+        const formatted = formatter.format(response, stats, error);
+        const expected = {
+            response,
+            stats,
+            error,
+        };
+        expect(JSON.parse(formatted)).toEqual(expected);
+    });
+    it('should handle error messages containing JSON content', () => {
+        const formatter = new JsonFormatter();
+        const errorWithJson = new Error('API returned: {"error": "Invalid request", "code": 400}');
+        const formatted = formatter.formatError(errorWithJson, 'API_ERROR');
+        const parsed = JSON.parse(formatted);
+        expect(parsed).toEqual({
+            error: {
+                type: 'Error',
+                message: 'API returned: {"error": "Invalid request", "code": 400}',
+                code: 'API_ERROR',
+            },
+        });
+        // Verify the entire output is valid JSON
+        expect(() => JSON.parse(formatted)).not.toThrow();
+    });
+    it('should handle error messages with quotes and special characters', () => {
+        const formatter = new JsonFormatter();
+        const errorWithQuotes = new Error('Error: "quoted text" and \\backslash');
+        const formatted = formatter.formatError(errorWithQuotes);
+        const parsed = JSON.parse(formatted);
+        expect(parsed).toEqual({
+            error: {
+                type: 'Error',
+                message: 'Error: "quoted text" and \\backslash',
+            },
+        });
+        // Verify the entire output is valid JSON
+        expect(() => JSON.parse(formatted)).not.toThrow();
+    });
+    it('should handle error messages with control characters', () => {
+        const formatter = new JsonFormatter();
+        const errorWithControlChars = new Error('Error with\n newline and\t tab');
+        const formatted = formatter.formatError(errorWithControlChars);
+        const parsed = JSON.parse(formatted);
+        // Should preserve newlines and tabs as they are common whitespace characters
+        expect(parsed.error.message).toBe('Error with\n newline and\t tab');
+        // Verify the entire output is valid JSON
+        expect(() => JSON.parse(formatted)).not.toThrow();
+    });
+    it('should strip ANSI escape sequences from error messages', () => {
+        const formatter = new JsonFormatter();
+        const errorWithAnsi = new Error('\x1B[31mRed error\x1B[0m message');
+        const formatted = formatter.formatError(errorWithAnsi);
+        const parsed = JSON.parse(formatted);
+        expect(parsed.error.message).toBe('Red error message');
+        expect(() => JSON.parse(formatted)).not.toThrow();
+    });
+    it('should strip unsafe control characters from error messages', () => {
+        const formatter = new JsonFormatter();
+        const errorWithControlChars = new Error('Error\x07 with\x08 control\x0B chars');
+        const formatted = formatter.formatError(errorWithControlChars);
+        const parsed = JSON.parse(formatted);
+        // Only ANSI codes are stripped, other control chars are preserved
+        expect(parsed.error.message).toBe('Error\x07 with\x08 control\x0B chars');
+        expect(() => JSON.parse(formatted)).not.toThrow();
+    });
+});
+//# sourceMappingURL=json-formatter.test.js.map

package/dist/src/output/json-formatter.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"json-formatter.test.js","sourceRoot":"","sources":["../../../src/output/json-formatter.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE9C,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAGpD,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,QAAQ,GAAG,0BAA0B,CAAC;QAC5C,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,QAAQ,GAAG;YACf,QAAQ;SACT,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,gBAAgB,GACpB,uDAAuD,CAAC;QAC1D,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,wBAAwB,GAC5B,6CAA6C,CAAC;QAChD,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACrC,kEAAkE;QAClE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,sBAAsB,GAAG,oCAAoC,CAAC;QACpE,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,QAAQ,GAAG,0BAA0B,CAAC;QAC5C,MAAM,KAAK,GAAmB;YAC5B,MAAM,EAAE;gBACN,gBAAgB,EAAE;oBAChB,GAAG,EAAE;wBACH,aAAa,EAAE,CAAC;wBAChB,WAAW,EAAE,CAAC;wBACd,cAAc,EAAE,IAAI;qBACrB;oBACD,MAAM,EAAE;wBACN,MAAM,EAAE,KAAK;wBACb,UAAU,EAAE,GAAG;wBACf,KAAK,EAAE,KAAK;wBACZ,MAAM,EAAE,KAAK;wBACb,QAAQ,EAAE,GAAG;wBACb,IAAI,EAAE,CAAC;qBACR;iBACF;gBACD,kBAAkB,EAAE;oBAClB,GAAG,EAAE;wBACH,aAAa,EAAE,CAAC;wBAChB,WAAW,EAAE,CAAC;wBACd,cAAc,EAAE,IAAI;qBACrB;oBACD,MAAM,EAAE;wBACN,MAAM,EAAE,KAAK;wBACb,UAAU,EAAE,GAAG;wBACf,KAAK,EAAE,KAAK;wBACZ,MAAM,EAAE,CAAC;wBACT,QAAQ,EAAE,GAAG;wBACb,IAAI,EAAE,CAAC;qBACR;iBACF;aACF;YACD,KAAK,EAAE;gBACL,UAAU,EAAE,CAAC;gBACb,YAAY,EAAE,CAAC;gBACf,SAAS,EAAE,CAAC;gBACZ,eAAe,EAAE,IAAI;gBACrB,cAAc,EAAE;oBACd,MAAM,EAAE,CAAC;oBACT,MAAM,EAAE,CAAC;oBACT,MAAM,EAAE,CAAC;oBACT,WAAW,EAAE,CAAC;iBACf;gBACD,MAAM,EAAE;oBACN,iBAAiB,EAAE;wBACjB,KAAK,EAAE,CAAC;wBACR,OAAO,EAAE,CAAC;wBACV,IAAI,EAAE,CAAC;wBACP,UAAU,EAAE,IAAI;wBAChB,SAAS,EAAE;4BACT,MAAM,EAAE,CAAC;4BACT,MAAM,EAAE,CAAC;4BACT,MAAM,EAAE,CAAC;4BACT,WAAW,EAAE,CAAC;yBACf;qBACF;iBACF;aACF;YACD,KAAK,EAAE;gBACL,eAAe,EAAE,CAAC;gBAClB,iBAAiB,EAAE,CAAC;aACrB;SACF,CAAC;QACF,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACpD,MAAM,QAAQ,GAAG;YACf,QAAQ;YACR,KAAK;SACN,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,KAAK,GAAc;YACvB,IAAI,EAAE,iBAAiB;YACvB,OAAO,EAAE,wBAAwB;YACjC,IAAI,EAAE,GAAG;SACV,CAAC;QACF,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,SAAS,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QAChE,MAAM,QAAQ,GAAG;YACf,KAAK;SACN,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,QAAQ,GAAG,kBAAkB,CAAC;QACpC,MAAM,KAAK,GAAc;YACvB,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,mBAAmB;YAC5B,IAAI,EAAE,SAAS;SAChB,CAAC;QACF,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QAC/D,MAAM,QAAQ,GAAG;YACf,QAAQ;YACR,KAAK;SACN,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAChD,MAAM,SAAS,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QACpD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAErC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,KAAK,EAAE;gBACL,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,sBAAsB;gBAC/B,IAAI,EAAE,GAAG;aACV;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,WAAY,SAAQ,KAAK;YAC7B,YAAY,OAAe;gBACzB,KAAK,CAAC,OAAO,CAAC,CAAC;gBACf,IAAI,CAAC,IAAI,GAAG,aAAa,CAAC;YAC5B,CAAC;SACF;QAED,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,uBAAuB,CAAC,CAAC;QACvD,MAAM,SAAS,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAErC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,KAAK,EAAE;gBACL,IAAI,EAAE,aAAa;gBACnB,OAAO,EAAE,uBAAuB;aACjC;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;QAC5E,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,QAAQ,GAAG,+BAA+B,CAAC;QACjD,MAAM,KAAK,GAAmB;YAC5B,MAAM,EAAE,EAAE;YACV,KAAK,EAAE;gBACL,UAAU,EAAE,CAAC;gBACb,YAAY,EAAE,CAAC;gBACf,SAAS,EAAE,CAAC;gBACZ,eAAe,EAAE,CAAC;gBAClB,cAAc,EAAE;oBACd,MAAM,EAAE,CAAC;oBACT,MAAM,EAAE,CAAC;oBACT,MAAM,EAAE,CAAC;oBACT,WAAW,EAAE,CAAC;iBACf;gBACD,MAAM,EAAE,EAAE;aACX;YACD,KAAK,EAAE;gBACL,eAAe,EAAE,CAAC;gBAClB,iBAAiB,EAAE,CAAC;aACrB;SACF,CAAC;QACF,MAAM,KAAK,GAAc;YACvB,IAAI,EAAE,UAAU;YAChB,OAAO,EAAE,qBAAqB;YAC9B,IAAI,EAAE,GAAG;SACV,CAAC;QAEF,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QAC3D,MAAM,QAAQ,GAAG;YACf,QAAQ;YACR,KAAK;YACL,KAAK;SACN,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,aAAa,GAAG,IAAI,KAAK,CAC7B,yDAAyD,CAC1D,CAAC;QACF,MAAM,SAAS,GAAG,SAAS,CAAC,WAAW,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC;QACpE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAErC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,KAAK,EAAE;gBACL,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,yDAAyD;gBAClE,IAAI,EAAE,WAAW;aAClB;SACF,CAAC,CAAC;QAEH,yCAAyC;QACzC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACzE,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,eAAe,GAAG,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1E,MAAM,SAAS,GAAG,SAAS,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;QACzD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAErC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,KAAK,EAAE;gBACL,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,sCAAsC;aAChD;SACF,CAAC,CAAC;QAEH,yCAAyC;QACzC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,qBAAqB,GAAG,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QAC1E,MAAM,SAAS,GAAG,SAAS,CAAC,WAAW,CAAC,qBAAqB,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAErC,6EAA6E;QAC7E,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;QAEpE,yCAAyC;QACzC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,aAAa,GAAG,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACpE,MAAM,SAAS,GAAG,SAAS,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;QACvD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAErC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACvD,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,SAAS,GAAG,IAAI,aAAa,EAAE,CAAC;QACtC,MAAM,qBAAqB,GAAG,IAAI,KAAK,CACrC,sCAAsC,CACvC,CAAC;QACF,MAAM,SAAS,GAAG,SAAS,CAAC,WAAW,CAAC,qBAAqB,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAErC,kEAAkE;QAClE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAC;QAC1E,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/output/types.d.ts

@@ -0,0 +1,25 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { SessionMetrics } from '../telemetry/uiTelemetry.js';
+export declare enum InputFormat {
+    TEXT = "text",
+    STREAM_JSON = "stream-json"
+}
+export declare enum OutputFormat {
+    TEXT = "text",
+    JSON = "json",
+    STREAM_JSON = "stream-json"
+}
+export interface JsonError {
+    type: string;
+    message: string;
+    code?: string | number;
+}
+export interface JsonOutput {
+    response?: string;
+    stats?: SessionMetrics;
+    error?: JsonError;
+}

package/dist/src/output/types.js

@@ -0,0 +1,17 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export var InputFormat;
+(function (InputFormat) {
+    InputFormat["TEXT"] = "text";
+    InputFormat["STREAM_JSON"] = "stream-json";
+})(InputFormat || (InputFormat = {}));
+export var OutputFormat;
+(function (OutputFormat) {
+    OutputFormat["TEXT"] = "text";
+    OutputFormat["JSON"] = "json";
+    OutputFormat["STREAM_JSON"] = "stream-json";
+})(OutputFormat || (OutputFormat = {}));
+//# sourceMappingURL=types.js.map

package/dist/src/output/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/output/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,MAAM,CAAN,IAAY,WAGX;AAHD,WAAY,WAAW;IACrB,4BAAa,CAAA;IACb,0CAA2B,CAAA;AAC7B,CAAC,EAHW,WAAW,KAAX,WAAW,QAGtB;AAED,MAAM,CAAN,IAAY,YAIX;AAJD,WAAY,YAAY;IACtB,6BAAa,CAAA;IACb,6BAAa,CAAA;IACb,2CAA2B,CAAA;AAC7B,CAAC,EAJW,YAAY,KAAZ,YAAY,QAIvB"}

package/dist/src/prompts/mcp-prompts.d.ts

@@ -0,0 +1,8 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config } from '../config/config.js';
+import type { DiscoveredMCPPrompt } from '../tools/mcp-client.js';
+export declare function getMCPServerPrompts(config: Config, serverName: string): DiscoveredMCPPrompt[];

package/dist/src/prompts/mcp-prompts.js

@@ -0,0 +1,13 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export function getMCPServerPrompts(config, serverName) {
+    const promptRegistry = config.getPromptRegistry();
+    if (!promptRegistry) {
+        return [];
+    }
+    return promptRegistry.getPromptsByServer(serverName);
+}
+//# sourceMappingURL=mcp-prompts.js.map

package/dist/src/prompts/mcp-prompts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-prompts.js","sourceRoot":"","sources":["../../../src/prompts/mcp-prompts.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH,MAAM,UAAU,mBAAmB,CACjC,MAAc,EACd,UAAkB;IAElB,MAAM,cAAc,GAAG,MAAM,CAAC,iBAAiB,EAAE,CAAC;IAClD,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,cAAc,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC;AACvD,CAAC"}

package/dist/src/prompts/prompt-registry.d.ts

@@ -0,0 +1,34 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { DiscoveredMCPPrompt } from '../tools/mcp-client.js';
+export declare class PromptRegistry {
+    private prompts;
+    /**
+     * Registers a prompt definition.
+     * @param prompt - The prompt object containing schema and execution logic.
+     */
+    registerPrompt(prompt: DiscoveredMCPPrompt): void;
+    /**
+     * Returns an array of all registered and discovered prompt instances.
+     */
+    getAllPrompts(): DiscoveredMCPPrompt[];
+    /**
+     * Get the definition of a specific prompt.
+     */
+    getPrompt(name: string): DiscoveredMCPPrompt | undefined;
+    /**
+     * Returns an array of prompts registered from a specific MCP server.
+     */
+    getPromptsByServer(serverName: string): DiscoveredMCPPrompt[];
+    /**
+     * Clears all the prompts from the registry.
+     */
+    clear(): void;
+    /**
+     * Removes all prompts from a specific server.
+     */
+    removePromptsByServer(serverName: string): void;
+}

package/dist/src/prompts/prompt-registry.js

@@ -0,0 +1,63 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export class PromptRegistry {
+    prompts = new Map();
+    /**
+     * Registers a prompt definition.
+     * @param prompt - The prompt object containing schema and execution logic.
+     */
+    registerPrompt(prompt) {
+        if (this.prompts.has(prompt.name)) {
+            const newName = `${prompt.serverName}_${prompt.name}`;
+            console.warn(`Prompt with name "${prompt.name}" is already registered. Renaming to "${newName}".`);
+            this.prompts.set(newName, { ...prompt, name: newName });
+        }
+        else {
+            this.prompts.set(prompt.name, prompt);
+        }
+    }
+    /**
+     * Returns an array of all registered and discovered prompt instances.
+     */
+    getAllPrompts() {
+        return Array.from(this.prompts.values()).sort((a, b) => a.name.localeCompare(b.name));
+    }
+    /**
+     * Get the definition of a specific prompt.
+     */
+    getPrompt(name) {
+        return this.prompts.get(name);
+    }
+    /**
+     * Returns an array of prompts registered from a specific MCP server.
+     */
+    getPromptsByServer(serverName) {
+        const serverPrompts = [];
+        for (const prompt of this.prompts.values()) {
+            if (prompt.serverName === serverName) {
+                serverPrompts.push(prompt);
+            }
+        }
+        return serverPrompts.sort((a, b) => a.name.localeCompare(b.name));
+    }
+    /**
+     * Clears all the prompts from the registry.
+     */
+    clear() {
+        this.prompts.clear();
+    }
+    /**
+     * Removes all prompts from a specific server.
+     */
+    removePromptsByServer(serverName) {
+        for (const [name, prompt] of this.prompts.entries()) {
+            if (prompt.serverName === serverName) {
+                this.prompts.delete(name);
+            }
+        }
+    }
+}
+//# sourceMappingURL=prompt-registry.js.map

package/dist/src/prompts/prompt-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-registry.js","sourceRoot":"","sources":["../../../src/prompts/prompt-registry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,MAAM,OAAO,cAAc;IACjB,OAAO,GAAqC,IAAI,GAAG,EAAE,CAAC;IAE9D;;;OAGG;IACH,cAAc,CAAC,MAA2B;QACxC,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,GAAG,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YACtD,OAAO,CAAC,IAAI,CACV,qBAAqB,MAAM,CAAC,IAAI,yCAAyC,OAAO,IAAI,CACrF,CAAC;YACF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC1D,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACrD,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAC7B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,IAAY;QACpB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,UAAkB;QACnC,MAAM,aAAa,GAA0B,EAAE,CAAC;QAChD,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,IAAI,MAAM,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;gBACrC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QACD,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IACpE,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,qBAAqB,CAAC,UAAkB;QACtC,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACpD,IAAI,MAAM,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;gBACrC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;CACF"}

package/dist/src/qwen/qwenContentGenerator.d.ts

@@ -0,0 +1,70 @@
+/**
+ * @license
+ * Copyright 2025 Qwen
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { OpenAIContentGenerator } from '../core/openaiContentGenerator/index.js';
+import type { IQwenOAuth2Client } from './qwenOAuth2.js';
+import { type Config } from '../config/config.js';
+import type { GenerateContentParameters, GenerateContentResponse, CountTokensParameters, CountTokensResponse, EmbedContentParameters, EmbedContentResponse } from '@google/genai';
+import type { ContentGeneratorConfig } from '../core/contentGenerator.js';
+/**
+ * Qwen Content Generator that uses Qwen OAuth tokens with automatic refresh
+ */
+export declare class QwenContentGenerator extends OpenAIContentGenerator {
+    private qwenClient;
+    private sharedManager;
+    private currentToken?;
+    constructor(qwenClient: IQwenOAuth2Client, contentGeneratorConfig: ContentGeneratorConfig, cliConfig: Config);
+    /**
+     * Get the current endpoint URL with proper protocol and /v1 suffix
+     */
+    private getCurrentEndpoint;
+    /**
+     * Override error logging behavior to suppress auth errors during token refresh
+     */
+    protected shouldSuppressErrorLogging(error: unknown, _request: GenerateContentParameters): boolean;
+    /**
+     * Get valid token and endpoint using the shared token manager
+     */
+    private getValidToken;
+    /**
+     * Execute an operation with automatic credential management and retry logic.
+     * This method handles:
+     * - Dynamic token and endpoint retrieval
+     * - Client configuration updates
+     * - Retry logic on authentication errors with token refresh
+     *
+     * @param operation - The operation to execute with updated client configuration
+     * @returns The result of the operation
+     */
+    private executeWithCredentialManagement;
+    /**
+     * Override to use dynamic token and endpoint with automatic retry
+     */
+    generateContent(request: GenerateContentParameters, userPromptId: string): Promise<GenerateContentResponse>;
+    /**
+     * Override to use dynamic token and endpoint with automatic retry
+     */
+    generateContentStream(request: GenerateContentParameters, userPromptId: string): Promise<AsyncGenerator<GenerateContentResponse>>;
+    /**
+     * Override to use dynamic token and endpoint with automatic retry
+     */
+    countTokens(request: CountTokensParameters): Promise<CountTokensResponse>;
+    /**
+     * Override to use dynamic token and endpoint with automatic retry
+     */
+    embedContent(request: EmbedContentParameters): Promise<EmbedContentResponse>;
+    /**
+     * Check if an error is related to authentication/authorization
+     */
+    private isAuthError;
+    /**
+     * Get the current cached token (may be expired)
+     */
+    getCurrentToken(): string | null;
+    /**
+     * Clear the cached token
+     */
+    clearToken(): void;
+}