@reverbia/sdk 1.0.0-next.20251202085711 → 1.0.0-next.20251202090922

package/dist/react/index.d.mts

@@ -339,12 +339,20 @@ type UseMemoryOptions = {
     /**
      * The model to use for fact extraction (default: "openai/gpt-4o")
      */
-    memoryModel?: string;
+    completionsModel?: string;
     /**
-     * The model to use for generating embeddings (default: "openai/text-embedding-3-small")
-     * Set to null/undefined to disable embedding generation
+     * The model to use for generating embeddings
+     * For local: default is "Snowflake/snowflake-arctic-embed-xs"
+     * For api: default is "openai/text-embedding-3-small"
+     * Set to null to disable embedding generation
      */
     embeddingModel?: string | null;
+    /**
+     * The provider to use for generating embeddings (default: "local")
+     * "local": Uses a local HuggingFace model (in-browser)
+     * "api": Uses the backend API
+     */
+    embeddingProvider?: "local" | "api";
     /**
      * Whether to automatically generate embeddings for extracted memories (default: true)
      */

package/dist/react/index.d.ts

@@ -339,12 +339,20 @@ type UseMemoryOptions = {
     /**
      * The model to use for fact extraction (default: "openai/gpt-4o")
      */
-    memoryModel?: string;
+    completionsModel?: string;
     /**
-     * The model to use for generating embeddings (default: "openai/text-embedding-3-small")
-     * Set to null/undefined to disable embedding generation
+     * The model to use for generating embeddings
+     * For local: default is "Snowflake/snowflake-arctic-embed-xs"
+     * For api: default is "openai/text-embedding-3-small"
+     * Set to null to disable embedding generation
      */
     embeddingModel?: string | null;
+    /**
+     * The provider to use for generating embeddings (default: "local")
+     * "local": Uses a local HuggingFace model (in-browser)
+     * "api": Uses the backend API
+     */
+    embeddingProvider?: "local" | "api";
     /**
      * Whether to automatically generate embeddings for extracted memories (default: true)
      */

package/dist/react/index.mjs

@@ -1,3 +1,5 @@
+import "./chunk-FBCDBTKJ.mjs";
+
 // src/react/useChat.ts
 import { useCallback, useEffect, useRef, useState } from "react";
 
@@ -1372,37 +1374,61 @@ var getApiV1Models = (options) => {
   });
 };
 
+// src/lib/memory/constants.ts
+var DEFAULT_LOCAL_EMBEDDING_MODEL = "Snowflake/snowflake-arctic-embed-xs";
+var DEFAULT_API_EMBEDDING_MODEL = "openai/text-embedding-3-small";
+var DEFAULT_COMPLETION_MODEL = "openai/gpt-4o";
+
 // src/lib/memory/embeddings.ts
