first commit

packages/@n8n/nodes-langchain/nodes/vendors/GoogleGemini/transport/index.test.ts

@@ -0,0 +1,83 @@
+import type { IExecuteFunctions } from 'n8n-workflow';
+import { mockDeep } from 'jest-mock-extended';
+import { apiRequest } from '.';
+
+describe('GoogleGemini transport', () => {
+	const executeFunctionsMock = mockDeep<IExecuteFunctions>();
+
+	beforeEach(() => {
+		jest.clearAllMocks();
+	});
+
+	it('should call httpRequestWithAuthentication with correct parameters', async () => {
+		executeFunctionsMock.getCredentials.mockResolvedValue({
+			host: 'https://custom-url.com',
+		});
+
+		await apiRequest.call(executeFunctionsMock, 'GET', '/v1beta/models', {
+			headers: {
+				'Content-Type': 'application/json',
+			},
+			body: {
+				foo: 'bar',
+			},
+			qs: {
+				test: 123,
+			},
+		});
+
+		expect(executeFunctionsMock.helpers.httpRequestWithAuthentication).toHaveBeenCalledWith(
+			'googlePalmApi',
+			{
+				method: 'GET',
+				url: 'https://custom-url.com/v1beta/models',
+				json: true,
+				body: {
+					foo: 'bar',
+				},
+				qs: {
+					test: 123,
+				},
+				headers: {
+					'Content-Type': 'application/json',
+				},
+			},
+		);
+	});
+
+	it('should use the default url if no custom url is provided', async () => {
+		executeFunctionsMock.getCredentials.mockResolvedValue({});
+
+		await apiRequest.call(executeFunctionsMock, 'GET', '/v1beta/models');
+
+		expect(executeFunctionsMock.helpers.httpRequestWithAuthentication).toHaveBeenCalledWith(
+			'googlePalmApi',
+			{
+				method: 'GET',
+				url: 'https://generativelanguage.googleapis.com/v1beta/models',
+				json: true,
+			},
+		);
+	});
+
+	it('should override the values with `option`', async () => {
+		executeFunctionsMock.getCredentials.mockResolvedValue({});
+
+		await apiRequest.call(executeFunctionsMock, 'GET', '', {
+			option: {
+				url: 'https://custom-url.com',
+				returnFullResponse: true,
+			},
+		});
+
+		expect(executeFunctionsMock.helpers.httpRequestWithAuthentication).toHaveBeenCalledWith(
+			'googlePalmApi',
+			{
+				method: 'GET',
+				url: 'https://custom-url.com',
+				json: true,
+				returnFullResponse: true,
+			},
+		);
+	});
+});

packages/@n8n/nodes-langchain/nodes/vendors/GoogleGemini/transport/index.ts

@@ -0,0 +1,50 @@
+import type {
+	IDataObject,
+	IExecuteFunctions,
+	IHttpRequestMethods,
+	ILoadOptionsFunctions,
+} from 'n8n-workflow';
+
+type RequestParameters = {
+	headers?: IDataObject;
+	body?: IDataObject | string;
+	qs?: IDataObject;
+	option?: IDataObject;
+};
+
+type GooglePalmApiCredentials = {
+	host: string;
+	apiKey: string;
+};
+
+export async function apiRequest(
+	this: IExecuteFunctions | ILoadOptionsFunctions,
+	method: IHttpRequestMethods,
+	endpoint: string,
+	parameters?: RequestParameters,
+) {
+	const { body, qs, option, headers } = parameters ?? {};
+
+	const credentials = await this.getCredentials<GooglePalmApiCredentials>('googlePalmApi');
+
+	let url = `https://generativelanguage.googleapis.com${endpoint}`;
+
+	if (credentials.host) {
+		url = `${credentials.host}${endpoint}`;
+	}
+
+	const options = {
+		headers,
+		method,
+		body,
+		qs,
+		url,
+		json: true,
+	};
+
+	if (option && Object.keys(option).length !== 0) {
+		Object.assign(options, option);
+	}
+
+	return await this.helpers.httpRequestWithAuthentication.call(this, 'googlePalmApi', options);
+}