mudhorn/n8n
1
0
Fork 0
mirror of https://github.com/n8n-io/n8n.git synced 2025-01-12 13:27:31 -08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(Text Classifier Node): Default system prompt template (#11018)

Browse source
This commit is contained in:
Eugene 2024-09-30 15:07:20 +02:00 committed by GitHub
parent 3ff3e05e75
commit 77fec195d9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
packages/@n8n/nodes-langchain/nodes/chains/TextClassifier/TextClassifier.node.ts
View file

@ -1,3 +1,8 @@
import type { BaseLanguageModel } from '@langchain/core/language_models/base';
import { HumanMessage } from '@langchain/core/messages';
import { SystemMessagePromptTemplate, ChatPromptTemplate } from '@langchain/core/prompts';
import { OutputFixingParser, StructuredOutputParser } from 'langchain/output_parsers';
import { NodeOperationError, NodeConnectionType } from 'n8n-workflow';
import type { import type {
IDataObject, IDataObject,
IExecuteFunctions, IExecuteFunctions,
@ -6,14 +11,8 @@ import type {
INodeType, INodeType,
INodeTypeDescription, INodeTypeDescription,
} from 'n8n-workflow'; } from 'n8n-workflow';
import { NodeConnectionType } from 'n8n-workflow';
import type { BaseLanguageModel } from '@langchain/core/language_models/base';
import { HumanMessage } from '@langchain/core/messages';
import { SystemMessagePromptTemplate, ChatPromptTemplate } from '@langchain/core/prompts';
import { OutputFixingParser, StructuredOutputParser } from 'langchain/output_parsers';
import { z } from 'zod'; import { z } from 'zod';
import { getTracingConfig } from '../../../utils/tracing'; import { getTracingConfig } from '../../../utils/tracing';
const SYSTEM_PROMPT_TEMPLATE = const SYSTEM_PROMPT_TEMPLATE =
@ -172,11 +171,15 @@ export class TextClassifier implements INodeType {
0, 0,
)) as BaseLanguageModel; )) as BaseLanguageModel;
const categories = this.getNodeParameter('categories.categories', 0) as Array<{ const categories = this.getNodeParameter('categories.categories', 0, []) as Array<{
category: string; category: string;
description: string; description: string;
}>; }>;
if (categories.length === 0) {
throw new NodeOperationError(this.getNode(), 'At least one category must be defined');
}
const options = this.getNodeParameter('options', 0, {}) as { const options = this.getNodeParameter('options', 0, {}) as {
multiClass: boolean; multiClass: boolean;
fallback?: string; fallback?: string;
@ -229,6 +232,7 @@ export class TextClassifier implements INodeType {
const systemPromptTemplateOpt = this.getNodeParameter( const systemPromptTemplateOpt = this.getNodeParameter(
'options.systemPromptTemplate', 'options.systemPromptTemplate',
itemIdx, itemIdx,
SYSTEM_PROMPT_TEMPLATE,
) as string; ) as string;
const systemPromptTemplate = SystemMessagePromptTemplate.fromTemplate( const systemPromptTemplate = SystemMessagePromptTemplate.fromTemplate(
`${systemPromptTemplateOpt ?? SYSTEM_PROMPT_TEMPLATE} `${systemPromptTemplateOpt ?? SYSTEM_PROMPT_TEMPLATE}