add additional systemMessagePrompt into CSV Agent
This commit is contained in:
parent
d208eae868
commit
45ffa62f0f
|
|
@ -37,6 +37,16 @@ class CSV_Agents implements INode {
|
||||||
label: 'Language Model',
|
label: 'Language Model',
|
||||||
name: 'model',
|
name: 'model',
|
||||||
type: 'BaseLanguageModel'
|
type: 'BaseLanguageModel'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
label: 'System Message',
|
||||||
|
name: 'systemMessagePrompt',
|
||||||
|
type: 'string',
|
||||||
|
rows: 4,
|
||||||
|
additionalParams: true,
|
||||||
|
optional: true,
|
||||||
|
placeholder:
|
||||||
|
'I want you to act as a document that I am having a conversation with. Your name is "AI Assistant". You will provide me with answers from the given info. If the answer is not included, say exactly "Hmm, I am not sure." and stop after that. Refuse to answer any question not about the info. Never break character.'
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
|
|
@ -49,6 +59,7 @@ class CSV_Agents implements INode {
|
||||||
async run(nodeData: INodeData, input: string, options: ICommonObject): Promise<string> {
|
async run(nodeData: INodeData, input: string, options: ICommonObject): Promise<string> {
|
||||||
const csvFileBase64 = nodeData.inputs?.csvFile as string
|
const csvFileBase64 = nodeData.inputs?.csvFile as string
|
||||||
const model = nodeData.inputs?.model as BaseLanguageModel
|
const model = nodeData.inputs?.model as BaseLanguageModel
|
||||||
|
const systemMessagePrompt = nodeData.inputs?.systemMessagePrompt as string
|
||||||
|
|
||||||
const loggerHandler = new ConsoleCallbackHandler(options.logger)
|
const loggerHandler = new ConsoleCallbackHandler(options.logger)
|
||||||
const handler = new CustomChainHandler(options.socketIO, options.socketIOClientId)
|
const handler = new CustomChainHandler(options.socketIO, options.socketIOClientId)
|
||||||
|
|
@ -127,7 +138,9 @@ json.dumps(my_dict)`
|
||||||
if (finalResult) {
|
if (finalResult) {
|
||||||
const chain = new LLMChain({
|
const chain = new LLMChain({
|
||||||
llm: model,
|
llm: model,
|
||||||
prompt: PromptTemplate.fromTemplate(finalSystemPrompt),
|
prompt: PromptTemplate.fromTemplate(
|
||||||
|
systemMessagePrompt ? `${systemMessagePrompt}\n${finalSystemPrompt}` : finalSystemPrompt
|
||||||
|
),
|
||||||
verbose: process.env.DEBUG === 'true' ? true : false
|
verbose: process.env.DEBUG === 'true' ? true : false
|
||||||
})
|
})
|
||||||
const inputs = {
|
const inputs = {
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue