@kyuda/n8n-nodes-databricks 1.0.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of @kyuda/n8n-nodes-databricks might be problematic. Click here for more details.
- package/LICENSE.md +19 -0
- package/README.md +478 -0
- package/dist/credentials/Databricks.credentials.d.ts +10 -0
- package/dist/credentials/Databricks.credentials.js +51 -0
- package/dist/credentials/Databricks.credentials.js.map +1 -0
- package/dist/credentials/databricks.svg +19 -0
- package/dist/nodes/Databricks/Databricks.node.d.ts +16 -0
- package/dist/nodes/Databricks/Databricks.node.js +1719 -0
- package/dist/nodes/Databricks/Databricks.node.js.map +1 -0
- package/dist/nodes/Databricks/databricks.dark.svg +19 -0
- package/dist/nodes/Databricks/databricks.svg +19 -0
- package/dist/nodes/Databricks/resources/databricksSql/operations.d.ts +2 -0
- package/dist/nodes/Databricks/resources/databricksSql/operations.js +24 -0
- package/dist/nodes/Databricks/resources/databricksSql/operations.js.map +1 -0
- package/dist/nodes/Databricks/resources/databricksSql/parameters.d.ts +2 -0
- package/dist/nodes/Databricks/resources/databricksSql/parameters.js +94 -0
- package/dist/nodes/Databricks/resources/databricksSql/parameters.js.map +1 -0
- package/dist/nodes/Databricks/resources/databricksSql.d.ts +3 -0
- package/dist/nodes/Databricks/resources/databricksSql.js +130 -0
- package/dist/nodes/Databricks/resources/databricksSql.js.map +1 -0
- package/dist/nodes/Databricks/resources/files/operations.d.ts +2 -0
- package/dist/nodes/Databricks/resources/files/operations.js +71 -0
- package/dist/nodes/Databricks/resources/files/operations.js.map +1 -0
- package/dist/nodes/Databricks/resources/files/parameters.d.ts +2 -0
- package/dist/nodes/Databricks/resources/files/parameters.js +124 -0
- package/dist/nodes/Databricks/resources/files/parameters.js.map +1 -0
- package/dist/nodes/Databricks/resources/files.d.ts +3 -0
- package/dist/nodes/Databricks/resources/files.js +183 -0
- package/dist/nodes/Databricks/resources/files.js.map +1 -0
- package/dist/nodes/Databricks/resources/genie/operations.d.ts +2 -0
- package/dist/nodes/Databricks/resources/genie/operations.js +96 -0
- package/dist/nodes/Databricks/resources/genie/operations.js.map +1 -0
- package/dist/nodes/Databricks/resources/genie/parameters.d.ts +2 -0
- package/dist/nodes/Databricks/resources/genie/parameters.js +96 -0
- package/dist/nodes/Databricks/resources/genie/parameters.js.map +1 -0
- package/dist/nodes/Databricks/resources/index.d.ts +12 -0
- package/dist/nodes/Databricks/resources/index.js +29 -0
- package/dist/nodes/Databricks/resources/index.js.map +1 -0
- package/dist/nodes/Databricks/resources/modelServing/operations.d.ts +2 -0
- package/dist/nodes/Databricks/resources/modelServing/operations.js +24 -0
- package/dist/nodes/Databricks/resources/modelServing/operations.js.map +1 -0
- package/dist/nodes/Databricks/resources/modelServing/parameters.d.ts +2 -0
- package/dist/nodes/Databricks/resources/modelServing/parameters.js +76 -0
- package/dist/nodes/Databricks/resources/modelServing/parameters.js.map +1 -0
- package/dist/nodes/Databricks/resources/modelServing.d.ts +3 -0
- package/dist/nodes/Databricks/resources/modelServing.js +176 -0
- package/dist/nodes/Databricks/resources/modelServing.js.map +1 -0
- package/dist/nodes/Databricks/resources/unityCatalog/operations.d.ts +2 -0
- package/dist/nodes/Databricks/resources/unityCatalog/operations.js +233 -0
- package/dist/nodes/Databricks/resources/unityCatalog/operations.js.map +1 -0
- package/dist/nodes/Databricks/resources/unityCatalog/parameters.d.ts +2 -0
- package/dist/nodes/Databricks/resources/unityCatalog/parameters.js +371 -0
- package/dist/nodes/Databricks/resources/unityCatalog/parameters.js.map +1 -0
- package/dist/nodes/Databricks/resources/unityCatalog.d.ts +3 -0
- package/dist/nodes/Databricks/resources/unityCatalog.js +239 -0
- package/dist/nodes/Databricks/resources/unityCatalog.js.map +1 -0
- package/dist/nodes/Databricks/resources/vectorSearch/operations.d.ts +2 -0
- package/dist/nodes/Databricks/resources/vectorSearch/operations.js +48 -0
- package/dist/nodes/Databricks/resources/vectorSearch/operations.js.map +1 -0
- package/dist/nodes/Databricks/resources/vectorSearch/parameters.d.ts +2 -0
- package/dist/nodes/Databricks/resources/vectorSearch/parameters.js +224 -0
- package/dist/nodes/Databricks/resources/vectorSearch/parameters.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/CallbackHandler.d.ts +66 -0
- package/dist/nodes/agents/DatabricksAiAgent/CallbackHandler.js +531 -0
- package/dist/nodes/agents/DatabricksAiAgent/CallbackHandler.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/DatabricksAiAgent.node.d.ts +5 -0
- package/dist/nodes/agents/DatabricksAiAgent/DatabricksAiAgent.node.js +133 -0
- package/dist/nodes/agents/DatabricksAiAgent/DatabricksAiAgent.node.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/DatabricksAiAgentDark.icon.svg +9 -0
- package/dist/nodes/agents/DatabricksAiAgent/DatabricksAiAgentLight.icon.svg +9 -0
- package/dist/nodes/agents/DatabricksAiAgent/description.d.ts +4 -0
- package/dist/nodes/agents/DatabricksAiAgent/description.js +28 -0
- package/dist/nodes/agents/DatabricksAiAgent/description.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/execute.d.ts +2 -0
- package/dist/nodes/agents/DatabricksAiAgent/execute.js +418 -0
- package/dist/nodes/agents/DatabricksAiAgent/execute.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/constants.d.ts +82 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/constants.js +105 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/constants.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/types/types.d.ts +6 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/types/types.js +3 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/types/types.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/N8nOutputParser.d.ts +5 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/N8nOutputParser.js +13 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/N8nOutputParser.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/N8nTool.d.ts +10 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/N8nTool.js +88 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/N8nTool.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/common.d.ts +28 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/common.js +207 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/common.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/descriptions.d.ts +17 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/descriptions.js +135 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/descriptions.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/helpers.d.ts +24 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/helpers.js +221 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/helpers.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/options.d.ts +2 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/options.js +38 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/options.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/output_parsers/N8nStructuredOutputParser.d.ts +12 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/output_parsers/N8nStructuredOutputParser.js +106 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/output_parsers/N8nStructuredOutputParser.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/prompt.d.ts +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/prompt.js +5 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/prompt.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/retry.d.ts +16 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/retry.js +132 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/retry.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/security.d.ts +18 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/security.js +150 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/security.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/sharedFields.d.ts +7 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/sharedFields.js +150 -0
- package/dist/nodes/agents/DatabricksAiAgent/src/utils/sharedFields.js.map +1 -0
- package/dist/nodes/agents/DatabricksAiAgent/utils.d.ts +2 -0
- package/dist/nodes/agents/DatabricksAiAgent/utils.js +52 -0
- package/dist/nodes/agents/DatabricksAiAgent/utils.js.map +1 -0
- package/dist/nodes/embeddings/EmbeddingsDatabricks/EmbeddingsDatabricks.node.d.ts +5 -0
- package/dist/nodes/embeddings/EmbeddingsDatabricks/EmbeddingsDatabricks.node.js +155 -0
- package/dist/nodes/embeddings/EmbeddingsDatabricks/EmbeddingsDatabricks.node.js.map +1 -0
- package/dist/nodes/embeddings/EmbeddingsDatabricks/databricks.dark.svg +19 -0
- package/dist/nodes/embeddings/EmbeddingsDatabricks/databricks.svg +19 -0
- package/dist/nodes/llms/LmChatDatabricks/LmChatDatabricks.node.d.ts +10 -0
- package/dist/nodes/llms/LmChatDatabricks/LmChatDatabricks.node.js +265 -0
- package/dist/nodes/llms/LmChatDatabricks/LmChatDatabricks.node.js.map +1 -0
- package/dist/nodes/llms/LmChatDatabricks/databricks.dark.svg +19 -0
- package/dist/nodes/llms/LmChatDatabricks/databricks.svg +19 -0
- package/dist/nodes/llms/N8nLlmTracing.d.ts +44 -0
- package/dist/nodes/llms/N8nLlmTracing.js +140 -0
- package/dist/nodes/llms/N8nLlmTracing.js.map +1 -0
- package/dist/nodes/llms/n8nDefaultFailedAttemptHandler.d.ts +1 -0
- package/dist/nodes/llms/n8nDefaultFailedAttemptHandler.js +31 -0
- package/dist/nodes/llms/n8nDefaultFailedAttemptHandler.js.map +1 -0
- package/dist/nodes/llms/n8nDefaultFailedAttemptHandler.test.d.ts +1 -0
- package/dist/nodes/llms/n8nDefaultFailedAttemptHandler.test.js +53 -0
- package/dist/nodes/llms/n8nDefaultFailedAttemptHandler.test.js.map +1 -0
- package/dist/nodes/llms/n8nLlmFailedAttemptHandler.d.ts +3 -0
- package/dist/nodes/llms/n8nLlmFailedAttemptHandler.js +28 -0
- package/dist/nodes/llms/n8nLlmFailedAttemptHandler.js.map +1 -0
- package/dist/nodes/llms/n8nLlmFailedAttemptHandler.test.d.ts +1 -0
- package/dist/nodes/llms/n8nLlmFailedAttemptHandler.test.js +50 -0
- package/dist/nodes/llms/n8nLlmFailedAttemptHandler.test.js.map +1 -0
- package/dist/nodes/vector_store/VectorStoreDatabricks/VectorStoreDatabricks.node.d.ts +15 -0
- package/dist/nodes/vector_store/VectorStoreDatabricks/VectorStoreDatabricks.node.js +98 -0
- package/dist/nodes/vector_store/VectorStoreDatabricks/VectorStoreDatabricks.node.js.map +1 -0
- package/dist/nodes/vector_store/VectorStoreDatabricks/databricks.dark.svg +19 -0
- package/dist/nodes/vector_store/VectorStoreDatabricks/databricks.svg +19 -0
- package/dist/nodes/vector_store/shared/MemoryManager/MemoryCalculator.d.ts +7 -0
- package/dist/nodes/vector_store/shared/MemoryManager/MemoryCalculator.js +50 -0
- package/dist/nodes/vector_store/shared/MemoryManager/MemoryCalculator.js.map +1 -0
- package/dist/nodes/vector_store/shared/MemoryManager/MemoryVectorStoreManager.d.ts +32 -0
- package/dist/nodes/vector_store/shared/MemoryManager/MemoryVectorStoreManager.js +175 -0
- package/dist/nodes/vector_store/shared/MemoryManager/MemoryVectorStoreManager.js.map +1 -0
- package/dist/nodes/vector_store/shared/MemoryManager/StoreCleanupService.d.ts +16 -0
- package/dist/nodes/vector_store/shared/MemoryManager/StoreCleanupService.js +95 -0
- package/dist/nodes/vector_store/shared/MemoryManager/StoreCleanupService.js.map +1 -0
- package/dist/nodes/vector_store/shared/MemoryManager/config.d.ts +4 -0
- package/dist/nodes/vector_store/shared/MemoryManager/config.js +38 -0
- package/dist/nodes/vector_store/shared/MemoryManager/config.js.map +1 -0
- package/dist/nodes/vector_store/shared/MemoryManager/test/MemoryCalculator.test.d.ts +1 -0
- package/dist/nodes/vector_store/shared/MemoryManager/test/MemoryCalculator.test.js +158 -0
- package/dist/nodes/vector_store/shared/MemoryManager/test/MemoryCalculator.test.js.map +1 -0
- package/dist/nodes/vector_store/shared/MemoryManager/test/MemoryVectorStoreManager.test.d.ts +1 -0
- package/dist/nodes/vector_store/shared/MemoryManager/test/MemoryVectorStoreManager.test.js +204 -0
- package/dist/nodes/vector_store/shared/MemoryManager/test/MemoryVectorStoreManager.test.js.map +1 -0
- package/dist/nodes/vector_store/shared/MemoryManager/test/StoreCleanupService.test.d.ts +1 -0
- package/dist/nodes/vector_store/shared/MemoryManager/test/StoreCleanupService.test.js +138 -0
- package/dist/nodes/vector_store/shared/MemoryManager/test/StoreCleanupService.test.js.map +1 -0
- package/dist/nodes/vector_store/shared/MemoryManager/test/config.test.d.ts +1 -0
- package/dist/nodes/vector_store/shared/MemoryManager/test/config.test.js +57 -0
- package/dist/nodes/vector_store/shared/MemoryManager/test/config.test.js.map +1 -0
- package/dist/nodes/vector_store/shared/MemoryManager/types.d.ts +39 -0
- package/dist/nodes/vector_store/shared/MemoryManager/types.js +3 -0
- package/dist/nodes/vector_store/shared/MemoryManager/types.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/__tests__/utils.test.d.ts +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/__tests__/utils.test.js +156 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/__tests__/utils.test.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/constants.d.ts +4 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/constants.js +45 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/constants.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/createVectorStoreNode.d.ts +15 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/createVectorStoreNode.js +227 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/createVectorStoreNode.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/createVectorStoreNode.test.d.ts +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/createVectorStoreNode.test.js +131 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/createVectorStoreNode.test.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/methods/listSearch.d.ts +7 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/methods/listSearch.js +32 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/methods/listSearch.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/insertOperation.test.d.ts +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/insertOperation.test.js +190 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/insertOperation.test.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/loadOperation.test.d.ts +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/loadOperation.test.js +101 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/loadOperation.test.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/operationHandlers.test.d.ts +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/operationHandlers.test.js +93 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/operationHandlers.test.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/retrieveAsToolOperation.test.d.ts +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/retrieveAsToolOperation.test.js +109 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/retrieveAsToolOperation.test.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/retrieveOperation.test.d.ts +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/retrieveOperation.test.js +60 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/retrieveOperation.test.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/updateOperation.test.d.ts +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/updateOperation.test.js +108 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/updateOperation.test.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/index.d.ts +5 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/index.js +22 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/index.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/insertOperation.d.ts +5 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/insertOperation.js +36 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/insertOperation.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/loadOperation.d.ts +5 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/loadOperation.js +34 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/loadOperation.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/retrieveAsToolOperation.d.ts +5 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/retrieveAsToolOperation.js +43 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/retrieveAsToolOperation.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/retrieveOperation.d.ts +5 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/retrieveOperation.js +17 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/retrieveOperation.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/updateOperation.d.ts +5 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/updateOperation.js +40 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/updateOperation.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/test-helpers.d.ts +0 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/test-helpers.js +2 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/test-helpers.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/types.d.ts +31 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/types.js +3 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/types.js.map +1 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/utils.d.ts +6 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/utils.js +22 -0
- package/dist/nodes/vector_store/shared/createVectorStoreNode/utils.js.map +1 -0
- package/dist/nodes/vector_store/shared/descriptions.d.ts +4 -0
- package/dist/nodes/vector_store/shared/descriptions.js +70 -0
- package/dist/nodes/vector_store/shared/descriptions.js.map +1 -0
- package/dist/nodes/vector_store/shared/processDocuments.d.ts +25 -0
- package/dist/nodes/vector_store/shared/processDocuments.js +42 -0
- package/dist/nodes/vector_store/shared/processDocuments.js.map +1 -0
- package/dist/package.json +93 -0
- package/dist/tsconfig.tsbuildinfo +1 -0
- package/dist/utils/DatabricksVectorStoreLangChain.d.ts +26 -0
- package/dist/utils/DatabricksVectorStoreLangChain.js +114 -0
- package/dist/utils/DatabricksVectorStoreLangChain.js.map +1 -0
- package/dist/utils/N8nBinaryLoader.d.ts +18 -0
- package/dist/utils/N8nBinaryLoader.js +159 -0
- package/dist/utils/N8nBinaryLoader.js.map +1 -0
- package/dist/utils/N8nJsonLoader.d.ts +11 -0
- package/dist/utils/N8nJsonLoader.js +67 -0
- package/dist/utils/N8nJsonLoader.js.map +1 -0
- package/dist/utils/N8nTool.d.ts +10 -0
- package/dist/utils/N8nTool.js +88 -0
- package/dist/utils/N8nTool.js.map +1 -0
- package/dist/utils/descriptions.d.ts +13 -0
- package/dist/utils/descriptions.js +112 -0
- package/dist/utils/descriptions.js.map +1 -0
- package/dist/utils/helpers.d.ts +23 -0
- package/dist/utils/helpers.js +172 -0
- package/dist/utils/helpers.js.map +1 -0
- package/dist/utils/logWrapper.d.ts +27 -0
- package/dist/utils/logWrapper.js +297 -0
- package/dist/utils/logWrapper.js.map +1 -0
- package/dist/utils/output_parsers/N8nItemListOutputParser.d.ts +13 -0
- package/dist/utils/output_parsers/N8nItemListOutputParser.js +43 -0
- package/dist/utils/output_parsers/N8nItemListOutputParser.js.map +1 -0
- package/dist/utils/output_parsers/N8nOutputFixingParser.d.ts +18 -0
- package/dist/utils/output_parsers/N8nOutputFixingParser.js +68 -0
- package/dist/utils/output_parsers/N8nOutputFixingParser.js.map +1 -0
- package/dist/utils/output_parsers/N8nOutputParser.d.ts +7 -0
- package/dist/utils/output_parsers/N8nOutputParser.js +19 -0
- package/dist/utils/output_parsers/N8nOutputParser.js.map +1 -0
- package/dist/utils/output_parsers/N8nStructuredOutputParser.d.ts +12 -0
- package/dist/utils/output_parsers/N8nStructuredOutputParser.js +84 -0
- package/dist/utils/output_parsers/N8nStructuredOutputParser.js.map +1 -0
- package/dist/utils/output_parsers/prompt.d.ts +3 -0
- package/dist/utils/output_parsers/prompt.js +22 -0
- package/dist/utils/output_parsers/prompt.js.map +1 -0
- package/dist/utils/schemaParsing.d.ts +6 -0
- package/dist/utils/schemaParsing.js +23 -0
- package/dist/utils/schemaParsing.js.map +1 -0
- package/dist/utils/sharedFields.d.ts +6 -0
- package/dist/utils/sharedFields.js +122 -0
- package/dist/utils/sharedFields.js.map +1 -0
- package/dist/utils/tracing.d.ts +7 -0
- package/dist/utils/tracing.js +20 -0
- package/dist/utils/tracing.js.map +1 -0
- package/package.json +93 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"n8nLlmFailedAttemptHandler.js","sourceRoot":"","sources":["../../../nodes/llms/n8nLlmFailedAttemptHandler.ts"],"names":[],"mappings":";;;AAEA,+CAA4C;AAE5C,qFAAkF;AAQ3E,MAAM,8BAA8B,GAAG,CAC7C,GAAyB,EACzB,OAA8B,EACP,EAAE;IACzB,OAAO,CAAC,KAAU,EAAE,EAAE;QACrB,IAAI,CAAC;YAEJ,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAG,KAAK,CAAC,CAAC;YAGjB,IAAA,+DAA8B,EAAC,KAAK,CAAC,CAAC;QACvC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YAEZ,MAAM,QAAQ,GAAG,IAAI,2BAAY,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,CAA0B,EAAE;gBAC5E,aAAa,EAAE,oBAAoB;aACnC,CAAC,CAAC;YAEH,MAAM,QAAQ,CAAC;QAChB,CAAC;QAID,IAAI,CAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,WAAW,IAAG,CAAC,EAAE,CAAC;YAC5B,OAAO;QACR,CAAC;QAGD,MAAM,QAAQ,GAAG,IAAI,2BAAY,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,KAA8B,EAAE;YAChF,aAAa,EAAE,oBAAoB;SACnC,CAAC,CAAC;QAEH,MAAM,QAAQ,CAAC;IAChB,CAAC,CAAC;AACH,CAAC,CAAC;AAjCW,QAAA,8BAA8B,kCAiCzC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,50 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
const jest_mock_extended_1 = require("jest-mock-extended");
|
|
4
|
+
const n8n_workflow_1 = require("n8n-workflow");
|
|
5
|
+
const n8nLlmFailedAttemptHandler_1 = require("./n8nLlmFailedAttemptHandler");
|
|
6
|
+
describe('makeN8nLlmFailedAttemptHandler', () => {
|
|
7
|
+
const ctx = (0, jest_mock_extended_1.mock)({
|
|
8
|
+
getNode: jest.fn(),
|
|
9
|
+
});
|
|
10
|
+
it('should throw a wrapped error, when NO custom handler is provided', () => {
|
|
11
|
+
const handler = (0, n8nLlmFailedAttemptHandler_1.makeN8nLlmFailedAttemptHandler)(ctx);
|
|
12
|
+
expect(() => handler(new Error('Test error'))).toThrow(n8n_workflow_1.NodeApiError);
|
|
13
|
+
});
|
|
14
|
+
it('should wrapped error when custom handler is provided', () => {
|
|
15
|
+
const customHandler = jest.fn();
|
|
16
|
+
const handler = (0, n8nLlmFailedAttemptHandler_1.makeN8nLlmFailedAttemptHandler)(ctx, customHandler);
|
|
17
|
+
expect(() => handler(new Error('Test error'))).toThrow(n8n_workflow_1.NodeApiError);
|
|
18
|
+
expect(customHandler).toHaveBeenCalled();
|
|
19
|
+
});
|
|
20
|
+
it('should throw wrapped exception from custom handler', () => {
|
|
21
|
+
const customHandler = jest.fn(() => {
|
|
22
|
+
throw new n8n_workflow_1.ApplicationError('Custom handler error');
|
|
23
|
+
});
|
|
24
|
+
const handler = (0, n8nLlmFailedAttemptHandler_1.makeN8nLlmFailedAttemptHandler)(ctx, customHandler);
|
|
25
|
+
expect(() => handler(new Error('Test error'))).toThrow('Custom handler error');
|
|
26
|
+
expect(customHandler).toHaveBeenCalled();
|
|
27
|
+
});
|
|
28
|
+
it('should not throw if retries are left', () => {
|
|
29
|
+
const customHandler = jest.fn();
|
|
30
|
+
const handler = (0, n8nLlmFailedAttemptHandler_1.makeN8nLlmFailedAttemptHandler)(ctx, customHandler);
|
|
31
|
+
const error = new Error('Test error');
|
|
32
|
+
error.retriesLeft = 1;
|
|
33
|
+
expect(() => handler(error)).not.toThrow();
|
|
34
|
+
});
|
|
35
|
+
it('should throw NodeApiError if no retries are left', () => {
|
|
36
|
+
const handler = (0, n8nLlmFailedAttemptHandler_1.makeN8nLlmFailedAttemptHandler)(ctx);
|
|
37
|
+
const error = new Error('Test error');
|
|
38
|
+
error.retriesLeft = 0;
|
|
39
|
+
expect(() => handler(error)).toThrow(n8n_workflow_1.NodeApiError);
|
|
40
|
+
});
|
|
41
|
+
it('should throw NodeApiError if no retries are left with custom handler', () => {
|
|
42
|
+
const customHandler = jest.fn();
|
|
43
|
+
const handler = (0, n8nLlmFailedAttemptHandler_1.makeN8nLlmFailedAttemptHandler)(ctx, customHandler);
|
|
44
|
+
const error = new Error('Test error');
|
|
45
|
+
error.retriesLeft = 0;
|
|
46
|
+
expect(() => handler(error)).toThrow(n8n_workflow_1.NodeApiError);
|
|
47
|
+
expect(customHandler).toHaveBeenCalled();
|
|
48
|
+
});
|
|
49
|
+
});
|
|
50
|
+
//# sourceMappingURL=n8nLlmFailedAttemptHandler.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"n8nLlmFailedAttemptHandler.test.js","sourceRoot":"","sources":["../../../nodes/llms/n8nLlmFailedAttemptHandler.test.ts"],"names":[],"mappings":";;AAAA,2DAA0C;AAE1C,+CAA8D;AAE9D,6EAA8E;AAE9E,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC/C,MAAM,GAAG,GAAG,IAAA,yBAAI,EAAuB;QACtC,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;KAClB,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC3E,MAAM,OAAO,GAAG,IAAA,2DAA8B,EAAC,GAAG,CAAC,CAAC;QAEpD,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,2BAAY,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC/D,MAAM,aAAa,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,IAAA,2DAA8B,EAAC,GAAG,EAAE,aAAa,CAAC,CAAC;QAEnE,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,2BAAY,CAAC,CAAC;QACrE,MAAM,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC7D,MAAM,aAAa,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE;YAClC,MAAM,IAAI,+BAAgB,CAAC,sBAAsB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAA,2DAA8B,EAAC,GAAG,EAAE,aAAa,CAAC,CAAC;QAEnE,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;QAC/E,MAAM,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC/C,MAAM,aAAa,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,IAAA,2DAA8B,EAAC,GAAG,EAAE,aAAa,CAAC,CAAC;QAEnE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QACrC,KAAa,CAAC,WAAW,GAAG,CAAC,CAAC;QAE/B,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC3D,MAAM,OAAO,GAAG,IAAA,2DAA8B,EAAC,GAAG,CAAC,CAAC;QAEpD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QACrC,KAAa,CAAC,WAAW,GAAG,CAAC,CAAC;QAE/B,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,2BAAY,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;QAC/E,MAAM,aAAa,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,IAAA,2DAA8B,EAAC,GAAG,EAAE,aAAa,CAAC,CAAC;QAEnE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QACrC,KAAa,CAAC,WAAW,GAAG,CAAC,CAAC;QAE/B,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,2BAAY,CAAC,CAAC;QACnD,MAAM,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
declare const VectorStoreDatabricks_base: {
|
|
2
|
+
new (): {
|
|
3
|
+
description: import("n8n-workflow/dist").INodeTypeDescription;
|
|
4
|
+
methods: {
|
|
5
|
+
listSearch?: {
|
|
6
|
+
[key: string]: (this: import("n8n-workflow/dist").ILoadOptionsFunctions, filter?: string, paginationToken?: string) => Promise<import("n8n-workflow/dist").INodeListSearchResult>;
|
|
7
|
+
};
|
|
8
|
+
} | undefined;
|
|
9
|
+
execute(this: import("n8n-workflow/dist").IExecuteFunctions): Promise<import("n8n-workflow/dist").INodeExecutionData[][]>;
|
|
10
|
+
supplyData(this: import("n8n-workflow/dist").ISupplyDataFunctions, itemIndex: number): Promise<import("n8n-workflow/dist").SupplyData>;
|
|
11
|
+
};
|
|
12
|
+
};
|
|
13
|
+
export declare class VectorStoreDatabricks extends VectorStoreDatabricks_base {
|
|
14
|
+
}
|
|
15
|
+
export default VectorStoreDatabricks;
|
|
@@ -0,0 +1,98 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.VectorStoreDatabricks = void 0;
|
|
4
|
+
const createVectorStoreNode_1 = require("../shared/createVectorStoreNode/createVectorStoreNode");
|
|
5
|
+
const DatabricksVectorStoreLangChain_1 = require("../../../utils/DatabricksVectorStoreLangChain");
|
|
6
|
+
class VectorStoreDatabricks extends (0, createVectorStoreNode_1.createVectorStoreNode)({
|
|
7
|
+
meta: {
|
|
8
|
+
displayName: 'Databricks Vector Store',
|
|
9
|
+
name: 'databricksVectorStore',
|
|
10
|
+
description: 'Operations for Databricks Vector Store',
|
|
11
|
+
docsUrl: 'https://docs.databricks.com/aws/en/generative-ai/vector-search',
|
|
12
|
+
icon: 'file:databricks.svg',
|
|
13
|
+
operationModes: ['load', 'insert', 'retrieve', 'retrieve-as-tool'],
|
|
14
|
+
credentials: [
|
|
15
|
+
{
|
|
16
|
+
name: 'databricks',
|
|
17
|
+
required: true,
|
|
18
|
+
},
|
|
19
|
+
],
|
|
20
|
+
},
|
|
21
|
+
sharedFields: [
|
|
22
|
+
{
|
|
23
|
+
displayName: 'Index Name',
|
|
24
|
+
name: 'indexName',
|
|
25
|
+
type: 'string',
|
|
26
|
+
default: '',
|
|
27
|
+
required: true,
|
|
28
|
+
description: 'Name of the vector search index in Databricks',
|
|
29
|
+
},
|
|
30
|
+
{
|
|
31
|
+
displayName: 'Text Column',
|
|
32
|
+
name: 'textColumn',
|
|
33
|
+
type: 'string',
|
|
34
|
+
default: 'text',
|
|
35
|
+
required: true,
|
|
36
|
+
description: 'Name of the column containing the document text',
|
|
37
|
+
},
|
|
38
|
+
{
|
|
39
|
+
displayName: 'Metadata Columns',
|
|
40
|
+
name: 'metadataColumns',
|
|
41
|
+
type: 'string',
|
|
42
|
+
default: '',
|
|
43
|
+
description: 'Comma-separated list of columns to include as metadata',
|
|
44
|
+
},
|
|
45
|
+
{
|
|
46
|
+
displayName: 'Score Threshold',
|
|
47
|
+
name: 'scoreThreshold',
|
|
48
|
+
type: 'number',
|
|
49
|
+
default: 0.8,
|
|
50
|
+
description: 'Minimum similarity score threshold',
|
|
51
|
+
},
|
|
52
|
+
],
|
|
53
|
+
retrieveFields: [],
|
|
54
|
+
async getVectorStoreClient(context, filter, embeddings, itemIndex) {
|
|
55
|
+
const credentials = await context.getCredentials('databricks');
|
|
56
|
+
const indexName = context.getNodeParameter('indexName', itemIndex);
|
|
57
|
+
const textColumn = context.getNodeParameter('textColumn', itemIndex);
|
|
58
|
+
const metadataColumns = context.getNodeParameter('metadataColumns', itemIndex);
|
|
59
|
+
const scoreThreshold = context.getNodeParameter('scoreThreshold', itemIndex, 0.8);
|
|
60
|
+
if (!indexName || !textColumn) {
|
|
61
|
+
throw new Error('Missing required input properties: indexName or textColumn');
|
|
62
|
+
}
|
|
63
|
+
const metadataColumnsList = metadataColumns
|
|
64
|
+
? metadataColumns.split(',').map((s) => s.trim())
|
|
65
|
+
: [];
|
|
66
|
+
return await DatabricksVectorStoreLangChain_1.DatabricksVectorStoreLangChain.fromExistingIndex(embeddings, {
|
|
67
|
+
workspaceUrl: credentials.host,
|
|
68
|
+
token: credentials.token,
|
|
69
|
+
indexName,
|
|
70
|
+
textColumn,
|
|
71
|
+
metadataColumns: metadataColumnsList,
|
|
72
|
+
scoreThreshold,
|
|
73
|
+
});
|
|
74
|
+
},
|
|
75
|
+
async populateVectorStore(context, embeddings, documents, itemIndex) {
|
|
76
|
+
const credentials = await context.getCredentials('databricks');
|
|
77
|
+
const inputData = context.getInputData()[itemIndex].json;
|
|
78
|
+
if (!(inputData === null || inputData === void 0 ? void 0 : inputData.indexName) || !(inputData === null || inputData === void 0 ? void 0 : inputData.textColumn)) {
|
|
79
|
+
throw new Error('Missing required input properties: indexName or textColumn');
|
|
80
|
+
}
|
|
81
|
+
const metadataColumnsList = inputData.metadataColumns ?
|
|
82
|
+
inputData.metadataColumns.split(',').map((s) => s.trim()) :
|
|
83
|
+
[];
|
|
84
|
+
await DatabricksVectorStoreLangChain_1.DatabricksVectorStoreLangChain.fromDocuments(documents, embeddings, {
|
|
85
|
+
workspaceUrl: credentials.host,
|
|
86
|
+
token: credentials.token,
|
|
87
|
+
indexName: inputData.indexName,
|
|
88
|
+
textColumn: inputData.textColumn,
|
|
89
|
+
metadataColumns: metadataColumnsList,
|
|
90
|
+
});
|
|
91
|
+
},
|
|
92
|
+
async releaseVectorStoreClient() {
|
|
93
|
+
},
|
|
94
|
+
}) {
|
|
95
|
+
}
|
|
96
|
+
exports.VectorStoreDatabricks = VectorStoreDatabricks;
|
|
97
|
+
exports.default = VectorStoreDatabricks;
|
|
98
|
+
//# sourceMappingURL=VectorStoreDatabricks.node.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"VectorStoreDatabricks.node.js","sourceRoot":"","sources":["../../../../nodes/vector_store/VectorStoreDatabricks/VectorStoreDatabricks.node.ts"],"names":[],"mappings":";;;AAKA,iGAA8F;AAC9F,0FAAuF;AASvF,MAAa,qBAAsB,SAAQ,IAAA,6CAAqB,EAAiC;IAC7F,IAAI,EAAE;QACF,WAAW,EAAE,yBAAyB;QACtC,IAAI,EAAE,uBAAuB;QAC7B,WAAW,EAAE,wCAAwC;QACrD,OAAO,EAAE,gEAAgE;QACzE,IAAI,EAAE,qBAAqB;QAC3B,cAAc,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,kBAAkB,CAAC;QAClE,WAAW,EAAE;YACT;gBACI,IAAI,EAAE,YAAY;gBAClB,QAAQ,EAAE,IAAI;aACjB;SACJ;KACJ;IACD,YAAY,EAAE;QACV;YACI,WAAW,EAAE,YAAY;YACzB,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,EAAE;YACX,QAAQ,EAAE,IAAI;YACd,WAAW,EAAE,+CAA+C;SAC/D;QACD;YACI,WAAW,EAAE,aAAa;YAC1B,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,MAAM;YACf,QAAQ,EAAE,IAAI;YACd,WAAW,EAAE,iDAAiD;SACjE;QACD;YACI,WAAW,EAAE,kBAAkB;YAC/B,IAAI,EAAE,iBAAiB;YACvB,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,EAAE;YACX,WAAW,EAAE,wDAAwD;SACxE;QACD;YACI,WAAW,EAAE,iBAAiB;YAC9B,IAAI,EAAE,gBAAgB;YACtB,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,GAAG;YACZ,WAAW,EAAE,oCAAoC;SACpD;KAEJ;IACD,cAAc,EAAE,EAEf;IACD,KAAK,CAAC,oBAAoB,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS;QAC7D,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAC/D,MAAM,SAAS,GAAG,OAAO,CAAC,gBAAgB,CAAC,WAAW,EAAE,SAAS,CAAW,CAAC;QAC7E,MAAM,UAAU,GAAG,OAAO,CAAC,gBAAgB,CAAC,YAAY,EAAE,SAAS,CAAW,CAAC;QAC/E,MAAM,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC,iBAAiB,EAAE,SAAS,CAAW,CAAC;QACzF,MAAM,cAAc,GAAG,OAAO,CAAC,gBAAgB,CAAC,gBAAgB,EAAE,SAAS,EAAE,GAAG,CAAW,CAAC;QAE5F,IAAI,CAAC,SAAS,IAAI,CAAC,UAAU,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;QAClF,CAAC;QAED,MAAM,mBAAmB,GAAG,eAAe;YACvC,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YACzD,CAAC,CAAC,EAAE,CAAC;QAET,OAAO,MAAM,+DAA8B,CAAC,iBAAiB,CAAC,UAAU,EAAE;YACtE,YAAY,EAAE,WAAW,CAAC,IAAc;YACxC,KAAK,EAAE,WAAW,CAAC,KAAe;YAClC,SAAS;YACT,UAAU;YACV,eAAe,EAAE,mBAAmB;YACpC,cAAc;SACjB,CAAC,CAAC;IACP,CAAC;IACD,KAAK,CAAC,mBAAmB,CAAC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS;QAC/D,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAC/D,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC,SAAS,CAAC,CAAC,IAAyB,CAAC;QAE9E,IAAI,CAAC,CAAA,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,SAAS,CAAA,IAAI,CAAC,CAAA,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,UAAU,CAAA,EAAE,CAAC;YAClD,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;QAClF,CAAC;QAED,MAAM,mBAAmB,GAAG,SAAS,CAAC,eAAe,CAAC,CAAC;YACnD,SAAS,CAAC,eAAe,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YACnE,EAAE,CAAC;QAEP,MAAM,+DAA8B,CAAC,aAAa,CAAC,SAAS,EAAE,UAAU,EAAE;YACtE,YAAY,EAAE,WAAW,CAAC,IAAc;YACxC,KAAK,EAAE,WAAW,CAAC,KAAe;YAClC,SAAS,EAAE,SAAS,CAAC,SAAS;YAC9B,UAAU,EAAE,SAAS,CAAC,UAAU;YAChC,eAAe,EAAE,mBAAmB;SACvC,CAAC,CAAC;IACP,CAAC;IACD,KAAK,CAAC,wBAAwB;IAE9B,CAAC;CACJ,CAAC;CAAG;AAlGL,sDAkGK;AAEL,kBAAe,qBAAqB,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
<svg version="1.1" id="Layer_1" xmlns:x="ns_extend;" xmlns:i="ns_ai;" xmlns:graph="ns_graphs;" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" x="0px" y="0px" viewBox="0 0 40.1 42" style="enable-background:new 0 0 40.1 42;" xml:space="preserve">
|
|
2
|
+
<style type="text/css">
|
|
3
|
+
.st0{fill:#FF3621;}
|
|
4
|
+
</style>
|
|
5
|
+
<metadata>
|
|
6
|
+
<sfw xmlns="ns_sfw;">
|
|
7
|
+
<slices>
|
|
8
|
+
</slices>
|
|
9
|
+
<sliceSourceBounds bottomLeftOrigin="true" height="42" width="40.1" x="-69.1" y="-10.5">
|
|
10
|
+
</sliceSourceBounds>
|
|
11
|
+
</sfw>
|
|
12
|
+
</metadata>
|
|
13
|
+
<g>
|
|
14
|
+
<path class="st0" d="M40.1,31.1v-7.4l-0.8-0.5L20.1,33.7l-18.2-10l0-4.3l18.2,9.9l20.1-10.9v-7.3l-0.8-0.5L20.1,21.2L2.6,11.6
|
|
15
|
+
L20.1,2l14.1,7.7l1.1-0.6V8.3L20.1,0L0,10.9V12L20.1,23l18.2-10v4.4l-18.2,10L0.8,16.8L0,17.3v7.4l20.1,10.9l18.2-9.9v4.3l-18.2,10
|
|
16
|
+
L0.8,29.5L0,30v1.1L20.1,42L40.1,31.1z">
|
|
17
|
+
</path>
|
|
18
|
+
</g>
|
|
19
|
+
</svg>
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
<svg version="1.1" id="Layer_1" xmlns:x="ns_extend;" xmlns:i="ns_ai;" xmlns:graph="ns_graphs;" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" x="0px" y="0px" viewBox="0 0 40.1 42" style="enable-background:new 0 0 40.1 42;" xml:space="preserve">
|
|
2
|
+
<style type="text/css">
|
|
3
|
+
.st0{fill:#FF3621;}
|
|
4
|
+
</style>
|
|
5
|
+
<metadata>
|
|
6
|
+
<sfw xmlns="ns_sfw;">
|
|
7
|
+
<slices>
|
|
8
|
+
</slices>
|
|
9
|
+
<sliceSourceBounds bottomLeftOrigin="true" height="42" width="40.1" x="-69.1" y="-10.5">
|
|
10
|
+
</sliceSourceBounds>
|
|
11
|
+
</sfw>
|
|
12
|
+
</metadata>
|
|
13
|
+
<g>
|
|
14
|
+
<path class="st0" d="M40.1,31.1v-7.4l-0.8-0.5L20.1,33.7l-18.2-10l0-4.3l18.2,9.9l20.1-10.9v-7.3l-0.8-0.5L20.1,21.2L2.6,11.6
|
|
15
|
+
L20.1,2l14.1,7.7l1.1-0.6V8.3L20.1,0L0,10.9V12L20.1,23l18.2-10v4.4l-18.2,10L0.8,16.8L0,17.3v7.4l20.1,10.9l18.2-9.9v4.3l-18.2,10
|
|
16
|
+
L0.8,29.5L0,30v1.1L20.1,42L40.1,31.1z">
|
|
17
|
+
</path>
|
|
18
|
+
</g>
|
|
19
|
+
</svg>
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import type { Document } from '@langchain/core/documents';
|
|
2
|
+
import type { MemoryVectorStore } from '@langchain/classic/vectorstores/memory';
|
|
3
|
+
import type { IMemoryCalculator } from './types';
|
|
4
|
+
export declare class MemoryCalculator implements IMemoryCalculator {
|
|
5
|
+
estimateBatchSize(documents: Document[]): number;
|
|
6
|
+
calculateVectorStoreSize(vectorStore: MemoryVectorStore): number;
|
|
7
|
+
}
|
|
@@ -0,0 +1,50 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.MemoryCalculator = void 0;
|
|
4
|
+
const FLOAT_SIZE_BYTES = 8;
|
|
5
|
+
const CHAR_SIZE_BYTES = 2;
|
|
6
|
+
const VECTOR_OVERHEAD_BYTES = 200;
|
|
7
|
+
const EMBEDDING_DIMENSIONS = 1536;
|
|
8
|
+
const EMBEDDING_SIZE_BYTES = EMBEDDING_DIMENSIONS * FLOAT_SIZE_BYTES;
|
|
9
|
+
const AVG_METADATA_SIZE_BYTES = 100;
|
|
10
|
+
class MemoryCalculator {
|
|
11
|
+
estimateBatchSize(documents) {
|
|
12
|
+
if (documents.length === 0)
|
|
13
|
+
return 0;
|
|
14
|
+
let totalContentSize = 0;
|
|
15
|
+
let totalMetadataSize = 0;
|
|
16
|
+
for (const doc of documents) {
|
|
17
|
+
if (doc.pageContent) {
|
|
18
|
+
totalContentSize += doc.pageContent.length * CHAR_SIZE_BYTES;
|
|
19
|
+
}
|
|
20
|
+
if (doc.metadata) {
|
|
21
|
+
const metadataKeys = Object.keys(doc.metadata).length;
|
|
22
|
+
if (metadataKeys > 0) {
|
|
23
|
+
totalMetadataSize += metadataKeys * AVG_METADATA_SIZE_BYTES;
|
|
24
|
+
}
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
const embeddingSize = documents.length * EMBEDDING_SIZE_BYTES;
|
|
28
|
+
const overhead = documents.length * VECTOR_OVERHEAD_BYTES;
|
|
29
|
+
const calculatedSize = totalContentSize + totalMetadataSize + embeddingSize + overhead;
|
|
30
|
+
return Math.ceil(calculatedSize);
|
|
31
|
+
}
|
|
32
|
+
calculateVectorStoreSize(vectorStore) {
|
|
33
|
+
if (!vectorStore.memoryVectors || vectorStore.memoryVectors.length === 0) {
|
|
34
|
+
return 0;
|
|
35
|
+
}
|
|
36
|
+
let storeSize = 0;
|
|
37
|
+
for (const vector of vectorStore.memoryVectors) {
|
|
38
|
+
storeSize += vector.embedding.length * FLOAT_SIZE_BYTES;
|
|
39
|
+
storeSize += vector.content ? vector.content.length * CHAR_SIZE_BYTES : 0;
|
|
40
|
+
if (vector.metadata) {
|
|
41
|
+
const metadataStr = JSON.stringify(vector.metadata);
|
|
42
|
+
storeSize += metadataStr.length * CHAR_SIZE_BYTES;
|
|
43
|
+
}
|
|
44
|
+
storeSize += VECTOR_OVERHEAD_BYTES;
|
|
45
|
+
}
|
|
46
|
+
return Math.ceil(storeSize);
|
|
47
|
+
}
|
|
48
|
+
}
|
|
49
|
+
exports.MemoryCalculator = MemoryCalculator;
|
|
50
|
+
//# sourceMappingURL=MemoryCalculator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"MemoryCalculator.js","sourceRoot":"","sources":["../../../../../nodes/vector_store/shared/MemoryManager/MemoryCalculator.ts"],"names":[],"mappings":";;;AAMA,MAAM,gBAAgB,GAAG,CAAC,CAAC;AAC3B,MAAM,eAAe,GAAG,CAAC,CAAC;AAC1B,MAAM,qBAAqB,GAAG,GAAG,CAAC;AAClC,MAAM,oBAAoB,GAAG,IAAI,CAAC;AAClC,MAAM,oBAAoB,GAAG,oBAAoB,GAAG,gBAAgB,CAAC;AACrE,MAAM,uBAAuB,GAAG,GAAG,CAAC;AAKpC,MAAa,gBAAgB;IAI5B,iBAAiB,CAAC,SAAqB;QACtC,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAErC,IAAI,gBAAgB,GAAG,CAAC,CAAC;QACzB,IAAI,iBAAiB,GAAG,CAAC,CAAC;QAG1B,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;YAC7B,IAAI,GAAG,CAAC,WAAW,EAAE,CAAC;gBACrB,gBAAgB,IAAI,GAAG,CAAC,WAAW,CAAC,MAAM,GAAG,eAAe,CAAC;YAC9D,CAAC;YAGD,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;gBAElB,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC;gBACtD,IAAI,YAAY,GAAG,CAAC,EAAE,CAAC;oBAGtB,iBAAiB,IAAI,YAAY,GAAG,uBAAuB,CAAC;gBAC7D,CAAC;YACF,CAAC;QACF,CAAC;QAID,MAAM,aAAa,GAAG,SAAS,CAAC,MAAM,GAAG,oBAAoB,CAAC;QAG9D,MAAM,QAAQ,GAAG,SAAS,CAAC,MAAM,GAAG,qBAAqB,CAAC;QAG1D,MAAM,cAAc,GAAG,gBAAgB,GAAG,iBAAiB,GAAG,aAAa,GAAG,QAAQ,CAAC;QAEvF,OAAO,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAClC,CAAC;IAKD,wBAAwB,CAAC,WAA8B;QACtD,IAAI,CAAC,WAAW,CAAC,aAAa,IAAI,WAAW,CAAC,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1E,OAAO,CAAC,CAAC;QACV,CAAC;QAED,IAAI,SAAS,GAAG,CAAC,CAAC;QAGlB,KAAK,MAAM,MAAM,IAAI,WAAW,CAAC,aAAa,EAAE,CAAC;YAEhD,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,gBAAgB,CAAC;YAGxD,SAAS,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;YAG1E,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBAErB,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACpD,SAAS,IAAI,WAAW,CAAC,MAAM,GAAG,eAAe,CAAC;YACnD,CAAC;YAGD,SAAS,IAAI,qBAAqB,CAAC;QACpC,CAAC;QAED,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;CACD;AAxED,4CAwEC"}
|
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
import type { Document } from '@langchain/core/documents';
|
|
2
|
+
import type { Embeddings } from '@langchain/core/embeddings';
|
|
3
|
+
import { MemoryVectorStore } from '@langchain/classic/vectorstores/memory';
|
|
4
|
+
import type { Logger } from 'n8n-workflow';
|
|
5
|
+
import { MemoryCalculator } from './MemoryCalculator';
|
|
6
|
+
import { StoreCleanupService } from './StoreCleanupService';
|
|
7
|
+
import type { VectorStoreMetadata, VectorStoreStats } from './types';
|
|
8
|
+
export declare class MemoryVectorStoreManager {
|
|
9
|
+
protected embeddings: Embeddings;
|
|
10
|
+
protected logger: Logger;
|
|
11
|
+
private static instance;
|
|
12
|
+
protected vectorStoreBuffer: Map<string, MemoryVectorStore>;
|
|
13
|
+
protected storeMetadata: Map<string, VectorStoreMetadata>;
|
|
14
|
+
protected memoryUsageBytes: number;
|
|
15
|
+
protected memoryCalculator: MemoryCalculator;
|
|
16
|
+
protected cleanupService: StoreCleanupService;
|
|
17
|
+
protected static logger: Logger;
|
|
18
|
+
protected maxMemorySizeBytes: number;
|
|
19
|
+
protected inactiveTtlMs: number;
|
|
20
|
+
protected ttlCleanupIntervalId: NodeJS.Timeout | null;
|
|
21
|
+
protected constructor(embeddings: Embeddings, logger: Logger);
|
|
22
|
+
static getInstance(embeddings: Embeddings, logger: Logger): MemoryVectorStoreManager;
|
|
23
|
+
private setupTtlCleanup;
|
|
24
|
+
private handleCleanup;
|
|
25
|
+
getVectorStore(memoryKey: string): Promise<MemoryVectorStore>;
|
|
26
|
+
protected clearStoreMetadata(memoryKey: string): void;
|
|
27
|
+
getMemoryUsage(): number;
|
|
28
|
+
getMemoryUsageFormatted(): string;
|
|
29
|
+
recalculateMemoryUsage(): void;
|
|
30
|
+
addDocuments(memoryKey: string, documents: Document[], clearStore?: boolean): Promise<void>;
|
|
31
|
+
getStats(): VectorStoreStats;
|
|
32
|
+
}
|
|
@@ -0,0 +1,175 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.MemoryVectorStoreManager = void 0;
|
|
4
|
+
const memory_1 = require("@langchain/classic/vectorstores/memory");
|
|
5
|
+
const config_1 = require("./config");
|
|
6
|
+
const MemoryCalculator_1 = require("./MemoryCalculator");
|
|
7
|
+
const StoreCleanupService_1 = require("./StoreCleanupService");
|
|
8
|
+
class MemoryVectorStoreManager {
|
|
9
|
+
constructor(embeddings, logger) {
|
|
10
|
+
this.embeddings = embeddings;
|
|
11
|
+
this.logger = logger;
|
|
12
|
+
this.memoryUsageBytes = 0;
|
|
13
|
+
this.ttlCleanupIntervalId = null;
|
|
14
|
+
this.vectorStoreBuffer = new Map();
|
|
15
|
+
this.storeMetadata = new Map();
|
|
16
|
+
this.logger = logger;
|
|
17
|
+
const config = (0, config_1.getConfig)();
|
|
18
|
+
this.maxMemorySizeBytes = (0, config_1.mbToBytes)(config.maxMemoryMB);
|
|
19
|
+
this.inactiveTtlMs = (0, config_1.hoursToMs)(config.ttlHours);
|
|
20
|
+
this.memoryCalculator = new MemoryCalculator_1.MemoryCalculator();
|
|
21
|
+
this.cleanupService = new StoreCleanupService_1.StoreCleanupService(this.maxMemorySizeBytes, this.inactiveTtlMs, this.vectorStoreBuffer, this.storeMetadata, this.handleCleanup.bind(this));
|
|
22
|
+
this.setupTtlCleanup();
|
|
23
|
+
}
|
|
24
|
+
static getInstance(embeddings, logger) {
|
|
25
|
+
if (!MemoryVectorStoreManager.instance) {
|
|
26
|
+
MemoryVectorStoreManager.instance = new MemoryVectorStoreManager(embeddings, logger);
|
|
27
|
+
}
|
|
28
|
+
else {
|
|
29
|
+
MemoryVectorStoreManager.instance.embeddings = embeddings;
|
|
30
|
+
MemoryVectorStoreManager.instance.vectorStoreBuffer.forEach((vectorStoreInstance) => {
|
|
31
|
+
vectorStoreInstance.embeddings = embeddings;
|
|
32
|
+
});
|
|
33
|
+
}
|
|
34
|
+
return MemoryVectorStoreManager.instance;
|
|
35
|
+
}
|
|
36
|
+
setupTtlCleanup() {
|
|
37
|
+
if (this.inactiveTtlMs <= 0) {
|
|
38
|
+
return;
|
|
39
|
+
}
|
|
40
|
+
const CLEANUP_INTERVAL_MS = 60 * 60 * 1000;
|
|
41
|
+
if (this.ttlCleanupIntervalId) {
|
|
42
|
+
clearInterval(this.ttlCleanupIntervalId);
|
|
43
|
+
}
|
|
44
|
+
this.ttlCleanupIntervalId = setInterval(() => {
|
|
45
|
+
this.cleanupService.cleanupInactiveStores();
|
|
46
|
+
}, CLEANUP_INTERVAL_MS);
|
|
47
|
+
}
|
|
48
|
+
handleCleanup(removedKeys, freedBytes, reason) {
|
|
49
|
+
this.memoryUsageBytes -= freedBytes;
|
|
50
|
+
if (reason === 'ttl') {
|
|
51
|
+
const ttlHours = Math.round(this.inactiveTtlMs / (60 * 60 * 1000));
|
|
52
|
+
this.logger.info(`TTL cleanup: removed ${removedKeys.length} inactive vector stores (${ttlHours}h TTL) to free ${Math.round(freedBytes / (1024 * 1024))}MB of memory`);
|
|
53
|
+
}
|
|
54
|
+
else {
|
|
55
|
+
this.logger.info(`Memory cleanup: removed ${removedKeys.length} oldest vector stores to free ${Math.round(freedBytes / (1024 * 1024))}MB of memory`);
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
async getVectorStore(memoryKey) {
|
|
59
|
+
let vectorStoreInstance = this.vectorStoreBuffer.get(memoryKey);
|
|
60
|
+
if (!vectorStoreInstance) {
|
|
61
|
+
vectorStoreInstance = await memory_1.MemoryVectorStore.fromExistingIndex(this.embeddings);
|
|
62
|
+
this.vectorStoreBuffer.set(memoryKey, vectorStoreInstance);
|
|
63
|
+
this.storeMetadata.set(memoryKey, {
|
|
64
|
+
size: 0,
|
|
65
|
+
createdAt: new Date(),
|
|
66
|
+
lastAccessed: new Date(),
|
|
67
|
+
});
|
|
68
|
+
}
|
|
69
|
+
else {
|
|
70
|
+
const metadata = this.storeMetadata.get(memoryKey);
|
|
71
|
+
if (metadata) {
|
|
72
|
+
metadata.lastAccessed = new Date();
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
return vectorStoreInstance;
|
|
76
|
+
}
|
|
77
|
+
clearStoreMetadata(memoryKey) {
|
|
78
|
+
const metadata = this.storeMetadata.get(memoryKey);
|
|
79
|
+
if (metadata) {
|
|
80
|
+
this.memoryUsageBytes -= metadata.size;
|
|
81
|
+
metadata.size = 0;
|
|
82
|
+
metadata.lastAccessed = new Date();
|
|
83
|
+
}
|
|
84
|
+
}
|
|
85
|
+
getMemoryUsage() {
|
|
86
|
+
return this.memoryUsageBytes;
|
|
87
|
+
}
|
|
88
|
+
getMemoryUsageFormatted() {
|
|
89
|
+
return `${Math.round(this.memoryUsageBytes / (1024 * 1024))}MB`;
|
|
90
|
+
}
|
|
91
|
+
recalculateMemoryUsage() {
|
|
92
|
+
this.memoryUsageBytes = 0;
|
|
93
|
+
for (const [key, vectorStore] of this.vectorStoreBuffer.entries()) {
|
|
94
|
+
const storeSize = this.memoryCalculator.calculateVectorStoreSize(vectorStore);
|
|
95
|
+
const metadata = this.storeMetadata.get(key);
|
|
96
|
+
if (metadata) {
|
|
97
|
+
metadata.size = storeSize;
|
|
98
|
+
this.memoryUsageBytes += storeSize;
|
|
99
|
+
}
|
|
100
|
+
}
|
|
101
|
+
this.logger.debug(`Recalculated vector store memory: ${this.getMemoryUsageFormatted()}`);
|
|
102
|
+
}
|
|
103
|
+
async addDocuments(memoryKey, documents, clearStore) {
|
|
104
|
+
var _a, _b;
|
|
105
|
+
if (clearStore) {
|
|
106
|
+
this.clearStoreMetadata(memoryKey);
|
|
107
|
+
this.vectorStoreBuffer.delete(memoryKey);
|
|
108
|
+
}
|
|
109
|
+
const estimatedAddedSize = this.memoryCalculator.estimateBatchSize(documents);
|
|
110
|
+
this.cleanupService.cleanupOldestStores(estimatedAddedSize);
|
|
111
|
+
const vectorStoreInstance = await this.getVectorStore(memoryKey);
|
|
112
|
+
const vectorCountBefore = ((_a = vectorStoreInstance.memoryVectors) === null || _a === void 0 ? void 0 : _a.length) || 0;
|
|
113
|
+
await vectorStoreInstance.addDocuments(documents);
|
|
114
|
+
const metadata = this.storeMetadata.get(memoryKey);
|
|
115
|
+
if (metadata) {
|
|
116
|
+
metadata.size += estimatedAddedSize;
|
|
117
|
+
metadata.lastAccessed = new Date();
|
|
118
|
+
this.memoryUsageBytes += estimatedAddedSize;
|
|
119
|
+
}
|
|
120
|
+
const vectorCount = ((_b = vectorStoreInstance.memoryVectors) === null || _b === void 0 ? void 0 : _b.length) || 0;
|
|
121
|
+
if ((vectorCount > 0 && vectorCount % 100 === 0) ||
|
|
122
|
+
documents.length > 20 ||
|
|
123
|
+
(vectorCountBefore === 0 && vectorCount > 0)) {
|
|
124
|
+
this.recalculateMemoryUsage();
|
|
125
|
+
}
|
|
126
|
+
const maxMemoryMB = this.maxMemorySizeBytes > 0
|
|
127
|
+
? (this.maxMemorySizeBytes / (1024 * 1024)).toFixed(0)
|
|
128
|
+
: 'unlimited';
|
|
129
|
+
this.logger.debug(`Vector store memory: ${this.getMemoryUsageFormatted()}/${maxMemoryMB}MB (${vectorCount} vectors in ${this.vectorStoreBuffer.size} stores)`);
|
|
130
|
+
}
|
|
131
|
+
getStats() {
|
|
132
|
+
var _a;
|
|
133
|
+
const now = Date.now();
|
|
134
|
+
let inactiveStoreCount = 0;
|
|
135
|
+
this.recalculateMemoryUsage();
|
|
136
|
+
const stats = {
|
|
137
|
+
totalSizeBytes: this.memoryUsageBytes,
|
|
138
|
+
totalSizeMB: Math.round((this.memoryUsageBytes / (1024 * 1024)) * 100) / 100,
|
|
139
|
+
percentOfLimit: this.maxMemorySizeBytes > 0
|
|
140
|
+
? Math.round((this.memoryUsageBytes / this.maxMemorySizeBytes) * 100)
|
|
141
|
+
: 0,
|
|
142
|
+
maxMemoryMB: this.maxMemorySizeBytes > 0 ? this.maxMemorySizeBytes / (1024 * 1024) : -1,
|
|
143
|
+
storeCount: this.vectorStoreBuffer.size,
|
|
144
|
+
inactiveStoreCount: 0,
|
|
145
|
+
ttlHours: this.inactiveTtlMs > 0 ? this.inactiveTtlMs / (60 * 60 * 1000) : -1,
|
|
146
|
+
stores: {},
|
|
147
|
+
};
|
|
148
|
+
for (const [key, metadata] of this.storeMetadata.entries()) {
|
|
149
|
+
const store = this.vectorStoreBuffer.get(key);
|
|
150
|
+
if (store) {
|
|
151
|
+
const lastAccessedTime = metadata.lastAccessed.getTime();
|
|
152
|
+
const inactiveTimeMs = now - lastAccessedTime;
|
|
153
|
+
const isInactive = this.cleanupService.isStoreInactive(metadata);
|
|
154
|
+
if (isInactive) {
|
|
155
|
+
inactiveStoreCount++;
|
|
156
|
+
}
|
|
157
|
+
stats.stores[key] = {
|
|
158
|
+
sizeBytes: metadata.size,
|
|
159
|
+
sizeMB: Math.round((metadata.size / (1024 * 1024)) * 100) / 100,
|
|
160
|
+
percentOfTotal: Math.round((metadata.size / this.memoryUsageBytes) * 100) || 0,
|
|
161
|
+
vectors: ((_a = store.memoryVectors) === null || _a === void 0 ? void 0 : _a.length) || 0,
|
|
162
|
+
createdAt: metadata.createdAt.toISOString(),
|
|
163
|
+
lastAccessed: metadata.lastAccessed.toISOString(),
|
|
164
|
+
inactive: isInactive,
|
|
165
|
+
inactiveForHours: Math.round(inactiveTimeMs / (60 * 60 * 1000)),
|
|
166
|
+
};
|
|
167
|
+
}
|
|
168
|
+
}
|
|
169
|
+
stats.inactiveStoreCount = inactiveStoreCount;
|
|
170
|
+
return stats;
|
|
171
|
+
}
|
|
172
|
+
}
|
|
173
|
+
exports.MemoryVectorStoreManager = MemoryVectorStoreManager;
|
|
174
|
+
MemoryVectorStoreManager.instance = null;
|
|
175
|
+
//# sourceMappingURL=MemoryVectorStoreManager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"MemoryVectorStoreManager.js","sourceRoot":"","sources":["../../../../../nodes/vector_store/shared/MemoryManager/MemoryVectorStoreManager.ts"],"names":[],"mappings":";;;AAEA,mEAA2E;AAG3E,qCAA2D;AAC3D,yDAAsD;AACtD,+DAA4D;AAM5D,MAAa,wBAAwB;IAyBpC,YACW,UAAsB,EACtB,MAAc;QADd,eAAU,GAAV,UAAU,CAAY;QACtB,WAAM,GAAN,MAAM,CAAQ;QAnBf,qBAAgB,GAAW,CAAC,CAAC;QAe7B,yBAAoB,GAA0B,IAAI,CAAC;QAO5D,IAAI,CAAC,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;QACnC,IAAI,CAAC,aAAa,GAAG,IAAI,GAAG,EAAE,CAAC;QAC/B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,MAAM,MAAM,GAAG,IAAA,kBAAS,GAAE,CAAC;QAC3B,IAAI,CAAC,kBAAkB,GAAG,IAAA,kBAAS,EAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACxD,IAAI,CAAC,aAAa,GAAG,IAAA,kBAAS,EAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAGhD,IAAI,CAAC,gBAAgB,GAAG,IAAI,mCAAgB,EAAE,CAAC;QAC/C,IAAI,CAAC,cAAc,GAAG,IAAI,yCAAmB,CAC5C,IAAI,CAAC,kBAAkB,EACvB,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,iBAAiB,EACtB,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAC7B,CAAC;QAEF,IAAI,CAAC,eAAe,EAAE,CAAC;IACxB,CAAC;IAKD,MAAM,CAAC,WAAW,CAAC,UAAsB,EAAE,MAAc;QACxD,IAAI,CAAC,wBAAwB,CAAC,QAAQ,EAAE,CAAC;YACxC,wBAAwB,CAAC,QAAQ,GAAG,IAAI,wBAAwB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QACtF,CAAC;aAAM,CAAC;YAIP,wBAAwB,CAAC,QAAQ,CAAC,UAAU,GAAG,UAAU,CAAC;YAC1D,wBAAwB,CAAC,QAAQ,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,mBAAmB,EAAE,EAAE;gBACnF,mBAAmB,CAAC,UAAU,GAAG,UAAU,CAAC;YAC7C,CAAC,CAAC,CAAC;QACJ,CAAC;QAED,OAAO,wBAAwB,CAAC,QAAQ,CAAC;IAC1C,CAAC;IAKO,eAAe;QAEtB,IAAI,IAAI,CAAC,aAAa,IAAI,CAAC,EAAE,CAAC;YAC7B,OAAO;QACR,CAAC;QAGD,MAAM,mBAAmB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QAG3C,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC/B,aAAa,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC1C,CAAC;QAGD,IAAI,CAAC,oBAAoB,GAAG,WAAW,CAAC,GAAG,EAAE;YAC5C,IAAI,CAAC,cAAc,CAAC,qBAAqB,EAAE,CAAC;QAC7C,CAAC,EAAE,mBAAmB,CAAC,CAAC;IACzB,CAAC;IAKO,aAAa,CAAC,WAAqB,EAAE,UAAkB,EAAE,MAAwB;QAExF,IAAI,CAAC,gBAAgB,IAAI,UAAU,CAAC;QAGpC,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;YACtB,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;YACnE,IAAI,CAAC,MAAM,CAAC,IAAI,CACf,wBAAwB,WAAW,CAAC,MAAM,4BAA4B,QAAQ,kBAAkB,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,cAAc,CACpJ,CAAC;QACH,CAAC;aAAM,CAAC;YACP,IAAI,CAAC,MAAM,CAAC,IAAI,CACf,2BAA2B,WAAW,CAAC,MAAM,iCAAiC,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,cAAc,CAClI,CAAC;QACH,CAAC;IACF,CAAC;IAKD,KAAK,CAAC,cAAc,CAAC,SAAiB;QACrC,IAAI,mBAAmB,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAEhE,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC1B,mBAAmB,GAAG,MAAM,0BAAiB,CAAC,iBAAiB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACjF,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;YAE3D,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,EAAE;gBACjC,IAAI,EAAE,CAAC;gBACP,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,YAAY,EAAE,IAAI,IAAI,EAAE;aACxB,CAAC,CAAC;QACJ,CAAC;aAAM,CAAC;YACP,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACnD,IAAI,QAAQ,EAAE,CAAC;gBACd,QAAQ,CAAC,YAAY,GAAG,IAAI,IAAI,EAAE,CAAC;YACpC,CAAC;QACF,CAAC;QAED,OAAO,mBAAmB,CAAC;IAC5B,CAAC;IAKS,kBAAkB,CAAC,SAAiB;QAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACnD,IAAI,QAAQ,EAAE,CAAC;YACd,IAAI,CAAC,gBAAgB,IAAI,QAAQ,CAAC,IAAI,CAAC;YACvC,QAAQ,CAAC,IAAI,GAAG,CAAC,CAAC;YAClB,QAAQ,CAAC,YAAY,GAAG,IAAI,IAAI,EAAE,CAAC;QACpC,CAAC;IACF,CAAC;IAKD,cAAc;QACb,OAAO,IAAI,CAAC,gBAAgB,CAAC;IAC9B,CAAC;IAKD,uBAAuB;QACtB,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,gBAAgB,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC;IACjE,CAAC;IAMD,sBAAsB;QACrB,IAAI,CAAC,gBAAgB,GAAG,CAAC,CAAC;QAG1B,KAAK,MAAM,CAAC,GAAG,EAAE,WAAW,CAAC,IAAI,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,EAAE,CAAC;YACnE,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,wBAAwB,CAAC,WAAW,CAAC,CAAC;YAG9E,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC7C,IAAI,QAAQ,EAAE,CAAC;gBACd,QAAQ,CAAC,IAAI,GAAG,SAAS,CAAC;gBAC1B,IAAI,CAAC,gBAAgB,IAAI,SAAS,CAAC;YACpC,CAAC;QACF,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qCAAqC,IAAI,CAAC,uBAAuB,EAAE,EAAE,CAAC,CAAC;IAC1F,CAAC;IAKD,KAAK,CAAC,YAAY,CACjB,SAAiB,EACjB,SAAqB,EACrB,UAAoB;;QAEpB,IAAI,UAAU,EAAE,CAAC;YAChB,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;YACnC,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC1C,CAAC;QAGD,MAAM,kBAAkB,GAAG,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAG9E,IAAI,CAAC,cAAc,CAAC,mBAAmB,CAAC,kBAAkB,CAAC,CAAC;QAE5D,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAGjE,MAAM,iBAAiB,GAAG,CAAA,MAAA,mBAAmB,CAAC,aAAa,0CAAE,MAAM,KAAI,CAAC,CAAC;QAEzE,MAAM,mBAAmB,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAGlD,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACnD,IAAI,QAAQ,EAAE,CAAC;YACd,QAAQ,CAAC,IAAI,IAAI,kBAAkB,CAAC;YACpC,QAAQ,CAAC,YAAY,GAAG,IAAI,IAAI,EAAE,CAAC;YACnC,IAAI,CAAC,gBAAgB,IAAI,kBAAkB,CAAC;QAC7C,CAAC;QAGD,MAAM,WAAW,GAAG,CAAA,MAAA,mBAAmB,CAAC,aAAa,0CAAE,MAAM,KAAI,CAAC,CAAC;QAGnE,IACC,CAAC,WAAW,GAAG,CAAC,IAAI,WAAW,GAAG,GAAG,KAAK,CAAC,CAAC;YAC5C,SAAS,CAAC,MAAM,GAAG,EAAE;YACrB,CAAC,iBAAiB,KAAK,CAAC,IAAI,WAAW,GAAG,CAAC,CAAC,EAC3C,CAAC;YACF,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC/B,CAAC;QAGD,MAAM,WAAW,GAChB,IAAI,CAAC,kBAAkB,GAAG,CAAC;YAC1B,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YACtD,CAAC,CAAC,WAAW,CAAC;QAEhB,IAAI,CAAC,MAAM,CAAC,KAAK,CAChB,wBAAwB,IAAI,CAAC,uBAAuB,EAAE,IAAI,WAAW,OAAO,WAAW,eAAe,IAAI,CAAC,iBAAiB,CAAC,IAAI,UAAU,CAC3I,CAAC;IACH,CAAC;IAKD,QAAQ;;QACP,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,kBAAkB,GAAG,CAAC,CAAC;QAG3B,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAE9B,MAAM,KAAK,GAAqB;YAC/B,cAAc,EAAE,IAAI,CAAC,gBAAgB;YACrC,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,gBAAgB,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG;YAC5E,cAAc,EACb,IAAI,CAAC,kBAAkB,GAAG,CAAC;gBAC1B,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,GAAG,CAAC;gBACrE,CAAC,CAAC,CAAC;YACL,WAAW,EAAE,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACvF,UAAU,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI;YACvC,kBAAkB,EAAE,CAAC;YACrB,QAAQ,EAAE,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC7E,MAAM,EAAE,EAAE;SACV,CAAC;QAGF,KAAK,MAAM,CAAC,GAAG,EAAE,QAAQ,CAAC,IAAI,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,EAAE,CAAC;YAC5D,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAE9C,IAAI,KAAK,EAAE,CAAC;gBACX,MAAM,gBAAgB,GAAG,QAAQ,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;gBACzD,MAAM,cAAc,GAAG,GAAG,GAAG,gBAAgB,CAAC;gBAC9C,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;gBAEjE,IAAI,UAAU,EAAE,CAAC;oBAChB,kBAAkB,EAAE,CAAC;gBACtB,CAAC;gBAED,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG;oBACnB,SAAS,EAAE,QAAQ,CAAC,IAAI;oBACxB,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG;oBAC/D,cAAc,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC;oBAC9E,OAAO,EAAE,CAAA,MAAA,KAAK,CAAC,aAAa,0CAAE,MAAM,KAAI,CAAC;oBACzC,SAAS,EAAE,QAAQ,CAAC,SAAS,CAAC,WAAW,EAAE;oBAC3C,YAAY,EAAE,QAAQ,CAAC,YAAY,CAAC,WAAW,EAAE;oBACjD,QAAQ,EAAE,UAAU;oBACpB,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;iBAC/D,CAAC;YACH,CAAC;QACF,CAAC;QAED,KAAK,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;QAE9C,OAAO,KAAK,CAAC;IACd,CAAC;;AAxSF,4DAySC;AAxSe,iCAAQ,GAAoC,IAAI,AAAxC,CAAyC"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { MemoryVectorStore } from '@langchain/classic/vectorstores/memory';
|
|
2
|
+
import type { VectorStoreMetadata, IStoreCleanupService } from './types';
|
|
3
|
+
export declare class StoreCleanupService implements IStoreCleanupService {
|
|
4
|
+
private readonly maxMemorySizeBytes;
|
|
5
|
+
private readonly inactiveTtlMs;
|
|
6
|
+
private readonly vectorStores;
|
|
7
|
+
private readonly storeMetadata;
|
|
8
|
+
private readonly onCleanup;
|
|
9
|
+
private oldestStoreKeys;
|
|
10
|
+
private lastSortTime;
|
|
11
|
+
private readonly CACHE_TTL_MS;
|
|
12
|
+
constructor(maxMemorySizeBytes: number, inactiveTtlMs: number, vectorStores: Map<string, MemoryVectorStore>, storeMetadata: Map<string, VectorStoreMetadata>, onCleanup: (removedKeys: string[], freedBytes: number, reason: 'ttl' | 'memory') => void);
|
|
13
|
+
isStoreInactive(metadata: VectorStoreMetadata): boolean;
|
|
14
|
+
cleanupInactiveStores(): void;
|
|
15
|
+
cleanupOldestStores(requiredBytes: number): void;
|
|
16
|
+
}
|
|
@@ -0,0 +1,95 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.StoreCleanupService = void 0;
|
|
4
|
+
class StoreCleanupService {
|
|
5
|
+
constructor(maxMemorySizeBytes, inactiveTtlMs, vectorStores, storeMetadata, onCleanup) {
|
|
6
|
+
this.maxMemorySizeBytes = maxMemorySizeBytes;
|
|
7
|
+
this.inactiveTtlMs = inactiveTtlMs;
|
|
8
|
+
this.vectorStores = vectorStores;
|
|
9
|
+
this.storeMetadata = storeMetadata;
|
|
10
|
+
this.onCleanup = onCleanup;
|
|
11
|
+
this.oldestStoreKeys = [];
|
|
12
|
+
this.lastSortTime = 0;
|
|
13
|
+
this.CACHE_TTL_MS = 5000;
|
|
14
|
+
}
|
|
15
|
+
isStoreInactive(metadata) {
|
|
16
|
+
if (this.inactiveTtlMs <= 0) {
|
|
17
|
+
return false;
|
|
18
|
+
}
|
|
19
|
+
const now = Date.now();
|
|
20
|
+
const lastAccessedTime = metadata.lastAccessed.getTime();
|
|
21
|
+
return now - lastAccessedTime > this.inactiveTtlMs;
|
|
22
|
+
}
|
|
23
|
+
cleanupInactiveStores() {
|
|
24
|
+
if (this.inactiveTtlMs <= 0) {
|
|
25
|
+
return;
|
|
26
|
+
}
|
|
27
|
+
let freedBytes = 0;
|
|
28
|
+
const removedStores = [];
|
|
29
|
+
for (const [key, metadata] of this.storeMetadata.entries()) {
|
|
30
|
+
if (this.isStoreInactive(metadata)) {
|
|
31
|
+
this.vectorStores.delete(key);
|
|
32
|
+
freedBytes += metadata.size;
|
|
33
|
+
removedStores.push(key);
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
for (const key of removedStores) {
|
|
37
|
+
this.storeMetadata.delete(key);
|
|
38
|
+
}
|
|
39
|
+
if (removedStores.length > 0) {
|
|
40
|
+
this.oldestStoreKeys = [];
|
|
41
|
+
this.onCleanup(removedStores, freedBytes, 'ttl');
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
cleanupOldestStores(requiredBytes) {
|
|
45
|
+
if (this.maxMemorySizeBytes <= 0) {
|
|
46
|
+
return;
|
|
47
|
+
}
|
|
48
|
+
let currentMemoryUsage = 0;
|
|
49
|
+
for (const metadata of this.storeMetadata.values()) {
|
|
50
|
+
currentMemoryUsage += metadata.size;
|
|
51
|
+
}
|
|
52
|
+
this.cleanupInactiveStores();
|
|
53
|
+
currentMemoryUsage = 0;
|
|
54
|
+
for (const metadata of this.storeMetadata.values()) {
|
|
55
|
+
currentMemoryUsage += metadata.size;
|
|
56
|
+
}
|
|
57
|
+
if (currentMemoryUsage + requiredBytes <= this.maxMemorySizeBytes) {
|
|
58
|
+
return;
|
|
59
|
+
}
|
|
60
|
+
const now = Date.now();
|
|
61
|
+
if (this.oldestStoreKeys.length === 0 || now - this.lastSortTime > this.CACHE_TTL_MS) {
|
|
62
|
+
const stores = [];
|
|
63
|
+
for (const [key, metadata] of this.storeMetadata.entries()) {
|
|
64
|
+
stores.push([key, metadata.createdAt.getTime()]);
|
|
65
|
+
}
|
|
66
|
+
stores.sort((a, b) => a[1] - b[1]);
|
|
67
|
+
this.oldestStoreKeys = stores.map(([key]) => key);
|
|
68
|
+
this.lastSortTime = now;
|
|
69
|
+
}
|
|
70
|
+
let freedBytes = 0;
|
|
71
|
+
const removedStores = [];
|
|
72
|
+
for (const key of this.oldestStoreKeys) {
|
|
73
|
+
if (!this.storeMetadata.has(key))
|
|
74
|
+
continue;
|
|
75
|
+
if (currentMemoryUsage - freedBytes + requiredBytes <= this.maxMemorySizeBytes) {
|
|
76
|
+
break;
|
|
77
|
+
}
|
|
78
|
+
const metadata = this.storeMetadata.get(key);
|
|
79
|
+
if (metadata) {
|
|
80
|
+
this.vectorStores.delete(key);
|
|
81
|
+
freedBytes += metadata.size;
|
|
82
|
+
removedStores.push(key);
|
|
83
|
+
}
|
|
84
|
+
}
|
|
85
|
+
for (const key of removedStores) {
|
|
86
|
+
this.storeMetadata.delete(key);
|
|
87
|
+
}
|
|
88
|
+
if (removedStores.length > 0) {
|
|
89
|
+
this.oldestStoreKeys = this.oldestStoreKeys.filter((key) => !removedStores.includes(key));
|
|
90
|
+
this.onCleanup(removedStores, freedBytes, 'memory');
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
exports.StoreCleanupService = StoreCleanupService;
|
|
95
|
+
//# sourceMappingURL=StoreCleanupService.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"StoreCleanupService.js","sourceRoot":"","sources":["../../../../../nodes/vector_store/shared/MemoryManager/StoreCleanupService.ts"],"names":[],"mappings":";;;AAOA,MAAa,mBAAmB;IAQ/B,YACkB,kBAA0B,EAC1B,aAAqB,EACrB,YAA4C,EAC5C,aAA+C,EAC/C,SAIR;QARQ,uBAAkB,GAAlB,kBAAkB,CAAQ;QAC1B,kBAAa,GAAb,aAAa,CAAQ;QACrB,iBAAY,GAAZ,YAAY,CAAgC;QAC5C,kBAAa,GAAb,aAAa,CAAkC;QAC/C,cAAS,GAAT,SAAS,CAIjB;QAfF,oBAAe,GAAa,EAAE,CAAC;QAE/B,iBAAY,GAAG,CAAC,CAAC;QAER,iBAAY,GAAG,IAAI,CAAC;IAYlC,CAAC;IAKJ,eAAe,CAAC,QAA6B;QAE5C,IAAI,IAAI,CAAC,aAAa,IAAI,CAAC,EAAE,CAAC;YAC7B,OAAO,KAAK,CAAC;QACd,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,gBAAgB,GAAG,QAAQ,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;QACzD,OAAO,GAAG,GAAG,gBAAgB,GAAG,IAAI,CAAC,aAAa,CAAC;IACpD,CAAC;IAKD,qBAAqB;QAEpB,IAAI,IAAI,CAAC,aAAa,IAAI,CAAC,EAAE,CAAC;YAC7B,OAAO;QACR,CAAC;QAED,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,MAAM,aAAa,GAAa,EAAE,CAAC;QAGnC,KAAK,MAAM,CAAC,GAAG,EAAE,QAAQ,CAAC,IAAI,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,EAAE,CAAC;YAC5D,IAAI,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAEpC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAC9B,UAAU,IAAI,QAAQ,CAAC,IAAI,CAAC;gBAC5B,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACzB,CAAC;QACF,CAAC;QAGD,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;YACjC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;QAGD,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,eAAe,GAAG,EAAE,CAAC;YAC1B,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;QAClD,CAAC;IACF,CAAC;IAKD,mBAAmB,CAAC,aAAqB;QAExC,IAAI,IAAI,CAAC,kBAAkB,IAAI,CAAC,EAAE,CAAC;YAClC,OAAO;QACR,CAAC;QAGD,IAAI,kBAAkB,GAAG,CAAC,CAAC;QAC3B,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC;YACpD,kBAAkB,IAAI,QAAQ,CAAC,IAAI,CAAC;QACrC,CAAC;QAGD,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAG7B,kBAAkB,GAAG,CAAC,CAAC;QACvB,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC;YACpD,kBAAkB,IAAI,QAAQ,CAAC,IAAI,CAAC;QACrC,CAAC;QAGD,IAAI,kBAAkB,GAAG,aAAa,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACnE,OAAO;QACR,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAGvB,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,KAAK,CAAC,IAAI,GAAG,GAAG,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;YAEtF,MAAM,MAAM,GAA4B,EAAE,CAAC;YAE3C,KAAK,MAAM,CAAC,GAAG,EAAE,QAAQ,CAAC,IAAI,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,EAAE,CAAC;gBAC5D,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAClD,CAAC;YAGD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAGnC,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;YAClD,IAAI,CAAC,YAAY,GAAG,GAAG,CAAC;QACzB,CAAC;QAED,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,MAAM,aAAa,GAAa,EAAE,CAAC;QAGnC,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YAExC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC;gBAAE,SAAS;YAG3C,IAAI,kBAAkB,GAAG,UAAU,GAAG,aAAa,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAChF,MAAM;YACP,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC7C,IAAI,QAAQ,EAAE,CAAC;gBACd,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAC9B,UAAU,IAAI,QAAQ,CAAC,IAAI,CAAC;gBAC5B,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACzB,CAAC;QACF,CAAC;QAGD,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;YACjC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;QAGD,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAE9B,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,aAAa,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1F,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;QACrD,CAAC;IACF,CAAC;CACD;AArJD,kDAqJC"}
|