@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,418 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
3
|
+
if (k2 === undefined) k2 = k;
|
|
4
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
5
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
6
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
7
|
+
}
|
|
8
|
+
Object.defineProperty(o, k2, desc);
|
|
9
|
+
}) : (function(o, m, k, k2) {
|
|
10
|
+
if (k2 === undefined) k2 = k;
|
|
11
|
+
o[k2] = m[k];
|
|
12
|
+
}));
|
|
13
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
14
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
15
|
+
}) : function(o, v) {
|
|
16
|
+
o["default"] = v;
|
|
17
|
+
});
|
|
18
|
+
var __importStar = (this && this.__importStar) || (function () {
|
|
19
|
+
var ownKeys = function(o) {
|
|
20
|
+
ownKeys = Object.getOwnPropertyNames || function (o) {
|
|
21
|
+
var ar = [];
|
|
22
|
+
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
|
|
23
|
+
return ar;
|
|
24
|
+
};
|
|
25
|
+
return ownKeys(o);
|
|
26
|
+
};
|
|
27
|
+
return function (mod) {
|
|
28
|
+
if (mod && mod.__esModule) return mod;
|
|
29
|
+
var result = {};
|
|
30
|
+
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
|
|
31
|
+
__setModuleDefault(result, mod);
|
|
32
|
+
return result;
|
|
33
|
+
};
|
|
34
|
+
})();
|
|
35
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
36
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
37
|
+
};
|
|
38
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
39
|
+
exports.toolsAgentExecute = toolsAgentExecute;
|
|
40
|
+
const runnables_1 = require("@langchain/core/runnables");
|
|
41
|
+
const agents_1 = require("@langchain/classic/agents");
|
|
42
|
+
const omit_1 = __importDefault(require("lodash/omit"));
|
|
43
|
+
const n8n_workflow_1 = require("n8n-workflow");
|
|
44
|
+
const node_assert_1 = __importDefault(require("node:assert"));
|
|
45
|
+
const CallbackHandler_1 = require("./CallbackHandler");
|
|
46
|
+
const mlflow = __importStar(require("mlflow-tracing"));
|
|
47
|
+
const helpers_1 = require("./src/utils/helpers");
|
|
48
|
+
const N8nOutputParser_1 = require("./src/utils/N8nOutputParser");
|
|
49
|
+
const common_1 = require("./src/utils/common");
|
|
50
|
+
const prompt_1 = require("./src/utils/prompt");
|
|
51
|
+
const constants_1 = require("./src/constants");
|
|
52
|
+
const retry_1 = require("./src/utils/retry");
|
|
53
|
+
const security_1 = require("./src/utils/security");
|
|
54
|
+
function createAgentExecutor(model, tools, prompt, options, outputParser, memory, fallbackModel, mlflowHandler) {
|
|
55
|
+
var _a;
|
|
56
|
+
const callbacks = mlflowHandler ? [mlflowHandler] : [];
|
|
57
|
+
const agent = (0, agents_1.createToolCallingAgent)({
|
|
58
|
+
llm: model,
|
|
59
|
+
tools,
|
|
60
|
+
prompt,
|
|
61
|
+
streamRunnable: false,
|
|
62
|
+
});
|
|
63
|
+
let fallbackAgent;
|
|
64
|
+
if (fallbackModel) {
|
|
65
|
+
fallbackAgent = (0, agents_1.createToolCallingAgent)({
|
|
66
|
+
llm: fallbackModel,
|
|
67
|
+
tools,
|
|
68
|
+
prompt,
|
|
69
|
+
streamRunnable: false,
|
|
70
|
+
});
|
|
71
|
+
}
|
|
72
|
+
const runnableAgent = runnables_1.RunnableSequence.from([
|
|
73
|
+
fallbackAgent ? agent.withFallbacks([fallbackAgent]) : agent,
|
|
74
|
+
(0, common_1.getAgentStepsParser)(outputParser, memory),
|
|
75
|
+
common_1.fixEmptyContentMessage,
|
|
76
|
+
]);
|
|
77
|
+
runnableAgent.singleAction = false;
|
|
78
|
+
runnableAgent.streamRunnable = false;
|
|
79
|
+
return agents_1.AgentExecutor.fromAgentAndTools({
|
|
80
|
+
agent: runnableAgent,
|
|
81
|
+
memory,
|
|
82
|
+
tools,
|
|
83
|
+
returnIntermediateSteps: options.returnIntermediateSteps === true,
|
|
84
|
+
maxIterations: (_a = options.maxIterations) !== null && _a !== void 0 ? _a : 10,
|
|
85
|
+
callbacks,
|
|
86
|
+
});
|
|
87
|
+
}
|
|
88
|
+
async function processEventStream(ctx, eventStream, itemIndex, returnIntermediateSteps = false) {
|
|
89
|
+
var _a;
|
|
90
|
+
const agentResult = {
|
|
91
|
+
output: '',
|
|
92
|
+
};
|
|
93
|
+
if (returnIntermediateSteps) {
|
|
94
|
+
agentResult.intermediateSteps = [];
|
|
95
|
+
}
|
|
96
|
+
if (ctx.sendChunk)
|
|
97
|
+
ctx.sendChunk('begin', itemIndex);
|
|
98
|
+
for await (const event of eventStream) {
|
|
99
|
+
switch (event.event) {
|
|
100
|
+
case 'on_chat_model_stream':
|
|
101
|
+
const chunk = (_a = event.data) === null || _a === void 0 ? void 0 : _a.chunk;
|
|
102
|
+
if (chunk === null || chunk === void 0 ? void 0 : chunk.content) {
|
|
103
|
+
const chunkContent = chunk.content;
|
|
104
|
+
let chunkText = '';
|
|
105
|
+
if (Array.isArray(chunkContent)) {
|
|
106
|
+
for (const message of chunkContent) {
|
|
107
|
+
chunkText += message === null || message === void 0 ? void 0 : message.text;
|
|
108
|
+
}
|
|
109
|
+
}
|
|
110
|
+
else if (typeof chunkContent === 'string') {
|
|
111
|
+
chunkText = chunkContent;
|
|
112
|
+
}
|
|
113
|
+
if (ctx.sendChunk)
|
|
114
|
+
ctx.sendChunk('item', itemIndex, chunkText);
|
|
115
|
+
agentResult.output += chunkText;
|
|
116
|
+
}
|
|
117
|
+
break;
|
|
118
|
+
case 'on_chat_model_end':
|
|
119
|
+
if (returnIntermediateSteps && event.data) {
|
|
120
|
+
const chatModelData = event.data;
|
|
121
|
+
const output = chatModelData.output;
|
|
122
|
+
if ((output === null || output === void 0 ? void 0 : output.tool_calls) && output.tool_calls.length > 0) {
|
|
123
|
+
for (const toolCall of output.tool_calls) {
|
|
124
|
+
agentResult.intermediateSteps.push({
|
|
125
|
+
action: {
|
|
126
|
+
tool: toolCall.name,
|
|
127
|
+
toolInput: toolCall.args,
|
|
128
|
+
log: output.content ||
|
|
129
|
+
`Calling ${toolCall.name} with input: ${JSON.stringify(toolCall.args)}`,
|
|
130
|
+
messageLog: [output],
|
|
131
|
+
toolCallId: toolCall.id,
|
|
132
|
+
type: toolCall.type,
|
|
133
|
+
},
|
|
134
|
+
});
|
|
135
|
+
}
|
|
136
|
+
}
|
|
137
|
+
}
|
|
138
|
+
break;
|
|
139
|
+
case 'on_tool_end':
|
|
140
|
+
if (returnIntermediateSteps && event.data && agentResult.intermediateSteps.length > 0) {
|
|
141
|
+
const toolData = event.data;
|
|
142
|
+
const matchingStep = agentResult.intermediateSteps.find((step) => !step.observation && step.action.tool === event.name);
|
|
143
|
+
if (matchingStep) {
|
|
144
|
+
matchingStep.observation = toolData.output;
|
|
145
|
+
}
|
|
146
|
+
}
|
|
147
|
+
break;
|
|
148
|
+
default:
|
|
149
|
+
break;
|
|
150
|
+
}
|
|
151
|
+
}
|
|
152
|
+
if (ctx.sendChunk)
|
|
153
|
+
ctx.sendChunk('end', itemIndex);
|
|
154
|
+
return agentResult;
|
|
155
|
+
}
|
|
156
|
+
async function toolsAgentExecute() {
|
|
157
|
+
const enableMLflow = this.getNodeParameter('enableMLflow', 0, false);
|
|
158
|
+
if (enableMLflow) {
|
|
159
|
+
const credentials = await this.getCredentials('databricks');
|
|
160
|
+
if (!credentials) {
|
|
161
|
+
throw new n8n_workflow_1.NodeOperationError(this.getNode(), constants_1.ERROR_MESSAGES.MISSING_CREDENTIALS);
|
|
162
|
+
}
|
|
163
|
+
if (!credentials.host || typeof credentials.host !== 'string' || credentials.host.trim() === '') {
|
|
164
|
+
throw new n8n_workflow_1.NodeOperationError(this.getNode(), constants_1.ERROR_MESSAGES.EMPTY_HOST);
|
|
165
|
+
}
|
|
166
|
+
if (!credentials.token || typeof credentials.token !== 'string' || credentials.token.trim() === '') {
|
|
167
|
+
throw new n8n_workflow_1.NodeOperationError(this.getNode(), constants_1.ERROR_MESSAGES.EMPTY_TOKEN);
|
|
168
|
+
}
|
|
169
|
+
const token = credentials.token;
|
|
170
|
+
if (token.length < constants_1.MLFLOW_CONSTANTS.MIN_DATABRICKS_TOKEN_LENGTH) {
|
|
171
|
+
this.logger.warn(constants_1.ERROR_MESSAGES.SHORT_TOKEN_WARNING(token.length));
|
|
172
|
+
}
|
|
173
|
+
const rawHost = credentials.host;
|
|
174
|
+
let databricksHost;
|
|
175
|
+
try {
|
|
176
|
+
const cleanedHost = rawHost.trim().replace(/\/$/, '');
|
|
177
|
+
const hostWithProtocol = cleanedHost.startsWith('http')
|
|
178
|
+
? cleanedHost
|
|
179
|
+
: `https://${cleanedHost}`;
|
|
180
|
+
const hostValidation = (0, security_1.validateDatabricksHost)(hostWithProtocol);
|
|
181
|
+
if (!hostValidation.isValid) {
|
|
182
|
+
throw new Error(hostValidation.error);
|
|
183
|
+
}
|
|
184
|
+
const url = new URL(hostWithProtocol);
|
|
185
|
+
if (url.protocol === 'http:') {
|
|
186
|
+
this.logger.warn(constants_1.ERROR_MESSAGES.HTTP_INSECURE_WARNING(url.href));
|
|
187
|
+
}
|
|
188
|
+
else if (url.protocol !== 'https:') {
|
|
189
|
+
throw new Error(`Host must use HTTP or HTTPS protocol, got: ${url.protocol}`);
|
|
190
|
+
}
|
|
191
|
+
if (!url.hostname) {
|
|
192
|
+
throw new Error('Invalid host URL: missing hostname');
|
|
193
|
+
}
|
|
194
|
+
databricksHost = hostValidation.sanitized;
|
|
195
|
+
}
|
|
196
|
+
catch (validationError) {
|
|
197
|
+
const errorMsg = validationError instanceof Error ? validationError.message : 'Invalid URL format';
|
|
198
|
+
throw new n8n_workflow_1.NodeOperationError(this.getNode(), constants_1.ERROR_MESSAGES.INVALID_HOST_URL(rawHost, errorMsg));
|
|
199
|
+
}
|
|
200
|
+
const workflowId = this.getWorkflow().id;
|
|
201
|
+
const experimentName = `/Shared/n8n-workflows-${workflowId}`;
|
|
202
|
+
let experimentId;
|
|
203
|
+
try {
|
|
204
|
+
const client = new mlflow.MlflowClient({
|
|
205
|
+
trackingUri: 'databricks',
|
|
206
|
+
host: databricksHost,
|
|
207
|
+
databricksToken: credentials.token,
|
|
208
|
+
});
|
|
209
|
+
try {
|
|
210
|
+
experimentId = await client.createExperiment(experimentName);
|
|
211
|
+
}
|
|
212
|
+
catch (createError) {
|
|
213
|
+
const getResponse = await this.helpers.request({
|
|
214
|
+
method: 'GET',
|
|
215
|
+
url: `${databricksHost}/api/2.0/mlflow/experiments/get-by-name`,
|
|
216
|
+
headers: {
|
|
217
|
+
'Authorization': `Bearer ${credentials.token}`,
|
|
218
|
+
'Content-Type': 'application/json',
|
|
219
|
+
},
|
|
220
|
+
qs: {
|
|
221
|
+
experiment_name: experimentName,
|
|
222
|
+
},
|
|
223
|
+
json: true,
|
|
224
|
+
});
|
|
225
|
+
if (getResponse && getResponse.experiment && getResponse.experiment.experiment_id) {
|
|
226
|
+
experimentId = getResponse.experiment.experiment_id;
|
|
227
|
+
}
|
|
228
|
+
else {
|
|
229
|
+
throw createError;
|
|
230
|
+
}
|
|
231
|
+
}
|
|
232
|
+
if (!experimentId) {
|
|
233
|
+
throw new Error('Failed to get or create experiment ID');
|
|
234
|
+
}
|
|
235
|
+
mlflow.init({
|
|
236
|
+
trackingUri: 'databricks',
|
|
237
|
+
experimentId: experimentId,
|
|
238
|
+
host: databricksHost,
|
|
239
|
+
databricksToken: credentials.token,
|
|
240
|
+
});
|
|
241
|
+
}
|
|
242
|
+
catch (error) {
|
|
243
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
244
|
+
throw new n8n_workflow_1.NodeOperationError(this.getNode(), `Failed to setup MLflow experiment: ${(0, retry_1.maskTokensInText)(errorMessage)}`);
|
|
245
|
+
}
|
|
246
|
+
}
|
|
247
|
+
const returnData = [];
|
|
248
|
+
const items = this.getInputData();
|
|
249
|
+
if (!items || items.length === 0) {
|
|
250
|
+
return [returnData];
|
|
251
|
+
}
|
|
252
|
+
const rawBatchSize = this.getNodeParameter('options.batching.batchSize', 0, 1);
|
|
253
|
+
let batchSize;
|
|
254
|
+
if (!Number.isFinite(rawBatchSize) || rawBatchSize < 1) {
|
|
255
|
+
this.logger.warn(`Invalid batch size ${rawBatchSize}, using default: 1`);
|
|
256
|
+
batchSize = 1;
|
|
257
|
+
}
|
|
258
|
+
else if (rawBatchSize > items.length) {
|
|
259
|
+
this.logger.warn(`Batch size ${rawBatchSize} exceeds items count ${items.length}, using ${items.length}`);
|
|
260
|
+
batchSize = items.length;
|
|
261
|
+
}
|
|
262
|
+
else {
|
|
263
|
+
batchSize = Math.floor(rawBatchSize);
|
|
264
|
+
if (batchSize !== rawBatchSize) {
|
|
265
|
+
this.logger.warn(`Non-integer batch size ${rawBatchSize}, rounded to ${batchSize}`);
|
|
266
|
+
}
|
|
267
|
+
}
|
|
268
|
+
const rawDelayBetweenBatches = this.getNodeParameter('options.batching.delayBetweenBatches', 0, 0);
|
|
269
|
+
let delayBetweenBatches;
|
|
270
|
+
if (!Number.isFinite(rawDelayBetweenBatches) || rawDelayBetweenBatches < 0) {
|
|
271
|
+
this.logger.warn(`Invalid delay ${rawDelayBetweenBatches}, using 0`);
|
|
272
|
+
delayBetweenBatches = 0;
|
|
273
|
+
}
|
|
274
|
+
else if (rawDelayBetweenBatches > constants_1.MLFLOW_CONSTANTS.MAX_DELAY_BETWEEN_BATCHES_MS) {
|
|
275
|
+
this.logger.warn(`Delay ${rawDelayBetweenBatches}ms exceeds maximum ${constants_1.MLFLOW_CONSTANTS.MAX_DELAY_BETWEEN_BATCHES_MS}ms, capping to maximum`);
|
|
276
|
+
delayBetweenBatches = constants_1.MLFLOW_CONSTANTS.MAX_DELAY_BETWEEN_BATCHES_MS;
|
|
277
|
+
}
|
|
278
|
+
else {
|
|
279
|
+
delayBetweenBatches = Math.floor(rawDelayBetweenBatches);
|
|
280
|
+
if (delayBetweenBatches !== rawDelayBetweenBatches) {
|
|
281
|
+
this.logger.warn(`Non-integer delay ${rawDelayBetweenBatches}, rounded to ${delayBetweenBatches}`);
|
|
282
|
+
}
|
|
283
|
+
}
|
|
284
|
+
const needsFallback = this.getNodeParameter('needsFallback', 0, false);
|
|
285
|
+
const memory = await (0, common_1.getOptionalMemory)(this);
|
|
286
|
+
const model = await (0, common_1.getChatModel)(this, 0);
|
|
287
|
+
(0, node_assert_1.default)(model, constants_1.ERROR_MESSAGES.MODEL_NOT_CONNECTED);
|
|
288
|
+
const fallbackModel = needsFallback ? await (0, common_1.getChatModel)(this, 1) : null;
|
|
289
|
+
if (needsFallback && !fallbackModel) {
|
|
290
|
+
throw new n8n_workflow_1.NodeOperationError(this.getNode(), constants_1.ERROR_MESSAGES.FALLBACK_MODEL_NOT_CONNECTED);
|
|
291
|
+
}
|
|
292
|
+
const enableStreaming = this.getNodeParameter('options.enableStreaming', 0, true);
|
|
293
|
+
const mlflowHandlers = [];
|
|
294
|
+
try {
|
|
295
|
+
for (let i = 0; i < items.length; i += batchSize) {
|
|
296
|
+
const cancelSignal = this.getExecutionCancelSignal();
|
|
297
|
+
if (cancelSignal === null || cancelSignal === void 0 ? void 0 : cancelSignal.aborted) {
|
|
298
|
+
const currentBatch = Math.floor(i / batchSize) + 1;
|
|
299
|
+
const totalBatches = Math.ceil(items.length / batchSize);
|
|
300
|
+
this.logger.info(`Execution cancelled at batch ${currentBatch}`);
|
|
301
|
+
throw new n8n_workflow_1.NodeOperationError(this.getNode(), constants_1.ERROR_MESSAGES.EXECUTION_CANCELLED(currentBatch, totalBatches));
|
|
302
|
+
}
|
|
303
|
+
const batch = items.slice(i, i + batchSize);
|
|
304
|
+
const batchPromises = batch.map(async (_item, batchItemIndex) => {
|
|
305
|
+
var _a, _b, _c;
|
|
306
|
+
const itemIndex = i + batchItemIndex;
|
|
307
|
+
const input = (0, helpers_1.getPromptInputByType)({
|
|
308
|
+
ctx: this,
|
|
309
|
+
i: itemIndex,
|
|
310
|
+
inputKey: 'text',
|
|
311
|
+
promptTypeKey: 'promptType',
|
|
312
|
+
});
|
|
313
|
+
if (input === undefined) {
|
|
314
|
+
throw new n8n_workflow_1.NodeOperationError(this.getNode(), 'The "text" parameter is empty.');
|
|
315
|
+
}
|
|
316
|
+
const outputParser = await (0, N8nOutputParser_1.getOptionalOutputParser)(this, itemIndex);
|
|
317
|
+
const tools = await (0, common_1.getTools)(this, outputParser);
|
|
318
|
+
const options = this.getNodeParameter('options', itemIndex, {});
|
|
319
|
+
const mlflowHandler = enableMLflow ? new CallbackHandler_1.CallbackHandler({}) : undefined;
|
|
320
|
+
if (mlflowHandler) {
|
|
321
|
+
mlflowHandlers.push(mlflowHandler);
|
|
322
|
+
}
|
|
323
|
+
const messages = await (0, common_1.prepareMessages)(this, itemIndex, {
|
|
324
|
+
systemMessage: options.systemMessage,
|
|
325
|
+
passthroughBinaryImages: (_a = options.passthroughBinaryImages) !== null && _a !== void 0 ? _a : true,
|
|
326
|
+
outputParser,
|
|
327
|
+
});
|
|
328
|
+
const prompt = (0, common_1.preparePrompt)(messages);
|
|
329
|
+
const executor = createAgentExecutor(model, tools, prompt, options, outputParser, memory, fallbackModel, mlflowHandler);
|
|
330
|
+
const invokeParams = {
|
|
331
|
+
input,
|
|
332
|
+
system_message: (_b = options.systemMessage) !== null && _b !== void 0 ? _b : prompt_1.SYSTEM_MESSAGE,
|
|
333
|
+
};
|
|
334
|
+
if (outputParser) {
|
|
335
|
+
invokeParams.formatting_instructions =
|
|
336
|
+
'IMPORTANT: For your response to user, you MUST use the `format_final_json_response` tool with your complete answer formatted according to the required schema. Do not attempt to format the JSON manually - always use this tool. Your response will be rejected if it is not properly formatted through this tool. Only use this tool once you are ready to provide your final answer.';
|
|
337
|
+
}
|
|
338
|
+
const executeOptions = {
|
|
339
|
+
signal: this.getExecutionCancelSignal(),
|
|
340
|
+
callbacks: mlflowHandler ? [mlflowHandler] : []
|
|
341
|
+
};
|
|
342
|
+
const isStreamingAvailable = 'isStreaming' in this ? (_c = this.isStreaming) === null || _c === void 0 ? void 0 : _c.call(this) : undefined;
|
|
343
|
+
if ('isStreaming' in this &&
|
|
344
|
+
enableStreaming &&
|
|
345
|
+
isStreamingAvailable &&
|
|
346
|
+
this.getNode().typeVersion >= 2.1) {
|
|
347
|
+
const chatHistory = await (memory === null || memory === void 0 ? void 0 : memory.chatHistory.getMessages());
|
|
348
|
+
const eventStream = executor.streamEvents({
|
|
349
|
+
...invokeParams,
|
|
350
|
+
chat_history: chatHistory !== null && chatHistory !== void 0 ? chatHistory : undefined,
|
|
351
|
+
}, {
|
|
352
|
+
version: 'v2',
|
|
353
|
+
...executeOptions,
|
|
354
|
+
});
|
|
355
|
+
return await processEventStream(this, eventStream, itemIndex, options.returnIntermediateSteps);
|
|
356
|
+
}
|
|
357
|
+
else {
|
|
358
|
+
return await executor.invoke(invokeParams, executeOptions);
|
|
359
|
+
}
|
|
360
|
+
});
|
|
361
|
+
const batchResults = await Promise.allSettled(batchPromises);
|
|
362
|
+
const outputParser = await (0, N8nOutputParser_1.getOptionalOutputParser)(this, 0);
|
|
363
|
+
batchResults.forEach((result, index) => {
|
|
364
|
+
var _a;
|
|
365
|
+
const itemIndex = i + index;
|
|
366
|
+
if (result.status === 'rejected') {
|
|
367
|
+
const error = result.reason;
|
|
368
|
+
const maskedMessage = (0, retry_1.maskTokensInText)(error.message);
|
|
369
|
+
const maskedStack = error.stack ? (0, retry_1.maskTokensInText)(error.stack) : undefined;
|
|
370
|
+
if (this.continueOnFail()) {
|
|
371
|
+
returnData.push({
|
|
372
|
+
json: { error: maskedMessage, stack: maskedStack },
|
|
373
|
+
pairedItem: { item: itemIndex },
|
|
374
|
+
});
|
|
375
|
+
return;
|
|
376
|
+
}
|
|
377
|
+
else {
|
|
378
|
+
const enhancedError = new Error(`Agent execution failed: ${maskedMessage}\n\nOriginal stack:\n${maskedStack || 'N/A'}`);
|
|
379
|
+
throw new n8n_workflow_1.NodeOperationError(this.getNode(), enhancedError);
|
|
380
|
+
}
|
|
381
|
+
}
|
|
382
|
+
const response = result.value;
|
|
383
|
+
if (Array.isArray(response.output)) {
|
|
384
|
+
const textItems = response.output.filter((item) => (item === null || item === void 0 ? void 0 : item.type) === 'text');
|
|
385
|
+
if (textItems.length > 0) {
|
|
386
|
+
response.output = textItems.map((item) => item.text).join('\n');
|
|
387
|
+
}
|
|
388
|
+
}
|
|
389
|
+
if (memory && outputParser) {
|
|
390
|
+
const parsedOutput = (0, n8n_workflow_1.jsonParse)(response.output);
|
|
391
|
+
response.output = (_a = parsedOutput === null || parsedOutput === void 0 ? void 0 : parsedOutput.output) !== null && _a !== void 0 ? _a : parsedOutput;
|
|
392
|
+
}
|
|
393
|
+
const itemResult = {
|
|
394
|
+
json: (0, omit_1.default)(response, 'system_message', 'formatting_instructions', 'input', 'chat_history', 'agent_scratchpad'),
|
|
395
|
+
pairedItem: { item: itemIndex },
|
|
396
|
+
};
|
|
397
|
+
returnData.push(itemResult);
|
|
398
|
+
});
|
|
399
|
+
if (i + batchSize < items.length && delayBetweenBatches > 0) {
|
|
400
|
+
await (0, n8n_workflow_1.sleep)(delayBetweenBatches);
|
|
401
|
+
}
|
|
402
|
+
}
|
|
403
|
+
return [returnData];
|
|
404
|
+
}
|
|
405
|
+
finally {
|
|
406
|
+
await Promise.allSettled(mlflowHandlers.map(async (handler) => {
|
|
407
|
+
try {
|
|
408
|
+
await new Promise(resolve => setTimeout(resolve, constants_1.MLFLOW_CONSTANTS.SPAN_CLEANUP_DELAY_MS));
|
|
409
|
+
handler.cleanup();
|
|
410
|
+
}
|
|
411
|
+
catch (cleanupError) {
|
|
412
|
+
const errorMsg = cleanupError instanceof Error ? cleanupError.message : String(cleanupError);
|
|
413
|
+
this.logger.debug(`Error cleaning up MLflow handler: ${(0, retry_1.maskTokensInText)(errorMsg)}`);
|
|
414
|
+
}
|
|
415
|
+
}));
|
|
416
|
+
}
|
|
417
|
+
}
|
|
418
|
+
//# sourceMappingURL=execute.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"execute.js","sourceRoot":"","sources":["../../../../nodes/agents/DatabricksAiAgent/execute.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAuPA,8CA0YC;AA5nBD,yDAA6D;AAC7D,sDAImC;AAGnC,uDAA+B;AAC/B,+CAAoE;AAEpE,8DAAiC;AACjC,uDAAoD;AACpD,uDAAyC;AAEzC,iDAA2D;AAE3D,iEAGqC;AAErC,+CAQ4B;AAE5B,+CAAoD;AACpD,+CAAmE;AACnE,6CAAqD;AACrD,mDAE8B;AAkC9B,SAAS,mBAAmB,CACxB,KAAoB,EACpB,KAA0C,EAC1C,MAA0B,EAC1B,OAAsE,EACtE,YAA8B,EAC9B,MAAuB,EACvB,aAAoC,EACpC,aAA+B;;IAE/B,MAAM,SAAS,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAEvD,MAAM,KAAK,GAAG,IAAA,+BAAsB,EAAC;QACjC,GAAG,EAAE,KAAK;QACV,KAAK;QACL,MAAM;QACN,cAAc,EAAE,KAAK;KACxB,CAAC,CAAC;IAEH,IAAI,aAAgD,CAAC;IACrD,IAAI,aAAa,EAAE,CAAC;QAChB,aAAa,GAAG,IAAA,+BAAsB,EAAC;YACnC,GAAG,EAAE,aAAa;YAClB,KAAK;YACL,MAAM;YACN,cAAc,EAAE,KAAK;SACxB,CAAC,CAAC;IACP,CAAC;IACD,MAAM,aAAa,GAAG,4BAAgB,CAAC,IAAI,CAAC;QACxC,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK;QAC5D,IAAA,4BAAmB,EAAC,YAAY,EAAE,MAAM,CAAC;QACzC,+BAAsB;KACzB,CAA0B,CAAC;IAE5B,aAAa,CAAC,YAAY,GAAG,KAAK,CAAC;IACnC,aAAa,CAAC,cAAc,GAAG,KAAK,CAAC;IAErC,OAAO,sBAAa,CAAC,iBAAiB,CAAC;QACnC,KAAK,EAAE,aAAa;QACpB,MAAM;QACN,KAAK;QACL,uBAAuB,EAAE,OAAO,CAAC,uBAAuB,KAAK,IAAI;QACjE,aAAa,EAAE,MAAA,OAAO,CAAC,aAAa,mCAAI,EAAE;QAC1C,SAAS;KACZ,CAAC,CAAC;AACP,CAAC;AA2BD,KAAK,UAAU,kBAAkB,CAC7B,GAAsB,EACtB,WAAgD,EAChD,SAAiB,EACjB,0BAAmC,KAAK;;IAExC,MAAM,WAAW,GAAkD;QAC/D,MAAM,EAAE,EAAE;KACb,CAAC;IAEF,IAAI,uBAAuB,EAAE,CAAC;QAC1B,WAAW,CAAC,iBAAiB,GAAG,EAAE,CAAC;IACvC,CAAC;IAGD,IAAI,GAAG,CAAC,SAAS;QAAE,GAAG,CAAC,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IACrD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;QAEpC,QAAQ,KAAK,CAAC,KAAK,EAAE,CAAC;YAClB,KAAK,sBAAsB;gBACvB,MAAM,KAAK,GAAG,MAAA,KAAK,CAAC,IAAI,0CAAE,KAAuB,CAAC;gBAClD,IAAI,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,OAAO,EAAE,CAAC;oBACjB,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC;oBACnC,IAAI,SAAS,GAAG,EAAE,CAAC;oBACnB,IAAI,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;wBAC9B,KAAK,MAAM,OAAO,IAAI,YAAY,EAAE,CAAC;4BACjC,SAAS,IAAK,OAA8B,aAA9B,OAAO,uBAAP,OAAO,CAAyB,IAAI,CAAC;wBACvD,CAAC;oBACL,CAAC;yBAAM,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;wBAC1C,SAAS,GAAG,YAAY,CAAC;oBAC7B,CAAC;oBAED,IAAI,GAAG,CAAC,SAAS;wBAAE,GAAG,CAAC,SAAS,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;oBAE/D,WAAW,CAAC,MAAM,IAAI,SAAS,CAAC;gBACpC,CAAC;gBACD,MAAM;YACV,KAAK,mBAAmB;gBAEpB,IAAI,uBAAuB,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;oBACxC,MAAM,aAAa,GAAG,KAAK,CAAC,IAAwH,CAAC;oBACrJ,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC;oBAGpC,IAAI,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,UAAU,KAAI,MAAM,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACrD,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;4BACvC,WAAW,CAAC,iBAAkB,CAAC,IAAI,CAAC;gCAChC,MAAM,EAAE;oCACJ,IAAI,EAAE,QAAQ,CAAC,IAAI;oCACnB,SAAS,EAAE,QAAQ,CAAC,IAAI;oCACxB,GAAG,EACC,MAAM,CAAC,OAAO;wCACd,WAAW,QAAQ,CAAC,IAAI,gBAAgB,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;oCAC3E,UAAU,EAAE,CAAC,MAAM,CAAC;oCACpB,UAAU,EAAE,QAAQ,CAAC,EAAE;oCACvB,IAAI,EAAE,QAAQ,CAAC,IAAI;iCACtB;6BACJ,CAAC,CAAC;wBACP,CAAC;oBACL,CAAC;gBACL,CAAC;gBACD,MAAM;YACV,KAAK,aAAa;gBAEd,IAAI,uBAAuB,IAAI,KAAK,CAAC,IAAI,IAAI,WAAW,CAAC,iBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACrF,MAAM,QAAQ,GAAG,KAAK,CAAC,IAA4B,CAAC;oBAEpD,MAAM,YAAY,GAAG,WAAW,CAAC,iBAAkB,CAAC,IAAI,CACpD,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI,CACjE,CAAC;oBACF,IAAI,YAAY,EAAE,CAAC;wBACf,YAAY,CAAC,WAAW,GAAG,QAAQ,CAAC,MAAM,CAAC;oBAC/C,CAAC;gBACL,CAAC;gBACD,MAAM;YACV;gBACI,MAAM;QACd,CAAC;IACL,CAAC;IAED,IAAI,GAAG,CAAC,SAAS;QAAE,GAAG,CAAC,SAAS,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IAEnD,OAAO,WAAW,CAAC;AACvB,CAAC;AAgBM,KAAK,UAAU,iBAAiB;IAInC,MAAM,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,CAAC,EAAE,KAAK,CAAY,CAAC;IAEhF,IAAI,YAAY,EAAE,CAAC;QAEf,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAE5D,IAAI,CAAC,WAAW,EAAE,CAAC;YACf,MAAM,IAAI,iCAAkB,CACxB,IAAI,CAAC,OAAO,EAAE,EACd,0BAAc,CAAC,mBAAmB,CACrC,CAAC;QACN,CAAC;QAGD,IAAI,CAAC,WAAW,CAAC,IAAI,IAAI,OAAO,WAAW,CAAC,IAAI,KAAK,QAAQ,IAAI,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YAC9F,MAAM,IAAI,iCAAkB,CACxB,IAAI,CAAC,OAAO,EAAE,EACd,0BAAc,CAAC,UAAU,CAC5B,CAAC;QACN,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,KAAK,IAAI,OAAO,WAAW,CAAC,KAAK,KAAK,QAAQ,IAAI,WAAW,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YACjG,MAAM,IAAI,iCAAkB,CACxB,IAAI,CAAC,OAAO,EAAE,EACd,0BAAc,CAAC,WAAW,CAC7B,CAAC;QACN,CAAC;QAGD,MAAM,KAAK,GAAG,WAAW,CAAC,KAAe,CAAC;QAC1C,IAAI,KAAK,CAAC,MAAM,GAAG,4BAAgB,CAAC,2BAA2B,EAAE,CAAC;YAC9D,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0BAAc,CAAC,mBAAmB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;QACvE,CAAC;QAGD,MAAM,OAAO,GAAG,WAAW,CAAC,IAAc,CAAC;QAC3C,IAAI,cAAsB,CAAC;QAC3B,IAAI,CAAC;YACD,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YAGtD,MAAM,gBAAgB,GAAG,WAAW,CAAC,UAAU,CAAC,MAAM,CAAC;gBACnD,CAAC,CAAC,WAAW;gBACb,CAAC,CAAC,WAAW,WAAW,EAAE,CAAC;YAG/B,MAAM,cAAc,GAAG,IAAA,iCAAsB,EAAC,gBAAgB,CAAC,CAAC;YAChE,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YAC1C,CAAC;YAED,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,gBAAgB,CAAC,CAAC;YAGtC,IAAI,GAAG,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;gBAC3B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0BAAc,CAAC,qBAAqB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;YACrE,CAAC;iBAAM,IAAI,GAAG,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACnC,MAAM,IAAI,KAAK,CAAC,8CAA8C,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;YAClF,CAAC;YAED,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;gBAChB,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;YAC1D,CAAC;YAED,cAAc,GAAG,cAAc,CAAC,SAAU,CAAC;QAC/C,CAAC;QAAC,OAAO,eAAwB,EAAE,CAAC;YAChC,MAAM,QAAQ,GAAG,eAAe,YAAY,KAAK,CAAC,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,oBAAoB,CAAC;YACnG,MAAM,IAAI,iCAAkB,CACxB,IAAI,CAAC,OAAO,EAAE,EACd,0BAAc,CAAC,gBAAgB,CAAC,OAAO,EAAE,QAAQ,CAAC,CACrD,CAAC;QACN,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;QACzC,MAAM,cAAc,GAAG,yBAAyB,UAAU,EAAE,CAAC;QAE7D,IAAI,YAAgC,CAAC;QAErC,IAAI,CAAC;YAED,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,YAAY,CAAC;gBACnC,WAAW,EAAE,YAAY;gBACzB,IAAI,EAAE,cAAc;gBACpB,eAAe,EAAE,WAAW,CAAC,KAAe;aAC/C,CAAC,CAAC;YAEH,IAAI,CAAC;gBAED,YAAY,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;YACjE,CAAC;YAAC,OAAO,WAAoB,EAAE,CAAC;gBAG5B,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC;oBAC3C,MAAM,EAAE,KAAK;oBACb,GAAG,EAAE,GAAG,cAAc,yCAAyC;oBAC/D,OAAO,EAAE;wBACL,eAAe,EAAE,UAAU,WAAW,CAAC,KAAK,EAAE;wBAC9C,cAAc,EAAE,kBAAkB;qBACrC;oBACD,EAAE,EAAE;wBACA,eAAe,EAAE,cAAc;qBAClC;oBACD,IAAI,EAAE,IAAI;iBACb,CAAC,CAAC;gBAEH,IAAI,WAAW,IAAI,WAAW,CAAC,UAAU,IAAI,WAAW,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC;oBAChF,YAAY,GAAG,WAAW,CAAC,UAAU,CAAC,aAAa,CAAC;gBACxD,CAAC;qBAAM,CAAC;oBACJ,MAAM,WAAW,CAAC;gBACtB,CAAC;YACL,CAAC;YAED,IAAI,CAAC,YAAY,EAAE,CAAC;gBAChB,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;YAC7D,CAAC;YAED,MAAM,CAAC,IAAI,CAAC;gBACR,WAAW,EAAE,YAAY;gBACzB,YAAY,EAAE,YAAY;gBAC1B,IAAI,EAAE,cAAc;gBACpB,eAAe,EAAE,WAAW,CAAC,KAAe;aAC/C,CAAC,CAAC;QACP,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,MAAM,IAAI,iCAAkB,CACxB,IAAI,CAAC,OAAO,EAAE,EACd,sCAAsC,IAAA,wBAAgB,EAAC,YAAY,CAAC,EAAE,CACzE,CAAC;QACN,CAAC;IACL,CAAC;IAED,MAAM,UAAU,GAAyB,EAAE,CAAC;IAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;IAElC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,OAAO,CAAC,UAAU,CAAC,CAAC;IACxB,CAAC;IAGD,MAAM,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,4BAA4B,EAAE,CAAC,EAAE,CAAC,CAAW,CAAC;IAGzF,IAAI,SAAiB,CAAC;IACtB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,YAAY,GAAG,CAAC,EAAE,CAAC;QACrD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,YAAY,oBAAoB,CAAC,CAAC;QACzE,SAAS,GAAG,CAAC,CAAC;IAClB,CAAC;SAAM,IAAI,YAAY,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QACrC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,YAAY,wBAAwB,KAAK,CAAC,MAAM,WAAW,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QAC1G,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC;IAC7B,CAAC;SAAM,CAAC;QACJ,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACrC,IAAI,SAAS,KAAK,YAAY,EAAE,CAAC;YAC7B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0BAA0B,YAAY,gBAAgB,SAAS,EAAE,CAAC,CAAC;QACxF,CAAC;IACL,CAAC;IAGD,MAAM,sBAAsB,GAAG,IAAI,CAAC,gBAAgB,CAChD,sCAAsC,EACtC,CAAC,EACD,CAAC,CACM,CAAC;IAEZ,IAAI,mBAA2B,CAAC;IAChC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,CAAC,IAAI,sBAAsB,GAAG,CAAC,EAAE,CAAC;QACzE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,sBAAsB,WAAW,CAAC,CAAC;QACrE,mBAAmB,GAAG,CAAC,CAAC;IAC5B,CAAC;SAAM,IAAI,sBAAsB,GAAG,4BAAgB,CAAC,4BAA4B,EAAE,CAAC;QAChF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,sBAAsB,sBAAsB,4BAAgB,CAAC,4BAA4B,wBAAwB,CAAC,CAAC;QAC7I,mBAAmB,GAAG,4BAAgB,CAAC,4BAA4B,CAAC;IACxE,CAAC;SAAM,CAAC;QACJ,mBAAmB,GAAG,IAAI,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;QACzD,IAAI,mBAAmB,KAAK,sBAAsB,EAAE,CAAC;YACjD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,sBAAsB,gBAAgB,mBAAmB,EAAE,CAAC,CAAC;QACvG,CAAC;IACL,CAAC;IACD,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,eAAe,EAAE,CAAC,EAAE,KAAK,CAAY,CAAC;IAClF,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAiB,EAAC,IAAI,CAAC,CAAC;IAC7C,MAAM,KAAK,GAAG,MAAM,IAAA,qBAAY,EAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAC1C,IAAA,qBAAM,EAAC,KAAK,EAAE,0BAAc,CAAC,mBAAmB,CAAC,CAAC;IAClD,MAAM,aAAa,GAAG,aAAa,CAAC,CAAC,CAAC,MAAM,IAAA,qBAAY,EAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAEzE,IAAI,aAAa,IAAI,CAAC,aAAa,EAAE,CAAC;QAClC,MAAM,IAAI,iCAAkB,CACxB,IAAI,CAAC,OAAO,EAAE,EACd,0BAAc,CAAC,4BAA4B,CAC9C,CAAC;IACN,CAAC;IAGD,MAAM,eAAe,GAAG,IAAI,CAAC,gBAAgB,CAAC,yBAAyB,EAAE,CAAC,EAAE,IAAI,CAAY,CAAC;IAG7F,MAAM,cAAc,GAAsB,EAAE,CAAC;IAE7C,IAAI,CAAC;QACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;YAEnD,MAAM,YAAY,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAC;YACrD,IAAI,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,OAAO,EAAE,CAAC;gBACxB,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;gBACnD,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;gBACzD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gCAAgC,YAAY,EAAE,CAAC,CAAC;gBACjE,MAAM,IAAI,iCAAkB,CACxB,IAAI,CAAC,OAAO,EAAE,EACd,0BAAc,CAAC,mBAAmB,CAAC,YAAY,EAAE,YAAY,CAAC,CACjE,CAAC;YACN,CAAC;YAED,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;YAC5C,MAAM,aAAa,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;;gBAC5D,MAAM,SAAS,GAAG,CAAC,GAAG,cAAc,CAAC;gBAErC,MAAM,KAAK,GAAG,IAAA,8BAAoB,EAAC;oBAC/B,GAAG,EAAE,IAAI;oBACT,CAAC,EAAE,SAAS;oBACZ,QAAQ,EAAE,MAAM;oBAChB,aAAa,EAAE,YAAY;iBAC9B,CAAC,CAAC;gBACH,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACtB,MAAM,IAAI,iCAAkB,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,gCAAgC,CAAC,CAAC;gBACnF,CAAC;gBACD,MAAM,YAAY,GAAG,MAAM,IAAA,yCAAuB,EAAC,IAAI,EAAE,SAAS,CAAC,CAAC;gBACpE,MAAM,KAAK,GAAG,MAAM,IAAA,iBAAQ,EAAC,IAAI,EAAE,YAAY,CAAC,CAAC;gBACjD,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,SAAS,EAAE,EAAE,CAK7D,CAAC;gBAEF,MAAM,aAAa,GAAG,YAAY,CAAC,CAAC,CAAC,IAAI,iCAAe,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBAGzE,IAAI,aAAa,EAAE,CAAC;oBAChB,cAAc,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBACvC,CAAC;gBAGD,MAAM,QAAQ,GAAG,MAAM,IAAA,wBAAe,EAAC,IAAI,EAAE,SAAS,EAAE;oBACpD,aAAa,EAAE,OAAO,CAAC,aAAa;oBACpC,uBAAuB,EAAE,MAAA,OAAO,CAAC,uBAAuB,mCAAI,IAAI;oBAChE,YAAY;iBACf,CAAC,CAAC;gBACH,MAAM,MAAM,GAAuB,IAAA,sBAAa,EAAC,QAAQ,CAAC,CAAC;gBAG3D,MAAM,QAAQ,GAAG,mBAAmB,CAChC,KAAK,EACL,KAAK,EACL,MAAM,EACN,OAAO,EACP,YAAY,EACZ,MAAM,EACN,aAAa,EACb,aAAa,CAChB,CAAC;gBAEF,MAAM,YAAY,GAAQ;oBACtB,KAAK;oBACL,cAAc,EAAE,MAAA,OAAO,CAAC,aAAa,mCAAI,uBAAc;iBAC1D,CAAC;gBAGF,IAAI,YAAY,EAAE,CAAC;oBACf,YAAY,CAAC,uBAAuB;wBAChC,yXAAyX,CAAC;gBAClY,CAAC;gBACD,MAAM,cAAc,GAAG;oBACnB,MAAM,EAAE,IAAI,CAAC,wBAAwB,EAAE;oBACvC,SAAS,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,EAAE;iBAClD,CAAC;gBAIF,MAAM,oBAAoB,GAAG,aAAa,IAAI,IAAI,CAAC,CAAC,CAAC,MAAA,IAAI,CAAC,WAAW,oDAAI,CAAC,CAAC,CAAC,SAAS,CAAC;gBAEtF,IACI,aAAa,IAAI,IAAI;oBACrB,eAAe;oBACf,oBAAoB;oBACpB,IAAI,CAAC,OAAO,EAAE,CAAC,WAAW,IAAI,GAAG,EACnC,CAAC;oBACC,MAAM,WAAW,GAAG,MAAM,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,WAAW,CAAC,WAAW,EAAE,CAAA,CAAC;oBAC5D,MAAM,WAAW,GAAG,QAAQ,CAAC,YAAY,CACrC;wBACI,GAAG,YAAY;wBACf,YAAY,EAAE,WAAW,aAAX,WAAW,cAAX,WAAW,GAAI,SAAS;qBACzC,EACD;wBACI,OAAO,EAAE,IAAI;wBACb,GAAG,cAAc;qBACpB,CACJ,CAAC;oBAEF,OAAO,MAAM,kBAAkB,CAEvB,IAAyB,EACzB,WAAW,EACX,SAAS,EACT,OAAO,CAAC,uBAAuB,CACtC,CAAC;gBAEN,CAAC;qBAAM,CAAC;oBACJ,OAAO,MAAM,QAAQ,CAAC,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC;gBAC/D,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;YAG7D,MAAM,YAAY,GAAG,MAAM,IAAA,yCAAuB,EAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YAC5D,YAAY,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;;gBACnC,MAAM,SAAS,GAAG,CAAC,GAAG,KAAK,CAAC;gBAC5B,IAAI,MAAM,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;oBAC/B,MAAM,KAAK,GAAG,MAAM,CAAC,MAAe,CAAC;oBAErC,MAAM,aAAa,GAAG,IAAA,wBAAgB,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBACtD,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,IAAA,wBAAgB,EAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;oBAE5E,IAAI,IAAI,CAAC,cAAc,EAAE,EAAE,CAAC;wBACxB,UAAU,CAAC,IAAI,CAAC;4BACZ,IAAI,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE,WAAW,EAAE;4BAClD,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;yBAClC,CAAC,CAAC;wBACH,OAAO;oBACX,CAAC;yBAAM,CAAC;wBAEJ,MAAM,aAAa,GAAG,IAAI,KAAK,CAAC,2BAA2B,aAAa,wBAAwB,WAAW,IAAI,KAAK,EAAE,CAAC,CAAC;wBACxH,MAAM,IAAI,iCAAkB,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,CAAC,CAAC;oBAChE,CAAC;gBACL,CAAC;gBACD,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC;gBAK9B,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;oBACjC,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,MAAK,MAAM,CAAC,CAAC;oBAC/E,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACvB,QAAQ,CAAC,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACzE,CAAC;gBACL,CAAC;gBAGD,IAAI,MAAM,IAAI,YAAY,EAAE,CAAC;oBACzB,MAAM,YAAY,GAAG,IAAA,wBAAS,EAC1B,QAAQ,CAAC,MAAgB,CAC5B,CAAC;oBACF,QAAQ,CAAC,MAAM,GAAG,MAAA,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,MAAM,mCAAI,YAAY,CAAC;gBAC3D,CAAC;gBAGD,MAAM,UAAU,GAAG;oBACf,IAAI,EAAE,IAAA,cAAI,EACN,QAAQ,EACR,gBAAgB,EAChB,yBAAyB,EACzB,OAAO,EACP,cAAc,EACd,kBAAkB,CACrB;oBACD,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;iBAClC,CAAC;gBAEF,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAChC,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,GAAG,SAAS,GAAG,KAAK,CAAC,MAAM,IAAI,mBAAmB,GAAG,CAAC,EAAE,CAAC;gBAC1D,MAAM,IAAA,oBAAK,EAAC,mBAAmB,CAAC,CAAC;YACrC,CAAC;QACD,CAAC;QAED,OAAO,CAAC,UAAU,CAAC,CAAC;IACxB,CAAC;YAAS,CAAC;QAEP,MAAM,OAAO,CAAC,UAAU,CACpB,cAAc,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;YACjC,IAAI,CAAC;gBAED,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,4BAAgB,CAAC,qBAAqB,CAAC,CAAC,CAAC;gBAC1F,OAAO,CAAC,OAAO,EAAE,CAAC;YACtB,CAAC;YAAC,OAAO,YAAqB,EAAE,CAAC;gBAC7B,MAAM,QAAQ,GAAG,YAAY,YAAY,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;gBAE7F,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qCAAqC,IAAA,wBAAgB,EAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YACzF,CAAC;QACL,CAAC,CAAC,CACL,CAAC;IACN,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1,82 @@
|
|
|
1
|
+
export declare const MLFLOW_CONSTANTS: {
|
|
2
|
+
readonly API_REQUEST_TIMEOUT_MS: 10000;
|
|
3
|
+
readonly MLFLOW_INIT_TIMEOUT_MS: 30000;
|
|
4
|
+
readonly SPAN_CLEANUP_DELAY_MS: 100;
|
|
5
|
+
readonly MAX_BATCH_SIZE: 1000;
|
|
6
|
+
readonly MAX_DELAY_BETWEEN_BATCHES_MS: 300000;
|
|
7
|
+
readonly MAX_INPUT_LENGTH: 100000;
|
|
8
|
+
readonly MAX_OUTPUT_SIZE: 1000000;
|
|
9
|
+
readonly MAX_TOOL_NAME_LENGTH: 64;
|
|
10
|
+
readonly MAX_TOOL_DESCRIPTION_LENGTH: 1000;
|
|
11
|
+
readonly MIN_TOOL_DESCRIPTION_LENGTH: 10;
|
|
12
|
+
readonly MIN_DATABRICKS_TOKEN_LENGTH: 20;
|
|
13
|
+
readonly EXPERIMENT_ID_PATTERN: RegExp;
|
|
14
|
+
readonly TOOL_NAME_PATTERN: RegExp;
|
|
15
|
+
readonly MAX_RUN_MAP_SIZE: 1000;
|
|
16
|
+
readonly RUN_MAP_CHECK_THRESHOLD: 500;
|
|
17
|
+
readonly RUN_MAP_CLEANUP_INTERVAL_MS: 60000;
|
|
18
|
+
readonly MAX_SPAN_AGE_MS: 300000;
|
|
19
|
+
readonly DEFAULT_MAX_RETRIES: 3;
|
|
20
|
+
readonly DEFAULT_RETRY_BASE_DELAY_MS: 1000;
|
|
21
|
+
readonly MAX_RETRY_DELAY_MS: 10000;
|
|
22
|
+
readonly MAX_ATTRIBUTE_DEPTH: 10;
|
|
23
|
+
readonly MAX_ARRAY_SIZE_IN_ATTRIBUTES: 100;
|
|
24
|
+
readonly MAX_OBJECT_KEYS_IN_ATTRIBUTES: 50;
|
|
25
|
+
};
|
|
26
|
+
export declare const DATABRICKS_PATTERNS: {
|
|
27
|
+
readonly AZURE: RegExp;
|
|
28
|
+
readonly AWS: RegExp;
|
|
29
|
+
readonly GCP: RegExp;
|
|
30
|
+
};
|
|
31
|
+
export declare const ERROR_MESSAGES: {
|
|
32
|
+
readonly MISSING_CREDENTIALS: "Databricks credentials are required when MLflow logging is enabled";
|
|
33
|
+
readonly EMPTY_HOST: "Databricks host is missing or empty in credentials. Please configure valid Databricks credentials.";
|
|
34
|
+
readonly EMPTY_TOKEN: "Databricks token is missing or empty in credentials. Please configure valid Databricks credentials.";
|
|
35
|
+
readonly INVALID_HOST_URL: (host: string, reason: string) => string;
|
|
36
|
+
readonly SHORT_TOKEN_WARNING: (length: number) => string;
|
|
37
|
+
readonly EMPTY_EXPERIMENT_NAME: "Experiment name cannot be empty. Please provide a valid experiment name.";
|
|
38
|
+
readonly INVALID_EXPERIMENT_ID: (id: string) => string;
|
|
39
|
+
readonly INVALID_EXPERIMENT_SELECTION: "Invalid experiment selection. Please select an experiment from the list or enter a custom name.";
|
|
40
|
+
readonly INVALID_EXPERIMENT_ID_FROM_LIST: (id: string) => string;
|
|
41
|
+
readonly EXPERIMENT_NOT_FOUND: (path: string) => string;
|
|
42
|
+
readonly EXPERIMENT_CREATE_FAILED: (path: string, reason: string) => string;
|
|
43
|
+
readonly EMPTY_INPUT: "The \"text\" parameter is empty.";
|
|
44
|
+
readonly WHITESPACE_INPUT: "The \"text\" parameter contains only whitespace. Please provide a valid prompt.";
|
|
45
|
+
readonly LONG_INPUT_WARNING: (length: number, max: number) => string;
|
|
46
|
+
readonly EXECUTION_CANCELLED: (batch: number, total: number) => string;
|
|
47
|
+
readonly HTTP_REQUEST_TIMEOUT: "HTTP request timeout. Please check if Databricks host is reachable.";
|
|
48
|
+
readonly MAX_RETRIES_EXCEEDED: (retries: number, lastError: string) => string;
|
|
49
|
+
readonly MLFLOW_INIT_FAILED: (reason: string) => string;
|
|
50
|
+
readonly MLFLOW_INIT_TIMEOUT: "MLflow client initialization timeout";
|
|
51
|
+
readonly MODEL_NOT_CONNECTED: "Please connect a model to the Chat Model input";
|
|
52
|
+
readonly FALLBACK_MODEL_NOT_CONNECTED: "Please connect a model to the Fallback Model input or disable the fallback option";
|
|
53
|
+
readonly HTTP_INSECURE_WARNING: (url: string) => string;
|
|
54
|
+
};
|
|
55
|
+
export declare const NON_RETRYABLE_STATUS_CODES: readonly [400, 401, 403, 404, 405, 422];
|
|
56
|
+
export declare const RETRYABLE_STATUS_CODES: readonly [429, 500, 502, 503, 504];
|
|
57
|
+
export declare const LOG_LEVELS: {
|
|
58
|
+
readonly BATCH_SIZE_ADJUSTED: "warn";
|
|
59
|
+
readonly DELAY_ADJUSTED: "warn";
|
|
60
|
+
readonly TOKEN_SHORT: "warn";
|
|
61
|
+
readonly HTTP_INSECURE: "warn";
|
|
62
|
+
readonly USER_NOT_FOUND: "warn";
|
|
63
|
+
readonly RETRY_ATTEMPT: "debug";
|
|
64
|
+
readonly CLEANUP: "debug";
|
|
65
|
+
readonly EXECUTION_START: "info";
|
|
66
|
+
readonly EXECUTION_COMPLETE: "info";
|
|
67
|
+
readonly MLFLOW_INITIALIZED: "info";
|
|
68
|
+
};
|
|
69
|
+
export declare const SECURITY_PATTERNS: {
|
|
70
|
+
readonly PATH_TRAVERSAL: RegExp;
|
|
71
|
+
readonly VALID_DATABRICKS_HOST: RegExp;
|
|
72
|
+
readonly PRIVATE_IP: RegExp;
|
|
73
|
+
readonly SAFE_EXPERIMENT_NAME: RegExp;
|
|
74
|
+
readonly MAX_EXPERIMENT_NAME_LENGTH: 200;
|
|
75
|
+
readonly MAX_HOST_LENGTH: 500;
|
|
76
|
+
};
|
|
77
|
+
export declare const TOKEN_PATTERNS: {
|
|
78
|
+
readonly DATABRICKS_TOKEN: RegExp;
|
|
79
|
+
readonly GENERIC_LONG_TOKEN: RegExp;
|
|
80
|
+
readonly BEARER_TOKEN: RegExp;
|
|
81
|
+
readonly MASK: "****";
|
|
82
|
+
};
|
|
@@ -0,0 +1,105 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.TOKEN_PATTERNS = exports.SECURITY_PATTERNS = exports.LOG_LEVELS = exports.RETRYABLE_STATUS_CODES = exports.NON_RETRYABLE_STATUS_CODES = exports.ERROR_MESSAGES = exports.DATABRICKS_PATTERNS = exports.MLFLOW_CONSTANTS = void 0;
|
|
4
|
+
exports.MLFLOW_CONSTANTS = {
|
|
5
|
+
API_REQUEST_TIMEOUT_MS: 10000,
|
|
6
|
+
MLFLOW_INIT_TIMEOUT_MS: 30000,
|
|
7
|
+
SPAN_CLEANUP_DELAY_MS: 100,
|
|
8
|
+
MAX_BATCH_SIZE: 1000,
|
|
9
|
+
MAX_DELAY_BETWEEN_BATCHES_MS: 300000,
|
|
10
|
+
MAX_INPUT_LENGTH: 100000,
|
|
11
|
+
MAX_OUTPUT_SIZE: 1000000,
|
|
12
|
+
MAX_TOOL_NAME_LENGTH: 64,
|
|
13
|
+
MAX_TOOL_DESCRIPTION_LENGTH: 1000,
|
|
14
|
+
MIN_TOOL_DESCRIPTION_LENGTH: 10,
|
|
15
|
+
MIN_DATABRICKS_TOKEN_LENGTH: 20,
|
|
16
|
+
EXPERIMENT_ID_PATTERN: /^\d+$/,
|
|
17
|
+
TOOL_NAME_PATTERN: /^[a-zA-Z0-9_-]+$/,
|
|
18
|
+
MAX_RUN_MAP_SIZE: 1000,
|
|
19
|
+
RUN_MAP_CHECK_THRESHOLD: 500,
|
|
20
|
+
RUN_MAP_CLEANUP_INTERVAL_MS: 60000,
|
|
21
|
+
MAX_SPAN_AGE_MS: 300000,
|
|
22
|
+
DEFAULT_MAX_RETRIES: 3,
|
|
23
|
+
DEFAULT_RETRY_BASE_DELAY_MS: 1000,
|
|
24
|
+
MAX_RETRY_DELAY_MS: 10000,
|
|
25
|
+
MAX_ATTRIBUTE_DEPTH: 10,
|
|
26
|
+
MAX_ARRAY_SIZE_IN_ATTRIBUTES: 100,
|
|
27
|
+
MAX_OBJECT_KEYS_IN_ATTRIBUTES: 50,
|
|
28
|
+
};
|
|
29
|
+
exports.DATABRICKS_PATTERNS = {
|
|
30
|
+
AZURE: /\.azuredatabricks\.net$/,
|
|
31
|
+
AWS: /\.cloud\.databricks\.com$/,
|
|
32
|
+
GCP: /\.gcp\.databricks\.com$/,
|
|
33
|
+
};
|
|
34
|
+
exports.ERROR_MESSAGES = {
|
|
35
|
+
MISSING_CREDENTIALS: 'Databricks credentials are required when MLflow logging is enabled',
|
|
36
|
+
EMPTY_HOST: 'Databricks host is missing or empty in credentials. Please configure valid Databricks credentials.',
|
|
37
|
+
EMPTY_TOKEN: 'Databricks token is missing or empty in credentials. Please configure valid Databricks credentials.',
|
|
38
|
+
INVALID_HOST_URL: (host, reason) => `Invalid Databricks host URL: "${host}". ${reason}. Expected format: https://adb-xxxxx.xx.azuredatabricks.net`,
|
|
39
|
+
SHORT_TOKEN_WARNING: (length) => `Databricks token seems unusually short (${length} characters). ` +
|
|
40
|
+
`Databricks personal access tokens are typically 40+ characters. ` +
|
|
41
|
+
`Please verify your token is correct.`,
|
|
42
|
+
EMPTY_EXPERIMENT_NAME: 'Experiment name cannot be empty. Please provide a valid experiment name.',
|
|
43
|
+
INVALID_EXPERIMENT_ID: (id) => `Invalid experiment ID format: "${id}". Expected numeric ID (e.g., 1427538817675103)`,
|
|
44
|
+
INVALID_EXPERIMENT_SELECTION: 'Invalid experiment selection. Please select an experiment from the list or enter a custom name.',
|
|
45
|
+
INVALID_EXPERIMENT_ID_FROM_LIST: (id) => `Invalid experiment ID from list: "${id}". Expected numeric ID. ` +
|
|
46
|
+
`This may indicate a problem with the experiment list. Try refreshing or using a custom name.`,
|
|
47
|
+
EXPERIMENT_NOT_FOUND: (path) => `Experiment "${path}" not found and "Create If Not Exists" is disabled`,
|
|
48
|
+
EXPERIMENT_CREATE_FAILED: (path, reason) => `Failed to create experiment "${path}": ${reason}`,
|
|
49
|
+
EMPTY_INPUT: 'The "text" parameter is empty.',
|
|
50
|
+
WHITESPACE_INPUT: 'The "text" parameter contains only whitespace. Please provide a valid prompt.',
|
|
51
|
+
LONG_INPUT_WARNING: (length, max) => `Input is very long (${length} characters). ` +
|
|
52
|
+
`This may cause performance issues or token limit errors. Maximum recommended: ${max} characters.`,
|
|
53
|
+
EXECUTION_CANCELLED: (batch, total) => `Execution was cancelled by user at batch ${batch} of ${total}`,
|
|
54
|
+
HTTP_REQUEST_TIMEOUT: 'HTTP request timeout. Please check if Databricks host is reachable.',
|
|
55
|
+
MAX_RETRIES_EXCEEDED: (retries, lastError) => `HTTP request failed after ${retries} attempts. Last error: ${lastError}`,
|
|
56
|
+
MLFLOW_INIT_FAILED: (reason) => `Failed to initialize MLflow: ${reason}. Please check your configuration.`,
|
|
57
|
+
MLFLOW_INIT_TIMEOUT: 'MLflow client initialization timeout',
|
|
58
|
+
MODEL_NOT_CONNECTED: 'Please connect a model to the Chat Model input',
|
|
59
|
+
FALLBACK_MODEL_NOT_CONNECTED: 'Please connect a model to the Fallback Model input or disable the fallback option',
|
|
60
|
+
HTTP_INSECURE_WARNING: (url) => `Databricks host is using HTTP instead of HTTPS: ${url}. ` +
|
|
61
|
+
`This is insecure and may not work with Databricks. ` +
|
|
62
|
+
`Consider using HTTPS instead.`,
|
|
63
|
+
};
|
|
64
|
+
exports.NON_RETRYABLE_STATUS_CODES = [
|
|
65
|
+
400,
|
|
66
|
+
401,
|
|
67
|
+
403,
|
|
68
|
+
404,
|
|
69
|
+
405,
|
|
70
|
+
422,
|
|
71
|
+
];
|
|
72
|
+
exports.RETRYABLE_STATUS_CODES = [
|
|
73
|
+
429,
|
|
74
|
+
500,
|
|
75
|
+
502,
|
|
76
|
+
503,
|
|
77
|
+
504,
|
|
78
|
+
];
|
|
79
|
+
exports.LOG_LEVELS = {
|
|
80
|
+
BATCH_SIZE_ADJUSTED: 'warn',
|
|
81
|
+
DELAY_ADJUSTED: 'warn',
|
|
82
|
+
TOKEN_SHORT: 'warn',
|
|
83
|
+
HTTP_INSECURE: 'warn',
|
|
84
|
+
USER_NOT_FOUND: 'warn',
|
|
85
|
+
RETRY_ATTEMPT: 'debug',
|
|
86
|
+
CLEANUP: 'debug',
|
|
87
|
+
EXECUTION_START: 'info',
|
|
88
|
+
EXECUTION_COMPLETE: 'info',
|
|
89
|
+
MLFLOW_INITIALIZED: 'info',
|
|
90
|
+
};
|
|
91
|
+
exports.SECURITY_PATTERNS = {
|
|
92
|
+
PATH_TRAVERSAL: /\.\.|\/\.\.|\.\.\/|%2e%2e|\.\.%2f|%2e%2e%2f/i,
|
|
93
|
+
VALID_DATABRICKS_HOST: /^https:\/\/([a-zA-Z0-9.-]+\.(azuredatabricks\.net|cloud\.databricks\.com|gcp\.databricks\.com)|localhost(:[0-9]+)?)$/,
|
|
94
|
+
PRIVATE_IP: /^(127\.|10\.|172\.(1[6-9]|2[0-9]|3[01])\.|192\.168\.|169\.254\.|::1|fc00:|fe80:)/,
|
|
95
|
+
SAFE_EXPERIMENT_NAME: /^[a-zA-Z0-9\s\-_\/]+$/,
|
|
96
|
+
MAX_EXPERIMENT_NAME_LENGTH: 200,
|
|
97
|
+
MAX_HOST_LENGTH: 500,
|
|
98
|
+
};
|
|
99
|
+
exports.TOKEN_PATTERNS = {
|
|
100
|
+
DATABRICKS_TOKEN: /dapi[a-zA-Z0-9]{20,}/g,
|
|
101
|
+
GENERIC_LONG_TOKEN: /\b[a-zA-Z0-9]{40,}\b/g,
|
|
102
|
+
BEARER_TOKEN: /Bearer\s+[a-zA-Z0-9._-]+/gi,
|
|
103
|
+
MASK: '****',
|
|
104
|
+
};
|
|
105
|
+
//# sourceMappingURL=constants.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../../../nodes/agents/DatabricksAiAgent/src/constants.ts"],"names":[],"mappings":";;;AAUa,QAAA,gBAAgB,GAAG;IAE5B,sBAAsB,EAAE,KAAK;IAC7B,sBAAsB,EAAE,KAAK;IAC7B,qBAAqB,EAAE,GAAG;IAG1B,cAAc,EAAE,IAAI;IACpB,4BAA4B,EAAE,MAAM;IACpC,gBAAgB,EAAE,MAAM;IACxB,eAAe,EAAE,OAAO;IACxB,oBAAoB,EAAE,EAAE;IACxB,2BAA2B,EAAE,IAAI;IACjC,2BAA2B,EAAE,EAAE;IAC/B,2BAA2B,EAAE,EAAE;IAG/B,qBAAqB,EAAE,OAAO;IAC9B,iBAAiB,EAAE,kBAAkB;IAGrC,gBAAgB,EAAE,IAAI;IACtB,uBAAuB,EAAE,GAAG;IAC5B,2BAA2B,EAAE,KAAK;IAClC,eAAe,EAAE,MAAM;IAGvB,mBAAmB,EAAE,CAAC;IACtB,2BAA2B,EAAE,IAAI;IACjC,kBAAkB,EAAE,KAAK;IAGzB,mBAAmB,EAAE,EAAE;IACvB,4BAA4B,EAAE,GAAG;IACjC,6BAA6B,EAAE,EAAE;CAC3B,CAAC;AAKE,QAAA,mBAAmB,GAAG;IAC/B,KAAK,EAAE,yBAAyB;IAChC,GAAG,EAAE,2BAA2B;IAChC,GAAG,EAAE,yBAAyB;CACxB,CAAC;AAKE,QAAA,cAAc,GAAG;IAE1B,mBAAmB,EAAE,oEAAoE;IACzF,UAAU,EAAE,oGAAoG;IAChH,WAAW,EAAE,qGAAqG;IAClH,gBAAgB,EAAE,CAAC,IAAY,EAAE,MAAc,EAAE,EAAE,CAC/C,iCAAiC,IAAI,MAAM,MAAM,6DAA6D;IAClH,mBAAmB,EAAE,CAAC,MAAc,EAAE,EAAE,CACpC,2CAA2C,MAAM,gBAAgB;QACjE,kEAAkE;QAClE,sCAAsC;IAG1C,qBAAqB,EAAE,0EAA0E;IACjG,qBAAqB,EAAE,CAAC,EAAU,EAAE,EAAE,CAClC,kCAAkC,EAAE,iDAAiD;IACzF,4BAA4B,EAAE,iGAAiG;IAC/H,+BAA+B,EAAE,CAAC,EAAU,EAAE,EAAE,CAC5C,qCAAqC,EAAE,0BAA0B;QACjE,8FAA8F;IAClG,oBAAoB,EAAE,CAAC,IAAY,EAAE,EAAE,CACnC,eAAe,IAAI,oDAAoD;IAC3E,wBAAwB,EAAE,CAAC,IAAY,EAAE,MAAc,EAAE,EAAE,CACvD,gCAAgC,IAAI,MAAM,MAAM,EAAE;IAGtD,WAAW,EAAE,gCAAgC;IAC7C,gBAAgB,EAAE,+EAA+E;IACjG,kBAAkB,EAAE,CAAC,MAAc,EAAE,GAAW,EAAE,EAAE,CAChD,uBAAuB,MAAM,gBAAgB;QAC7C,iFAAiF,GAAG,cAAc;IAGtG,mBAAmB,EAAE,CAAC,KAAa,EAAE,KAAa,EAAE,EAAE,CAClD,4CAA4C,KAAK,OAAO,KAAK,EAAE;IAGnE,oBAAoB,EAAE,qEAAqE;IAC3F,oBAAoB,EAAE,CAAC,OAAe,EAAE,SAAiB,EAAE,EAAE,CACzD,6BAA6B,OAAO,0BAA0B,SAAS,EAAE;IAG7E,kBAAkB,EAAE,CAAC,MAAc,EAAE,EAAE,CACnC,gCAAgC,MAAM,oCAAoC;IAC9E,mBAAmB,EAAE,sCAAsC;IAG3D,mBAAmB,EAAE,gDAAgD;IACrE,4BAA4B,EAAE,mFAAmF;IAGjH,qBAAqB,EAAE,CAAC,GAAW,EAAE,EAAE,CACnC,mDAAmD,GAAG,IAAI;QAC1D,qDAAqD;QACrD,+BAA+B;CAC7B,CAAC;AAKE,QAAA,0BAA0B,GAAG;IACtC,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;CACG,CAAC;AAKE,QAAA,sBAAsB,GAAG;IAClC,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;IACH,GAAG;CACG,CAAC;AAKE,QAAA,UAAU,GAAG;IACtB,mBAAmB,EAAE,MAAM;IAC3B,cAAc,EAAE,MAAM;IACtB,WAAW,EAAE,MAAM;IACnB,aAAa,EAAE,MAAM;IACrB,cAAc,EAAE,MAAM;IACtB,aAAa,EAAE,OAAO;IACtB,OAAO,EAAE,OAAO;IAChB,eAAe,EAAE,MAAM;IACvB,kBAAkB,EAAE,MAAM;IAC1B,kBAAkB,EAAE,MAAM;CACpB,CAAC;AAKE,QAAA,iBAAiB,GAAG;IAE7B,cAAc,EAAE,8CAA8C;IAG9D,qBAAqB,EAAE,sHAAsH;IAG7I,UAAU,EAAE,kFAAkF;IAG9F,oBAAoB,EAAE,uBAAuB;IAG7C,0BAA0B,EAAE,GAAG;IAC/B,eAAe,EAAE,GAAG;CACd,CAAC;AAKE,QAAA,cAAc,GAAG;IAE1B,gBAAgB,EAAE,uBAAuB;IAGzC,kBAAkB,EAAE,uBAAuB;IAG3C,YAAY,EAAE,4BAA4B;IAG1C,IAAI,EAAE,MAAM;CACN,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../../nodes/agents/DatabricksAiAgent/src/types/types.ts"],"names":[],"mappings":""}
|