first commit

packages/@n8n/nodes-langchain/nodes/vector_store/shared/descriptions.ts

@@ -0,0 +1,138 @@
+import type { INodeProperties } from 'n8n-workflow';
+
+export const pineconeIndexRLC: INodeProperties = {
+	displayName: 'Pinecone Index',
+	name: 'pineconeIndex',
+	type: 'resourceLocator',
+	default: { mode: 'list', value: '' },
+	required: true,
+	modes: [
+		{
+			displayName: 'From List',
+			name: 'list',
+			type: 'list',
+			typeOptions: {
+				searchListMethod: 'pineconeIndexSearch',
+			},
+		},
+		{
+			displayName: 'ID',
+			name: 'id',
+			type: 'string',
+		},
+	],
+};
+
+export const supabaseTableNameRLC: INodeProperties = {
+	displayName: 'Table Name',
+	name: 'tableName',
+	type: 'resourceLocator',
+	default: { mode: 'list', value: '' },
+	required: true,
+	modes: [
+		{
+			displayName: 'From List',
+			name: 'list',
+			type: 'list',
+			typeOptions: {
+				searchListMethod: 'supabaseTableNameSearch',
+			},
+		},
+		{
+			displayName: 'ID',
+			name: 'id',
+			type: 'string',
+		},
+	],
+};
+
+export const qdrantCollectionRLC: INodeProperties = {
+	displayName: 'Qdrant Collection',
+	name: 'qdrantCollection',
+	type: 'resourceLocator',
+	default: { mode: 'list', value: '' },
+	required: true,
+	modes: [
+		{
+			displayName: 'From List',
+			name: 'list',
+			type: 'list',
+			typeOptions: {
+				searchListMethod: 'qdrantCollectionsSearch',
+			},
+		},
+		{
+			displayName: 'ID',
+			name: 'id',
+			type: 'string',
+		},
+	],
+};
+
+export const milvusCollectionRLC: INodeProperties = {
+	displayName: 'Milvus Collection',
+	name: 'milvusCollection',
+	type: 'resourceLocator',
+	default: { mode: 'list', value: '' },
+	required: true,
+	modes: [
+		{
+			displayName: 'From List',
+			name: 'list',
+			type: 'list',
+			typeOptions: {
+				searchListMethod: 'milvusCollectionsSearch',
+			},
+		},
+		{
+			displayName: 'ID',
+			name: 'id',
+			type: 'string',
+		},
+	],
+};
+
+export const weaviateCollectionRLC: INodeProperties = {
+	displayName: 'Weaviate Collection',
+	name: 'weaviateCollection',
+	type: 'resourceLocator',
+	default: { mode: 'list', value: '' },
+	required: true,
+	modes: [
+		{
+			displayName: 'From List',
+			name: 'list',
+			type: 'list',
+			typeOptions: {
+				searchListMethod: 'weaviateCollectionsSearch',
+			},
+		},
+		{
+			displayName: 'ID',
+			name: 'id',
+			type: 'string',
+		},
+	],
+};
+export const chromaCollectionRLC: INodeProperties = {
+	displayName: 'Chroma Collection',
+	name: 'chromaCollection',
+	type: 'resourceLocator',
+	default: { mode: 'list', value: '' },
+	required: true,
+	modes: [
+		{
+			displayName: 'From List',
+			name: 'list',
+			type: 'list',
+			typeOptions: {
+				searchListMethod: 'chromaCollectionsSearch',
+			},
+		},
+		{
+			displayName: 'ID',
+			name: 'id',
+			type: 'string',
+		},
+	],
+};

packages/@n8n/nodes-langchain/nodes/vector_store/shared/methods/listSearch.ts

@@ -0,0 +1,108 @@
+import { Pinecone } from '@pinecone-database/pinecone';
+import { MilvusClient } from '@zilliz/milvus2-sdk-node';
+import { ApplicationError, type IDataObject, type ILoadOptionsFunctions } from 'n8n-workflow';
+
+import type { QdrantCredential } from '../../VectorStoreQdrant/Qdrant.utils';
+import { createQdrantClient } from '../../VectorStoreQdrant/Qdrant.utils';
+import type { WeaviateCredential } from '../../VectorStoreWeaviate/Weaviate.utils';
+import { createWeaviateClient } from '../../VectorStoreWeaviate/Weaviate.utils';
+
+export async function pineconeIndexSearch(this: ILoadOptionsFunctions) {
+	const credentials = await this.getCredentials('pineconeApi');
+
+	const client = new Pinecone({
+		apiKey: credentials.apiKey as string,
+	});
+
+	const indexes = await client.listIndexes();
+
+	const results = (indexes.indexes ?? []).map((index) => ({
+		name: index.name,
+		value: index.name,
+	}));
+
+	return { results };
+}
+
+export async function supabaseTableNameSearch(this: ILoadOptionsFunctions) {
+	const credentials = await this.getCredentials('supabaseApi');
+
+	const results = [];
+
+	if (typeof credentials.host !== 'string') {
+		throw new ApplicationError('Expected Supabase credentials host to be a string');
+	}
+
+	const { paths } = (await this.helpers.requestWithAuthentication.call(this, 'supabaseApi', {
+		headers: {
+			Prefer: 'return=representation',
+		},
+		method: 'GET',
+		uri: `${credentials.host}/rest/v1/`,
+		json: true,
+	})) as { paths: IDataObject };
+
+	for (const path of Object.keys(paths)) {
+		//omit introspection path
+		if (path === '/') continue;
+
+		results.push({
+			name: path.replace('/', ''),
+			value: path.replace('/', ''),
+		});
+	}
+
+	return { results };
+}
+
+export async function qdrantCollectionsSearch(this: ILoadOptionsFunctions) {
+	const credentials = await this.getCredentials('qdrantApi');
+
+	const client = createQdrantClient(credentials as QdrantCredential);
+
+	const response = await client.getCollections();
+
+	const results = response.collections.map((collection) => ({
+		name: collection.name,
+		value: collection.name,
+	}));
+
+	return { results };
+}
+
+export async function milvusCollectionsSearch(this: ILoadOptionsFunctions) {
+	const credentials = await this.getCredentials<{
+		baseUrl: string;
+		username: string;
+		password: string;
+	}>('milvusApi');
+
+	const client = new MilvusClient({
+		address: credentials.baseUrl,
+		token: `${credentials.username}:${credentials.password}`,
+	});
+
+	const response = await client.listCollections();
+
+	const results = response.data.map((collection) => ({
+		name: collection.name,
+		value: collection.name,
+	}));
+
+	return { results };
+}
+
+export async function weaviateCollectionsSearch(this: ILoadOptionsFunctions) {
+	const credentials = await this.getCredentials('weaviateApi');
+
+	const client = await createWeaviateClient(credentials as WeaviateCredential);
+
+	const collections = await client.collections.listAll();
+
+	const results = collections.map((collection: { name: string }) => ({
+		name: collection.name,
+		value: collection.name,
+	}));
+
+	return { results };
+}