@vybestack/llxprt-code-core 0.1.18-nightly.250811.b0db22c6 → 0.1.18-nightly.250812.12fa8ad2

package/dist/src/code_assist/converter.test.js

@@ -1,279 +0,0 @@
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-import { describe, it, expect } from 'vitest';
-import { toGenerateContentRequest, fromGenerateContentResponse, toContents, } from './converter.js';
-import { GenerateContentResponse, FinishReason, BlockedReason, } from '@google/genai';
-describe('converter', () => {
-    describe('toCodeAssistRequest', () => {
-        it('should convert a simple request with project', () => {
-            const genaiReq = {
-                model: 'gemini-pro',
-                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
-            };
-            const codeAssistReq = toGenerateContentRequest(genaiReq, 'my-prompt', 'my-project', 'my-session');
-            expect(codeAssistReq).toEqual({
-                model: 'gemini-pro',
-                project: 'my-project',
-                request: {
-                    contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
-                    systemInstruction: undefined,
-                    cachedContent: undefined,
-                    tools: undefined,
-                    toolConfig: undefined,
-                    labels: undefined,
-                    safetySettings: undefined,
-                    generationConfig: undefined,
-                    session_id: 'my-session',
-                },
-                user_prompt_id: 'my-prompt',
-            });
-        });
-        it('should convert a request without a project', () => {
-            const genaiReq = {
-                model: 'gemini-pro',
-                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
-            };
-            const codeAssistReq = toGenerateContentRequest(genaiReq, 'my-prompt', undefined, 'my-session');
-            expect(codeAssistReq).toEqual({
-                model: 'gemini-pro',
-                project: undefined,
-                request: {
-                    contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
-                    systemInstruction: undefined,
-                    cachedContent: undefined,
-                    tools: undefined,
-                    toolConfig: undefined,
-                    labels: undefined,
-                    safetySettings: undefined,
-                    generationConfig: undefined,
-                    session_id: 'my-session',
-                },
-                user_prompt_id: 'my-prompt',
-            });
-        });
-        it('should convert a request with sessionId', () => {
-            const genaiReq = {
-                model: 'gemini-pro',
-                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
-            };
-            const codeAssistReq = toGenerateContentRequest(genaiReq, 'my-prompt', 'my-project', 'session-123');
-            expect(codeAssistReq).toEqual({
-                model: 'gemini-pro',
-                project: 'my-project',
-                request: {
-                    contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
-                    systemInstruction: undefined,
-                    cachedContent: undefined,
-                    tools: undefined,
-                    toolConfig: undefined,
-                    labels: undefined,
-                    safetySettings: undefined,
-                    generationConfig: undefined,
-                    session_id: 'session-123',
-                },
-                user_prompt_id: 'my-prompt',
-            });
-        });
-        it('should handle string content', () => {
-            const genaiReq = {
-                model: 'gemini-pro',
-                contents: 'Hello',
-            };
-            const codeAssistReq = toGenerateContentRequest(genaiReq, 'my-prompt', 'my-project', 'my-session');
-            expect(codeAssistReq.request.contents).toEqual([
-                { role: 'user', parts: [{ text: 'Hello' }] },
-            ]);
-        });
-        it('should handle Part[] content', () => {
-            const genaiReq = {
-                model: 'gemini-pro',
-                contents: [{ text: 'Hello' }, { text: 'World' }],
-            };
-            const codeAssistReq = toGenerateContentRequest(genaiReq, 'my-prompt', 'my-project', 'my-session');
-            expect(codeAssistReq.request.contents).toEqual([
-                { role: 'user', parts: [{ text: 'Hello' }] },
-                { role: 'user', parts: [{ text: 'World' }] },
-            ]);
-        });
-        it('should handle system instructions', () => {
-            const genaiReq = {
-                model: 'gemini-pro',
-                contents: 'Hello',
-                config: {
-                    systemInstruction: 'You are a helpful assistant.',
-                },
-            };
-            const codeAssistReq = toGenerateContentRequest(genaiReq, 'my-prompt', 'my-project', 'my-session');
-            expect(codeAssistReq.request.systemInstruction).toEqual({
-                role: 'user',
-                parts: [{ text: 'You are a helpful assistant.' }],
-            });
-        });
-        it('should handle generation config', () => {
-            const genaiReq = {
-                model: 'gemini-pro',
-                contents: 'Hello',
-                config: {
-                    temperature: 0.8,
-                    topK: 40,
-                },
-            };
-            const codeAssistReq = toGenerateContentRequest(genaiReq, 'my-prompt', 'my-project', 'my-session');
-            expect(codeAssistReq.request.generationConfig).toEqual({
-                temperature: 0.8,
-                topK: 40,
-            });
-        });
-        it('should handle all generation config fields', () => {
-            const genaiReq = {
-                model: 'gemini-pro',
-                contents: 'Hello',
-                config: {
-                    temperature: 0.1,
-                    topP: 0.2,
-                    topK: 3,
-                    candidateCount: 4,
-                    maxOutputTokens: 5,
-                    stopSequences: ['a'],
-                    responseLogprobs: true,
-                    logprobs: 6,
-                    presencePenalty: 0.7,
-                    frequencyPenalty: 0.8,
-                    seed: 9,
-                    responseMimeType: 'application/json',
-                },
-            };
-            const codeAssistReq = toGenerateContentRequest(genaiReq, 'my-prompt', 'my-project', 'my-session');
-            expect(codeAssistReq.request.generationConfig).toEqual({
-                temperature: 0.1,
-                topP: 0.2,
-                topK: 3,
-                candidateCount: 4,
-                maxOutputTokens: 5,
-                stopSequences: ['a'],
-                responseLogprobs: true,
-                logprobs: 6,
-                presencePenalty: 0.7,
-                frequencyPenalty: 0.8,
-                seed: 9,
-                responseMimeType: 'application/json',
-            });
-        });
-    });
-    describe('fromCodeAssistResponse', () => {
-        it('should convert a simple response', () => {
-            const codeAssistRes = {
-                response: {
-                    candidates: [
-                        {
-                            index: 0,
-                            content: {
-                                role: 'model',
-                                parts: [{ text: 'Hi there!' }],
-                            },
-                            finishReason: FinishReason.STOP,
-                            safetyRatings: [],
-                        },
-                    ],
-                },
-            };
-            const genaiRes = fromGenerateContentResponse(codeAssistRes);
-            expect(genaiRes).toBeInstanceOf(GenerateContentResponse);
-            expect(genaiRes.candidates).toEqual(codeAssistRes.response.candidates);
-        });
-        it('should handle prompt feedback and usage metadata', () => {
-            const codeAssistRes = {
-                response: {
-                    candidates: [],
-                    promptFeedback: {
-                        blockReason: BlockedReason.SAFETY,
-                        safetyRatings: [],
-                    },
-                    usageMetadata: {
-                        promptTokenCount: 10,
-                        candidatesTokenCount: 20,
-                        totalTokenCount: 30,
-                    },
-                },
-            };
-            const genaiRes = fromGenerateContentResponse(codeAssistRes);
-            expect(genaiRes.promptFeedback).toEqual(codeAssistRes.response.promptFeedback);
-            expect(genaiRes.usageMetadata).toEqual(codeAssistRes.response.usageMetadata);
-        });
-        it('should handle automatic function calling history', () => {
-            const codeAssistRes = {
-                response: {
-                    candidates: [],
-                    automaticFunctionCallingHistory: [
-                        {
-                            role: 'model',
-                            parts: [
-                                {
-                                    functionCall: {
-                                        name: 'test_function',
-                                        args: {
-                                            foo: 'bar',
-                                        },
-                                    },
-                                },
-                            ],
-                        },
-                    ],
-                },
-            };
-            const genaiRes = fromGenerateContentResponse(codeAssistRes);
-            expect(genaiRes.automaticFunctionCallingHistory).toEqual(codeAssistRes.response.automaticFunctionCallingHistory);
-        });
-    });
-    describe('toContents', () => {
-        it('should handle Content', () => {
-            const content = {
-                role: 'user',
-                parts: [{ text: 'hello' }],
-            };
-            expect(toContents(content)).toEqual([
-                { role: 'user', parts: [{ text: 'hello' }] },
-            ]);
-        });
-        it('should handle array of Contents', () => {
-            const contents = [
-                { role: 'user', parts: [{ text: 'hello' }] },
-                { role: 'model', parts: [{ text: 'hi' }] },
-            ];
-            expect(toContents(contents)).toEqual([
-                { role: 'user', parts: [{ text: 'hello' }] },
-                { role: 'model', parts: [{ text: 'hi' }] },
-            ]);
-        });
-        it('should handle Part', () => {
-            const part = { text: 'a part' };
-            expect(toContents(part)).toEqual([
-                { role: 'user', parts: [{ text: 'a part' }] },
-            ]);
-        });
-        it('should handle array of Parts', () => {
-            const parts = [{ text: 'part 1' }, 'part 2'];
-            expect(toContents(parts)).toEqual([
-                { role: 'user', parts: [{ text: 'part 1' }] },
-                { role: 'user', parts: [{ text: 'part 2' }] },
-            ]);
-        });
-        it('should handle string', () => {
-            const str = 'a string';
-            expect(toContents(str)).toEqual([
-                { role: 'user', parts: [{ text: 'a string' }] },
-            ]);
-        });
-        it('should handle array of strings', () => {
-            const strings = ['string 1', 'string 2'];
-            expect(toContents(strings)).toEqual([
-                { role: 'user', parts: [{ text: 'string 1' }] },
-                { role: 'user', parts: [{ text: 'string 2' }] },
-            ]);
-        });
-    });
-});
-//# sourceMappingURL=converter.test.js.map

