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.
Merge branch 'custom' of https://github.com/Hostcomm/Flowise
- Loading branch information
Showing
14 changed files
with
579 additions
and
260 deletions.
There are no files selected for viewing
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
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,25 @@ | ||
import { INodeParams, INodeCredential } from '../src/Interface' | ||
|
||
class CerebrasAPIAuth implements INodeCredential { | ||
label: string | ||
name: string | ||
version: number | ||
description: string | ||
inputs: INodeParams[] | ||
|
||
constructor() { | ||
this.label = 'Cerebras API Key' | ||
this.name = 'cerebrasAIApi' | ||
this.version = 1.0 | ||
this.inputs = [ | ||
{ | ||
label: 'Cerebras API Key', | ||
name: 'cerebrasApiKey', | ||
type: 'password', | ||
description: 'API Key (cloud.cerebras.ai)' | ||
} | ||
] | ||
} | ||
} | ||
|
||
module.exports = { credClass: CerebrasAPIAuth } |
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
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
237 changes: 147 additions & 90 deletions
237
packages/components/nodes/agents/OpenAIAssistant/OpenAIAssistant.ts
Large diffs are not rendered by default.
Oops, something went wrong.
161 changes: 161 additions & 0 deletions
161
packages/components/nodes/chatmodels/ChatCerebras/ChatCerebras.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,161 @@ | ||
import { ChatOpenAI, OpenAIChatInput } from '@langchain/openai' | ||
import { BaseCache } from '@langchain/core/caches' | ||
import { BaseLLMParams } from '@langchain/core/language_models/llms' | ||
import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface' | ||
import { getBaseClasses, getCredentialData, getCredentialParam } from '../../../src/utils' | ||
|
||
class ChatCerebras_ChatModels 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 = 'ChatCerebras' | ||
this.name = 'chatCerebras' | ||
this.version = 1.0 | ||
this.type = 'ChatCerebras' | ||
this.icon = 'cerebras.png' | ||
this.category = 'Chat Models' | ||
this.description = 'Models available via Cerebras' | ||
this.baseClasses = [this.type, ...getBaseClasses(ChatOpenAI)] | ||
this.credential = { | ||
label: 'Connect Credential', | ||
name: 'credential', | ||
type: 'credential', | ||
credentialNames: ['cerebrasAIApi'], | ||
optional: true | ||
} | ||
this.inputs = [ | ||
{ | ||
label: 'Cache', | ||
name: 'cache', | ||
type: 'BaseCache', | ||
optional: true | ||
}, | ||
{ | ||
label: 'Model Name', | ||
name: 'modelName', | ||
type: 'string', | ||
placeholder: 'llama3.1-8b' | ||
}, | ||
{ | ||
label: 'Temperature', | ||
name: 'temperature', | ||
type: 'number', | ||
step: 0.1, | ||
default: 0.9, | ||
optional: true | ||
}, | ||
{ | ||
label: 'Max Tokens', | ||
name: 'maxTokens', | ||
type: 'number', | ||
step: 1, | ||
optional: true, | ||
additionalParams: true | ||
}, | ||
{ | ||
label: 'Top Probability', | ||
name: 'topP', | ||
type: 'number', | ||
step: 0.1, | ||
optional: true, | ||
additionalParams: true | ||
}, | ||
{ | ||
label: 'Frequency Penalty', | ||
name: 'frequencyPenalty', | ||
type: 'number', | ||
step: 0.1, | ||
optional: true, | ||
additionalParams: true | ||
}, | ||
{ | ||
label: 'Presence Penalty', | ||
name: 'presencePenalty', | ||
type: 'number', | ||
step: 0.1, | ||
optional: true, | ||
additionalParams: true | ||
}, | ||
{ | ||
label: 'Timeout', | ||
name: 'timeout', | ||
type: 'number', | ||
step: 1, | ||
optional: true, | ||
additionalParams: true | ||
}, | ||
{ | ||
label: 'BasePath', | ||
name: 'basepath', | ||
type: 'string', | ||
optional: true, | ||
default: 'https://api.cerebras.ai/v1', | ||
additionalParams: true | ||
}, | ||
{ | ||
label: 'BaseOptions', | ||
name: 'baseOptions', | ||
type: 'json', | ||
optional: true, | ||
additionalParams: true | ||
} | ||
] | ||
} | ||
|
||
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> { | ||
const temperature = nodeData.inputs?.temperature as string | ||
const modelName = nodeData.inputs?.modelName as string | ||
const maxTokens = nodeData.inputs?.maxTokens as string | ||
const topP = nodeData.inputs?.topP as string | ||
const frequencyPenalty = nodeData.inputs?.frequencyPenalty as string | ||
const presencePenalty = nodeData.inputs?.presencePenalty as string | ||
const timeout = nodeData.inputs?.timeout as string | ||
const streaming = nodeData.inputs?.streaming as boolean | ||
const basePath = nodeData.inputs?.basepath as string | ||
const baseOptions = nodeData.inputs?.baseOptions | ||
const cache = nodeData.inputs?.cache as BaseCache | ||
|
||
const credentialData = await getCredentialData(nodeData.credential ?? '', options) | ||
const cerebrasAIApiKey = getCredentialParam('cerebrasApiKey', credentialData, nodeData) | ||
|
||
const obj: Partial<OpenAIChatInput> & BaseLLMParams & { cerebrasAIApiKey?: string } = { | ||
temperature: parseFloat(temperature), | ||
modelName, | ||
openAIApiKey: cerebrasAIApiKey, | ||
streaming: streaming ?? true | ||
} | ||
|
||
if (maxTokens) obj.maxTokens = parseInt(maxTokens, 10) | ||
if (topP) obj.topP = parseFloat(topP) | ||
if (frequencyPenalty) obj.frequencyPenalty = parseFloat(frequencyPenalty) | ||
if (presencePenalty) obj.presencePenalty = parseFloat(presencePenalty) | ||
if (timeout) obj.timeout = parseInt(timeout, 10) | ||
if (cache) obj.cache = cache | ||
|
||
let parsedBaseOptions: any | undefined = undefined | ||
|
||
if (baseOptions) { | ||
try { | ||
parsedBaseOptions = typeof baseOptions === 'object' ? baseOptions : JSON.parse(baseOptions) | ||
} catch (exception) { | ||
throw new Error("Invalid JSON in the ChatCerebras's BaseOptions: " + exception) | ||
} | ||
} | ||
const model = new ChatOpenAI(obj, { | ||
basePath, | ||
baseOptions: parsedBaseOptions | ||
}) | ||
return model | ||
} | ||
} | ||
|
||
module.exports = { nodeClass: ChatCerebras_ChatModels } |
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
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
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
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
18 changes: 18 additions & 0 deletions
18
...es/server/src/database/migrations/mariadb/1715861032479-AddVectorStoreConfigToDocStore.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,18 @@ | ||
import { MigrationInterface, QueryRunner } from 'typeorm' | ||
|
||
export class AddVectorStoreConfigToDocStore1715861032479 implements MigrationInterface { | ||
public async up(queryRunner: QueryRunner): Promise<void> { | ||
const columnExists = await queryRunner.hasColumn('document_store', 'vectorStoreConfig') | ||
if (!columnExists) { | ||
await queryRunner.query(`ALTER TABLE \`document_store\` ADD COLUMN \`vectorStoreConfig\` TEXT;`) | ||
await queryRunner.query(`ALTER TABLE \`document_store\` ADD COLUMN \`embeddingConfig\` TEXT;`) | ||
await queryRunner.query(`ALTER TABLE \`document_store\` ADD COLUMN \`recordManagerConfig\` TEXT;`) | ||
} | ||
} | ||
|
||
public async down(queryRunner: QueryRunner): Promise<void> { | ||
await queryRunner.query(`ALTER TABLE \`document_store\` DROP COLUMN \`vectorStoreConfig\`;`) | ||
await queryRunner.query(`ALTER TABLE \`document_store\` DROP COLUMN \`embeddingConfig\`;`) | ||
await queryRunner.query(`ALTER TABLE \`document_store\` DROP COLUMN \`recordManagerConfig\`;`) | ||
} | ||
} |
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
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
Oops, something went wrong.