@nebula-ai/sdk 1.2.0 → 1.2.2

package/dist/index.mjs

@@ -339,7 +339,7 @@ var _Nebula = class _Nebula {
    */
   async storeMemory(memory, name) {
     let mem;
-    if ("collection_id" in memory) {
+    if ("collection_id" in memory || "snapshot" in memory) {
       mem = memory;
     } else {
       const memRecord2 = memory;
@@ -348,9 +348,23 @@ var _Nebula = class _Nebula {
         content: memRecord2.content || "",
         role: memRecord2.role,
         memory_id: memRecord2.memory_id || memRecord2.memoryId || void 0,
-        metadata: memRecord2.metadata || {}
+        metadata: memRecord2.metadata || {},
+        snapshot: memRecord2.snapshot
       };
     }
+    if (mem.snapshot) {
+      const contentText = await this._serializeContentAsText(mem.content);
+      const payload2 = {
+        snapshot: mem.snapshot,
+        raw_text: contentText
+      };
+      const response2 = await this._makeRequest("POST", "/v1/memories", payload2);
+      const snapshot = response2?.results?.snapshot;
+      if (snapshot) {
+        return snapshot;
+      }
+      return response2?.results ?? {};
+    }
     if (mem.memory_id) {
       return await this._appendToMemory(mem.memory_id, mem);
     }
@@ -819,6 +833,27 @@ var _Nebula = class _Nebula {
    * https://docs.trynebula.ai/guides/metadata-filtering
    */
   async search(options) {
+    if (options.snapshot) {
+      const snapshotData = {
+        snapshot: options.snapshot,
+        query: options.query
+      };
+      if (options.effort) {
+        snapshotData.effort = options.effort;
+      }
+      const response2 = await this._makeRequest("POST", "/v1/memories/search", snapshotData);
+      const memoryResponseData2 = response2.results;
+      return {
+        query: memoryResponseData2.query || options.query,
+        semantics: memoryResponseData2.semantics || memoryResponseData2.knowledge || [],
+        procedures: memoryResponseData2.procedures || [],
+        episodes: memoryResponseData2.episodes || [],
+        sources: memoryResponseData2.sources || [],
+        total_traversal_time_ms: memoryResponseData2.total_traversal_time_ms,
+        token_count: memoryResponseData2.token_count,
+        entities: memoryResponseData2.entities || []
+      };
+    }
     const data = {
       query: options.query
     };
@@ -839,7 +874,12 @@ var _Nebula = class _Nebula {
       data.search_settings = options.searchSettings;
     }
     const collectionIds = data.collection_ids;
-    const extraHeaders = collectionIds && collectionIds.length === 1 ? { "X-Nebula-Collection-Id": collectionIds[0] } : void 0;
+    let extraHeaders;
+    if (collectionIds && collectionIds.length === 1) {
+      extraHeaders = { "X-Nebula-Collection-Id": collectionIds[0] };
+    } else if (collectionIds && collectionIds.length > 1) {
+      extraHeaders = { "X-Nebula-Collection-Id": [...collectionIds].sort().join(",") };
+    }
     const response = await this._makeRequest("POST", "/v1/memories/search", data, void 0, extraHeaders);
     const memoryResponseData = response.results;
     const memoryResponse = {
@@ -1174,6 +1214,29 @@ var _Nebula = class _Nebula {
     }
     return response;
   }
+  // ------------------------------------------------------------------
+  // Device Memory
+  // ------------------------------------------------------------------
+  /**
+   * Export a collection's full graph state as a portable snapshot.
+   */
+  async exportSnapshot(collectionId) {
+    const response = await this._makeRequest("POST", "/v1/device-memory/snapshot/export", {
+      collection_id: collectionId
+    });
+    return response.results ?? response;
+  }
+  /**
+   * Import a snapshot into an ephemeral server-side collection.
+   * @returns The ephemeral collection ID.
+   */
+  async importSnapshot(snapshot) {
+    const response = await this._makeRequest("POST", "/v1/device-memory/snapshot/import", {
+      snapshot
+    });
+    const result = response.results ?? response;
+    return result.ephemeral_collection_id ?? "";
+  }
 };
 // Files larger than 5MB are automatically uploaded to S3
 _Nebula.MAX_INLINE_SIZE = 5 * 1024 * 1024;