Support for Google Gemini Models: Minor fixes

This commit is contained in:
vinodkiran
2023-12-15 22:20:51 +05:30
parent e7f13c87d8
commit f475732e6b
2 changed files with 5 additions and 5 deletions
@@ -18,7 +18,7 @@ class GoogleGenerativeAI_ChatModels implements INode {
constructor() { constructor() {
this.label = 'ChatGoogleGenerativeAI' this.label = 'ChatGoogleGenerativeAI'
this.name = 'chatGoogleGenerativeAI' this.name = 'chatGoogleGenerativeAI'
this.version = 2.0 this.version = 1.0
this.type = 'ChatGoogleGenerativeAI' this.type = 'ChatGoogleGenerativeAI'
this.icon = 'gemini.png' this.icon = 'gemini.png'
this.category = 'Chat Models' this.category = 'Chat Models'
@@ -29,7 +29,7 @@ class GoogleGenerativeAI_ChatModels implements INode {
name: 'credential', name: 'credential',
type: 'credential', type: 'credential',
credentialNames: ['googleGenerativeAI'], credentialNames: ['googleGenerativeAI'],
optional: true, optional: false,
description: 'Google Generative AI credential.' description: 'Google Generative AI credential.'
} }
this.inputs = [ this.inputs = [
@@ -100,7 +100,7 @@ class GoogleGenerativeAI_ChatModels implements INode {
const model = new ChatGoogleGenerativeAI(obj) const model = new ChatGoogleGenerativeAI(obj)
if (topP) model.topP = parseFloat(topP) if (topP) model.topP = parseFloat(topP)
if (cache) model.cache = cache if (cache) model.cache = cache
if (temperature) model.temperature = parseInt(temperature) if (temperature) model.temperature = parseFloat(temperature)
return model return model
} }
} }
@@ -24,13 +24,13 @@ class GoogleGenerativeAIEmbedding_Embeddings implements INode {
this.icon = 'gemini.png' this.icon = 'gemini.png'
this.category = 'Embeddings' this.category = 'Embeddings'
this.description = 'Google Generative API to generate embeddings for a given text' this.description = 'Google Generative API to generate embeddings for a given text'
this.baseClasses = [this.type, ...getBaseClasses(GoogleVertexAIEmbeddings)] this.baseClasses = [this.type, ...getBaseClasses(GoogleGenerativeAIEmbeddings)]
this.credential = { this.credential = {
label: 'Connect Credential', label: 'Connect Credential',
name: 'credential', name: 'credential',
type: 'credential', type: 'credential',
credentialNames: ['googleGenerativeAI'], credentialNames: ['googleGenerativeAI'],
optional: true, optional: false,
description: 'Google Generative AI credential.' description: 'Google Generative AI credential.'
} }
this.inputs = [ this.inputs = [