@maximem/synap-js-sdk 0.1.3 → 0.1.4

package/bridge/synap_bridge.py

@@ -168,11 +168,12 @@ async def handle_add_memory(params: dict) -> dict:
     start = time.perf_counter()
 
     step = time.perf_counter()
+    mode = params.get("mode", "long-range")
     create_result = await sdk.memories.create(
         document=transcript,
         document_type="ai-chat-conversation",
         user_id=user_id,
-        mode="long-range",
+        mode=mode,
     )
     append_step(timings, "memories_create", step)
 
@@ -243,6 +244,7 @@ async def handle_search_memory(params: dict) -> dict:
     user_id = params["user_id"]
     query = params["query"]
     max_results = params.get("max_results", 10)
+    mode = params.get("mode", "fast")
 
     start = time.perf_counter()
 
@@ -252,7 +254,7 @@ async def handle_search_memory(params: dict) -> dict:
         search_query=[query],
         max_results=max_results,
         types=["all"],
-        mode="fast",
+        mode=mode,
     )
     append_step(timings, "context_fetch", step)
 
@@ -293,6 +295,7 @@ async def handle_get_memories(params: dict) -> dict:
     timings: List[dict] = []
 
     user_id = params["user_id"]
+    mode = params.get("mode", "fast")
 
     start = time.perf_counter()
 
@@ -302,7 +305,7 @@ async def handle_get_memories(params: dict) -> dict:
         search_query=[],
         max_results=100,
         types=["all"],
-        mode="fast",
+        mode=mode,
     )
     append_step(timings, "context_fetch_all", step)
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@maximem/synap-js-sdk",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "JavaScript wrapper around the Synap Python SDK",
   "main": "src/index.js",
   "types": "types/index.d.ts",

package/src/synap-client.js

@@ -14,32 +14,33 @@ class SynapClient {
     await this.bridge.ensureStarted();
   }
 
-  async addMemory({ userId, messages }) {
+  async addMemory({ userId, messages, mode }) {
     this.#assert(userId, 'userId is required');
     this.#assert(Array.isArray(messages), 'messages must be an array');
 
-    return this.bridge.call(
-      'add_memory',
-      { user_id: userId, messages },
-      this.options.ingestTimeoutMs
-    );
+    const params = { user_id: userId, messages };
+    if (mode !== undefined) params.mode = mode;
+
+    return this.bridge.call('add_memory', params, this.options.ingestTimeoutMs);
   }
 
-  async searchMemory({ userId, query, maxResults = 10 }) {
+  async searchMemory({ userId, query, maxResults = 10, mode }) {
     this.#assert(userId, 'userId is required');
     this.#assert(query, 'query is required');
 
-    return this.bridge.call('search_memory', {
-      user_id: userId,
-      query,
-      max_results: maxResults,
-    });
+    const params = { user_id: userId, query, max_results: maxResults };
+    if (mode !== undefined) params.mode = mode;
+
+    return this.bridge.call('search_memory', params);
   }
 
-  async getMemories({ userId }) {
+  async getMemories({ userId, mode }) {
     this.#assert(userId, 'userId is required');
 
-    return this.bridge.call('get_memories', { user_id: userId });
+    const params = { user_id: userId };
+    if (mode !== undefined) params.mode = mode;
+
+    return this.bridge.call('get_memories', params);
   }
 
   async deleteMemory({ userId, memoryId = null }) {