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

package/dist/react/index.mjs

@@ -1419,6 +1419,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",
@@ -1432,23 +1433,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);
@@ -1670,7 +1700,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) {
@@ -1759,6 +1790,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);
@@ -2049,6 +2081,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",
@@ -2120,8 +2191,9 @@ function useChatStorage(options) {
 
 // src/lib/chatStorage/schema.ts
 import { appSchema, tableSchema } from "@nozbe/watermelondb";
+import { schemaMigrations, addColumns } from "@nozbe/watermelondb/Schema/migrations";
 var chatStorageSchema = appSchema({
-  version: 1,
+  version: 2,
   tables: [
     tableSchema({
       name: "history",
@@ -2144,7 +2216,8 @@ var chatStorageSchema = 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 }
       ]
     }),
     tableSchema({
@@ -2159,6 +2232,21 @@ var chatStorageSchema = appSchema({
     })
   ]
 });
+var chatStorageMigrations = schemaMigrations({
+  migrations: [
+    {
+      toVersion: 2,
+      steps: [
+        addColumns({
+          table: "history",
+          columns: [
+            { name: "was_stopped", type: "boolean", isOptional: true }
+          ]
+        })
+      ]
+    }
+  ]
+});
 
 // src/lib/chatStorage/models.ts
 import { Model } from "@nozbe/watermelondb";
@@ -2242,6 +2330,10 @@ var Message = class extends 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 = {
@@ -3444,8 +3536,180 @@ var Memory = class extends Model2 {
 };
 Memory.table = "memories";
 
+// src/react/useSettings.ts
+import { useCallback as useCallback4, useState as useState4, useMemo as useMemo3, useEffect as useEffect2 } from "react";
+
+// src/lib/settingsStorage/operations.ts
+import { Q as Q3 } from "@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(Q3.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(Q3.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(Q3.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] = useState4(null);
+  const [isLoading, setIsLoading] = useState4(false);
+  const modelPreferencesCollection = useMemo3(
+    () => database.get("modelPreferences"),
+    [database]
+  );
+  const storageCtx = useMemo3(
+    () => ({
+      database,
+      modelPreferencesCollection
+    }),
+    [database, modelPreferencesCollection]
+  );
+  const getModelPreference = useCallback4(
+    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 = useCallback4(
+    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 = useCallback4(
+    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]
+  );
+  useEffect2(() => {
+    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
+import { appSchema as appSchema3, tableSchema as tableSchema3 } from "@nozbe/watermelondb";
+var settingsStorageSchema = appSchema3({
+  version: 1,
+  tables: [
+    tableSchema3({
+      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
+import { Model as Model3 } from "@nozbe/watermelondb";
+var ModelPreference = class extends Model3 {
+  /** 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
-import { useCallback as useCallback4, useState as useState4 } from "react";
+import { useCallback as useCallback5, useState as useState5 } from "react";
 
 // src/lib/pdf.ts
 import * as pdfjs from "pdfjs-dist";
@@ -3498,9 +3762,9 @@ async function convertPdfToImages(pdfDataUrl) {
 // src/react/usePdf.ts
 var PDF_MIME_TYPE = "application/pdf";
 function usePdf() {
-  const [isProcessing, setIsProcessing] = useState4(false);
-  const [error, setError] = useState4(null);
-  const extractPdfContext = useCallback4(
+  const [isProcessing, setIsProcessing] = useState5(false);
+  const [error, setError] = useState5(null);
+  const extractPdfContext = useCallback5(
     async (files) => {
       setIsProcessing(true);
       setError(null);
@@ -3547,12 +3811,12 @@ ${text}`;
 }
 
 // src/react/useOCR.ts
-import { useCallback as useCallback5, useState as useState5 } from "react";
+import { useCallback as useCallback6, useState as useState6 } from "react";
 import Tesseract from "tesseract.js";
 function useOCR() {
-  const [isProcessing, setIsProcessing] = useState5(false);
-  const [error, setError] = useState5(null);
-  const extractOCRContext = useCallback5(
+  const [isProcessing, setIsProcessing] = useState6(false);
+  const [error, setError] = useState6(null);
+  const extractOCRContext = useCallback6(
     async (files) => {
       setIsProcessing(true);
       setError(null);
@@ -3638,22 +3902,22 @@ ${text}`;
 }
 
 // src/react/useModels.ts
-import { useCallback as useCallback6, useEffect as useEffect2, useRef as useRef3, useState as useState6 } from "react";
+import { useCallback as useCallback7, useEffect as useEffect3, useRef as useRef3, useState as useState7 } from "react";
 function useModels(options = {}) {
   const { getToken, baseUrl = BASE_URL, provider, autoFetch = true } = options;
-  const [models, setModels] = useState6([]);
-  const [isLoading, setIsLoading] = useState6(false);
-  const [error, setError] = useState6(null);
+  const [models, setModels] = useState7([]);
+  const [isLoading, setIsLoading] = useState7(false);
+  const [error, setError] = useState7(null);
   const getTokenRef = useRef3(getToken);
   const baseUrlRef = useRef3(baseUrl);
   const providerRef = useRef3(provider);
   const abortControllerRef = useRef3(null);
-  useEffect2(() => {
+  useEffect3(() => {
     getTokenRef.current = getToken;
     baseUrlRef.current = baseUrl;
     providerRef.current = provider;
   });
-  useEffect2(() => {
+  useEffect3(() => {
     return () => {
       if (abortControllerRef.current) {
         abortControllerRef.current.abort();
@@ -3661,7 +3925,7 @@ function useModels(options = {}) {
       }
     };
   }, []);
-  const fetchModels = useCallback6(async () => {
+  const fetchModels = useCallback7(async () => {
     if (abortControllerRef.current) {
       abortControllerRef.current.abort();
     }
@@ -3719,12 +3983,12 @@ function useModels(options = {}) {
       }
     }
   }, []);
-  const refetch = useCallback6(async () => {
+  const refetch = useCallback7(async () => {
     setModels([]);
     await fetchModels();
   }, [fetchModels]);
   const hasFetchedRef = useRef3(false);
-  useEffect2(() => {
+  useEffect3(() => {
     if (autoFetch && !hasFetchedRef.current) {
       hasFetchedRef.current = true;
       fetchModels();
@@ -3742,15 +4006,15 @@ function useModels(options = {}) {
 }
 
 // src/react/useSearch.ts
-import { useCallback as useCallback7, useEffect as useEffect3, useRef as useRef4, useState as useState7 } from "react";
+import { useCallback as useCallback8, useEffect as useEffect4, useRef as useRef4, useState as useState8 } from "react";
 function useSearch(options = {}) {
   const { getToken, baseUrl = BASE_URL, onError } = options;
-  const [isLoading, setIsLoading] = useState7(false);
-  const [results, setResults] = useState7(null);
-  const [response, setResponse] = useState7(null);
-  const [error, setError] = useState7(null);
+  const [isLoading, setIsLoading] = useState8(false);
+  const [results, setResults] = useState8(null);
+  const [response, setResponse] = useState8(null);
+  const [error, setError] = useState8(null);
   const abortControllerRef = useRef4(null);
-  useEffect3(() => {
+  useEffect4(() => {
     return () => {
       if (abortControllerRef.current) {
         abortControllerRef.current.abort();
@@ -3758,7 +4022,7 @@ function useSearch(options = {}) {
       }
     };
   }, []);
-  const search = useCallback7(
+  const search = useCallback8(
     async (query, searchOptions = {}) => {
       if (abortControllerRef.current) {
         abortControllerRef.current.abort();
@@ -3826,12 +4090,12 @@ function useSearch(options = {}) {
 }
 
 // src/react/useImageGeneration.ts
-import { useCallback as useCallback8, useEffect as useEffect4, useRef as useRef5, useState as useState8 } from "react";
+import { useCallback as useCallback9, useEffect as useEffect5, useRef as useRef5, useState as useState9 } from "react";
 function useImageGeneration(options = {}) {
   const { getToken, baseUrl = BASE_URL, onFinish, onError } = options;
-  const [isLoading, setIsLoading] = useState8(false);
+  const [isLoading, setIsLoading] = useState9(false);
   const abortControllerRef = useRef5(null);
-  useEffect4(() => {
+  useEffect5(() => {
     return () => {
       if (abortControllerRef.current) {
         abortControllerRef.current.abort();
@@ -3839,13 +4103,13 @@ function useImageGeneration(options = {}) {
       }
     };
   }, []);
-  const stop = useCallback8(() => {
+  const stop = useCallback9(() => {
     if (abortControllerRef.current) {
       abortControllerRef.current.abort();
       abortControllerRef.current = null;
     }
   }, []);
-  const generateImage = useCallback8(
+  const generateImage = useCallback9(
     async (args) => {
       if (abortControllerRef.current) {
         abortControllerRef.current.abort();
@@ -3966,6 +4230,8 @@ export {
   Message as ChatMessage,
   DEFAULT_TOOL_SELECTOR_MODEL,
   Memory as StoredMemoryModel,
+  ModelPreference as StoredModelPreferenceModel,
+  chatStorageMigrations,
   chatStorageSchema,
   createMemoryContextSystemMessage,
   decryptData,
@@ -3981,6 +4247,7 @@ export {
   memoryStorageSchema,
   requestEncryptionKey,
   selectTool,
+  settingsStorageSchema,
   useChat,
   useChatStorage,
   useEncryption,
@@ -3989,5 +4256,6 @@ export {
   useModels,
   useOCR,
   usePdf,
-  useSearch
+  useSearch,
+  useSettings
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@reverbia/sdk",
-  "version": "1.0.0-next.20251215193957",
+  "version": "1.0.0-next.20251217134403",
   "description": "",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",