refactor: Migrate NodeConnectionType to const object type (no-changelog) (#14078)

Co-authored-by: कारतोफ्फेलस्क्रिप्ट™ <aditya@netroy.in>
This commit is contained in:
Alex Grozav
2025-03-21 14:01:26 +02:00
committed by GitHub
parent 7e8179b848
commit 8215e0b59f
703 changed files with 3104 additions and 3018 deletions

View File

@@ -5,7 +5,7 @@ import type { BaseRetriever } from '@langchain/core/retrievers';
import { ContextualCompressionRetriever } from 'langchain/retrievers/contextual_compression';
import { LLMChainExtractor } from 'langchain/retrievers/document_compressors/chain_extract';
import {
NodeConnectionType,
NodeConnectionTypes,
type INodeType,
type INodeTypeDescription,
type ISupplyDataFunctions,
@@ -44,13 +44,13 @@ export class RetrieverContextualCompression implements INodeType {
{
displayName: 'Model',
maxConnections: 1,
type: NodeConnectionType.AiLanguageModel,
type: NodeConnectionTypes.AiLanguageModel,
required: true,
},
{
displayName: 'Retriever',
maxConnections: 1,
type: NodeConnectionType.AiRetriever,
type: NodeConnectionTypes.AiRetriever,
required: true,
},
],
@@ -58,7 +58,7 @@ export class RetrieverContextualCompression implements INodeType {
{
displayName: 'Retriever',
maxConnections: 1,
type: NodeConnectionType.AiRetriever,
type: NodeConnectionTypes.AiRetriever,
},
],
properties: [],
@@ -68,12 +68,12 @@ export class RetrieverContextualCompression implements INodeType {
this.logger.debug('Supplying data for Contextual Compression Retriever');
const model = (await this.getInputConnectionData(
NodeConnectionType.AiLanguageModel,
NodeConnectionTypes.AiLanguageModel,
itemIndex,
)) as BaseLanguageModel;
const baseRetriever = (await this.getInputConnectionData(
NodeConnectionType.AiRetriever,
NodeConnectionTypes.AiRetriever,
itemIndex,
)) as BaseRetriever;

View File

@@ -4,7 +4,7 @@ import type { BaseLanguageModel } from '@langchain/core/language_models/base';
import type { BaseRetriever } from '@langchain/core/retrievers';
import { MultiQueryRetriever } from 'langchain/retrievers/multi_query';
import {
NodeConnectionType,
NodeConnectionTypes,
type INodeType,
type INodeTypeDescription,
type ISupplyDataFunctions,
@@ -44,13 +44,13 @@ export class RetrieverMultiQuery implements INodeType {
{
displayName: 'Model',
maxConnections: 1,
type: NodeConnectionType.AiLanguageModel,
type: NodeConnectionTypes.AiLanguageModel,
required: true,
},
{
displayName: 'Retriever',
maxConnections: 1,
type: NodeConnectionType.AiRetriever,
type: NodeConnectionTypes.AiRetriever,
required: true,
},
],
@@ -58,7 +58,7 @@ export class RetrieverMultiQuery implements INodeType {
{
displayName: 'Retriever',
maxConnections: 1,
type: NodeConnectionType.AiRetriever,
type: NodeConnectionTypes.AiRetriever,
},
],
properties: [
@@ -89,12 +89,12 @@ export class RetrieverMultiQuery implements INodeType {
const options = this.getNodeParameter('options', itemIndex, {}) as { queryCount?: number };
const model = (await this.getInputConnectionData(
NodeConnectionType.AiLanguageModel,
NodeConnectionTypes.AiLanguageModel,
itemIndex,
)) as BaseLanguageModel;
const baseRetriever = (await this.getInputConnectionData(
NodeConnectionType.AiRetriever,
NodeConnectionTypes.AiRetriever,
itemIndex,
)) as BaseRetriever;

View File

@@ -1,7 +1,7 @@
/* eslint-disable n8n-nodes-base/node-dirname-against-convention */
import type { VectorStore } from '@langchain/core/vectorstores';
import {
NodeConnectionType,
NodeConnectionTypes,
type INodeType,
type INodeTypeDescription,
type ISupplyDataFunctions,
@@ -40,12 +40,12 @@ export class RetrieverVectorStore implements INodeType {
{
displayName: 'Vector Store',
maxConnections: 1,
type: NodeConnectionType.AiVectorStore,
type: NodeConnectionTypes.AiVectorStore,
required: true,
},
],
// eslint-disable-next-line n8n-nodes-base/node-class-description-outputs-wrong
outputs: [NodeConnectionType.AiRetriever],
outputs: [NodeConnectionTypes.AiRetriever],
outputNames: ['Retriever'],
properties: [
{
@@ -63,7 +63,7 @@ export class RetrieverVectorStore implements INodeType {
const topK = this.getNodeParameter('topK', itemIndex, 4) as number;
const vectorStore = (await this.getInputConnectionData(
NodeConnectionType.AiVectorStore,
NodeConnectionTypes.AiVectorStore,
itemIndex,
)) as VectorStore;

View File

@@ -4,7 +4,7 @@ import { Document } from '@langchain/core/documents';
import { BaseRetriever, type BaseRetrieverInput } from '@langchain/core/retrievers';
import type { SetField, SetNodeOptions } from 'n8n-nodes-base/dist/nodes/Set/v2/helpers/interfaces';
import * as manual from 'n8n-nodes-base/dist/nodes/Set/v2/manual.mode';
import { NodeConnectionType, NodeOperationError } from 'n8n-workflow';
import { NodeConnectionTypes, NodeOperationError } from 'n8n-workflow';
import type {
IDataObject,
IExecuteWorkflowInfo,
@@ -66,7 +66,7 @@ export class RetrieverWorkflow implements INodeType {
{
displayName: 'Retriever',
maxConnections: 1,
type: NodeConnectionType.AiRetriever,
type: NodeConnectionTypes.AiRetriever,
},
],
properties: [