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 {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<any> {
async init(nodeData: INodeData, _: string): Promise<any> {
const system_prompt = nodeData.inputs?.system_prompt as string
const obj: Partial<BittensorInput> = {
systemPrompt: system_prompt,
systemPrompt: system_prompt
}
const model = new NIBittensorChatModel(obj)
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 {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<any> {
async init(nodeData: INodeData, _: string): Promise<any> {
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> = {
systemPrompt: system_prompt,
topResponses: topResponses,
topResponses: topResponses
}
const model = new NIBittensorLLM(obj)