@vybestack/llxprt-code-core 0.1.12

package/dist/index.d.ts

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export * from './src/index.js';
+export { DEFAULT_GEMINI_MODEL, DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_EMBEDDING_MODEL, } from './src/config/models.js';

package/dist/index.js

@@ -0,0 +1,8 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export * from './src/index.js';
+export { DEFAULT_GEMINI_MODEL, DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_EMBEDDING_MODEL, } from './src/config/models.js';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,cAAc,gBAAgB,CAAC;AAC/B,OAAO,EACL,oBAAoB,EACpB,0BAA0B,EAC1B,8BAA8B,GAC/B,MAAM,wBAAwB,CAAC"}

package/dist/src/__mocks__/fs/promises.d.ts

@@ -0,0 +1,11 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import * as actualFsPromises from 'node:fs/promises';
+export declare const mockControl: {
+    mockReadFile: import("vitest").Mock<(...args: any[]) => any>;
+};
+export declare const access: typeof actualFsPromises.access, appendFile: typeof actualFsPromises.appendFile, chmod: typeof actualFsPromises.chmod, chown: typeof actualFsPromises.chown, copyFile: typeof actualFsPromises.copyFile, cp: typeof actualFsPromises.cp, lchmod: typeof actualFsPromises.lchmod, lchown: typeof actualFsPromises.lchown, link: typeof actualFsPromises.link, lstat: typeof actualFsPromises.lstat, mkdir: typeof actualFsPromises.mkdir, open: typeof actualFsPromises.open, opendir: typeof actualFsPromises.opendir, readdir: typeof actualFsPromises.readdir, readlink: typeof actualFsPromises.readlink, realpath: typeof actualFsPromises.realpath, rename: typeof actualFsPromises.rename, rmdir: typeof actualFsPromises.rmdir, rm: typeof actualFsPromises.rm, stat: typeof actualFsPromises.stat, symlink: typeof actualFsPromises.symlink, truncate: typeof actualFsPromises.truncate, unlink: typeof actualFsPromises.unlink, utimes: typeof actualFsPromises.utimes, watch: typeof actualFsPromises.watch, writeFile: typeof actualFsPromises.writeFile;
+export declare const readFile: import("vitest").Mock<(...args: any[]) => any>;

package/dist/src/__mocks__/fs/promises.js

@@ -0,0 +1,17 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi } from 'vitest';
+import * as actualFsPromises from 'node:fs/promises';
+const readFileMock = vi.fn();
+// Export a control object so tests can access and manipulate the mock
+export const mockControl = {
+    mockReadFile: readFileMock,
+};
+// Export all other functions from the actual fs/promises module
+export const { access, appendFile, chmod, chown, copyFile, cp, lchmod, lchown, link, lstat, mkdir, open, opendir, readdir, readlink, realpath, rename, rmdir, rm, stat, symlink, truncate, unlink, utimes, watch, writeFile, } = actualFsPromises;
+// Override readFile with our mock
+export const readFile = readFileMock;
+//# sourceMappingURL=promises.js.map

package/dist/src/__mocks__/fs/promises.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"promises.js","sourceRoot":"","sources":["../../../../src/__mocks__/fs/promises.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC5B,OAAO,KAAK,gBAAgB,MAAM,kBAAkB,CAAC;AAErD,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAE7B,sEAAsE;AACtE,MAAM,CAAC,MAAM,WAAW,GAAG;IACzB,YAAY,EAAE,YAAY;CAC3B,CAAC;AAEF,gEAAgE;AAChE,MAAM,CAAC,MAAM,EACX,MAAM,EACN,UAAU,EACV,KAAK,EACL,KAAK,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,MAAM,EACN,IAAI,EACJ,KAAK,EACL,KAAK,EACL,IAAI,EACJ,OAAO,EACP,OAAO,EACP,QAAQ,EACR,QAAQ,EACR,MAAM,EACN,KAAK,EACL,EAAE,EACF,IAAI,EACJ,OAAO,EACP,QAAQ,EACR,MAAM,EACN,MAAM,EACN,KAAK,EACL,SAAS,GACV,GAAG,gBAAgB,CAAC;AAErB,kCAAkC;AAClC,MAAM,CAAC,MAAM,QAAQ,GAAG,YAAY,CAAC"}

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

@@ -0,0 +1,9 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AuthType, ContentGenerator } from '../core/contentGenerator.js';
+import { HttpOptions } from './server.js';
+import { Config } from '../config/config.js';
+export declare function createCodeAssistContentGenerator(httpOptions: HttpOptions, authType: AuthType, config: Config, sessionId?: string): Promise<ContentGenerator>;

package/dist/src/code_assist/codeAssist.js

