@utaba/deep-memory-storage-cosmosdb 0.12.1 → 0.14.2

package/dist/index.d.cts

@@ -147,7 +147,7 @@ declare class CosmosDbProvider implements StorageProvider, GraphTraversalProvide
      * For agent-facing graph queries use {@link traverse}, which enforces the
      * repositoryId partition predicate.
      */
-    executeNativeQuery(_repositoryId: string, query: string, params?: Record<string, unknown>): Promise<TraversalResult>;
+    executeNativeQuery(_repositoryId: string, query: string, params?: Record<string, unknown>): Promise<unknown[]>;
     private executeNativeQueryImpl;
 }
 

package/dist/index.d.ts

@@ -147,7 +147,7 @@ declare class CosmosDbProvider implements StorageProvider, GraphTraversalProvide
      * For agent-facing graph queries use {@link traverse}, which enforces the
      * repositoryId partition predicate.
      */
-    executeNativeQuery(_repositoryId: string, query: string, params?: Record<string, unknown>): Promise<TraversalResult>;
+    executeNativeQuery(_repositoryId: string, query: string, params?: Record<string, unknown>): Promise<unknown[]>;
     private executeNativeQueryImpl;
 }
 

package/dist/index.js

@@ -1095,6 +1095,7 @@ var DEFAULT_START = 5;
 var DEFAULT_MAX = 32;
 var DEFAULT_INCREASE_AFTER = 50;
 var DEFAULT_COOLDOWN_MS = 1e3;
+var DEFAULT_RAMP_UP_COOLDOWN_MS = 5e3;
 var DEFAULT_MAX_CONSECUTIVE_THROTTLES_AT_MIN = 10;
 var DEFAULT_THROTTLE_CEILING_MULTIPLIER = 3;
 var NO_SOFT_CEILING = Number.POSITIVE_INFINITY;
@@ -1104,6 +1105,7 @@ function resolveOptions(opts) {
   const start = Math.min(max, Math.max(min, opts?.start ?? DEFAULT_START));
   const increaseAfter = Math.max(1, opts?.increaseAfter ?? DEFAULT_INCREASE_AFTER);
   const cooldownMs = Math.max(0, opts?.cooldownMs ?? DEFAULT_COOLDOWN_MS);
+  const rampUpCooldownMs = Math.max(0, opts?.rampUpCooldownMs ?? DEFAULT_RAMP_UP_COOLDOWN_MS);
   const maxConsecutiveThrottlesAtMin = Math.max(
     1,
     opts?.maxConsecutiveThrottlesAtMin ?? DEFAULT_MAX_CONSECUTIVE_THROTTLES_AT_MIN
@@ -1118,6 +1120,7 @@ function resolveOptions(opts) {
     max,
     increaseAfter,
     cooldownMs,
+    rampUpCooldownMs,
     maxConsecutiveThrottlesAtMin,
     throttleCeilingMultiplier,
     onAdjust: opts?.onAdjust
@@ -1138,6 +1141,7 @@ var AdaptiveConcurrencyController = class {
   current;
   streak = 0;
   cooldownUntil = 0;
+  rampFrozenUntil = 0;
   completed = 0;
   throttled = 0;
   startEmitted = false;
@@ -1188,10 +1192,11 @@ var AdaptiveConcurrencyController = class {
     this.notify("start", this.current);
   }
   /** Record a throttle-free task completion. May trigger ramp-up. */
-  noteSuccess() {
+  noteSuccess(now = Date.now()) {
     this.completed++;
-    this.streak++;
     this.consecutiveThrottlesAtMin = 0;
+    if (now < this.rampFrozenUntil) return;
+    this.streak++;
     if (this.current >= this.opts.max) return;
     const target = this.current + 1;
     const needed = target >= this.softCeiling ? this.opts.increaseAfter * this.opts.throttleCeilingMultiplier : this.opts.increaseAfter;
@@ -1213,6 +1218,7 @@ var AdaptiveConcurrencyController = class {
     const previous = this.current;
     const next = Math.max(this.opts.min, Math.floor(this.current / 2));
     this.cooldownUntil = now + this.opts.cooldownMs;
+    this.rampFrozenUntil = now + this.opts.rampUpCooldownMs;
     if (next !== previous) {
       this.current = next;
       this.softCeiling = previous;
@@ -2178,41 +2184,8 @@ var CosmosDbProvider = class {
     return this.track("executeNativeQuery", void 0, () => this.executeNativeQueryImpl(query, params));
   }
   async executeNativeQueryImpl(query, params) {
-    const startTime = Date.now();
-    try {
-      const result = await this.conn.submit(query, params);
-      const executionTimeMs = Date.now() - startTime;
-      const entities = [];
-      for (const item of result.items) {
-        try {
-          const entity = entityFromGremlin(item);
-          if (entity.id) {
-            entities.push(entity);
-          }
-        } catch {
-        }
-      }
-      const queryMetadata = {
-        executionTimeMs,
-        resourceCost: result.requestCharge != null ? { units: "RU", value: result.requestCharge } : void 0,
-        compiledQuery: query,
-        compiledQueryLanguage: "gremlin",
-        appliedLimits: { maxResults: entities.length },
-        truncated: false
-      };
-      return {
-        entities,
-        total: entities.length,
-        returned: entities.length,
-        hasMore: false,
-        queryMetadata
-      };
-    } catch (err) {
-      throw new ProviderError(
-        `Native Gremlin query failed: ${err instanceof Error ? err.message : String(err)}`,
-        "Verify the Gremlin query syntax is valid for CosmosDB."
-      );
-    }
+    const result = await this.conn.submit(query, params);
+    return result.items;
   }
 };
 function unwrapGremlinValue(val) {