@blackbox_ai/blackbox-cli-core 0.0.4

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

@@ -0,0 +1,362 @@
+/**
+ * @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' }] },
+            ]);
+        });
+        it('should convert thought parts to text parts for API compatibility', () => {
+            const contentWithThought = {
+                role: 'model',
+                parts: [
+                    { text: 'regular text' },
+                    { thought: 'thinking about the problem' },
+                    { text: 'more text' },
+                ],
+            };
+            expect(toContents(contentWithThought)).toEqual([
+                {
+                    role: 'model',
+                    parts: [
+                        { text: 'regular text' },
+                        { text: '[Thought: thinking about the problem]' },
+                        { text: 'more text' },
+                    ],
+                },
+            ]);
+        });
+        it('should combine text and thought for text parts with thoughts', () => {
+            const contentWithTextAndThought = {
+                role: 'model',
+                parts: [
+                    {
+                        text: 'Here is my response',
+                        thought: 'I need to be careful here',
+                    },
+                ],
+            };
+            expect(toContents(contentWithTextAndThought)).toEqual([
+                {
+                    role: 'model',
+                    parts: [
+                        {
+                            text: 'Here is my response\n[Thought: I need to be careful here]',
+                        },
+                    ],
+                },
+            ]);
+        });
+        it('should preserve non-thought properties while removing thought', () => {
+            const contentWithComplexPart = {
+                role: 'model',
+                parts: [
+                    {
+                        functionCall: { name: 'calculate', args: { x: 5, y: 10 } },
+                        thought: 'Performing calculation',
+                    },
+                ],
+            };
+            expect(toContents(contentWithComplexPart)).toEqual([
+                {
+                    role: 'model',
+                    parts: [
+                        {
+                            functionCall: { name: 'calculate', args: { x: 5, y: 10 } },
+                        },
+                    ],
+                },
+            ]);
+        });
+        it('should convert invalid text content to valid text part with thought', () => {
+            const contentWithInvalidText = {
+                role: 'model',
+                parts: [
+                    {
+                        text: 123, // Invalid - should be string
+                        thought: 'Processing number',
+                    },
+                ],
+            };
+            expect(toContents(contentWithInvalidText)).toEqual([
+                {
+                    role: 'model',
+                    parts: [
+                        {
+                            text: '123\n[Thought: Processing number]',
+                        },
+                    ],
+                },
+            ]);
+        });
+    });
+});
+//# sourceMappingURL=converter.test.js.map

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

@@ -0,0 +1 @@
+{"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;AAE9C,OAAO,EACL,wBAAwB,EACxB,2BAA2B,EAC3B,UAAU,GACX,MAAM,gBAAgB,CAAC;AAKxB,OAAO,EACL,uBAAuB,EACvB,YAAY,EACZ,aAAa,GAEd,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;QAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YAC1E,MAAM,kBAAkB,GAAqB;gBAC3C,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,EAAE,IAAI,EAAE,cAAc,EAAE;oBACxB,EAAE,OAAO,EAAE,4BAA4B,EAEtC;oBACD,EAAE,IAAI,EAAE,WAAW,EAAE;iBACtB;aACF,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,CAAC,OAAO,CAAC;gBAC7C;oBACE,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE;wBACL,EAAE,IAAI,EAAE,cAAc,EAAE;wBACxB,EAAE,IAAI,EAAE,uCAAuC,EAAE;wBACjD,EAAE,IAAI,EAAE,WAAW,EAAE;qBACtB;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,MAAM,yBAAyB,GAAqB;gBAClD,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL;wBACE,IAAI,EAAE,qBAAqB;wBAC3B,OAAO,EAAE,2BAA2B;qBACP;iBAChC;aACF,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,yBAAyB,CAAC,CAAC,CAAC,OAAO,CAAC;gBACpD;oBACE,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE;wBACL;4BACE,IAAI,EAAE,2DAA2D;yBAClE;qBACF;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,sBAAsB,GAAqB;gBAC/C,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL;wBACE,YAAY,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE;wBAC1D,OAAO,EAAE,wBAAwB;qBACJ;iBAChC;aACF,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,CAAC,OAAO,CAAC;gBACjD;oBACE,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE;wBACL;4BACE,YAAY,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE;yBAC3D;qBACF;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;YAC7E,MAAM,sBAAsB,GAAqB;gBAC/C,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL;wBACE,IAAI,EAAE,GAAG,EAAE,6BAA6B;wBACxC,OAAO,EAAE,mBAAmB;qBACe;iBAC9C;aACF,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,CAAC,OAAO,CAAC;gBACjD;oBACE,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE;wBACL;4BACE,IAAI,EAAE,mCAAmC;yBAC1C;qBACF;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,22 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { OAuth2Client } from 'google-auth-library';
+import type { Config } from '../config/config.js';
+import { AuthType } from '../core/contentGenerator.js';
+/**
+ * An Authentication URL for updating the credentials of a Oauth2Client
+ * as well as a promise that will resolve when the credentials have
+ * been refreshed (or which throws error when refreshing credentials failed).
+ */
+export interface OauthWebLogin {
+    authUrl: string;
+    loginCompletePromise: Promise<void>;
+}
+export declare function getOauthClient(authType: AuthType, config: Config): Promise<OAuth2Client>;
+export declare function getAvailablePort(): Promise<number>;
+export declare function clearOauthClientCache(): void;
+export declare function clearCachedCredentialFile(): Promise<void>;
+export declare function resetOauthClientForTesting(): void;