@@ -0,0 +1,19 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AuthType } from '../core/contentGenerator.js';
+import { getOauthClient } from './oauth2.js';
+import { setupUser } from './setup.js';
+import { CodeAssistServer } from './server.js';
+export async function createCodeAssistContentGenerator(httpOptions, authType, config, sessionId) {
+    if (authType === AuthType.LOGIN_WITH_GOOGLE ||
+        authType === AuthType.CLOUD_SHELL) {
+        const authClient = await getOauthClient(authType, config);
+        const projectId = await setupUser(authClient);
+        return new CodeAssistServer(authClient, projectId, httpOptions, sessionId);
+    }
+    throw new Error(`Unsupported authType: ${authType}`);
+}
+//# sourceMappingURL=codeAssist.js.map

package/dist/src/code_assist/codeAssist.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codeAssist.js","sourceRoot":"","sources":["../../../src/code_assist/codeAssist.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAoB,MAAM,6BAA6B,CAAC;AACzE,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AACvC,OAAO,EAAE,gBAAgB,EAAe,MAAM,aAAa,CAAC;AAG5D,MAAM,CAAC,KAAK,UAAU,gCAAgC,CACpD,WAAwB,EACxB,QAAkB,EAClB,MAAc,EACd,SAAkB;IAElB,IACE,QAAQ,KAAK,QAAQ,CAAC,iBAAiB;QACvC,QAAQ,KAAK,QAAQ,CAAC,WAAW,EACjC,CAAC;QACD,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC1D,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,UAAU,CAAC,CAAC;QAC9C,OAAO,IAAI,gBAAgB,CAAC,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;IAC7E,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,EAAE,CAAC,CAAC;AACvD,CAAC"}

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

@@ -0,0 +1,68 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Content, GenerateContentParameters, CountTokensParameters, CountTokensResponse, GenerateContentResponse, GenerationConfigRoutingConfig, MediaResolution, Candidate, ModelSelectionConfig, GenerateContentResponsePromptFeedback, GenerateContentResponseUsageMetadata, SafetySetting, SchemaUnion, SpeechConfigUnion, ThinkingConfig, ToolListUnion, ToolConfig } from '@google/genai';
+export interface CAGenerateContentRequest {
+    model: string;
+    project?: string;
+    request: VertexGenerateContentRequest;
+}
+interface VertexGenerateContentRequest {
+    contents: Content[];
+    systemInstruction?: Content;
+    cachedContent?: string;
+    tools?: ToolListUnion;
+    toolConfig?: ToolConfig;
+    labels?: Record<string, string>;
+    safetySettings?: SafetySetting[];
+    generationConfig?: VertexGenerationConfig;
+    session_id?: string;
+}
+interface VertexGenerationConfig {
+    temperature?: number;
+    topP?: number;
+    topK?: number;
+    candidateCount?: number;
+    maxOutputTokens?: number;
+    stopSequences?: string[];
+    responseLogprobs?: boolean;
+    logprobs?: number;
+    presencePenalty?: number;
+    frequencyPenalty?: number;
+    seed?: number;
+    responseMimeType?: string;
+    responseSchema?: SchemaUnion;
+    routingConfig?: GenerationConfigRoutingConfig;
+    modelSelectionConfig?: ModelSelectionConfig;
+    responseModalities?: string[];
+    mediaResolution?: MediaResolution;
+    speechConfig?: SpeechConfigUnion;
+    audioTimestamp?: boolean;
+    thinkingConfig?: ThinkingConfig;
+}
+export interface CaGenerateContentResponse {
+    response: VertexGenerateContentResponse;
+}
+interface VertexGenerateContentResponse {
+    candidates: Candidate[];
+    automaticFunctionCallingHistory?: Content[];
+    promptFeedback?: GenerateContentResponsePromptFeedback;
+    usageMetadata?: GenerateContentResponseUsageMetadata;
+}
+export interface CaCountTokenRequest {
+    request: VertexCountTokenRequest;
+}
+interface VertexCountTokenRequest {
+    model: string;
+    contents: Content[];
+}
+export interface CaCountTokenResponse {
+    totalTokens: number;
+}
+export declare function toCountTokenRequest(req: CountTokensParameters): CaCountTokenRequest;
+export declare function fromCountTokenResponse(res: CaCountTokenResponse): CountTokensResponse;
+export declare function toGenerateContentRequest(req: GenerateContentParameters, project?: string, sessionId?: string): CAGenerateContentRequest;
+export declare function fromGenerateContentResponse(res: CaGenerateContentResponse): GenerateContentResponse;
+export {};

package/dist/src/code_assist/converter.js

