Skip to content

Commit

Permalink
fix(Google Vertex Chat Model Node): Clean service account private key (
Browse files Browse the repository at this point in the history
  • Loading branch information
burivuhster authored and riascho committed Sep 23, 2024
1 parent 2e33f8d commit 56ad669
Showing 1 changed file with 9 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@ import {
import { ChatVertexAI } from '@langchain/google-vertexai';
import type { SafetySetting } from '@google/generative-ai';
import { ProjectsClient } from '@google-cloud/resource-manager';
import { formatPrivateKey } from 'n8n-nodes-base/dist/utils/utilities';
import { getConnectionHintNoticeField } from '../../../utils/sharedFields';
import { N8nLlmTracing } from '../N8nLlmTracing';
import { additionalOptions } from '../gemini-common/additional-options';
Expand Down Expand Up @@ -97,11 +98,13 @@ export class LmChatGoogleVertex implements INodeType {
const results: Array<{ name: string; value: string }> = [];

const credentials = await this.getCredentials('googleApi');
const privateKey = formatPrivateKey(credentials.privateKey as string);
const email = (credentials.email as string).trim();

const client = new ProjectsClient({
credentials: {
client_email: credentials.email as string,
private_key: credentials.privateKey as string,
client_email: email,
private_key: privateKey,
},
});

Expand All @@ -123,6 +126,8 @@ export class LmChatGoogleVertex implements INodeType {

async supplyData(this: IExecuteFunctions, itemIndex: number): Promise<SupplyData> {
const credentials = await this.getCredentials('googleApi');
const privateKey = formatPrivateKey(credentials.privateKey as string);
const email = (credentials.email as string).trim();

const modelName = this.getNodeParameter('modelName', itemIndex) as string;

Expand Down Expand Up @@ -153,8 +158,8 @@ export class LmChatGoogleVertex implements INodeType {
authOptions: {
projectId,
credentials: {
client_email: credentials.email as string,
private_key: credentials.privateKey as string,
client_email: email,
private_key: privateKey,
},
},
model: modelName,
Expand Down

0 comments on commit 56ad669

Please sign in to comment.