@reverbia/sdk 1.0.0-next.20251215193957 → 1.0.0-next.20251217134403

package/dist/react/index.cjs

@@ -34,6 +34,8 @@ __export(index_exports, {
   ChatMessage: () => Message,
   DEFAULT_TOOL_SELECTOR_MODEL: () => DEFAULT_TOOL_SELECTOR_MODEL,
   StoredMemoryModel: () => Memory,
+  StoredModelPreferenceModel: () => ModelPreference,
+  chatStorageMigrations: () => chatStorageMigrations,
   chatStorageSchema: () => chatStorageSchema,
   createMemoryContextSystemMessage: () => createMemoryContextSystemMessage,
   decryptData: () => decryptData,
@@ -49,6 +51,7 @@ __export(index_exports, {
   memoryStorageSchema: () => memoryStorageSchema,
   requestEncryptionKey: () => requestEncryptionKey,
   selectTool: () => selectTool,
+  settingsStorageSchema: () => settingsStorageSchema,
   useChat: () => useChat,
   useChatStorage: () => useChatStorage,
   useEncryption: () => useEncryption,
@@ -57,7 +60,8 @@ __export(index_exports, {
   useModels: () => useModels,
   useOCR: () => useOCR,
   usePdf: () => usePdf,
-  useSearch: () => useSearch
+  useSearch: () => useSearch,
+  useSettings: () => useSettings
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -1482,6 +1486,7 @@ Please inform the user about this issue and try to help them alternatively.`
               toolExecution: toolExecutionResult
             };
           }
+          let sseError = null;
           const sseResult = await client.sse.post({
             baseUrl,
             url: "/api/v1/chat/completions",
@@ -1495,23 +1500,52 @@ Please inform the user about this issue and try to help them alternatively.`
               Authorization: `Bearer ${token}`,
               ...headers
             },
-            signal: abortController.signal
+            signal: abortController.signal,
+            sseMaxRetryAttempts: 1,
+            onSseError: (error) => {
+              sseError = error instanceof Error ? error : new Error(String(error));
+            }
           });
           const accumulator = createStreamAccumulator();
-          for await (const chunk of sseResult.stream) {
-            if (isDoneMarker(chunk)) {
-              continue;
-            }
-            if (chunk && typeof chunk === "object") {
-              const contentDelta = processStreamingChunk(
-                chunk,
-                accumulator
-              );
-              if (contentDelta) {
-                if (onData) onData(contentDelta);
-                if (globalOnData) globalOnData(contentDelta);
+          try {
+            for await (const chunk of sseResult.stream) {
+              if (isDoneMarker(chunk)) {
+                continue;
+              }
+              if (chunk && typeof chunk === "object") {
+                const contentDelta = processStreamingChunk(
+                  chunk,
+                  accumulator
+                );
+                if (contentDelta) {
+                  if (onData) onData(contentDelta);
+                  if (globalOnData) globalOnData(contentDelta);
+                }
               }
             }
+          } catch (streamErr) {
+            if (isAbortError(streamErr) || abortController.signal.aborted) {
+              setIsLoading(false);
+              const partialCompletion = buildCompletionResponse(accumulator);
+              return {
+                data: partialCompletion,
+                error: "Request aborted",
+                toolExecution: toolExecutionResult
+              };
+            }
+            throw streamErr;
+          }
+          if (abortController.signal.aborted) {
+            setIsLoading(false);
+            const partialCompletion = buildCompletionResponse(accumulator);
+            return {
+              data: partialCompletion,
+              error: "Request aborted",
+              toolExecution: toolExecutionResult
+            };
+          }
+          if (sseError) {
+            throw sseError;
           }
           const completion = buildCompletionResponse(accumulator);
           setIsLoading(false);
@@ -1733,7 +1767,8 @@ function messageToStored(message) {
     embeddingModel: message.embeddingModel,
     usage: message.usage,
     sources: message.sources,
-    responseDuration: message.responseDuration
+    responseDuration: message.responseDuration,
+    wasStopped: message.wasStopped
   };
 }
 function conversationToStored(conversation) {
@@ -1822,6 +1857,7 @@ async function createMessageOp(ctx, opts) {
         msg._setRaw("response_duration", opts.responseDuration);
       if (opts.vector) msg._setRaw("vector", JSON.stringify(opts.vector));
       if (opts.embeddingModel) msg._setRaw("embedding_model", opts.embeddingModel);
+      if (opts.wasStopped) msg._setRaw("was_stopped", opts.wasStopped);
     });
   });
   return messageToStored(created);
@@ -2112,6 +2148,45 @@ function useChatStorage(options) {
       });
       const responseDuration = (Date.now() - startTime) / 1e3;
       if (result.error || !result.data) {
+        const abortedResult = result;
+        if (abortedResult.error === "Request aborted") {
+          const assistantContent2 = abortedResult.data?.choices?.[0]?.message?.content?.map((part) => part.text || "").join("") || "";
+          const responseModel = abortedResult.data?.model || model || "";
+          let storedAssistantMessage2;
+          try {
+            storedAssistantMessage2 = await createMessageOp(storageCtx, {
+              conversationId: convId,
+              role: "assistant",
+              content: assistantContent2,
+              model: responseModel,
+              usage: convertUsageToStored(abortedResult.data?.usage),
+              responseDuration,
+              wasStopped: true
+            });
+            const completionData = abortedResult.data || {
+              id: `aborted-${Date.now()}`,
+              model: responseModel,
+              choices: [{
+                index: 0,
+                message: {
+                  role: "assistant",
+                  content: [{ type: "text", text: assistantContent2 }]
+                },
+                finish_reason: "stop"
+              }],
+              usage: void 0
+            };
+            return {
+              data: completionData,
+              error: null,
+              // Treat as success to the caller
+              toolExecution: abortedResult.toolExecution,
+              userMessage: storedUserMessage,
+              assistantMessage: storedAssistantMessage2
+            };
+          } catch (err) {
+          }
+        }
         return {
           data: null,
           error: result.error || "No response data received",
@@ -2183,8 +2258,9 @@ function useChatStorage(options) {
 
 // src/lib/chatStorage/schema.ts
 var import_watermelondb2 = require("@nozbe/watermelondb");
+var import_migrations = require("@nozbe/watermelondb/Schema/migrations");
 var chatStorageSchema = (0, import_watermelondb2.appSchema)({
-  version: 1,
+  version: 2,
   tables: [
     (0, import_watermelondb2.tableSchema)({
       name: "history",
@@ -2207,7 +2283,8 @@ var chatStorageSchema = (0, import_watermelondb2.appSchema)({
         // JSON stringified ChatCompletionUsage
         { name: "sources", type: "string", isOptional: true },
         // JSON stringified SearchSource[]
-        { name: "response_duration", type: "number", isOptional: true }
+        { name: "response_duration", type: "number", isOptional: true },
+        { name: "was_stopped", type: "boolean", isOptional: true }
       ]
     }),
     (0, import_watermelondb2.tableSchema)({
@@ -2222,6 +2299,21 @@ var chatStorageSchema = (0, import_watermelondb2.appSchema)({
     })
   ]
 });
+var chatStorageMigrations = (0, import_migrations.schemaMigrations)({
+  migrations: [
+    {
+      toVersion: 2,
+      steps: [
+        (0, import_migrations.addColumns)({
+          table: "history",
+          columns: [
+            { name: "was_stopped", type: "boolean", isOptional: true }
+          ]
+        })
+      ]
+    }
+  ]
+});
 
 // src/lib/chatStorage/models.ts
 var import_watermelondb3 = require("@nozbe/watermelondb");
@@ -2305,6 +2397,10 @@ var Message = class extends import_watermelondb3.Model {
     const value = this._getRaw("response_duration");
     return value !== null && value !== void 0 ? value : void 0;
   }
+  /** Whether the message generation was stopped by the user */
+  get wasStopped() {
+    return this._getRaw("was_stopped");
+  }
 };
 Message.table = "history";
 Message.associations = {
@@ -3507,9 +3603,181 @@ var Memory = class extends import_watermelondb6.Model {
 };
 Memory.table = "memories";
 
-// src/react/usePdf.ts
+// src/react/useSettings.ts
 var import_react4 = require("react");
 
+// src/lib/settingsStorage/operations.ts
+var import_watermelondb7 = require("@nozbe/watermelondb");
+function modelPreferenceToStored(preference) {
+  return {
+    uniqueId: preference.id,
+    walletAddress: preference.walletAddress,
+    models: preference.models
+  };
+}
+async function getModelPreferenceOp(ctx, walletAddress) {
+  const results = await ctx.modelPreferencesCollection.query(import_watermelondb7.Q.where("wallet_address", walletAddress)).fetch();
+  return results.length > 0 ? modelPreferenceToStored(results[0]) : null;
+}
+async function setModelPreferenceOp(ctx, walletAddress, models) {
+  const result = await ctx.database.write(async () => {
+    const results = await ctx.modelPreferencesCollection.query(import_watermelondb7.Q.where("wallet_address", walletAddress)).fetch();
+    if (results.length > 0) {
+      const preference = results[0];
+      await preference.update((pref) => {
+        if (models !== void 0) {
+          pref._setRaw("models", models || null);
+        }
+      });
+      return preference;
+    }
+    return await ctx.modelPreferencesCollection.create((pref) => {
+      pref._setRaw("wallet_address", walletAddress);
+      if (models) pref._setRaw("models", models);
+    });
+  });
+  return modelPreferenceToStored(result);
+}
+async function deleteModelPreferenceOp(ctx, walletAddress) {
+  const results = await ctx.modelPreferencesCollection.query(import_watermelondb7.Q.where("wallet_address", walletAddress)).fetch();
+  if (results.length === 0) return false;
+  await ctx.database.write(async () => {
+    await results[0].destroyPermanently();
+  });
+  return true;
+}
+
+// src/react/useSettings.ts
+function useSettings(options) {
+  const { database, walletAddress } = options;
+  const [modelPreference, setModelPreferenceState] = (0, import_react4.useState)(null);
+  const [isLoading, setIsLoading] = (0, import_react4.useState)(false);
+  const modelPreferencesCollection = (0, import_react4.useMemo)(
+    () => database.get("modelPreferences"),
+    [database]
+  );
+  const storageCtx = (0, import_react4.useMemo)(
+    () => ({
+      database,
+      modelPreferencesCollection
+    }),
+    [database, modelPreferencesCollection]
+  );
+  const getModelPreference = (0, import_react4.useCallback)(
+    async (address) => {
+      try {
+        if (!address) throw new Error("Wallet address is required");
+        const result = await getModelPreferenceOp(storageCtx, address);
+        return result;
+      } catch (error) {
+        throw new Error(
+          error instanceof Error ? error.message : "An unknown error occurred"
+        );
+      }
+    },
+    [storageCtx]
+  );
+  const setModelPreference = (0, import_react4.useCallback)(
+    async (address, models) => {
+      try {
+        if (!address) throw new Error("Wallet address is required");
+        const result = await setModelPreferenceOp(storageCtx, address, models);
+        if (walletAddress && address === walletAddress) {
+          setModelPreferenceState(result);
+        }
+        return result;
+      } catch (error) {
+        throw new Error(
+          error instanceof Error ? error.message : "An unknown error occurred"
+        );
+      }
+    },
+    [storageCtx, walletAddress]
+  );
+  const deleteModelPreference = (0, import_react4.useCallback)(
+    async (address) => {
+      try {
+        if (!address) throw new Error("Wallet address is required");
+        const deleted = await deleteModelPreferenceOp(storageCtx, address);
+        if (deleted && walletAddress && address === walletAddress) {
+          setModelPreferenceState(null);
+        }
+        return deleted;
+      } catch (error) {
+        throw new Error(
+          error instanceof Error ? error.message : "An unknown error occurred"
+        );
+      }
+    },
+    [storageCtx, walletAddress]
+  );
+  (0, import_react4.useEffect)(() => {
+    if (!walletAddress) {
+      setModelPreferenceState(null);
+      return;
+    }
+    let cancelled = false;
+    const loadPreference = async () => {
+      setIsLoading(true);
+      try {
+        const preference = await getModelPreference(walletAddress);
+        if (!cancelled) {
+          setModelPreferenceState(preference);
+        }
+      } finally {
+        if (!cancelled) {
+          setIsLoading(false);
+        }
+      }
+    };
+    loadPreference();
+    return () => {
+      cancelled = true;
+    };
+  }, [walletAddress, getModelPreference]);
+  return {
+    modelPreference,
+    isLoading,
+    getModelPreference,
+    setModelPreference,
+    deleteModelPreference
+  };
+}
+
+// src/lib/settingsStorage/schema.ts
+var import_watermelondb8 = require("@nozbe/watermelondb");
+var settingsStorageSchema = (0, import_watermelondb8.appSchema)({
+  version: 1,
+  tables: [
+    (0, import_watermelondb8.tableSchema)({
+      name: "modelPreferences",
+      columns: [
+        { name: "wallet_address", type: "string", isIndexed: true },
+        { name: "models", type: "string", isOptional: true }
+        // stored as JSON stringified ModelPreference[]
+      ]
+    })
+  ]
+});
+
+// src/lib/settingsStorage/models.ts
+var import_watermelondb9 = require("@nozbe/watermelondb");
+var ModelPreference = class extends import_watermelondb9.Model {
+  /** User's wallet address */
+  get walletAddress() {
+    return this._getRaw("wallet_address");
+  }
+  /** Preferred model identifier */
+  get models() {
+    const value = this._getRaw("models");
+    return value ? value : void 0;
+  }
+};
+ModelPreference.table = "modelPreferences";
+
+// src/react/usePdf.ts
+var import_react5 = require("react");
+
 // src/lib/pdf.ts
 var pdfjs = __toESM(require("pdfjs-dist"));
 pdfjs.GlobalWorkerOptions.workerSrc = `https://unpkg.com/pdfjs-dist@${pdfjs.version}/build/pdf.worker.min.mjs`;
@@ -3561,9 +3829,9 @@ async function convertPdfToImages(pdfDataUrl) {
 // src/react/usePdf.ts
 var PDF_MIME_TYPE = "application/pdf";
 function usePdf() {
-  const [isProcessing, setIsProcessing] = (0, import_react4.useState)(false);
-  const [error, setError] = (0, import_react4.useState)(null);
-  const extractPdfContext = (0, import_react4.useCallback)(
+  const [isProcessing, setIsProcessing] = (0, import_react5.useState)(false);
+  const [error, setError] = (0, import_react5.useState)(null);
+  const extractPdfContext = (0, import_react5.useCallback)(
     async (files) => {
       setIsProcessing(true);
       setError(null);
@@ -3610,12 +3878,12 @@ ${text}`;
 }
 
 // src/react/useOCR.ts
-var import_react5 = require("react");
+var import_react6 = require("react");
 var import_tesseract = __toESM(require("tesseract.js"));
 function useOCR() {
-  const [isProcessing, setIsProcessing] = (0, import_react5.useState)(false);
-  const [error, setError] = (0, import_react5.useState)(null);
-  const extractOCRContext = (0, import_react5.useCallback)(
+  const [isProcessing, setIsProcessing] = (0, import_react6.useState)(false);
+  const [error, setError] = (0, import_react6.useState)(null);
+  const extractOCRContext = (0, import_react6.useCallback)(
     async (files) => {
       setIsProcessing(true);
       setError(null);
@@ -3701,22 +3969,22 @@ ${text}`;
 }
 
 // src/react/useModels.ts
-var import_react6 = require("react");
+var import_react7 = require("react");
 function useModels(options = {}) {
   const { getToken, baseUrl = BASE_URL, provider, autoFetch = true } = options;
-  const [models, setModels] = (0, import_react6.useState)([]);
-  const [isLoading, setIsLoading] = (0, import_react6.useState)(false);
-  const [error, setError] = (0, import_react6.useState)(null);
-  const getTokenRef = (0, import_react6.useRef)(getToken);
-  const baseUrlRef = (0, import_react6.useRef)(baseUrl);
-  const providerRef = (0, import_react6.useRef)(provider);
-  const abortControllerRef = (0, import_react6.useRef)(null);
-  (0, import_react6.useEffect)(() => {
+  const [models, setModels] = (0, import_react7.useState)([]);
+  const [isLoading, setIsLoading] = (0, import_react7.useState)(false);
+  const [error, setError] = (0, import_react7.useState)(null);
+  const getTokenRef = (0, import_react7.useRef)(getToken);
+  const baseUrlRef = (0, import_react7.useRef)(baseUrl);
+  const providerRef = (0, import_react7.useRef)(provider);
+  const abortControllerRef = (0, import_react7.useRef)(null);
+  (0, import_react7.useEffect)(() => {
     getTokenRef.current = getToken;
     baseUrlRef.current = baseUrl;
     providerRef.current = provider;
   });
-  (0, import_react6.useEffect)(() => {
+  (0, import_react7.useEffect)(() => {
     return () => {
       if (abortControllerRef.current) {
         abortControllerRef.current.abort();
@@ -3724,7 +3992,7 @@ function useModels(options = {}) {
       }
     };
   }, []);
-  const fetchModels = (0, import_react6.useCallback)(async () => {
+  const fetchModels = (0, import_react7.useCallback)(async () => {
     if (abortControllerRef.current) {
       abortControllerRef.current.abort();
     }
@@ -3782,12 +4050,12 @@ function useModels(options = {}) {
       }
     }
   }, []);
-  const refetch = (0, import_react6.useCallback)(async () => {
+  const refetch = (0, import_react7.useCallback)(async () => {
     setModels([]);
     await fetchModels();
   }, [fetchModels]);
-  const hasFetchedRef = (0, import_react6.useRef)(false);
-  (0, import_react6.useEffect)(() => {
+  const hasFetchedRef = (0, import_react7.useRef)(false);
+  (0, import_react7.useEffect)(() => {
     if (autoFetch && !hasFetchedRef.current) {
       hasFetchedRef.current = true;
       fetchModels();
@@ -3805,15 +4073,15 @@ function useModels(options = {}) {
 }
 
 // src/react/useSearch.ts
-var import_react7 = require("react");
+var import_react8 = require("react");
 function useSearch(options = {}) {
   const { getToken, baseUrl = BASE_URL, onError } = options;
-  const [isLoading, setIsLoading] = (0, import_react7.useState)(false);
-  const [results, setResults] = (0, import_react7.useState)(null);
-  const [response, setResponse] = (0, import_react7.useState)(null);
-  const [error, setError] = (0, import_react7.useState)(null);
-  const abortControllerRef = (0, import_react7.useRef)(null);
-  (0, import_react7.useEffect)(() => {
+  const [isLoading, setIsLoading] = (0, import_react8.useState)(false);
+  const [results, setResults] = (0, import_react8.useState)(null);
+  const [response, setResponse] = (0, import_react8.useState)(null);
+  const [error, setError] = (0, import_react8.useState)(null);
+  const abortControllerRef = (0, import_react8.useRef)(null);
+  (0, import_react8.useEffect)(() => {
     return () => {
       if (abortControllerRef.current) {
         abortControllerRef.current.abort();
@@ -3821,7 +4089,7 @@ function useSearch(options = {}) {
       }
     };
   }, []);
-  const search = (0, import_react7.useCallback)(
+  const search = (0, import_react8.useCallback)(
     async (query, searchOptions = {}) => {
       if (abortControllerRef.current) {
         abortControllerRef.current.abort();
@@ -3889,12 +4157,12 @@ function useSearch(options = {}) {
 }
 
 // src/react/useImageGeneration.ts
-var import_react8 = require("react");
+var import_react9 = require("react");
 function useImageGeneration(options = {}) {
   const { getToken, baseUrl = BASE_URL, onFinish, onError } = options;
-  const [isLoading, setIsLoading] = (0, import_react8.useState)(false);
-  const abortControllerRef = (0, import_react8.useRef)(null);
-  (0, import_react8.useEffect)(() => {
+  const [isLoading, setIsLoading] = (0, import_react9.useState)(false);
+  const abortControllerRef = (0, import_react9.useRef)(null);
+  (0, import_react9.useEffect)(() => {
     return () => {
       if (abortControllerRef.current) {
         abortControllerRef.current.abort();
@@ -3902,13 +4170,13 @@ function useImageGeneration(options = {}) {
       }
     };
   }, []);
-  const stop = (0, import_react8.useCallback)(() => {
+  const stop = (0, import_react9.useCallback)(() => {
     if (abortControllerRef.current) {
       abortControllerRef.current.abort();
       abortControllerRef.current = null;
     }
   }, []);
-  const generateImage = (0, import_react8.useCallback)(
+  const generateImage = (0, import_react9.useCallback)(
     async (args) => {
       if (abortControllerRef.current) {
         abortControllerRef.current.abort();
@@ -4030,6 +4298,8 @@ var extractConversationContext = (messages, maxMessages = 3) => {
   ChatMessage,
   DEFAULT_TOOL_SELECTOR_MODEL,
   StoredMemoryModel,
+  StoredModelPreferenceModel,
+  chatStorageMigrations,
   chatStorageSchema,
   createMemoryContextSystemMessage,
   decryptData,
@@ -4045,6 +4315,7 @@ var extractConversationContext = (messages, maxMessages = 3) => {
   memoryStorageSchema,
   requestEncryptionKey,
   selectTool,
+  settingsStorageSchema,
   useChat,
   useChatStorage,
   useEncryption,
@@ -4053,5 +4324,6 @@ var extractConversationContext = (messages, maxMessages = 3) => {
   useModels,
   useOCR,
   usePdf,
-  useSearch
+  useSearch,
+  useSettings
 });