Sultan
/
Flowise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
packages/components/nodes/chatmodels/ChatGoogleGenerativeAI/ChatGoogleGenerativeAI.ts
View File

@ -18,7 +18,7 @@ class GoogleGenerativeAI_ChatModels implements INode {
constructor() {
this.label = 'ChatGoogleGenerativeAI'
this.name = 'chatGoogleGenerativeAI'
this.version = 2.0
this.version = 1.0
this.type = 'ChatGoogleGenerativeAI'
this.icon = 'gemini.png'
this.category = 'Chat Models'
@ -29,7 +29,7 @@ class GoogleGenerativeAI_ChatModels implements INode {
name: 'credential',
type: 'credential',
credentialNames: ['googleGenerativeAI'],
optional: true,
optional: false,
description: 'Google Generative AI credential.'
}
this.inputs = [
@ -100,7 +100,7 @@ class GoogleGenerativeAI_ChatModels implements INode {
const model = new ChatGoogleGenerativeAI(obj)
if (topP) model.topP = parseFloat(topP)
if (cache) model.cache = cache
if (temperature) model.temperature = parseInt(temperature)
if (temperature) model.temperature = parseFloat(temperature)
return model
}
}

4
packages/components/nodes/embeddings/GoogleGenerativeAIEmbedding/GoogleGenerativeAIEmbedding.ts
View File

@ -24,13 +24,13 @@ class GoogleGenerativeAIEmbedding_Embeddings implements INode {
this.icon = 'gemini.png'
this.category = 'Embeddings'
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 = {
label: 'Connect Credential',
name: 'credential',
type: 'credential',
credentialNames: ['googleGenerativeAI'],
optional: true,
optional: false,
description: 'Google Generative AI credential.'
}
this.inputs = [