farcasclaudiu/Flowise
1
0
Fork 0
mirror of https://github.com/farcasclaudiu/Flowise.git synced 2026-06-28 15:00:57 +03:00
Code Issues Packages Projects Releases Wiki Activity

modify google gemini based on request changes

Browse Source
This commit is contained in:
chungyau97
2024-01-16 16:19:21 +08:00
parent 4d15e886fe
commit f3a244a93c
2 changed files with 11 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/chatmodels/ChatGoogleGenerativeAI/ChatGoogleGenerativeAI.ts
+9 -50
View File
@@ -78,10 +78,11 @@ class GoogleGenerativeAI_ChatModels implements INode {
additionalParams: true
},
{
label: 'topK',
label: 'Top Next Highest Probability Tokens',
name: 'topK',
type: 'number',
step: 0.1,
description: `Decode using top-k sampling: consider the set of top_k most probable tokens. Must be positive`,
step: 1,
optional: true,
additionalParams: true
},
@@ -90,7 +91,7 @@ class GoogleGenerativeAI_ChatModels implements INode {
name: 'harmCategory',
type: 'multiOptions',
description:
'Refer to <a target="_blank" href="https://cloud.google.com/vertex-ai/docs/generative-ai/multimodal/configure-safety-attributes#gemini-TASK-samples-go">official guide</a> on how to use Harm Category',
'Refer to <a target="_blank" href="https://cloud.google.com/vertex-ai/docs/generative-ai/multimodal/configure-safety-attributes#safety_attribute_definitions">official guide</a> on how to use Harm Category',
options: [
{
label: 'Dangerous',
@@ -117,7 +118,7 @@ class GoogleGenerativeAI_ChatModels implements INode {
name: 'harmBlockThreshold',
type: 'multiOptions',
description:
'Refer to <a target="_blank" href="https://cloud.google.com/vertex-ai/docs/generative-ai/multimodal/configure-safety-attributes#gemini-TASK-samples-go">official guide</a> on how to use Harm Block Threshold',
'Refer to <a target="_blank" href="https://cloud.google.com/vertex-ai/docs/generative-ai/multimodal/configure-safety-attributes#safety_setting_thresholds">official guide</a> on how to use Harm Block Threshold',
options: [
{
label: 'Low and Above',
@@ -169,7 +170,7 @@ class GoogleGenerativeAI_ChatModels implements INode {
const model = new ChatGoogleGenerativeAI(obj)
if (topP) model.topP = parseFloat(topP)
if (topK) model.topP = parseFloat(topK)
if (topK) model.topK = parseFloat(topK)
if (cache) model.cache = cache
if (temperature) model.temperature = parseFloat(temperature)
@@ -178,10 +179,10 @@ class GoogleGenerativeAI_ChatModels implements INode {
let harmBlockThresholds: string[] = convertMultiOptionsToStringArray(harmBlockThreshold)
if (harmCategories.length != harmBlockThresholds.length)
throw new Error(`Harm Category & Harm Block Threshold are not the same length`)
const safetySettings: SafetySetting[] = harmCategories.map((value, index) => {
const safetySettings: SafetySetting[] = harmCategories.map((harmCategory, index) => {
return {
category: categoryInput(value),
threshold: thresholdInput(harmBlockThresholds[index])
category: harmCategory as HarmCategory,
threshold: harmBlockThresholds[index] as HarmBlockThreshold
}
})
if (safetySettings.length > 0) model.safetySettings = safetySettings
@@ -190,46 +191,4 @@ class GoogleGenerativeAI_ChatModels implements INode {
}
}
const categoryInput = (categoryInput: string): HarmCategory => {
let categoryOutput: HarmCategory
switch (categoryInput) {
case HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT:
categoryOutput = HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT
break
case HarmCategory.HARM_CATEGORY_HATE_SPEECH:
categoryOutput = HarmCategory.HARM_CATEGORY_HATE_SPEECH
break
case HarmCategory.HARM_CATEGORY_HARASSMENT:
categoryOutput = HarmCategory.HARM_CATEGORY_HARASSMENT
break
case HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT:
categoryOutput = HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT
break
default:
categoryOutput = HarmCategory.HARM_CATEGORY_UNSPECIFIED
}
return categoryOutput
}
const thresholdInput = (thresholdInput: string): HarmBlockThreshold => {
let thresholdOutput: HarmBlockThreshold
switch (thresholdInput) {
case HarmBlockThreshold.BLOCK_LOW_AND_ABOVE:
thresholdOutput = HarmBlockThreshold.BLOCK_LOW_AND_ABOVE
break
case HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE:
thresholdOutput = HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE
break
case HarmBlockThreshold.BLOCK_NONE:
thresholdOutput = HarmBlockThreshold.BLOCK_NONE
break
case HarmBlockThreshold.BLOCK_ONLY_HIGH:
thresholdOutput = HarmBlockThreshold.BLOCK_ONLY_HIGH
break
default:
thresholdOutput = HarmBlockThreshold.HARM_BLOCK_THRESHOLD_UNSPECIFIED
}
return thresholdOutput
}
module.exports = { nodeClass: GoogleGenerativeAI_ChatModels }