forked from FlowiseAI/Flowise
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* add TogetherAI.ts skeleton * add TogetherAI api links * fix constructing final obj * fix INodeData import * fix category * fix version & streaming * add stop * update constructor * update constructor with default values * add togetherAI logo * add credential * change logo * update streaming * disable streaming * remove model * clean up comment * delete unused icon * add togetherAI for streaming * Update index.ts --------- Co-authored-by: Henry Heng <henryheng@flowiseai.com>
- Loading branch information
1 parent
e32b643
commit 50c53de
Showing
2 changed files
with
157 additions
and
0 deletions.
There are no files selected for viewing
157 changes: 157 additions & 0 deletions
157
packages/components/nodes/llms/TogetherAI/TogetherAI.ts
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,157 @@ | ||
import { ICommonObject, INode, INodeData, INodeOptionsValue, INodeParams } from '../../../src' | ||
import { getBaseClasses, getCredentialData, getCredentialParam } from '../../../src' | ||
import { TogetherAI, TogetherAIInputs } from '@langchain/community/llms/togetherai' | ||
import { getModels, MODEL_TYPE } from '../../../src/modelLoader' | ||
import { BaseCache } from '@langchain/core/caches' | ||
|
||
class TogetherAI_LLMs implements INode { | ||
label: string | ||
name: string | ||
version: number | ||
type: string | ||
icon: string | ||
category: string | ||
description: string | ||
baseClasses: string[] | ||
credential: INodeParams | ||
inputs: INodeParams[] | ||
|
||
constructor() { | ||
this.label = 'TogetherAI' | ||
this.name = 'togetherAI' | ||
this.version = 1.0 | ||
this.type = 'TogetherAI' | ||
this.icon = 'togetherai.png' | ||
this.category = 'LLMs' | ||
this.description = 'Wrapper around TogetherAI large language models' | ||
this.baseClasses = [this.type, ...getBaseClasses(TogetherAI)] | ||
this.credential = { | ||
label: 'Connect Credential', | ||
name: 'credential', | ||
type: 'credential', | ||
credentialNames: ['togetherAIApi'] | ||
} | ||
this.inputs = [ | ||
{ | ||
label: 'Cache', | ||
name: 'cache', | ||
type: 'BaseCache', | ||
optional: true | ||
}, | ||
{ | ||
label: 'Model Name', | ||
name: 'modelName', | ||
type: 'string', | ||
description: 'The name of the model to query.' | ||
}, | ||
{ | ||
label: 'Top K', | ||
name: 'topK', | ||
type: 'number', | ||
description: | ||
'The topK parameter is used to limit the number of choices for the next predicted word or token. It specifies the maximum number of tokens to consider at each step, based on their probability of occurrence. This technique helps to speed up the generation process and can improve the quality of the generated text by focusing on the most likely options.', | ||
step: 1, | ||
default: 50 | ||
}, | ||
{ | ||
label: 'Top P', | ||
name: 'topP', | ||
type: 'number', | ||
description: | ||
'The topP (nucleus) parameter is used to dynamically adjust the number of choices for each predicted token based on the cumulative probabilities. It specifies a probability threshold, below which all less likely tokens are filtered out. This technique helps to maintain diversity and generate more fluent and natural-sounding text.', | ||
step: 0.1, | ||
default: 0.7 | ||
}, | ||
{ | ||
label: 'Temperature', | ||
name: 'temperature', | ||
type: 'number', | ||
description: | ||
'A decimal number that determines the degree of randomness in the response. A value of 1 will always yield the same output. A temperature less than 1 favors more correctness and is appropriate for question answering or summarization. A value greater than 1 introduces more randomness in the output.', | ||
step: 0.1, | ||
default: 0.7 | ||
}, | ||
{ | ||
label: 'Repeat Penalty', | ||
name: 'repeatPenalty', | ||
type: 'number', | ||
description: | ||
'A number that controls the diversity of generated text by reducing the likelihood of repeated sequences. Higher values decrease repetition.', | ||
step: 0.1, | ||
default: 1 | ||
}, | ||
{ | ||
label: 'Streaming', | ||
name: 'streaming', | ||
type: 'boolean', | ||
default: false, | ||
description: 'Whether or not to stream tokens as they are generated' | ||
}, | ||
{ | ||
label: 'Max Tokens', | ||
name: 'maxTokens', | ||
type: 'number', | ||
step: 1, | ||
description: 'Limit the number of tokens generated.', | ||
optional: true, | ||
additionalParams: true | ||
}, | ||
{ | ||
label: 'Stop Sequence', | ||
name: 'stop', | ||
type: 'string', | ||
rows: 4, | ||
placeholder: 'AI assistant:', | ||
description: 'A list of tokens at which the generation should stop.', | ||
optional: true, | ||
additionalParams: true | ||
} | ||
// todo: safetyModel? logprobs? | ||
] | ||
} | ||
|
||
//@ts-ignore | ||
loadMethods = { | ||
async listModels(): Promise<INodeOptionsValue[]> { | ||
return await getModels(MODEL_TYPE.LLM, 'togetherAI') | ||
} | ||
} | ||
|
||
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> { | ||
const temperature = nodeData.inputs?.temperature as string | ||
const maxTokens = nodeData.inputs?.maxTokens as string | ||
const topP = nodeData.inputs?.topP as string | ||
const topK = nodeData.inputs?.topK as string | ||
const repeatPenalty = nodeData.inputs?.repeatPenalty as string | ||
const modelName = nodeData.inputs?.modelName as string | ||
const stop = nodeData.inputs?.stop as string | ||
const streaming = nodeData.inputs?.streaming as boolean | ||
|
||
const cache = nodeData.inputs?.cache as BaseCache | ||
|
||
const credentialData = await getCredentialData(nodeData.credential ?? '', options) | ||
const togetherAiApiKey = getCredentialParam('togetherAIApiKey', credentialData, nodeData) | ||
|
||
const obj: TogetherAIInputs = { | ||
modelName, | ||
apiKey: togetherAiApiKey, | ||
streaming: streaming ?? false | ||
} | ||
|
||
if (temperature) obj.temperature = parseFloat(temperature) | ||
if (maxTokens) obj.maxTokens = parseInt(maxTokens, 10) | ||
if (topP) obj.topP = parseFloat(topP) | ||
if (topK) obj.topK = parseFloat(topK) | ||
if (repeatPenalty) obj.repetitionPenalty = parseFloat(repeatPenalty) | ||
if (streaming) obj.streaming = streaming | ||
if (stop) { | ||
obj.stop = stop.split(',') | ||
} | ||
if (cache) obj.cache = cache | ||
|
||
const togetherAI = new TogetherAI(obj) | ||
return togetherAI | ||
} | ||
} | ||
|
||
module.exports = { nodeClass: TogetherAI_LLMs } |
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.