@@ -0,0 +1,125 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { GenerateContentResponse, } from '@google/genai';
+export function toCountTokenRequest(req) {
+    return {
+        request: {
+            model: 'models/' + req.model,
+            contents: toContents(req.contents),
+        },
+    };
+}
+export function fromCountTokenResponse(res) {
+    return {
+        totalTokens: res.totalTokens,
+    };
+}
+export function toGenerateContentRequest(req, project, sessionId) {
+    return {
+        model: req.model,
+        project,
+        request: toVertexGenerateContentRequest(req, sessionId),
+    };
+}
+export function fromGenerateContentResponse(res) {
+    const inres = res.response;
+    const out = new GenerateContentResponse();
+    out.candidates = inres.candidates;
+    out.automaticFunctionCallingHistory = inres.automaticFunctionCallingHistory;
+    out.promptFeedback = inres.promptFeedback;
+    out.usageMetadata = inres.usageMetadata;
+    return out;
+}
+function toVertexGenerateContentRequest(req, sessionId) {
+    return {
+        contents: toContents(req.contents),
+        systemInstruction: maybeToContent(req.config?.systemInstruction),
+        cachedContent: req.config?.cachedContent,
+        tools: req.config?.tools,
+        toolConfig: req.config?.toolConfig,
+        labels: req.config?.labels,
+        safetySettings: req.config?.safetySettings,
+        generationConfig: toVertexGenerationConfig(req.config),
+        session_id: sessionId,
+    };
+}
+function toContents(contents) {
+    if (Array.isArray(contents)) {
+        // it's a Content[] or a PartsUnion[]
+        return contents.map(toContent);
+    }
+    // it's a Content or a PartsUnion
+    return [toContent(contents)];
+}
+function maybeToContent(content) {
+    if (!content) {
+        return undefined;
+    }
+    return toContent(content);
+}
+function toContent(content) {
+    if (Array.isArray(content)) {
+        // it's a PartsUnion[]
+        return {
+            role: 'user',
+            parts: toParts(content),
+        };
+    }
+    if (typeof content === 'string') {
+        // it's a string
+        return {
+            role: 'user',
+            parts: [{ text: content }],
+        };
+    }
+    if ('parts' in content) {
+        // it's a Content
+        return content;
+    }
+    // it's a Part
+    return {
+        role: 'user',
+        parts: [content],
+    };
+}
+function toParts(parts) {
+    return parts.map(toPart);
+}
+function toPart(part) {
+    if (typeof part === 'string') {
+        // it's a string
+        return { text: part };
+    }
+    return part;
+}
+function toVertexGenerationConfig(config) {
+    if (!config) {
+        return undefined;
+    }
+    return {
+        temperature: config.temperature,
+        topP: config.topP,
+        topK: config.topK,
+        candidateCount: config.candidateCount,
+        maxOutputTokens: config.maxOutputTokens,
+        stopSequences: config.stopSequences,
+        responseLogprobs: config.responseLogprobs,
+        logprobs: config.logprobs,
+        presencePenalty: config.presencePenalty,
+        frequencyPenalty: config.frequencyPenalty,
+        seed: config.seed,
+        responseMimeType: config.responseMimeType,
+        responseSchema: config.responseSchema,
+        routingConfig: config.routingConfig,
+        modelSelectionConfig: config.modelSelectionConfig,
+        responseModalities: config.responseModalities,
+        mediaResolution: config.mediaResolution,
+        speechConfig: config.speechConfig,
+        audioTimestamp: config.audioTimestamp,
+        thinkingConfig: config.thinkingConfig,
+    };
+}
+//# sourceMappingURL=converter.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"converter.js","sourceRoot":"","sources":["../../../src/code_assist/converter.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAQL,uBAAuB,GAexB,MAAM,eAAe,CAAC;AAmEvB,MAAM,UAAU,mBAAmB,CACjC,GAA0B;IAE1B,OAAO;QACL,OAAO,EAAE;YACP,KAAK,EAAE,SAAS,GAAG,GAAG,CAAC,KAAK;YAC5B,QAAQ,EAAE,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC;SACnC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,GAAyB;IAEzB,OAAO;QACL,WAAW,EAAE,GAAG,CAAC,WAAW;KAC7B,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,GAA8B,EAC9B,OAAgB,EAChB,SAAkB;IAElB,OAAO;QACL,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,OAAO;QACP,OAAO,EAAE,8BAA8B,CAAC,GAAG,EAAE,SAAS,CAAC;KACxD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,GAA8B;IAE9B,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,CAAC;IAC3B,MAAM,GAAG,GAAG,IAAI,uBAAuB,EAAE,CAAC;IAC1C,GAAG,CAAC,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC;IAClC,GAAG,CAAC,+BAA+B,GAAG,KAAK,CAAC,+BAA+B,CAAC;IAC5E,GAAG,CAAC,cAAc,GAAG,KAAK,CAAC,cAAc,CAAC;IAC1C,GAAG,CAAC,aAAa,GAAG,KAAK,CAAC,aAAa,CAAC;IACxC,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,8BAA8B,CACrC,GAA8B,EAC9B,SAAkB;IAElB,OAAO;QACL,QAAQ,EAAE,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClC,iBAAiB,EAAE,cAAc,CAAC,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC;QAChE,aAAa,EAAE,GAAG,CAAC,MAAM,EAAE,aAAa;QACxC,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,KAAK;QACxB,UAAU,EAAE,GAAG,CAAC,MAAM,EAAE,UAAU;QAClC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM;QAC1B,cAAc,EAAE,GAAG,CAAC,MAAM,EAAE,cAAc;QAC1C,gBAAgB,EAAE,wBAAwB,CAAC,GAAG,CAAC,MAAM,CAAC;QACtD,UAAU,EAAE,SAAS;KACtB,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CAAC,QAA0B;IAC5C,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5B,qCAAqC;QACrC,OAAO,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;IACD,iCAAiC;IACjC,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;AAC/B,CAAC;AAED,SAAS,cAAc,CAAC,OAAsB;IAC5C,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,SAAS,CAAC,OAAO,CAAC,CAAC;AAC5B,CAAC;AAED,SAAS,SAAS,CAAC,OAAqB;IACtC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,sBAAsB;QACtB,OAAO;YACL,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC;SACxB,CAAC;IACJ,CAAC;IACD,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,gBAAgB;QAChB,OAAO;YACL,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;SAC3B,CAAC;IACJ,CAAC;IACD,IAAI,OAAO,IAAI,OAAO,EAAE,CAAC;QACvB,iBAAiB;QACjB,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,cAAc;IACd,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE,CAAC,OAAe,CAAC;KACzB,CAAC;AACJ,CAAC;AAED,SAAS,OAAO,CAAC,KAAkB;IACjC,OAAO,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AAC3B,CAAC;AAED,SAAS,MAAM,CAAC,IAAe;IAC7B,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC7B,gBAAgB;QAChB,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IACxB,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,wBAAwB,CAC/B,MAA8B;IAE9B,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO;QACL,WAAW,EAAE,MAAM,CAAC,WAAW;QAC/B,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,cAAc,EAAE,MAAM,CAAC,cAAc;QACrC,eAAe,EAAE,MAAM,CAAC,eAAe;QACvC,aAAa,EAAE,MAAM,CAAC,aAAa;QACnC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;QACzC,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,eAAe,EAAE,MAAM,CAAC,eAAe;QACvC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;QACzC,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;QACzC,cAAc,EAAE,MAAM,CAAC,cAAc;QACrC,aAAa,EAAE,MAAM,CAAC,aAAa;QACnC,oBAAoB,EAAE,MAAM,CAAC,oBAAoB;QACjD,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;QAC7C,eAAe,EAAE,MAAM,CAAC,eAAe;QACvC,YAAY,EAAE,MAAM,CAAC,YAAY;QACjC,cAAc,EAAE,MAAM,CAAC,cAAc;QACrC,cAAc,EAAE,MAAM,CAAC,cAAc;KACtC,CAAC;AACJ,CAAC"}

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

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

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

