@ductape/sdk 0.3.7 → 0.3.8

package/dist/index.js

@@ -50,9 +50,9 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.validateDatabaseActionOutput = exports.deriveDatabaseActionOutputContract = exports.createMemoryAdapter = exports.createWeaviateAdapter = exports.createQdrantAdapter = exports.createPineconeAdapter = exports.createVectorStoreAdapter = exports.VectorServiceAdapter = exports.OpenAIProvider = exports.AnthropicProvider = exports.createLLMProvider = exports.MemoryManager = exports.ToolRegistry = exports.AgentContext = exports.AgentExecutor = exports.agentsService = exports.AgentError = exports.AgentsService = exports.verifyInvocationSignature = exports.PORTABLE_FUNCTION_HTTP_PATH = exports.handlePortableFunctionHttpRequest = exports.PortableFunctionError = exports.PortableFunctionsRegistry = exports.portableFunctions = exports.defineFunctions = exports.defineFunction = exports.FeatureExecutor = exports.createFeature = exports.FeatureBuilder = exports.featureService = exports.FeatureCompilationError = exports.FeatureError = exports.FeatureService = exports.GraphAdapterFactory = exports.MemgraphAdapter = exports.ArangoDBAdapter = exports.NeptuneAdapter = exports.Neo4jAdapter = exports.BaseGraphAdapter = exports.GraphError = exports.GraphSortOrder = exports.GraphFeature = exports.GraphErrorType = exports.GraphTransactionStatus = exports.GraphIsolationLevel = exports.NodeConstraintType = exports.NodeIndexType = exports.TraversalDirection = exports.GraphType = exports.GraphService = void 0;
-exports.SecretResolutionError = exports.SecretRevokedError = exports.SecretExpiredError = exports.SecretEnvironmentError = exports.SecretScopeError = exports.SecretNotFoundError = exports.SECRET_PATTERN_GLOBAL = exports.SECRET_PATTERN = exports.SecretsService = exports.VectorFeature = exports.VectorErrorType = exports.VectorIndexType = exports.DistanceMetric = exports.VectorDBType = exports.MemoryAdapter = exports.WeaviateAdapter = exports.QdrantAdapter = exports.PineconeAdapter = exports.BaseVectorAdapter = exports.VectorError = exports.VectorDatabaseService = exports.SmsProvider = exports.PushNotificationProvider = exports.NotificationChannelType = exports.notificationsService = exports.NotificationsService = exports.getSqsQueueUrl = exports.isValidSqsUrl = exports.parseEventString = exports.loadBrokerService = exports.GooglePubSubService = exports.RedisService = exports.SQSService = exports.KafkaService = exports.RabbitMQService = exports.BrokerError = exports.BrokersService = exports.CacheLogger = exports.cacheLogger = exports.CacheError = exports.CacheService = exports.SessionError = exports.SessionsService = exports.generateSignedUrl = exports.listBlobsInCloud = exports.deleteBlobFromCloud = exports.downloadBlobFromCloud = exports.uploadBlobToCloud = exports.StorageError = exports.StorageService = void 0;
-exports.getAssetSchemas = exports.buildVpcConnectorRelayWebSocketUrl = exports.parseTcpEndpointFromUrl = exports.buildVpcConnectorTunnelAdapterOptions = exports.openVpcConnectorTunnelStream = exports.resolveCloudLinkedEnvConfig = exports.resolveCloudLinkedConfig = exports.CloudError = exports.CloudService = exports.mightContainSecrets = exports.findAllSecretReferences = exports.extractSecretKeys = exports.containsSecretReferences = exports.isSecretReference = void 0;
+exports.deriveDatabaseActionOutputContract = exports.createMemoryAdapter = exports.createWeaviateAdapter = exports.createQdrantAdapter = exports.createPineconeAdapter = exports.createVectorStoreAdapter = exports.VectorServiceAdapter = exports.OpenAIProvider = exports.AnthropicProvider = exports.createLLMProvider = exports.MemoryManager = exports.ToolRegistry = exports.AgentContext = exports.AgentExecutor = exports.agentsService = exports.AgentError = exports.AgentsService = exports.verifyInvocationSignature = exports.PORTABLE_FUNCTION_HTTP_PATH = exports.handlePortableFunctionHttpRequest = exports.PortableFunctionError = exports.PortableFunctionsRegistry = exports.portableFunctions = exports.defineFunctions = exports.defineFunction = exports.FeatureExecutor = exports.createFeature = exports.FeatureBuilder = exports.FeaturePublicationError = exports.featureService = exports.FeatureCompilationError = exports.FeatureError = exports.FeatureService = exports.GraphAdapterFactory = exports.MemgraphAdapter = exports.ArangoDBAdapter = exports.NeptuneAdapter = exports.Neo4jAdapter = exports.BaseGraphAdapter = exports.GraphError = exports.GraphSortOrder = exports.GraphFeature = exports.GraphErrorType = exports.GraphTransactionStatus = exports.GraphIsolationLevel = exports.NodeConstraintType = exports.NodeIndexType = exports.TraversalDirection = exports.GraphType = exports.GraphService = void 0;
+exports.SecretRevokedError = exports.SecretExpiredError = exports.SecretEnvironmentError = exports.SecretScopeError = exports.SecretNotFoundError = exports.SECRET_PATTERN_GLOBAL = exports.SECRET_PATTERN = exports.SecretsService = exports.VectorFeature = exports.VectorErrorType = exports.VectorIndexType = exports.DistanceMetric = exports.VectorDBType = exports.MemoryAdapter = exports.WeaviateAdapter = exports.QdrantAdapter = exports.PineconeAdapter = exports.BaseVectorAdapter = exports.VectorError = exports.VectorDatabaseService = exports.SmsProvider = exports.PushNotificationProvider = exports.NotificationChannelType = exports.notificationsService = exports.NotificationsService = exports.getSqsQueueUrl = exports.isValidSqsUrl = exports.parseEventString = exports.loadBrokerService = exports.GooglePubSubService = exports.RedisService = exports.SQSService = exports.KafkaService = exports.RabbitMQService = exports.BrokerError = exports.BrokersService = exports.CacheLogger = exports.cacheLogger = exports.CacheError = exports.CacheService = exports.SessionError = exports.SessionsService = exports.generateSignedUrl = exports.listBlobsInCloud = exports.deleteBlobFromCloud = exports.downloadBlobFromCloud = exports.uploadBlobToCloud = exports.StorageError = exports.StorageService = exports.validateDatabaseActionOutput = void 0;
+exports.getAssetSchemas = exports.buildVpcConnectorRelayWebSocketUrl = exports.parseTcpEndpointFromUrl = exports.buildVpcConnectorTunnelAdapterOptions = exports.openVpcConnectorTunnelStream = exports.resolveCloudLinkedEnvConfig = exports.resolveCloudLinkedConfig = exports.CloudError = exports.CloudService = exports.mightContainSecrets = exports.findAllSecretReferences = exports.extractSecretKeys = exports.containsSecretReferences = exports.isSecretReference = exports.SecretResolutionError = void 0;
 const appApi_service_1 = require("./api/services/appApi.service");
 const userApi_service_1 = require("./api/services/userApi.service");
 const workspaceApi_service_1 = require("./api/services/workspaceApi.service");
