Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions containers/api-proxy/providers/gemini.js
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
* All configuration lives in GOOGLE_PROVIDER_SPECS.gemini (google-provider-specs.js).
*/

const { createGoogleProviderAdapter } = require('./google-adapter');
const { makeGoogleProviderFactory } = require('./google-adapter');

/**
* Create the Google Gemini provider adapter.
Expand All @@ -24,6 +24,6 @@ const { createGoogleProviderAdapter } = require('./google-adapter');
* @param {{ bodyTransform?: ((body: Buffer) => (Buffer | null | Promise<Buffer | null>))|null }} [deps={}] - Injected dependencies
* @returns {import('./index').ProviderAdapter}
*/
const createGeminiAdapter = createGoogleProviderAdapter.bind(null, 'gemini');
const createGeminiAdapter = makeGoogleProviderFactory('gemini');

module.exports = { createGeminiAdapter };
12 changes: 11 additions & 1 deletion containers/api-proxy/providers/google-adapter.js
Original file line number Diff line number Diff line change
Expand Up @@ -126,4 +126,14 @@ function createGoogleProviderAdapter(providerKey, env, deps = {}) {
});
}

module.exports = { createGoogleApiKeyAdapter, createGoogleProviderAdapter };
/**
* Create a named Google provider adapter factory.
*
* @param {string} providerKey - Key into GOOGLE_PROVIDER_SPECS (e.g. 'gemini')
* @returns {(env: Record<string, string|undefined>, deps?: { bodyTransform?: ((body: Buffer) => (Buffer | null | Promise<Buffer | null>))|null }) => import('./index').ProviderAdapter}
*/
function makeGoogleProviderFactory(providerKey) {
return (env, deps = {}) => createGoogleProviderAdapter(providerKey, env, deps);
}

module.exports = { createGoogleApiKeyAdapter, createGoogleProviderAdapter, makeGoogleProviderFactory };
9 changes: 8 additions & 1 deletion containers/api-proxy/providers/google-adapter.test.js
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
'use strict';

const { createGoogleProviderAdapter } = require('./google-adapter');
const { createGoogleProviderAdapter, makeGoogleProviderFactory } = require('./google-adapter');
const { GOOGLE_PROVIDER_SPECS } = require('./google-provider-specs');
const { createGeminiAdapter } = require('./gemini');
const { createVertexAdapter } = require('./vertex');
Expand All @@ -10,6 +10,13 @@ describe('createGoogleProviderAdapter', () => {
expect(() => createGoogleProviderAdapter('bogus', {})).toThrow(/Unknown Google provider spec: bogus/);
});

it('creates reusable provider factories from provider keys', () => {
const createGemini = makeGoogleProviderFactory('gemini');
expect(createGemini({ GEMINI_API_KEY: 'key' }).getAuthHeaders()).toEqual({
'x-goog-api-key': 'key',
});
});

it('derives gemini ports, targets and messages from the spec', () => {
const adapter = createGoogleProviderAdapter('gemini', {});
expect(adapter.name).toBe('gemini');
Expand Down
4 changes: 2 additions & 2 deletions containers/api-proxy/providers/vertex.js
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
* All configuration lives in GOOGLE_PROVIDER_SPECS.vertex (google-provider-specs.js).
*/

const { createGoogleProviderAdapter } = require('./google-adapter');
const { makeGoogleProviderFactory } = require('./google-adapter');

/**
* Create the Google Vertex AI provider adapter.
Expand All @@ -26,6 +26,6 @@ const { createGoogleProviderAdapter } = require('./google-adapter');
* @param {{ bodyTransform?: ((body: Buffer) => (Buffer | null | Promise<Buffer | null>))|null }} [deps={}] - Injected dependencies
* @returns {import('./index').ProviderAdapter}
*/
const createVertexAdapter = createGoogleProviderAdapter.bind(null, 'vertex');
const createVertexAdapter = makeGoogleProviderFactory('vertex');

module.exports = { createVertexAdapter };
Loading