@vybestack/llxprt-code-core 0.1.12 → 0.1.13

package/dist/src/providers/gemini/GeminiProvider.d.ts

@@ -0,0 +1,97 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { IProvider } from '../IProvider.js';
+import { IModel } from '../IModel.js';
+import { IMessage } from '../IMessage.js';
+import { ITool } from '../ITool.js';
+import { Config, AuthType } from '@vybestack/llxprt-code-core';
+/**
+ * Represents the default Gemini provider.
+ * This provider is implicitly active when no other provider is explicitly set.
+ *
+ * NOTE: This provider acts as a configuration layer for the native Gemini integration.
+ * It doesn't implement generateChatCompletion directly but instead configures the
+ * system to use the native Gemini client with the appropriate authentication.
+ */
+type GeminiAuthMode = 'oauth' | 'gemini-api-key' | 'vertex-ai' | 'none';
+export declare class GeminiProvider implements IProvider {
+    readonly name: string;
+    private apiKey?;
+    private authMode;
+    private config?;
+    private currentModel;
+    private modelExplicitlySet;
+    private authDetermined;
+    private toolSchemas;
+    constructor();
+    /**
+     * Determines the best available authentication method based on environment variables
+     * and existing configuration. Follows the hierarchy: Vertex AI → Gemini API key → OAuth
+     */
+    private determineBestAuth;
+    /**
+     * Checks if Vertex AI credentials are available
+     */
+    private hasVertexAICredentials;
+    /**
+     * Checks if Gemini API key is available
+     */
+    private hasGeminiAPIKey;
+    /**
+     * Sets up environment variables for Vertex AI authentication
+     */
+    private setupVertexAIAuth;
+    /**
+     * Sets the config instance for reading OAuth credentials
+     */
+    setConfig(config: Config): void;
+    getModels(): Promise<IModel[]>;
+    /**
+     * Checks if OAuth authentication is still valid
+     */
+    private isOAuthValid;
+    generateChatCompletion(messages: IMessage[], tools?: ITool[], _toolFormat?: string): AsyncIterableIterator<unknown>;
+    private convertMessagesToGeminiFormat;
+    private convertToolsToGeminiFormat;
+    setApiKey(apiKey: string): void;
+    /**
+     * Gets the current authentication mode
+     */
+    getAuthMode(): GeminiAuthMode;
+    /**
+     * Gets the appropriate AuthType for the core library
+     */
+    getCoreAuthType(): AuthType;
+    /**
+     * Gets the current model ID
+     */
+    getCurrentModel(): string;
+    /**
+     * Sets the current model ID
+     */
+    setModel(modelId: string): void;
+    /**
+     * Checks if the current auth mode requires payment
+     */
+    isPaidMode(): boolean;
+    /**
+     * Clears provider state but preserves explicitly set model
+     */
+    clearState(): void;
+    /**
+     * Forces re-determination of auth method
+     */
+    clearAuthCache(): void;
+    /**
+     * Get the list of server tools supported by this provider
+     */
+    getServerTools(): string[];
+    /**
+     * Invoke a server tool (native provider tool)
+     */
+    invokeServerTool(toolName: string, params: unknown, _config?: unknown): Promise<unknown>;
+}
+export {};

package/dist/src/providers/gemini/GeminiProvider.integration.test.d.ts

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

package/dist/src/providers/gemini/GeminiProvider.integration.test.js

@@ -0,0 +1,90 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { promises as fs } from 'fs';
+import path from 'path';
+import { processSingleFileContent, ContentGeneratorRole, } from '@vybestack/llxprt-code-core';
+import { GeminiProvider } from './GeminiProvider.js';
+describe('GeminiProvider Integration', () => {
+    it('should handle real PDF file content', async () => {
+        const provider = new GeminiProvider();
+        const testPdfPath = path.join(__dirname, '__fixtures__/test.pdf');
+        // Check if test PDF exists
+        try {
+            await fs.access(testPdfPath);
+        }
+        catch {
+            console.log('Skipping test - test PDF not found at:', testPdfPath);
+            return;
+        }
+        // Process the PDF file like the read tool would
+        const pdfResult = await processSingleFileContent(testPdfPath, path.dirname(testPdfPath));
+        // Create a message with the PDF content
+        const messages = [
+            {
+                role: ContentGeneratorRole.USER,
+                content: 'Please analyze this PDF document:',
+                parts: [
+                    { text: 'Please analyze this PDF document:' },
+                    pdfResult.llmContent,
+                ],
+            },
+        ];
+        // Convert to Gemini format
+        const result = provider['convertMessagesToGeminiFormat'](messages);
+        // Verify the structure
+        expect(result).toHaveLength(1);
+        expect(result[0].role).toBe('user');
+        expect(result[0].parts).toHaveLength(2);
+        // First part should be the text
+        expect(result[0].parts[0]).toEqual({
+            text: 'Please analyze this PDF document:',
+        });
+        // Second part should be the inline data
+        expect(result[0].parts[1]).toHaveProperty('inlineData');
+        expect(result[0].parts[1].inlineData).toHaveProperty('data');
+        expect(result[0].parts[1].inlineData).toHaveProperty('mimeType', 'application/pdf');
+        // Verify the base64 data is present and non-empty
+        const inlineData = result[0].parts[1].inlineData;
+        expect(inlineData?.data).toBeTruthy();
+        expect(inlineData?.data?.length).toBeGreaterThan(100); // PDF should have substantial base64 data
+    });
+    it('should handle mixed text and image content', async () => {
+        const provider = new GeminiProvider();
+        // Simulate mixed content with text and an image
+        const messages = [
+            {
+                role: ContentGeneratorRole.USER,
+                content: 'What do you see in this image? Is it red?',
+                parts: [
+                    { text: 'What do you see in this image?' },
+                    {
+                        inlineData: {
+                            data: 'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNkYPhfDwAChwGA60e6kgAAAABJRU5ErkJggg==', // 1x1 red pixel PNG
+                            mimeType: 'image/png',
+                        },
+                    },
+                    { text: 'Is it red?' },
+                ],
+            },
+        ];
+        const result = provider['convertMessagesToGeminiFormat'](messages);
+        expect(result).toHaveLength(1);
+        expect(result[0].role).toBe('user');
+        expect(result[0].parts).toHaveLength(3);
+        expect(result[0].parts[0]).toEqual({
+            text: 'What do you see in this image?',
+        });
+        expect(result[0].parts[1]).toEqual({
+            inlineData: {
+                data: 'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNkYPhfDwAChwGA60e6kgAAAABJRU5ErkJggg==',
+                mimeType: 'image/png',
+            },
+        });
+        expect(result[0].parts[2]).toEqual({ text: 'Is it red?' });
+    });
+});
+//# sourceMappingURL=GeminiProvider.integration.test.js.map

package/dist/src/providers/gemini/GeminiProvider.integration.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"GeminiProvider.integration.test.js","sourceRoot":"","sources":["../../../../src/providers/gemini/GeminiProvider.integration.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,IAAI,CAAC;AACpC,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EACL,wBAAwB,EACxB,oBAAoB,GACrB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAIrD,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;IAC1C,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;QACtC,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,uBAAuB,CAAC,CAAC;QAElE,2BAA2B;QAC3B,IAAI,CAAC;YACH,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAC/B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,GAAG,CAAC,wCAAwC,EAAE,WAAW,CAAC,CAAC;YACnE,OAAO;QACT,CAAC;QAED,gDAAgD;QAChD,MAAM,SAAS,GAAG,MAAM,wBAAwB,CAC9C,WAAW,EACX,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAC1B,CAAC;QAEF,wCAAwC;QACxC,MAAM,QAAQ,GAAe;YAC3B;gBACE,IAAI,EAAE,oBAAoB,CAAC,IAAI;gBAC/B,OAAO,EAAE,mCAAmC;gBAC5C,KAAK,EAAE;oBACL,EAAE,IAAI,EAAE,mCAAmC,EAAE;oBAC7C,SAAS,CAAC,UAAkB;iBAC7B;aACF;SACF,CAAC;QAEF,2BAA2B;QAC3B,MAAM,MAAM,GAAG,QAAQ,CAAC,+BAA+B,CAAC,CAAC,QAAQ,CAAC,CAAC;QAEnE,uBAAuB;QACvB,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAExC,gCAAgC;QAChC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACjC,IAAI,EAAE,mCAAmC;SAC1C,CAAC,CAAC;QAEH,wCAAwC;QACxC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,cAAc,CAClD,UAAU,EACV,iBAAiB,CAClB,CAAC;QAEF,kDAAkD;QAClD,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC;QACjD,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,UAAU,EAAE,CAAC;QACtC,MAAM,CAAC,UAAU,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,0CAA0C;IACnG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;QAEtC,gDAAgD;QAChD,MAAM,QAAQ,GAAe;YAC3B;gBACE,IAAI,EAAE,oBAAoB,CAAC,IAAI;gBAC/B,OAAO,EAAE,2CAA2C;gBACpD,KAAK,EAAE;oBACL,EAAE,IAAI,EAAE,gCAAgC,EAAE;oBAC1C;wBACE,UAAU,EAAE;4BACV,IAAI,EAAE,kGAAkG,EAAE,oBAAoB;4BAC9H,QAAQ,EAAE,WAAW;yBACtB;qBACF;oBACD,EAAE,IAAI,EAAE,YAAY,EAAE;iBACvB;aACF;SACF,CAAC;QAEF,MAAM,MAAM,GAAG,QAAQ,CAAC,+BAA+B,CAAC,CAAC,QAAQ,CAAC,CAAC;QAEnE,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACjC,IAAI,EAAE,gCAAgC;SACvC,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACjC,UAAU,EAAE;gBACV,IAAI,EAAE,kGAAkG;gBACxG,QAAQ,EAAE,WAAW;aACtB;SACF,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}