@@ -60,6 +60,7 @@ const app_service_1 = __importDefault(require("./apps/services/app.service"));
 const filePath = __importStar(require("path"));
 const products_service_1 = __importDefault(require("./products/services/products.service"));
 const fs_1 = require("fs");
+const crypto_1 = require("crypto");
 const types_1 = require("./types");
 const index_types_1 = require("./types/index.types");
 const runtime_defaults_1 = require("./runtime/runtime-defaults");
@@ -323,10 +324,12 @@ Object.defineProperty(exports, "FeatureError", { enumerable: true, get: function
 Object.defineProperty(exports, "FeatureCompilationError", { enumerable: true, get: function () { return features_2.FeatureCompilationError; } });
 Object.defineProperty(exports, "featureService", { enumerable: true, get: function () { return features_2.featureService; } });
 var features_3 = require("./features");
-Object.defineProperty(exports, "FeatureBuilder", { enumerable: true, get: function () { return features_3.FeatureBuilder; } });
-Object.defineProperty(exports, "createFeature", { enumerable: true, get: function () { return features_3.createFeature; } });
+Object.defineProperty(exports, "FeaturePublicationError", { enumerable: true, get: function () { return features_3.FeaturePublicationError; } });
 var features_4 = require("./features");
-Object.defineProperty(exports, "FeatureExecutor", { enumerable: true, get: function () { return features_4.FeatureExecutor; } });
+Object.defineProperty(exports, "FeatureBuilder", { enumerable: true, get: function () { return features_4.FeatureBuilder; } });
+Object.defineProperty(exports, "createFeature", { enumerable: true, get: function () { return features_4.createFeature; } });
+var features_5 = require("./features");
+Object.defineProperty(exports, "FeatureExecutor", { enumerable: true, get: function () { return features_5.FeatureExecutor; } });
 var functions_2 = require("./functions");
 Object.defineProperty(exports, "defineFunction", { enumerable: true, get: function () { return functions_2.defineFunction; } });
 Object.defineProperty(exports, "defineFunctions", { enumerable: true, get: function () { return functions_2.defineFunctions; } });
@@ -539,8 +542,27 @@ class Ductape {
         this.runtimeCredentialState = { status: 'idle', referenced: 0, cached: 0, missing: 0 };
         this.runtimeClosed = false;
         /** Register and inspect portable application functions used by Feature JSON. */
+        this.registeredFunctionContracts = new Map();
         this.functions = {
-            register: (contract) => functions_1.portableFunctions.register(contract),
+            register: (contract) => {
+                const client = functions_1.portableFunctions.register(contract);
+                this.registeredFunctionContracts.set(`${contract.namespace}@${contract.version}`, contract);
+                return client;
+            },
+            sync: async (options = {}) => {
+                var _a;
+                const product = (_a = options.product) !== null && _a !== void 0 ? _a : this.defaultProduct;
+                if (!product)
+                    throw new Error('Product is required to persist function metadata');
+                const builder = await this.getProductBuilder(product);
+                const results = [];
+                for (const contract of this.registeredFunctionContracts.values()) {
+                    if (options.namespaces && !options.namespaces.includes(contract.namespace))
+                        continue;
+                    results.push(await builder.syncPortableFunction(contract));
+                }
+                return results;
+            },
             use: (contract) => functions_1.portableFunctions.use(contract),
             list: () => functions_1.portableFunctions.list(),
             registerGrpcInvoker: (invoker) => functions_1.portableFunctions.registerGrpcInvoker(invoker),
@@ -548,15 +570,41 @@ class Ductape {
         };
         /** Atomic keyed request enforcement; distinct from weighted provider quotas. */
         this.rateLimit = {
+            list: () => { var _a, _b; return (_b = (_a = this._rateLimitService) === null || _a === void 0 ? void 0 : _a.list()) !== null && _b !== void 0 ? _b : []; },
+            sync: async (env, product) => {
+                var _a;
+                await this.initUserAuth(false);
+                (_a = this._rateLimitService) !== null && _a !== void 0 ? _a : (this._rateLimitService = new resilience_1.RateLimitService(this.redisClient));
+                this._rateLimitService.setRedis(this.redisClient);
+                this._rateLimitService.setContext({ workspace: this.workspace_id, product: product !== null && product !== void 0 ? product : this.defaultProduct });
+                return this._rateLimitService.sync(env, product);
+            },
             define: (definition) => {
                 var _a;
                 (_a = this._rateLimitService) !== null && _a !== void 0 ? _a : (this._rateLimitService = new resilience_1.RateLimitService(this.redisClient));
                 return this._rateLimitService.define(definition);
             },
             run: async (input) => {
-                var _a;
-                (_a = this._rateLimitService) !== null && _a !== void 0 ? _a : (this._rateLimitService = new resilience_1.RateLimitService(this.redisClient));
+                var _a, _b;
+                await this.initUserAuth(false);
+                let sessionFields = {};
+                if (input.session) {
+                    const builder = await this.getProductBuilder((_a = input.product) !== null && _a !== void 0 ? _a : this.defaultProduct);
+                    const { processSessionForExecution } = await Promise.resolve().then(() => __importStar(require('./sessions')));
+                    const resolved = await processSessionForExecution(input.session, builder.fetchPrivateKey(), input, input.env, {
+                        fetchSessionSelector: async (tag) => { var _a; return (_a = (await builder.fetchSession(tag))) === null || _a === void 0 ? void 0 : _a.selector; },
+                    });
+                    if (resolved.error)
+                        throw new Error('Invalid rate-limit session');
+                    sessionFields = resolved.logFields;
+                }
+                (_b = this._rateLimitService) !== null && _b !== void 0 ? _b : (this._rateLimitService = new resilience_1.RateLimitService(this.redisClient));
                 this._rateLimitService.setRedis(this.redisClient);
+                const logger = this.createNewLogger();
+                this._rateLimitService.setContext({ workspace: this.workspace_id, product: this.defaultProduct, observe: (event) => {
+                        logger.add(Object.assign({ process_id: (0, crypto_1.randomUUID)(), product_tag: event.product, env: event.env, parent_tag: event.tag, type: 'rate_limit', status: (event.outcome === 'allowed' ? 'success' : 'fail'), successful_execution: event.outcome === 'allowed', failed_execution: event.outcome !== 'allowed', name: 'Rate limit decision', action: event.outcome, data: event, latency: event.latency, parent_process_id: event.executionId, trace_id: event.correlationId }, sessionFields));
+                        void logger.publish();
+                    } });
                 return this._rateLimitService.run(input);
             },
         };