Sultan
/
Flowise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Addition of OutputParser for FewShotPromptTemplate

This commit is contained in:
vinodkiran 2023-10-25 19:52:50 +05:30
parent 4654f69951
commit b8b8f09bbc
2 changed files with 10 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
packages/components/nodes/chains/LLMChain/LLMChain.ts
View File

@ -4,7 +4,7 @@ import { LLMChain } from 'langchain/chains'
import { BaseLanguageModel } from 'langchain/base_language' import { BaseLanguageModel } from 'langchain/base_language'
import { ConsoleCallbackHandler, CustomChainHandler, additionalCallbacks } from '../../../src/handler' import { ConsoleCallbackHandler, CustomChainHandler, additionalCallbacks } from '../../../src/handler'
import { BaseOutputParser } from 'langchain/schema/output_parser' import { BaseOutputParser } from 'langchain/schema/output_parser'
import { ChatPromptTemplate, PromptTemplate, SystemMessagePromptTemplate } from 'langchain/prompts' import { ChatPromptTemplate, FewShotPromptTemplate, PromptTemplate, SystemMessagePromptTemplate } from 'langchain/prompts'
class LLMChain_Chains implements INode { class LLMChain_Chains implements INode {
label: string label: string
@ -99,29 +99,25 @@ class LLMChain_Chains implements INode {
const outputParser = nodeData.inputs?.outputParser as BaseOutputParser const outputParser = nodeData.inputs?.outputParser as BaseOutputParser
if (outputParser && chain.prompt) { if (outputParser && chain.prompt) {
const formatInstructions = outputParser.getFormatInstructions() const formatInstructions = outputParser.getFormatInstructions()
chain.prompt.inputVariables.push('format_instructions')
if (chain.prompt instanceof PromptTemplate) { if (chain.prompt instanceof PromptTemplate) {
let pt = chain.prompt let pt = chain.prompt
pt.template = pt.template + '\n{format_instructions}' pt.template = pt.template + '\n{format_instructions}'
chain.prompt.partialVariables = { format_instructions: formatInstructions } chain.prompt.partialVariables = { format_instructions: formatInstructions }
// eslint-disable-next-line no-console
console.log('prompt :: ', chain.prompt)
} else if (chain.prompt instanceof ChatPromptTemplate) { } else if (chain.prompt instanceof ChatPromptTemplate) {
let pt = chain.prompt let pt = chain.prompt
pt.promptMessages.forEach((msg) => { pt.promptMessages.forEach((msg) => {
if (msg instanceof SystemMessagePromptTemplate) { if (msg instanceof SystemMessagePromptTemplate) {
;(msg.prompt as any).partialVariables = { format_instructions: outputParser.getFormatInstructions() } ;(msg.prompt as any).partialVariables = { format_instructions: outputParser.getFormatInstructions() }
;(msg.prompt as any).template = ((msg.prompt as any).template + '\n{format_instructions}') as string ;(msg.prompt as any).template = ((msg.prompt as any).template + '\n{format_instructions}') as string
// eslint-disable-next-line no-console
console.log(msg)
} }
}) })
//pt.template = pt.template + '\n{format_instructions}' } else if (chain.prompt instanceof FewShotPromptTemplate) {
chain.prompt.examplePrompt.partialVariables = { format_instructions: formatInstructions }
chain.prompt.examplePrompt.template = chain.prompt.examplePrompt.template + '\n{format_instructions}'
} }
chain.prompt.inputVariables.push('format_instructions')
promptValues = { ...promptValues, format_instructions: outputParser.getFormatInstructions() } promptValues = { ...promptValues, format_instructions: outputParser.getFormatInstructions() }
// eslint-disable-next-line no-console
console.log('promptValues :: ', promptValues)
} }
const res = await runPrediction(inputVariables, chain, input, promptValues, options, nodeData, outputParser) const res = await runPrediction(inputVariables, chain, input, promptValues, options, nodeData, outputParser)
// eslint-disable-next-line no-console // eslint-disable-next-line no-console

10
packages/components/nodes/outputparsers/structured/StructuredOutputParser.ts
View File

@ -62,15 +62,15 @@ class StructuredOutputParser implements INode {
if (structure) { if (structure) {
try { try {
parsedStructure = JSON.parse(structure) parsedStructure = JSON.parse(structure)
if (structureType === 'fromZodSchema') {
return LangchainStructuredOutputParser.fromZodSchema(parsedStructure)
} else {
return LangchainStructuredOutputParser.fromNamesAndDescriptions(parsedStructure)
}
} catch (exception) { } catch (exception) {
throw new Error('Invalid JSON in StructuredOutputParser: ' + exception) throw new Error('Invalid JSON in StructuredOutputParser: ' + exception)
} }
} }
if (structureType === 'fromZodSchema') {
return LangchainStructuredOutputParser.fromZodSchema(parsedStructure)
} else {
return LangchainStructuredOutputParser.fromNamesAndDescriptions(parsedStructure)
}
} }
} }