mudhorn/n8n
1
0
Fork 0
mirror of https://github.com/n8n-io/n8n.git synced 2025-03-05 20:50:17 -08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixed lint problems on Question and Answer Chain

Browse source
This commit is contained in:
mp 2024-08-13 15:16:42 -03:00
parent 830ba21de0
commit 0773bcd997
1 changed files with 24 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
packages/@n8n/nodes-langchain/nodes/chains/ChainRetrievalQA/ChainRetrievalQa.node.ts
View file

@ -14,13 +14,14 @@ import {
ChatPromptTemplate, ChatPromptTemplate,
SystemMessagePromptTemplate, SystemMessagePromptTemplate,
HumanMessagePromptTemplate, HumanMessagePromptTemplate,
PromptTemplate PromptTemplate,
} from "@langchain/core/prompts"; } from '@langchain/core/prompts';
import { getTemplateNoticeField } from '../../../utils/sharedFields'; import { getTemplateNoticeField } from '../../../utils/sharedFields';
import { getPromptInputByType } from '../../../utils/helpers'; import { getPromptInputByType } from '../../../utils/helpers';
import { getTracingConfig } from '../../../utils/tracing'; import { getTracingConfig } from '../../../utils/tracing';
const QA_PROMPT_TEMPLATE = "Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\n\n{context}\n\nQuestion: {question}\nHelpful Answer:"; const QA_PROMPT_TEMPLATE =
"Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\n\n{context}\n\nQuestion: {question}\nHelpful Answer:";
const CHAT_PROMPT_TEMPLATE = `Use the following pieces of context to answer the users question. const CHAT_PROMPT_TEMPLATE = `Use the following pieces of context to answer the users question.
If you don't know the answer, just say that you don't know, don't try to make up an answer. If you don't know the answer, just say that you don't know, don't try to make up an answer.
---------------- ----------------
@ -169,9 +170,9 @@ export class ChainRetrievalQa implements INodeType {
description: 'Uses a standard prompt template (for non-Chat Models)', description: 'Uses a standard prompt template (for non-Chat Models)',
}, },
{ {
name: "Chat Prompt", name: 'Chat Prompt',
value: 'chatPrompt', value: 'chatPrompt',
description: "Uses a system message template (for Chat Models)", description: 'Uses a system message template (for Chat Models)',
}, },
], ],
displayOptions: { displayOptions: {
@ -200,7 +201,8 @@ export class ChainRetrievalQa implements INodeType {
name: 'chatPromptTemplate', name: 'chatPromptTemplate',
type: 'string', type: 'string',
default: CHAT_PROMPT_TEMPLATE, default: CHAT_PROMPT_TEMPLATE,
description: 'Template string for the Question and Answer prompt as a system message (for Chat Models)', description:
'Template string for the Question and Answer prompt as a system message (for Chat Models)',
typeOptions: { typeOptions: {
rows: 8, rows: 8,
}, },
@ -229,38 +231,43 @@ export class ChainRetrievalQa implements INodeType {
const items = this.getInputData(); const items = this.getInputData();
const customQAPrompt = this.getNodeParameter('customQAPrompt', 0, false) as boolean; const customQAPrompt = this.getNodeParameter('customQAPrompt', 0, false) as boolean;
const chainParameters = {} as { const chainParameters = {} as {
prompt?: PromptTemplate | ChatPromptTemplate; prompt?: PromptTemplate | ChatPromptTemplate;
}; };
if(customQAPrompt){ if (customQAPrompt) {
const qAPromptType = this.getNodeParameter('qAPromptType', 0) as string; const qAPromptType = this.getNodeParameter('qAPromptType', 0) as string;
if(qAPromptType == 'standardPrompt'){ if (qAPromptType == 'standardPrompt') {
const standardPromptTemplateParameter = this.getNodeParameter('standardPromptTemplate', 0) as string; const standardPromptTemplateParameter = this.getNodeParameter(
'standardPromptTemplate',
0,
) as string;
const standardPromptTemplate = new PromptTemplate({ const standardPromptTemplate = new PromptTemplate({
template: standardPromptTemplateParameter, template: standardPromptTemplateParameter,
inputVariables: ['context', 'question'] inputVariables: ['context', 'question'],
}); });
chainParameters.prompt = standardPromptTemplate; chainParameters.prompt = standardPromptTemplate;
} } else if (qAPromptType == 'chatPrompt') {
else if(qAPromptType == 'chatPrompt'){ const chatPromptTemplateParameter = this.getNodeParameter(
const chatPromptTemplateParameter = this.getNodeParameter('chatPromptTemplate', 0) as string; 'chatPromptTemplate',
0,
) as string;
const messages = [ const messages = [
SystemMessagePromptTemplate.fromTemplate(chatPromptTemplateParameter), SystemMessagePromptTemplate.fromTemplate(chatPromptTemplateParameter),
HumanMessagePromptTemplate.fromTemplate("{question}"), HumanMessagePromptTemplate.fromTemplate('{question}'),
]; ];
const chatPromptTemplate = ChatPromptTemplate.fromMessages(messages); const chatPromptTemplate = ChatPromptTemplate.fromMessages(messages);
chainParameters.prompt = chatPromptTemplate; chainParameters.prompt = chatPromptTemplate;
} }
} }
const chain = RetrievalQAChain.fromLLM(model, retriever, chainParameters); const chain = RetrievalQAChain.fromLLM(model, retriever, chainParameters);
const returnData: INodeExecutionData[] = []; const returnData: INodeExecutionData[] = [];