This commit is contained in:
BitVoyagerMan
2023-09-21 07:12:28 -04:00
parent 50903453f0
commit 37d8b3f310
2 changed files with 10 additions and 13 deletions
@@ -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 { 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 { class Bittensor_ChatModels implements INode {
label: string label: string
@@ -29,17 +29,15 @@ class Bittensor_ChatModels implements INode {
type: 'string', type: 'string',
additionalParams: true, additionalParams: true,
optional: true optional: true
}, }
] ]
} }
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> { async init(nodeData: INodeData, _: string): Promise<any> {
const system_prompt = nodeData.inputs?.system_prompt as string const system_prompt = nodeData.inputs?.system_prompt as string
const obj: Partial<BittensorInput> = { const obj: Partial<BittensorInput> = {
systemPrompt: system_prompt, systemPrompt: system_prompt
} }
const model = new NIBittensorChatModel(obj) const model = new NIBittensorChatModel(obj)
return model return model
} }
@@ -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 { 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 { class Bittensor_LLMs implements INode {
label: string label: string
@@ -39,15 +39,14 @@ class Bittensor_LLMs implements INode {
additionalParams: true additionalParams: true
} }
] ]
} }
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> { async init(nodeData: INodeData, _: string): Promise<any> {
const system_prompt = nodeData.inputs?.system_prompt as string 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<BittensorInput> = { const obj: Partial<BittensorInput> = {
systemPrompt: system_prompt, systemPrompt: system_prompt,
topResponses: topResponses, topResponses: topResponses
} }
const model = new NIBittensorLLM(obj) const model = new NIBittensorLLM(obj)