@reverbia/sdk 1.0.0-next.20251121094738 → 1.0.0-next.20251125084024

package/dist/react/index.cjs

@@ -40,287 +40,6 @@ module.exports = __toCommonJS(index_exports);
 
 // src/react/useChat.ts
 var import_react = require("react");
-var import_client = require("@reverbia/sdk");
-function useChat(options) {
-  const { getToken } = options || {};
-  const [isLoading, setIsLoading] = (0, import_react.useState)(false);
-  const sendMessage = (0, import_react.useCallback)(
-    async ({
-      messages,
-      model
-    }) => {
-      if (!messages?.length) {
-        const error = "messages are required to call sendMessage.";
-        return { data: null, error };
-      }
-      if (!model) {
-        const error = "model is required to call sendMessage.";
-        return { data: null, error };
-      }
-      if (!getToken) {
-        const error = "Token getter function is required.";
-        return { data: null, error };
-      }
-      setIsLoading(true);
-      try {
-        const token = await getToken();
-        if (!token) {
-          const error = "No access token available.";
-          setIsLoading(false);
-          return { data: null, error };
-        }
-        const completion = await (0, import_client.postApiV1ChatCompletions)({
-          body: {
-            messages,
-            model
-          }
-          // headers: {
-          //   Authorization: `Bearer ${token}`,
-          // },
-        });
-        if (!completion.data) {
-          const error = completion.error?.error ?? "API did not return a completion response.";
-          setIsLoading(false);
-          return { data: null, error };
-        }
-        if (typeof completion.data === "string") {
-          const error = "API returned a string response instead of a completion object.";
-          setIsLoading(false);
-          return { data: null, error };
-        }
-        setIsLoading(false);
-        return { data: completion.data, error: null };
-      } catch (err) {
-        const error = err instanceof Error ? err.message : "Failed to send message.";
-        setIsLoading(false);
-        return { data: null, error };
-      }
-    },
-    [getToken]
-  );
-  return {
-    isLoading,
-    sendMessage
-  };
-}
-
-// src/react/useMemory.ts
-var import_react2 = require("react");
-var import_client6 = require("@reverbia/sdk");
-
-// src/lib/memory/service.ts
-var FACT_EXTRACTION_PROMPT = `You are a memory extraction system. Extract durable user memories from chat messages.
-
-CRITICAL: You MUST respond with ONLY valid JSON. No explanations, no markdown, no code blocks, just pure JSON.
-
-Only extract facts that will be useful in future conversations, such as identity, stable preferences, ongoing projects, skills, and constraints.
-
-Do not extract sensitive attributes, temporary things, or single-use instructions.
-
-If there are no memories to extract, return: {"items": []}
-
-Response format (JSON only, no other text):
-
-{
-  "items": [
-    {
-      "type": "identity",
-      "namespace": "identity",
-      "key": "name",
-      "value": "Charlie",
-      "rawEvidence": "I'm Charlie",
-      "confidence": 0.98,
-      "pii": true
-    },
-    {
-      "type": "identity",
-      "namespace": "work",
-      "key": "company",
-      "value": "ZetaChain",
-      "rawEvidence": "called ZetaChain",
-      "confidence": 0.99,
-      "pii": false
-    },
-    {
-      "type": "preference",
-      "namespace": "answer_style",
-      "key": "verbosity",
-      "value": "concise_direct",
-      "rawEvidence": "I prefer concise, direct answers",
-      "confidence": 0.96,
-      "pii": false
-    },
-    {
-      "type": "identity",
-      "namespace": "timezone",
-      "key": "tz",
-      "value": "America/Los_Angeles",
-      "rawEvidence": "I'm in PST",
-      "confidence": 0.9,
-      "pii": false
-    }
-  ]
-}`;
-var preprocessMemories = (items, minConfidence = 0.6) => {
-  if (!items || !Array.isArray(items)) {
-    return [];
-  }
-  const validItems = items.filter((item) => {
-    if (item.namespace == null || item.key == null || item.value == null) {
-      console.warn(
-        "Dropping memory item with null/undefined namespace, key, or value:",
-        item
-      );
-      return false;
-    }
-    const namespace = String(item.namespace).trim();
-    const key = String(item.key).trim();
-    const value = String(item.value).trim();
-    if (namespace === "" || key === "" || value === "") {
-      console.warn(
-        "Dropping memory item with empty namespace, key, or value after trimming:",
-        item
-      );
-      return false;
-    }
-    if (typeof item.confidence !== "number" || item.confidence < minConfidence) {
-      console.warn(
-        `Dropping memory item with confidence ${item.confidence} below threshold ${minConfidence}:`,
-        item
-      );
-      return false;
-    }
-    return true;
-  });
-  const deduplicatedMap = /* @__PURE__ */ new Map();
-  for (const item of validItems) {
-    const uniqueKey = `${item.namespace}:${item.key}:${item.value}`;
-    const existing = deduplicatedMap.get(uniqueKey);
-    if (!existing || item.confidence > existing.confidence) {
-      deduplicatedMap.set(uniqueKey, item);
-    } else {
-      console.debug(
-        `Deduplicating memory item: keeping entry with higher confidence (${existing.confidence} > ${item.confidence})`,
-        { namespace: item.namespace, key: item.key, value: item.value }
-      );
-    }
-  }
-  return Array.from(deduplicatedMap.values());
-};
-
-// src/lib/memory/db.ts
-var import_dexie = __toESM(require("dexie"));
-var MemoryDatabase = class extends import_dexie.default {
-  constructor() {
-    super("MemoryDatabase");
-    this.version(2).stores({
-      memories: "++id, uniqueKey, compositeKey, namespace, key, type, createdAt, updatedAt"
-    });
-    this.version(3).stores({
-      memories: "++id, uniqueKey, compositeKey, namespace, key, type, createdAt, updatedAt"
-    });
-  }
-};
-var memoryDb = new MemoryDatabase();
-var saveMemory = async (memory) => {
-  const compositeKey = `${memory.namespace}:${memory.key}`;
-  const uniqueKey = `${memory.namespace}:${memory.key}:${memory.value}`;
-  const now = Date.now();
-  const existing = await memoryDb.memories.where("uniqueKey").equals(uniqueKey).first();
-  if (existing) {
-    const shouldPreserveEmbedding = existing.value === memory.value && existing.rawEvidence === memory.rawEvidence && existing.type === memory.type && existing.namespace === memory.namespace && existing.key === memory.key && existing.embedding !== void 0 && existing.embedding.length > 0;
-    const updateData = {
-      ...memory,
-      compositeKey,
-      uniqueKey,
-      updatedAt: now,
-      createdAt: existing.createdAt
-    };
-    if (shouldPreserveEmbedding) {
-      updateData.embedding = existing.embedding;
-      updateData.embeddingModel = existing.embeddingModel;
-    } else {
-      updateData.embedding = [];
-      updateData.embeddingModel = void 0;
-    }
-    await memoryDb.memories.update(existing.id, updateData);
-  } else {
-    await memoryDb.memories.add({
-      ...memory,
-      compositeKey,
-      uniqueKey,
-      createdAt: now,
-      updatedAt: now
-    });
-  }
-};
-var saveMemories = async (memories) => {
-  await Promise.all(memories.map((memory) => saveMemory(memory)));
-};
-var getAllMemories = async () => {
-  return memoryDb.memories.toArray();
-};
-var cosineSimilarity = (a, b) => {
-  if (a.length !== b.length) {
-    throw new Error("Vectors must have the same length");
-  }
-  let dotProduct = 0;
-  let normA = 0;
-  let normB = 0;
-  for (let i = 0; i < a.length; i++) {
-    dotProduct += a[i] * b[i];
-    normA += a[i] * a[i];
-    normB += b[i] * b[i];
-  }
-  const denominator = Math.sqrt(normA) * Math.sqrt(normB);
-  if (denominator === 0) {
-    return 0;
-  }
-  return dotProduct / denominator;
-};
-var searchSimilarMemories = async (queryEmbedding, limit = 10, minSimilarity = 0.6) => {
-  const allMemories = await getAllMemories();
-  const memoriesWithEmbeddings = allMemories.filter(
-    (m) => m.embedding && m.embedding.length > 0
-  );
-  console.log(
-    `[Memory Search] Total memories: ${allMemories.length}, memories with embeddings: ${memoriesWithEmbeddings.length}`
-  );
-  if (memoriesWithEmbeddings.length === 0) {
-    console.warn(
-      "[Memory Search] No memories with embeddings found. Memories may need embeddings generated. Use generateAndStoreEmbeddings() to generate embeddings for existing memories."
-    );
-    return [];
-  }
-  const allResults = memoriesWithEmbeddings.map((memory) => {
-    const similarity = cosineSimilarity(queryEmbedding, memory.embedding);
-    return {
-      ...memory,
-      similarity
-    };
-  }).sort((a, b) => b.similarity - a.similarity);
-  console.log(
-    `[Memory Search] All similarity scores:`,
-    allResults.map((r) => ({
-      key: `${r.namespace}:${r.key}`,
-      value: r.value,
-      similarity: r.similarity.toFixed(4)
-    }))
-  );
-  const results = allResults.filter((result) => result.similarity >= minSimilarity).slice(0, limit);
-  if (results.length === 0 && allResults.length > 0) {
-    const topSimilarity = allResults[0].similarity;
-    const suggestedThreshold = Math.max(0.3, topSimilarity - 0.1);
-    console.warn(
-      `[Memory Search] No memories above threshold ${minSimilarity}. Highest similarity was ${topSimilarity.toFixed(4)}. Consider lowering the threshold to ${suggestedThreshold.toFixed(2)}`
-    );
-  } else {
-    console.log(
-      `[Memory Search] Found ${results.length} memories above similarity threshold ${minSimilarity}. Top similarity: ${results[0]?.similarity.toFixed(4) || "N/A"}`
-    );
-  }
-  return results;
-};
 
 // src/client/core/bodySerializer.gen.ts
 var jsonBodySerializer = {
@@ -1136,6 +855,376 @@ var createClientConfig = (config) => ({
 // src/client/client.gen.ts
 var client = createClient(createClientConfig(createConfig()));
 
+// src/react/useChat.ts
+function useChat(options) {
+  const { getToken, onData: globalOnData, onFinish, onError } = options || {};
+  const [isLoading, setIsLoading] = (0, import_react.useState)(false);
+  const abortControllerRef = (0, import_react.useRef)(null);
+  const stop = (0, import_react.useCallback)(() => {
+    if (abortControllerRef.current) {
+      abortControllerRef.current.abort();
+      abortControllerRef.current = null;
+    }
+  }, []);
+  (0, import_react.useEffect)(() => {
+    return () => {
+      if (abortControllerRef.current) {
+        abortControllerRef.current.abort();
+        abortControllerRef.current = null;
+      }
+    };
+  }, []);
+  const sendMessage = (0, import_react.useCallback)(
+    async ({
+      messages,
+      model,
+      onData
+    }) => {
+      if (!messages?.length) {
+        const errorMsg = "messages are required to call sendMessage.";
+        if (onError) onError(new Error(errorMsg));
+        return { data: null, error: errorMsg };
+      }
+      if (!model) {
+        const errorMsg = "model is required to call sendMessage.";
+        if (onError) onError(new Error(errorMsg));
+        return { data: null, error: errorMsg };
+      }
+      if (!getToken) {
+        const errorMsg = "Token getter function is required.";
+        if (onError) onError(new Error(errorMsg));
+        return { data: null, error: errorMsg };
+      }
+      if (abortControllerRef.current) {
+        abortControllerRef.current.abort();
+      }
+      const abortController = new AbortController();
+      abortControllerRef.current = abortController;
+      setIsLoading(true);
+      try {
+        const token = await getToken();
+        if (!token) {
+          const errorMsg = "No access token available.";
+          setIsLoading(false);
+          if (onError) onError(new Error(errorMsg));
+          return { data: null, error: errorMsg };
+        }
+        const sseResult = await client.sse.post({
+          url: "/api/v1/chat/completions",
+          body: {
+            messages,
+            model,
+            stream: true
+          },
+          headers: {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${token}`
+          },
+          signal: abortController.signal
+        });
+        let accumulatedContent = "";
+        let completionId = "";
+        let completionModel = "";
+        let usage;
+        let finishReason;
+        for await (const chunk of sseResult.stream) {
+          if (typeof chunk === "string" && (chunk.trim() === "[DONE]" || chunk.includes("[DONE]"))) {
+            continue;
+          }
+          if (chunk && typeof chunk === "object") {
+            const chunkData = chunk;
+            if (chunkData.id && !completionId) {
+              completionId = chunkData.id;
+            }
+            if (chunkData.model && !completionModel) {
+              completionModel = chunkData.model;
+            }
+            if (chunkData.usage) {
+              usage = chunkData.usage;
+            }
+            if (chunkData.choices && Array.isArray(chunkData.choices) && chunkData.choices.length > 0) {
+              const choice = chunkData.choices[0];
+              if (choice.delta?.content) {
+                const content = choice.delta.content;
+                accumulatedContent += content;
+                if (onData) {
+                  onData(content);
+                }
+                if (globalOnData) {
+                  globalOnData(content);
+                }
+              }
+              if (choice.finish_reason) {
+                finishReason = choice.finish_reason;
+              }
+            }
+          }
+        }
+        const completion = {
+          id: completionId,
+          model: completionModel,
+          choices: [
+            {
+              index: 0,
+              message: {
+                role: "assistant",
+                content: accumulatedContent
+              },
+              finish_reason: finishReason
+            }
+          ],
+          usage
+        };
+        setIsLoading(false);
+        if (onFinish) {
+          onFinish(completion);
+        }
+        return { data: completion, error: null };
+      } catch (err) {
+        if (err instanceof Error && err.name === "AbortError") {
+          setIsLoading(false);
+          return { data: null, error: "Request aborted" };
+        }
+        const errorMsg = err instanceof Error ? err.message : "Failed to send message.";
+        const errorObj = err instanceof Error ? err : new Error(errorMsg);
+        setIsLoading(false);
+        if (onError) {
+          onError(errorObj);
+        }
+        return { data: null, error: errorMsg };
+      } finally {
+        if (abortControllerRef.current === abortController) {
+          abortControllerRef.current = null;
+        }
+      }
+    },
+    [getToken, globalOnData, onFinish, onError]
+  );
+  return {
+    isLoading,
+    sendMessage,
+    stop
+  };
+}
+
+// src/react/useMemory.ts
+var import_react2 = require("react");
+var import_client6 = require("@reverbia/sdk");
+
+// src/lib/memory/service.ts
+var FACT_EXTRACTION_PROMPT = `You are a memory extraction system. Extract durable user memories from chat messages.
+
+CRITICAL: You MUST respond with ONLY valid JSON. No explanations, no markdown, no code blocks, just pure JSON.
+
+Only extract facts that will be useful in future conversations, such as identity, stable preferences, ongoing projects, skills, and constraints.
+
+Do not extract sensitive attributes, temporary things, or single-use instructions.
+
+If there are no memories to extract, return: {"items": []}
+
+Response format (JSON only, no other text):
+
+{
+  "items": [
+    {
+      "type": "identity",
+      "namespace": "identity",
+      "key": "name",
+      "value": "Charlie",
+      "rawEvidence": "I'm Charlie",
+      "confidence": 0.98,
+      "pii": true
+    },
+    {
+      "type": "identity",
+      "namespace": "work",
+      "key": "company",
+      "value": "ZetaChain",
+      "rawEvidence": "called ZetaChain",
+      "confidence": 0.99,
+      "pii": false
+    },
+    {
+      "type": "preference",
+      "namespace": "answer_style",
+      "key": "verbosity",
+      "value": "concise_direct",
+      "rawEvidence": "I prefer concise, direct answers",
+      "confidence": 0.96,
+      "pii": false
+    },
+    {
+      "type": "identity",
+      "namespace": "timezone",
+      "key": "tz",
+      "value": "America/Los_Angeles",
+      "rawEvidence": "I'm in PST",
+      "confidence": 0.9,
+      "pii": false
+    }
+  ]
+}`;
+var preprocessMemories = (items, minConfidence = 0.6) => {
+  if (!items || !Array.isArray(items)) {
+    return [];
+  }
+  const validItems = items.filter((item) => {
+    if (item.namespace == null || item.key == null || item.value == null) {
+      console.warn(
+        "Dropping memory item with null/undefined namespace, key, or value:",
+        item
+      );
+      return false;
+    }
+    const namespace = String(item.namespace).trim();
+    const key = String(item.key).trim();
+    const value = String(item.value).trim();
+    if (namespace === "" || key === "" || value === "") {
+      console.warn(
+        "Dropping memory item with empty namespace, key, or value after trimming:",
+        item
+      );
+      return false;
+    }
+    if (typeof item.confidence !== "number" || item.confidence < minConfidence) {
+      console.warn(
+        `Dropping memory item with confidence ${item.confidence} below threshold ${minConfidence}:`,
+        item
+      );
+      return false;
+    }
+    return true;
+  });
+  const deduplicatedMap = /* @__PURE__ */ new Map();
+  for (const item of validItems) {
+    const uniqueKey = `${item.namespace}:${item.key}:${item.value}`;
+    const existing = deduplicatedMap.get(uniqueKey);
+    if (!existing || item.confidence > existing.confidence) {
+      deduplicatedMap.set(uniqueKey, item);
+    } else {
+      console.debug(
+        `Deduplicating memory item: keeping entry with higher confidence (${existing.confidence} > ${item.confidence})`,
+        { namespace: item.namespace, key: item.key, value: item.value }
+      );
+    }
+  }
+  return Array.from(deduplicatedMap.values());
+};
+
+// src/lib/memory/db.ts
+var import_dexie = __toESM(require("dexie"));
+var MemoryDatabase = class extends import_dexie.default {
+  constructor() {
+    super("MemoryDatabase");
+    this.version(2).stores({
+      memories: "++id, uniqueKey, compositeKey, namespace, key, type, createdAt, updatedAt"
+    });
+    this.version(3).stores({
+      memories: "++id, uniqueKey, compositeKey, namespace, key, type, createdAt, updatedAt"
+    });
+  }
+};
+var memoryDb = new MemoryDatabase();
+var saveMemory = async (memory) => {
+  const compositeKey = `${memory.namespace}:${memory.key}`;
+  const uniqueKey = `${memory.namespace}:${memory.key}:${memory.value}`;
+  const now = Date.now();
+  const existing = await memoryDb.memories.where("uniqueKey").equals(uniqueKey).first();
+  if (existing) {
+    const shouldPreserveEmbedding = existing.value === memory.value && existing.rawEvidence === memory.rawEvidence && existing.type === memory.type && existing.namespace === memory.namespace && existing.key === memory.key && existing.embedding !== void 0 && existing.embedding.length > 0;
+    const updateData = {
+      ...memory,
+      compositeKey,
+      uniqueKey,
+      updatedAt: now,
+      createdAt: existing.createdAt
+    };
+    if (shouldPreserveEmbedding) {
+      updateData.embedding = existing.embedding;
+      updateData.embeddingModel = existing.embeddingModel;
+    } else {
+      updateData.embedding = [];
+      updateData.embeddingModel = void 0;
+    }
+    await memoryDb.memories.update(existing.id, updateData);
+  } else {
+    await memoryDb.memories.add({
+      ...memory,
+      compositeKey,
+      uniqueKey,
+      createdAt: now,
+      updatedAt: now
+    });
+  }
+};
+var saveMemories = async (memories) => {
+  await Promise.all(memories.map((memory) => saveMemory(memory)));
+};
+var getAllMemories = async () => {
+  return memoryDb.memories.toArray();
+};
+var cosineSimilarity = (a, b) => {
+  if (a.length !== b.length) {
+    throw new Error("Vectors must have the same length");
+  }
+  let dotProduct = 0;
+  let normA = 0;
+  let normB = 0;
+  for (let i = 0; i < a.length; i++) {
+    dotProduct += a[i] * b[i];
+    normA += a[i] * a[i];
+    normB += b[i] * b[i];
+  }
+  const denominator = Math.sqrt(normA) * Math.sqrt(normB);
+  if (denominator === 0) {
+    return 0;
+  }
+  return dotProduct / denominator;
+};
+var searchSimilarMemories = async (queryEmbedding, limit = 10, minSimilarity = 0.6) => {
+  const allMemories = await getAllMemories();
+  const memoriesWithEmbeddings = allMemories.filter(
+    (m) => m.embedding && m.embedding.length > 0
+  );
+  console.log(
+    `[Memory Search] Total memories: ${allMemories.length}, memories with embeddings: ${memoriesWithEmbeddings.length}`
+  );
+  if (memoriesWithEmbeddings.length === 0) {
+    console.warn(
+      "[Memory Search] No memories with embeddings found. Memories may need embeddings generated. Use generateAndStoreEmbeddings() to generate embeddings for existing memories."
+    );
+    return [];
+  }
+  const allResults = memoriesWithEmbeddings.map((memory) => {
+    const similarity = cosineSimilarity(queryEmbedding, memory.embedding);
+    return {
+      ...memory,
+      similarity
+    };
+  }).sort((a, b) => b.similarity - a.similarity);
+  console.log(
+    `[Memory Search] All similarity scores:`,
+    allResults.map((r) => ({
+      key: `${r.namespace}:${r.key}`,
+      value: r.value,
+      similarity: r.similarity.toFixed(4)
+    }))
+  );
+  const results = allResults.filter((result) => result.similarity >= minSimilarity).slice(0, limit);
+  if (results.length === 0 && allResults.length > 0) {
+    const topSimilarity = allResults[0].similarity;
+    const suggestedThreshold = Math.max(0.3, topSimilarity - 0.1);
+    console.warn(
+      `[Memory Search] No memories above threshold ${minSimilarity}. Highest similarity was ${topSimilarity.toFixed(4)}. Consider lowering the threshold to ${suggestedThreshold.toFixed(2)}`
+    );
+  } else {
+    console.log(
+      `[Memory Search] Found ${results.length} memories above similarity threshold ${minSimilarity}. Top similarity: ${results[0]?.similarity.toFixed(4) || "N/A"}`
+    );
+  }
+  return results;
+};
+
 // src/client/sdk.gen.ts
 var postApiV1Embeddings = (options) => {
   return (options.client ?? client).post({
@@ -1161,8 +1250,8 @@ var generateEmbeddingForText = async (text, options = {}) => {
       body: {
         input: text,
         model
-      }
-      // headers,
+      },
+      headers
     });
     if (!response.data || !response.data.data || response.data.data.length === 0) {
       throw new Error(
@@ -1276,10 +1365,10 @@ function useMemory(options = {}) {
               ...messages
             ],
             model: model || memoryModel
+          },
+          headers: {
+            Authorization: `Bearer ${token}`
           }
-          // headers: {
-          //   Authorization: `Bearer ${token}`,
-          // },
         });
         if (!completion.data) {
           console.error(