@@ -0,0 +1,229 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { toGenerateContentRequest, fromGenerateContentResponse, } 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-project');
+            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: undefined,
+                },
+            });
+        });
+        it('should convert a request without a project', () => {
+            const genaiReq = {
+                model: 'gemini-pro',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            const codeAssistReq = toGenerateContentRequest(genaiReq);
+            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: undefined,
+                },
+            });
+        });
+        it('should convert a request with sessionId', () => {
+            const genaiReq = {
+                model: 'gemini-pro',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            const codeAssistReq = toGenerateContentRequest(genaiReq, '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',
+                },
+            });
+        });
+        it('should handle string content', () => {
+            const genaiReq = {
+                model: 'gemini-pro',
+                contents: 'Hello',
+            };
+            const codeAssistReq = toGenerateContentRequest(genaiReq);
+            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);
+            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);
+            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);
+            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);
+            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);
+        });
+    });
+});
+//# 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;AAC9C,OAAO,EACL,wBAAwB,EACxB,2BAA2B,GAE5B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAEL,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,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;YACvE,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,SAAS;iBACtB;aACF,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,CAAC,QAAQ,CAAC,CAAC;YACzD,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,SAAS;iBACtB;aACF,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,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;aACF,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,CAAC,QAAQ,CAAC,CAAC;YACzD,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,CAAC,QAAQ,CAAC,CAAC;YACzD,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,CAAC,QAAQ,CAAC,CAAC;YACzD,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,CAAC,QAAQ,CAAC,CAAC;YACzD,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,CAAC,QAAQ,CAAC,CAAC;YACzD,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;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,20 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { OAuth2Client } from 'google-auth-library';
+import { 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 clearCachedCredentialFile(): Promise<void>;