diff --git a/packages/components/nodes/chatmodels/Bittensor/Bittensor.ts b/packages/components/nodes/chatmodels/Bittensor/Bittensor.ts index 8ca05634..cfcac863 100644 --- a/packages/components/nodes/chatmodels/Bittensor/Bittensor.ts +++ b/packages/components/nodes/chatmodels/Bittensor/Bittensor.ts @@ -1,6 +1,6 @@ -import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface' +import { INode, INodeData, INodeParams } from '../../../src/Interface' import { getBaseClasses } from '../../../src/utils' -import {NIBittensorChatModel, BittensorInput} from 'langchain/experimental/chat_models/bittensor'; +import { NIBittensorChatModel, BittensorInput } from 'langchain/experimental/chat_models/bittensor' class Bittensor_ChatModels implements INode { label: string @@ -29,17 +29,15 @@ class Bittensor_ChatModels implements INode { type: 'string', additionalParams: true, optional: true - }, + } ] - } - async init(nodeData: INodeData, _: string, options: ICommonObject): Promise { + async init(nodeData: INodeData, _: string): Promise { const system_prompt = nodeData.inputs?.system_prompt as string const obj: Partial = { - systemPrompt: system_prompt, + systemPrompt: system_prompt } - const model = new NIBittensorChatModel(obj) return model } diff --git a/packages/components/nodes/llms/Bittensor/Bittensor.ts b/packages/components/nodes/llms/Bittensor/Bittensor.ts index 86b400a4..a87a7e48 100644 --- a/packages/components/nodes/llms/Bittensor/Bittensor.ts +++ b/packages/components/nodes/llms/Bittensor/Bittensor.ts @@ -1,6 +1,6 @@ -import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface' +import { INode, INodeData, INodeParams } from '../../../src/Interface' import { getBaseClasses } from '../../../src/utils' -import {NIBittensorLLM, BittensorInput} from 'langchain/experimental/llms/bittensor'; +import { NIBittensorLLM, BittensorInput } from 'langchain/experimental/llms/bittensor' class Bittensor_LLMs implements INode { label: string @@ -39,15 +39,14 @@ class Bittensor_LLMs implements INode { additionalParams: true } ] - } - async init(nodeData: INodeData, _: string, options: ICommonObject): Promise { + async init(nodeData: INodeData, _: string): Promise { const system_prompt = nodeData.inputs?.system_prompt as string - const topResponses = Number(nodeData.inputs?.topResponses as number); + const topResponses = Number(nodeData.inputs?.topResponses as number) const obj: Partial = { systemPrompt: system_prompt, - topResponses: topResponses, + topResponses: topResponses } const model = new NIBittensorLLM(obj)