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

refactor: move out metadata

Browse source
This commit is contained in:
Mutasem Aldmour 2024-12-17 17:52:01 +01:00
parent 13efb7fcce
commit 425d449445
No known key found for this signature in database
GPG key ID: 3DFA8122BB7FD6B8
1 changed files with 14 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
packages/@n8n/nodes-langchain/nodes/vector_store/shared/createVectorStoreNode.ts
View file

@ -217,6 +217,13 @@ export const createVectorStoreNode = (args: VectorStoreNodeConstructorArgs) =>
}, },
}, },
}, },
{
displayName: 'Include Metadata',
name: 'includeDocumentMetadata',
type: 'boolean',
default: true,
description: 'Whether or not to include document metadata',
},
// ID is always used for update operation // ID is always used for update operation
{ {
displayName: 'ID', displayName: 'ID',
@ -269,11 +276,16 @@ export const createVectorStoreNode = (args: VectorStoreNodeConstructorArgs) =>
topK, topK,
filter, filter,
); );
const includeDocumentMetadata = this.getNodeParameter(
'includeDocumentMetadata',
itemIndex,
true,
) as boolean;
const serializedDocs = docs.map(([doc, score]) => { const serializedDocs = docs.map(([doc, score]) => {
const document = { const document = {
metadata: doc.metadata,
pageContent: doc.pageContent, pageContent: doc.pageContent,
...(includeDocumentMetadata ? { metadata: doc.metadata } : {}),
}; };
return { return {
@ -409,7 +421,7 @@ export const createVectorStoreNode = (args: VectorStoreNodeConstructorArgs) =>
const includeDocumentMetadata = this.getNodeParameter( const includeDocumentMetadata = this.getNodeParameter(
'includeDocumentMetadata', 'includeDocumentMetadata',
itemIndex, itemIndex,
false, true,
) as boolean; ) as boolean;
const vectorStoreTool = new DynamicTool({ const vectorStoreTool = new DynamicTool({
@ -484,17 +496,6 @@ export const createVectorStoreNode = (args: VectorStoreNodeConstructorArgs) =>
// default: 'auto', // default: 'auto',
// }; // };
// todo move out
const metadataProp: INodeProperties = {
displayName: 'Include Metadata',
name: 'includeDocumentMetadata',
type: 'boolean',
default: false,
description: 'Whether or not to include document metadata',
};
description.properties.unshift(metadataProp);
const descProp: INodeProperties = { const descProp: INodeProperties = {
displayName: 'Description', displayName: 'Description',
name: 'toolDescription', name: 'toolDescription',