Fix data store name referential integrity in agent flows

Keep original format for backward compatibility while fetching current store name from database to prevent UI breaks when document store names are updated.

Author: naaa760

packages/components/nodes/agentflow/Agent/Agent.ts

@@ -433,8 +433,8 @@ class Agent_Agentflow implements INode {
             for (const store of stores) {
                 if (store.status === 'UPSERTED') {
                     const obj = {
-                        name: store.id, // Store only the ID, not the name
-                        label: store.name, // Display the current name
+                        name: `${store.id}:${store.name}`,
+                        label: store.name,
                         description: store.description
                     }
                     returnData.push(obj)
@@ -564,6 +564,12 @@ class Agent_Agentflow implements INode {
                         storeName = store?.name || storeId // Fallback to ID if store not found
                     }
 
+                    // Always fetch the current store name from database for referential integrity
+                    const appDataSource = options.appDataSource as DataSource
+                    const databaseEntities = options.databaseEntities as IDatabaseEntity
+                    const store = await appDataSource.getRepository(databaseEntities['DocumentStore']).findOneBy({ id: storeId })
+                    const currentStoreName = store?.name || storeId // Use current name from database
+
                     const docStoreVectorInstanceFilePath = options.componentNodes['documentStoreVS'].filePath as string
                     const docStoreVectorModule = await import(docStoreVectorInstanceFilePath)
                     const newDocStoreVectorInstance = new docStoreVectorModule.nodeClass()
@@ -586,7 +592,7 @@ class Agent_Agentflow implements INode {
                         ...nodeData,
                         inputs: {
                             ...nodeData.inputs,
-                            name: storeName
+                            name: currentStoreName
                                 .toLowerCase()
                                 .replace(/ /g, '_')
                                 .replace(/[^a-z0-9_-]/g, ''),
@@ -606,7 +612,7 @@ class Agent_Agentflow implements INode {
                     const componentNode = options.componentNodes['retrieverTool']
 
                     availableTools.push({
-                        name: storeName
+                        name: currentStoreName
                             .toLowerCase()
                             .replace(/ /g, '_')
                             .replace(/[^a-z0-9_-]/g, ''),

packages/components/nodes/agentflow/Retriever/Retriever.ts

@@ -124,8 +124,8 @@ class Retriever_Agentflow implements INode {
             for (const store of stores) {
                 if (store.status === 'UPSERTED') {
                     const obj = {
-                        name: store.id, // please store only the ID, not the name
-                        label: store.name, // display the current name
+                        name: `${store.id}:${store.name}`,
+                        label: store.name,
                         description: store.description
                     }
                     returnData.push(obj)
@@ -152,16 +152,7 @@ class Retriever_Agentflow implements INode {
         const knowledgeBases = nodeData.inputs?.retrieverKnowledgeDocumentStores as IKnowledgeBase[]
         if (knowledgeBases && knowledgeBases.length > 0) {
             for (const knowledgeBase of knowledgeBases) {
-                // Handle both old format (id:name) and new format (id only)
-                let storeId = knowledgeBase.documentStore
-
-                if (knowledgeBase.documentStore.includes(':')) {
-                    // Old format - split to get ID
-                    ;[storeId] = knowledgeBase.documentStore.split(':')
-                } else {
-                    // New format - only ID is stored
-                    storeId = knowledgeBase.documentStore
-                }
+                const [storeId] = knowledgeBase.documentStore.split(':')
 
                 const docStoreVectorInstanceFilePath = options.componentNodes['documentStoreVS'].filePath as string
                 const docStoreVectorModule = await import(docStoreVectorInstanceFilePath)