+var embeddingPipeline = null;
 var generateEmbeddingForText = async (text, options = {}) => {
-  const {
-    model = "openai/text-embedding-3-small",
-    getToken,
-    baseUrl = BASE_URL
-  } = options;
-  try {
-    const token = getToken ? await getToken() : null;
-    const headers = {};
-    if (token) {
-      headers.Authorization = `Bearer ${token}`;
+  const { baseUrl = BASE_URL, provider = "local" } = options;
+  if (provider === "api") {
+    const { getToken, model: model2 } = options;
+    if (!getToken) {
+      throw new Error("getToken is required for API embeddings");
+    }
+    const token = await getToken();
+    if (!token) {
+      throw new Error("No access token available for API embeddings");
     }
     const response = await postApiV1Embeddings({
       baseUrl,
       body: {
         input: text,
-        model
+        model: model2
       },
-      headers
+      headers: {
+        Authorization: `Bearer ${token}`
+      }
     });
-    if (!response.data || !response.data.data || response.data.data.length === 0) {
+    if (response.error) {
       throw new Error(
-        `Failed to generate embedding: ${response.error?.error ?? "No data returned"}`
+        typeof response.error === "object" && response.error && "error" in response.error ? response.error.error : "API embedding failed"
       );
     }
-    const embedding = response.data.data[0]?.embedding;
-    if (!embedding || !Array.isArray(embedding)) {
-      throw new Error("Invalid embedding format returned from API");
+    if (!response.data?.data?.[0]?.embedding) {
+      throw new Error("No embedding returned from API");
+    }
+    return response.data.data[0].embedding;
+  }
+  let { model } = options;
+  if (!model || model === DEFAULT_API_EMBEDDING_MODEL) {
+    model = DEFAULT_LOCAL_EMBEDDING_MODEL;
+  }
+  try {
+    if (!embeddingPipeline) {
+      const { pipeline } = await import("./transformers.node-BSHUG7OY.mjs");
+      embeddingPipeline = await pipeline("feature-extraction", model);
+    }
+    const output = await embeddingPipeline(text, {
+      pooling: "cls",
+      normalize: true
+    });
+    if (output?.data) {
+      return Array.from(output.data);
     }
-    return embedding;
+    throw new Error("Invalid embedding output from transformers.js");
   } catch (error) {
     console.error("Failed to generate embedding:", error);
     throw error;
@@ -1419,20 +1445,11 @@ var generateEmbeddingForMemory = async (memory, options = {}) => {
   return generateEmbeddingForText(text, options);
 };
 var generateEmbeddingsForMemories = async (memories, options = {}) => {
-  const {
-    model = "openai/text-embedding-3-small",
-    getToken,
-    baseUrl = BASE_URL
-  } = options;
   const embeddings = /* @__PURE__ */ new Map();
   for (const memory of memories) {
     const uniqueKey = `${memory.namespace}:${memory.key}:${memory.value}`;
     try {
-      const embedding = await generateEmbeddingForMemory(memory, {
-        model,
-        getToken,
-        baseUrl
-      });
+      const embedding = await generateEmbeddingForMemory(memory, options);
       embeddings.set(uniqueKey, embedding);
     } catch (error) {
       console.error(
@@ -1464,29 +1481,42 @@ var updateMemoriesWithEmbeddings = async (embeddings, embeddingModel) => {
   await Promise.all(updates);
 };
 var generateAndStoreEmbeddings = async (memories, options = {}) => {
-  const { model = "openai/text-embedding-3-small" } = options;
+  let { model } = options;
+  const { provider = "local" } = options;
+  if (!model) {
+    if (provider === "local") {
+      model = DEFAULT_LOCAL_EMBEDDING_MODEL;
+    } else {
+      model = DEFAULT_API_EMBEDDING_MODEL;
+    }
+  }
+  if (provider === "local" && model === DEFAULT_API_EMBEDDING_MODEL) {
+    model = DEFAULT_LOCAL_EMBEDDING_MODEL;
+  }
   if (memories.length === 0) {
     return;
   }
   console.log(`Generating embeddings for ${memories.length} memories...`);
-  const embeddings = await generateEmbeddingsForMemories(memories, options);
+  const embeddings = await generateEmbeddingsForMemories(memories, {
+    ...options,
+    model
+  });
   await updateMemoriesWithEmbeddings(embeddings, model);
   console.log(`Generated and stored ${embeddings.size} embeddings`);
 };
-var generateQueryEmbedding = async (query, options = {}) => {
-  return generateEmbeddingForText(query, options);
-};
 
 // src/react/useMemory.ts
 function useMemory(options = {}) {
   const {
-    memoryModel = "openai/gpt-4o",
-    embeddingModel = "openai/text-embedding-3-small",
+    completionsModel = DEFAULT_COMPLETION_MODEL,
+    embeddingModel: userEmbeddingModel,
+    embeddingProvider = "local",
     generateEmbeddings = true,
     onFactsExtracted,
     getToken,
     baseUrl = BASE_URL
   } = options;
+  const embeddingModel = userEmbeddingModel === void 0 ? embeddingProvider === "local" ? DEFAULT_LOCAL_EMBEDDING_MODEL : DEFAULT_API_EMBEDDING_MODEL : userEmbeddingModel;
   const extractionInProgressRef = useRef3(false);
   const extractMemoriesFromMessage = useCallback2(
     async (options2) => {
@@ -1511,7 +1541,7 @@ function useMemory(options = {}) {
               },
               ...messages
             ],
-            model: model || memoryModel
+            model: model || completionsModel
           },
           headers: {
             Authorization: `Bearer ${token}`
@@ -1626,6 +1656,7 @@ function useMemory(options = {}) {
               try {
                 await generateAndStoreEmbeddings(result.items, {
                   model: embeddingModel,
+                  provider: embeddingProvider,
                   getToken: getToken || void 0,
                   baseUrl
                 });
@@ -1652,8 +1683,9 @@ function useMemory(options = {}) {
       }
     },
     [
-      memoryModel,
+      completionsModel,
       embeddingModel,
+      embeddingProvider,
       generateEmbeddings,
       getToken,
       onFactsExtracted,
@@ -1662,16 +1694,15 @@ function useMemory(options = {}) {
   );
   const searchMemories = useCallback2(
     async (query, limit = 10, minSimilarity = 0.6) => {
-      if (!getToken || !embeddingModel) {
-        console.warn(
-          "Cannot search memories: getToken or embeddingModel not provided"
-        );
+      if (!embeddingModel) {
+        console.warn("Cannot search memories: embeddingModel not provided");
         return [];
       }
       try {
         console.log(`[Memory Search] Searching for: "${query}"`);
-        const queryEmbedding = await generateQueryEmbedding(query, {
+        const queryEmbedding = await generateEmbeddingForText(query, {
           model: embeddingModel,
+          provider: embeddingProvider,
           getToken,
           baseUrl
         });
@@ -1698,7 +1729,7 @@ function useMemory(options = {}) {
         return [];
       }
     },
-    [embeddingModel, getToken, baseUrl]
+    [embeddingModel, embeddingProvider, getToken, baseUrl]
   );
   return {
     extractMemoriesFromMessage,