package/dist/src/code_assist/converter.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"converter.test.js","sourceRoot":"","sources":["../../../src/code_assist/converter.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,wBAAwB,EACxB,2BAA2B,EAE3B,UAAU,GACX,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAGL,uBAAuB,EACvB,YAAY,EACZ,aAAa,GACd,MAAM,eAAe,CAAC;AAEvB,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,QAAQ,GAA8B;gBAC1C,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;aACzD,CAAC;YACF,MAAM,aAAa,GAAG,wBAAwB,CAC5C,QAAQ,EACR,WAAW,EACX,YAAY,EACZ,YAAY,CACb,CAAC;YACF,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC;gBAC5B,KAAK,EAAE,YAAY;gBACnB,OAAO,EAAE,YAAY;gBACrB,OAAO,EAAE;oBACP,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;oBACxD,iBAAiB,EAAE,SAAS;oBAC5B,aAAa,EAAE,SAAS;oBACxB,KAAK,EAAE,SAAS;oBAChB,UAAU,EAAE,SAAS;oBACrB,MAAM,EAAE,SAAS;oBACjB,cAAc,EAAE,SAAS;oBACzB,gBAAgB,EAAE,SAAS;oBAC3B,UAAU,EAAE,YAAY;iBACzB;gBACD,cAAc,EAAE,WAAW;aAC5B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,QAAQ,GAA8B;gBAC1C,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;aACzD,CAAC;YACF,MAAM,aAAa,GAAG,wBAAwB,CAC5C,QAAQ,EACR,WAAW,EACX,SAAS,EACT,YAAY,CACb,CAAC;YACF,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC;gBAC5B,KAAK,EAAE,YAAY;gBACnB,OAAO,EAAE,SAAS;gBAClB,OAAO,EAAE;oBACP,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;oBACxD,iBAAiB,EAAE,SAAS;oBAC5B,aAAa,EAAE,SAAS;oBACxB,KAAK,EAAE,SAAS;oBAChB,UAAU,EAAE,SAAS;oBACrB,MAAM,EAAE,SAAS;oBACjB,cAAc,EAAE,SAAS;oBACzB,gBAAgB,EAAE,SAAS;oBAC3B,UAAU,EAAE,YAAY;iBACzB;gBACD,cAAc,EAAE,WAAW;aAC5B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,QAAQ,GAA8B;gBAC1C,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;aACzD,CAAC;YACF,MAAM,aAAa,GAAG,wBAAwB,CAC5C,QAAQ,EACR,WAAW,EACX,YAAY,EACZ,aAAa,CACd,CAAC;YACF,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC;gBAC5B,KAAK,EAAE,YAAY;gBACnB,OAAO,EAAE,YAAY;gBACrB,OAAO,EAAE;oBACP,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;oBACxD,iBAAiB,EAAE,SAAS;oBAC5B,aAAa,EAAE,SAAS;oBACxB,KAAK,EAAE,SAAS;oBAChB,UAAU,EAAE,SAAS;oBACrB,MAAM,EAAE,SAAS;oBACjB,cAAc,EAAE,SAAS;oBACzB,gBAAgB,EAAE,SAAS;oBAC3B,UAAU,EAAE,aAAa;iBAC1B;gBACD,cAAc,EAAE,WAAW;aAC5B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,QAAQ,GAA8B;gBAC1C,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,OAAO;aAClB,CAAC;YACF,MAAM,aAAa,GAAG,wBAAwB,CAC5C,QAAQ,EACR,WAAW,EACX,YAAY,EACZ,YAAY,CACb,CAAC;YACF,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBAC7C,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE;aAC7C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,QAAQ,GAA8B;gBAC1C,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aACjD,CAAC;YACF,MAAM,aAAa,GAAG,wBAAwB,CAC5C,QAAQ,EACR,WAAW,EACX,YAAY,EACZ,YAAY,CACb,CAAC;YACF,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBAC7C,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE;gBAC5C,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE;aAC7C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,QAAQ,GAA8B;gBAC1C,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,OAAO;gBACjB,MAAM,EAAE;oBACN,iBAAiB,EAAE,8BAA8B;iBAClD;aACF,CAAC;YACF,MAAM,aAAa,GAAG,wBAAwB,CAC5C,QAAQ,EACR,WAAW,EACX,YAAY,EACZ,YAAY,CACb,CAAC;YACF,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC;gBACtD,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;aAClD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,QAAQ,GAA8B;gBAC1C,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,OAAO;gBACjB,MAAM,EAAE;oBACN,WAAW,EAAE,GAAG;oBAChB,IAAI,EAAE,EAAE;iBACT;aACF,CAAC;YACF,MAAM,aAAa,GAAG,wBAAwB,CAC5C,QAAQ,EACR,WAAW,EACX,YAAY,EACZ,YAAY,CACb,CAAC;YACF,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC;gBACrD,WAAW,EAAE,GAAG;gBAChB,IAAI,EAAE,EAAE;aACT,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,QAAQ,GAA8B;gBAC1C,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,OAAO;gBACjB,MAAM,EAAE;oBACN,WAAW,EAAE,GAAG;oBAChB,IAAI,EAAE,GAAG;oBACT,IAAI,EAAE,CAAC;oBACP,cAAc,EAAE,CAAC;oBACjB,eAAe,EAAE,CAAC;oBAClB,aAAa,EAAE,CAAC,GAAG,CAAC;oBACpB,gBAAgB,EAAE,IAAI;oBACtB,QAAQ,EAAE,CAAC;oBACX,eAAe,EAAE,GAAG;oBACpB,gBAAgB,EAAE,GAAG;oBACrB,IAAI,EAAE,CAAC;oBACP,gBAAgB,EAAE,kBAAkB;iBACrC;aACF,CAAC;YACF,MAAM,aAAa,GAAG,wBAAwB,CAC5C,QAAQ,EACR,WAAW,EACX,YAAY,EACZ,YAAY,CACb,CAAC;YACF,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC;gBACrD,WAAW,EAAE,GAAG;gBAChB,IAAI,EAAE,GAAG;gBACT,IAAI,EAAE,CAAC;gBACP,cAAc,EAAE,CAAC;gBACjB,eAAe,EAAE,CAAC;gBAClB,aAAa,EAAE,CAAC,GAAG,CAAC;gBACpB,gBAAgB,EAAE,IAAI;gBACtB,QAAQ,EAAE,CAAC;gBACX,eAAe,EAAE,GAAG;gBACpB,gBAAgB,EAAE,GAAG;gBACrB,IAAI,EAAE,CAAC;gBACP,gBAAgB,EAAE,kBAAkB;aACrC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,aAAa,GAA8B;gBAC/C,QAAQ,EAAE;oBACR,UAAU,EAAE;wBACV;4BACE,KAAK,EAAE,CAAC;4BACR,OAAO,EAAE;gCACP,IAAI,EAAE,OAAO;gCACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;6BAC/B;4BACD,YAAY,EAAE,YAAY,CAAC,IAAI;4BAC/B,aAAa,EAAE,EAAE;yBAClB;qBACF;iBACF;aACF,CAAC;YACF,MAAM,QAAQ,GAAG,2BAA2B,CAAC,aAAa,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,uBAAuB,CAAC,CAAC;YACzD,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,aAAa,GAA8B;gBAC/C,QAAQ,EAAE;oBACR,UAAU,EAAE,EAAE;oBACd,cAAc,EAAE;wBACd,WAAW,EAAE,aAAa,CAAC,MAAM;wBACjC,aAAa,EAAE,EAAE;qBAClB;oBACD,aAAa,EAAE;wBACb,gBAAgB,EAAE,EAAE;wBACpB,oBAAoB,EAAE,EAAE;wBACxB,eAAe,EAAE,EAAE;qBACpB;iBACF;aACF,CAAC;YACF,MAAM,QAAQ,GAAG,2BAA2B,CAAC,aAAa,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,OAAO,CACrC,aAAa,CAAC,QAAQ,CAAC,cAAc,CACtC,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,OAAO,CACpC,aAAa,CAAC,QAAQ,CAAC,aAAa,CACrC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,aAAa,GAA8B;gBAC/C,QAAQ,EAAE;oBACR,UAAU,EAAE,EAAE;oBACd,+BAA+B,EAAE;wBAC/B;4BACE,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE;gCACL;oCACE,YAAY,EAAE;wCACZ,IAAI,EAAE,eAAe;wCACrB,IAAI,EAAE;4CACJ,GAAG,EAAE,KAAK;yCACX;qCACF;iCACF;6BACF;yBACF;qBACF;iBACF;aACF,CAAC;YACF,MAAM,QAAQ,GAAG,2BAA2B,CAAC,aAAa,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,+BAA+B,CAAC,CAAC,OAAO,CACtD,aAAa,CAAC,QAAQ,CAAC,+BAA+B,CACvD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;YAC/B,MAAM,OAAO,GAAqB;gBAChC,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC3B,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;gBAClC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE;aAC7C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,QAAQ,GAAqB;gBACjC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE;gBAC5C,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE;aAC3C,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;gBACnC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE;gBAC5C,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE;aAC3C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;YAC5B,MAAM,IAAI,GAAqB,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YAClD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;gBAC/B,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE;aAC9C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,KAAK,GAAG,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,QAAQ,CAAC,CAAC;YAC7C,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;gBAChC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE;gBAC7C,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE;aAC9C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;YAC9B,MAAM,GAAG,GAAqB,UAAU,CAAC;YACzC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC;gBAC9B,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE;aAChD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,OAAO,GAAqB,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YAC3D,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;gBAClC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE;gBAC/C,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE;aAChD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/code_assist/oauth2.test.d.ts

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

package/dist/src/code_assist/oauth2.test.js

@@ -1,370 +0,0 @@
-/**
- * @license
- * Copyright 2025 Vybestack LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-import { describe, it, expect, vi, beforeEach, afterEach, } from 'vitest';
-import { getOauthClient } from './oauth2.js';
-import { getCachedGoogleAccount } from '../utils/user_account.js';
-import { OAuth2Client, Compute } from 'google-auth-library';
-import * as fs from 'fs';
-import * as path from 'path';
-import http from 'http';
-import open from 'open';
-import crypto from 'crypto';
-import * as os from 'os';
-import { AuthType } from '../core/contentGenerator.js';
-import readline from 'node:readline';
-vi.mock('os', async (importOriginal) => {
-    const os = await importOriginal();
-    return {
-        ...os,
-        homedir: vi.fn(),
-    };
-});
-vi.mock('google-auth-library');
-vi.mock('http');
-vi.mock('open');
-vi.mock('crypto');
-vi.mock('node:readline');
-const mockConfig = {
-    getNoBrowser: () => false,
-    getProxy: () => 'http://test.proxy.com:8080',
-    isBrowserLaunchSuppressed: () => false,
-};
-// Mock fetch globally
-global.fetch = vi.fn();
-describe('oauth2', () => {
-    let tempHomeDir;
-    let processExitSpy;
-    beforeEach(() => {
-        // Mock process.exit to prevent tests from actually exiting
-        processExitSpy = vi.spyOn(process, 'exit').mockImplementation((() => {
-            throw new Error(`process.exit unexpectedly called`);
-        }));
-        tempHomeDir = fs.mkdtempSync(path.join(os.tmpdir(), 'gemini-cli-test-home-'));
-        vi.mocked(os.homedir).mockReturnValue(tempHomeDir);
-    });
-    afterEach(() => {
-        fs.rmSync(tempHomeDir, { recursive: true, force: true });
-        vi.clearAllMocks();
-        delete process.env.CLOUD_SHELL;
-        delete process.env.GOOGLE_GENAI_USE_GCA;
-        delete process.env.GOOGLE_CLOUD_ACCESS_TOKEN;
-        processExitSpy.mockRestore();
-    });
-    it.skip('should perform a web login', { timeout: 20000 }, async () => {
-        const mockAuthUrl = 'https://example.com/auth';
-        const mockCode = 'test-code';
-        const mockState = 'test-state';
-        const mockTokens = {
-            access_token: 'test-access-token',
-            refresh_token: 'test-refresh-token',
-        };
-        const mockGenerateAuthUrl = vi.fn().mockReturnValue(mockAuthUrl);
-        const mockGetToken = vi.fn().mockResolvedValue({ tokens: mockTokens });
-        const mockSetCredentials = vi.fn();
-        const mockGetAccessToken = vi
-            .fn()
-            .mockResolvedValue({ token: 'mock-access-token' });
-        const mockGenerateCodeVerifierAsync = vi
-            .fn()
-            .mockResolvedValue('mock-code-verifier');
-        const mockOAuth2Client = {
-            generateAuthUrl: mockGenerateAuthUrl,
-            getToken: mockGetToken,
-            setCredentials: mockSetCredentials,
-            getAccessToken: mockGetAccessToken,
-            generateCodeVerifierAsync: mockGenerateCodeVerifierAsync,
-            credentials: mockTokens,
-            on: vi.fn(),
-        };
-        vi.mocked(OAuth2Client).mockImplementation(() => mockOAuth2Client);
-        vi.spyOn(crypto, 'randomBytes').mockReturnValue(mockState);
-        // Mock open to return a proper child process mock
-        const mockChildProcess = {
-            on: vi.fn(),
-            pid: 12345,
-        };
-        vi.mocked(open).mockImplementation(async () => mockChildProcess);
-        // Mock readline in case it falls back to user code
-        const mockReadline = {
-            question: vi.fn((_query, callback) => callback(mockCode)),
-            close: vi.fn(),
-        };
-        vi.mocked(readline.createInterface).mockReturnValue(mockReadline);
-        // Mock the UserInfo API response
-        vi.mocked(global.fetch).mockResolvedValue({
-            ok: true,
-            json: vi
-                .fn()
-                .mockResolvedValue({ email: 'test-google-account@gmail.com' }),
-        });
-        let requestCallback;
-        let serverListeningCallback;
-        const serverListeningPromise = new Promise((resolve) => (serverListeningCallback = resolve));
-        let capturedPort = 0;
-        const mockHttpServer = {
-            listen: vi.fn((port, host, callback) => {
-                capturedPort = port;
-                // The callback might be passed as second parameter when host is omitted
-                if (typeof host === 'function') {
-                    callback = host;
-                }
-                if (callback) {
-                    callback();
-                }
-                serverListeningCallback(undefined);
-            }),
-            close: vi.fn((callback) => {
-                if (callback) {
-                    callback();
-                }
-            }),
-            on: vi.fn(),
-            address: () => ({ port: capturedPort }),
-        };
-        vi.mocked(http.createServer).mockImplementation((cb) => {
-            requestCallback = cb;
-            return mockHttpServer;
-        });
-        const clientPromise = getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
-        // wait for server to start listening.
-        await serverListeningPromise;
-        const mockReq = {
-            url: `/oauth2callback?code=${mockCode}&state=${mockState}`,
-        };
-        const mockRes = {
-            writeHead: vi.fn(),
-            end: vi.fn(),
-        };
-        await requestCallback(mockReq, mockRes);
-        const client = await clientPromise;
-        expect(client).toBe(mockOAuth2Client);
-        expect(open).toHaveBeenCalledWith(mockAuthUrl);
-        expect(mockGetToken).toHaveBeenCalledWith({
-            code: mockCode,
-            redirect_uri: `http://localhost:${capturedPort}/oauth2callback`,
-        });
-        expect(mockSetCredentials).toHaveBeenCalledWith(mockTokens);
-        // Verify Google Account was cached
-        const googleAccountPath = path.join(tempHomeDir, '.llxprt', 'google_accounts.json');
-        expect(fs.existsSync(googleAccountPath)).toBe(true);
-        const cachedGoogleAccount = fs.readFileSync(googleAccountPath, 'utf-8');
-        expect(JSON.parse(cachedGoogleAccount)).toEqual({
-            active: 'test-google-account@gmail.com',
-            old: [],
-        });
-        // Verify the getCachedGoogleAccount function works
-        expect(getCachedGoogleAccount()).toBe('test-google-account@gmail.com');
-    });
-    it.skip('should perform login with user code', { timeout: 30000 }, async () => {
-        const mockConfigWithNoBrowser = {
-            getNoBrowser: () => true,
-            getProxy: () => 'http://test.proxy.com:8080',
-            isBrowserLaunchSuppressed: () => true,
-        };
-        const mockCodeVerifier = {
-            codeChallenge: 'test-challenge',
-            codeVerifier: 'test-verifier',
-        };
-        const mockAuthUrl = 'https://example.com/auth-user-code';
-        const mockCode = 'test-user-code';
-        const mockTokens = {
-            access_token: 'test-access-token-user-code',
-            refresh_token: 'test-refresh-token-user-code',
-        };
-        const mockGenerateAuthUrl = vi.fn().mockReturnValue(mockAuthUrl);
-        const mockGetToken = vi.fn().mockResolvedValue({ tokens: mockTokens });
-        const mockSetCredentials = vi.fn();
-        const mockGenerateCodeVerifierAsync = vi
-            .fn()
-            .mockResolvedValue(mockCodeVerifier);
-        const mockGetAccessToken = vi
-            .fn()
-            .mockResolvedValue({ token: 'mock-access-token-user-code' });
-        const mockOAuth2Client = {
-            generateAuthUrl: mockGenerateAuthUrl,
-            getToken: mockGetToken,
-            setCredentials: mockSetCredentials,
-            generateCodeVerifierAsync: mockGenerateCodeVerifierAsync,
-            getAccessToken: mockGetAccessToken,
-            credentials: mockTokens,
-            on: vi.fn(),
-        };
-        vi.mocked(OAuth2Client).mockImplementation(() => mockOAuth2Client);
-        // Mock open to immediately reject to simulate browser not available
-        vi.mocked(open).mockRejectedValue(new Error('Browser not available'));
-        const mockReadline = {
-            question: vi.fn((_query, callback) => callback(mockCode)),
-            close: vi.fn(),
-        };
-        vi.mocked(readline.createInterface).mockReturnValue(mockReadline);
-        const consoleLogSpy = vi
-            .spyOn(console, 'log')
-            .mockImplementation(() => { });
-        // Mock the UserInfo API response
-        vi.mocked(global.fetch).mockResolvedValue({
-            ok: true,
-            json: vi.fn().mockResolvedValue({ email: 'test-user-code@gmail.com' }),
-        });
-        const client = await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfigWithNoBrowser);
-        expect(client).toBe(mockOAuth2Client);
-        // Verify the auth flow
-        expect(mockGenerateCodeVerifierAsync).toHaveBeenCalled();
-        expect(mockGenerateAuthUrl).toHaveBeenCalled();
-        expect(consoleLogSpy).toHaveBeenCalledWith(expect.stringContaining(mockAuthUrl));
-        expect(mockReadline.question).toHaveBeenCalledWith('Enter the authorization code: ', expect.any(Function));
-        expect(mockGetToken).toHaveBeenCalledWith({
-            code: mockCode,
-            codeVerifier: mockCodeVerifier.codeVerifier,
-            redirect_uri: 'https://codeassist.google.com/authcode',
-        });
-        expect(mockSetCredentials).toHaveBeenCalledWith(mockTokens);
-        consoleLogSpy.mockRestore();
-    });
-    describe('in Cloud Shell', () => {
-        const mockGetAccessToken = vi.fn();
-        let mockComputeClient;
-        beforeEach(() => {
-            mockGetAccessToken.mockResolvedValue({ token: 'test-access-token' });
-            mockComputeClient = {
-                credentials: { refresh_token: 'test-refresh-token' },
-                getAccessToken: mockGetAccessToken,
-            };
-            vi.mocked(Compute).mockImplementation(() => mockComputeClient);
-        });
-        it('should attempt to load cached credentials first', async () => {
-            const cachedCreds = { refresh_token: 'cached-token' };
-            const credsPath = path.join(tempHomeDir, '.llxprt', 'oauth_creds.json');
-            await fs.promises.mkdir(path.dirname(credsPath), { recursive: true });
-            await fs.promises.writeFile(credsPath, JSON.stringify(cachedCreds));
-            const mockClient = {
-                setCredentials: vi.fn(),
-                getAccessToken: vi.fn().mockResolvedValue({ token: 'test-token' }),
-                getTokenInfo: vi.fn().mockResolvedValue({}),
-                on: vi.fn(),
-            };
-            // To mock the new OAuth2Client() inside the function
-            vi.mocked(OAuth2Client).mockImplementation(() => mockClient);
-            await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
-            expect(mockClient.setCredentials).toHaveBeenCalledWith(cachedCreds);
-            expect(mockClient.getAccessToken).toHaveBeenCalled();
-            expect(mockClient.getTokenInfo).toHaveBeenCalled();
-            expect(Compute).not.toHaveBeenCalled(); // Should not fetch new client if cache is valid
-        });
-        it('should use Compute to get a client if no cached credentials exist', async () => {
-            await getOauthClient(AuthType.CLOUD_SHELL, mockConfig);
-            expect(Compute).toHaveBeenCalledWith({});
-            expect(mockGetAccessToken).toHaveBeenCalled();
-        });
-        it('should not cache the credentials after fetching them via ADC', async () => {
-            const newCredentials = { refresh_token: 'new-adc-token' };
-            mockComputeClient.credentials = newCredentials;
-            mockGetAccessToken.mockResolvedValue({ token: 'new-adc-token' });
-            await getOauthClient(AuthType.CLOUD_SHELL, mockConfig);
-            const credsPath = path.join(tempHomeDir, '.llxprt', 'oauth_creds.json');
-            expect(fs.existsSync(credsPath)).toBe(false);
-        });
-        it('should return the Compute client on successful ADC authentication', async () => {
-            const client = await getOauthClient(AuthType.CLOUD_SHELL, mockConfig);
-            expect(client).toBe(mockComputeClient);
-        });
-        it('should throw an error if ADC fails', async () => {
-            const testError = new Error('ADC Failed');
-            mockGetAccessToken.mockRejectedValue(testError);
-            await expect(getOauthClient(AuthType.CLOUD_SHELL, mockConfig)).rejects.toThrow('Could not authenticate using Cloud Shell credentials. Please select a different authentication method or ensure you are in a properly configured environment. Error: ADC Failed');
-        });
-    });
-    describe('with GCP environment variables', () => {
-        it('should use GOOGLE_CLOUD_ACCESS_TOKEN when GOOGLE_GENAI_USE_GCA is true', async () => {
-            process.env.GOOGLE_GENAI_USE_GCA = 'true';
-            process.env.GOOGLE_CLOUD_ACCESS_TOKEN = 'gcp-access-token';
-            const mockSetCredentials = vi.fn();
-            const mockGetAccessToken = vi
-                .fn()
-                .mockResolvedValue({ token: 'gcp-access-token' });
-            const mockOAuth2Client = {
-                setCredentials: mockSetCredentials,
-                getAccessToken: mockGetAccessToken,
-                on: vi.fn(),
-            };
-            vi.mocked(OAuth2Client).mockImplementation(() => mockOAuth2Client);
-            // Mock the UserInfo API response for fetchAndCacheUserInfo
-            vi.mocked(global.fetch).mockResolvedValue({
-                ok: true,
-                json: vi
-                    .fn()
-                    .mockResolvedValue({ email: 'test-gcp-account@gmail.com' }),
-            });
-            const client = await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
-            expect(client).toBe(mockOAuth2Client);
-            expect(mockSetCredentials).toHaveBeenCalledWith({
-                access_token: 'gcp-access-token',
-            });
-            // Verify fetchAndCacheUserInfo was effectively called
-            expect(mockGetAccessToken).toHaveBeenCalled();
-            expect(global.fetch).toHaveBeenCalledWith('https://www.googleapis.com/oauth2/v2/userinfo', {
-                headers: {
-                    Authorization: 'Bearer gcp-access-token',
-                },
-            });
-            // Verify Google Account was cached
-            const googleAccountPath = path.join(tempHomeDir, '.llxprt', 'google_accounts.json');
-            const cachedContent = fs.readFileSync(googleAccountPath, 'utf-8');
-            expect(JSON.parse(cachedContent)).toEqual({
-                active: 'test-gcp-account@gmail.com',
-                old: [],
-            });
-        });
-        it('should not use GCP token if GOOGLE_CLOUD_ACCESS_TOKEN is not set', async () => {
-            process.env.GOOGLE_GENAI_USE_GCA = 'true';
-            const mockSetCredentials = vi.fn();
-            const mockGetAccessToken = vi
-                .fn()
-                .mockResolvedValue({ token: 'cached-access-token' });
-            const mockGetTokenInfo = vi.fn().mockResolvedValue({});
-            const mockOAuth2Client = {
-                setCredentials: mockSetCredentials,
-                getAccessToken: mockGetAccessToken,
-                getTokenInfo: mockGetTokenInfo,
-                on: vi.fn(),
-            };
-            vi.mocked(OAuth2Client).mockImplementation(() => mockOAuth2Client);
-            // Make it fall through to cached credentials path
-            const cachedCreds = { refresh_token: 'cached-token' };
-            const credsPath = path.join(tempHomeDir, '.llxprt', 'oauth_creds.json');
-            await fs.promises.mkdir(path.dirname(credsPath), { recursive: true });
-            await fs.promises.writeFile(credsPath, JSON.stringify(cachedCreds));
-            await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
-            // It should be called with the cached credentials, not the GCP access token.
-            expect(mockSetCredentials).toHaveBeenCalledTimes(1);
-            expect(mockSetCredentials).toHaveBeenCalledWith(cachedCreds);
-        });
-        it('should not use GCP token if GOOGLE_GENAI_USE_GCA is not set', async () => {
-            process.env.GOOGLE_CLOUD_ACCESS_TOKEN = 'gcp-access-token';
-            const mockSetCredentials = vi.fn();
-            const mockGetAccessToken = vi
-                .fn()
-                .mockResolvedValue({ token: 'cached-access-token' });
-            const mockGetTokenInfo = vi.fn().mockResolvedValue({});
-            const mockOAuth2Client = {
-                setCredentials: mockSetCredentials,
-                getAccessToken: mockGetAccessToken,
-                getTokenInfo: mockGetTokenInfo,
-                on: vi.fn(),
-            };
-            vi.mocked(OAuth2Client).mockImplementation(() => mockOAuth2Client);
-            // Make it fall through to cached credentials path
-            const cachedCreds = { refresh_token: 'cached-token' };
-            const credsPath = path.join(tempHomeDir, '.llxprt', 'oauth_creds.json');
-            await fs.promises.mkdir(path.dirname(credsPath), { recursive: true });
-            await fs.promises.writeFile(credsPath, JSON.stringify(cachedCreds));
-            await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
-            // It should be called with the cached credentials, not the GCP access token.
-            expect(mockSetCredentials).toHaveBeenCalledTimes(1);
-            expect(mockSetCredentials).toHaveBeenCalledWith(cachedCreds);
-        });
-    });
-});
-//# sourceMappingURL=oauth2.test.js.map