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

package/dist/expo/index.cjs

@@ -23,6 +23,7 @@ __export(index_exports, {
   ChatConversation: () => Conversation,
   ChatMessage: () => Message,
   StoredMemoryModel: () => Memory,
+  chatStorageMigrations: () => chatStorageMigrations,
   chatStorageSchema: () => chatStorageSchema,
   generateCompositeKey: () => generateCompositeKey,
   generateConversationId: () => generateConversationId,
@@ -352,7 +353,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) {
@@ -441,6 +443,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);
@@ -663,6 +666,44 @@ 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
+              userMessage: storedUserMessage,
+              assistantMessage: storedAssistantMessage2
+            };
+          } catch {
+          }
+        }
         return {
           data: null,
           error: result.error || "No response data received",
@@ -2692,8 +2733,9 @@ function useMemoryStorage(options) {
 
 // src/lib/chatStorage/schema.ts
 var import_watermelondb3 = require("@nozbe/watermelondb");
+var import_migrations = require("@nozbe/watermelondb/Schema/migrations");
 var chatStorageSchema = (0, import_watermelondb3.appSchema)({
-  version: 1,
+  version: 2,
   tables: [
     (0, import_watermelondb3.tableSchema)({
       name: "history",
@@ -2716,7 +2758,8 @@ var chatStorageSchema = (0, import_watermelondb3.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_watermelondb3.tableSchema)({
@@ -2731,6 +2774,21 @@ var chatStorageSchema = (0, import_watermelondb3.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_watermelondb4 = require("@nozbe/watermelondb");
@@ -2814,6 +2872,10 @@ var Message = class extends import_watermelondb4.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 = {
@@ -2957,6 +3019,7 @@ Memory.table = "memories";
   ChatConversation,
   ChatMessage,
   StoredMemoryModel,
+  chatStorageMigrations,
   chatStorageSchema,
   generateCompositeKey,
   generateConversationId,

package/dist/expo/index.d.mts

@@ -1,4 +1,5 @@
 import { Database, Model } from '@nozbe/watermelondb';
+import * as _nozbe_watermelondb_Schema_migrations from '@nozbe/watermelondb/Schema/migrations';
 import * as _nozbe_watermelondb_Schema from '@nozbe/watermelondb/Schema';
 import { Associations } from '@nozbe/watermelondb/Model';
 
@@ -480,6 +481,8 @@ interface StoredMessage {
     sources?: SearchSource[];
     /** Response time in seconds */
     responseDuration?: number;
+    /** Whether the message generation was stopped by the user */
+    wasStopped?: boolean;
 }
 /**
  * Stored conversation record
@@ -529,6 +532,8 @@ interface CreateMessageOptions {
     vector?: number[];
     /** Model used to generate the embedding */
     embeddingModel?: string;
+    /** Whether the message generation was stopped by the user */
+    wasStopped?: boolean;
 }
 /**
  * Options for creating a new conversation
@@ -1105,6 +1110,15 @@ declare const chatStorageSchema: Readonly<{
     tables: _nozbe_watermelondb_Schema.TableMap;
     unsafeSql?: (_: string, __: _nozbe_watermelondb_Schema.AppSchemaUnsafeSqlKind) => string;
 }>;
+/**
+ * Schema migrations
+ */
+declare const chatStorageMigrations: Readonly<{
+    validated: true;
+    minVersion: _nozbe_watermelondb_Schema.SchemaVersion;
+    maxVersion: _nozbe_watermelondb_Schema.SchemaVersion;
+    sortedMigrations: _nozbe_watermelondb_Schema_migrations.Migration[];
+}>;
 
 /**
  * Message model representing a single chat message
@@ -1141,6 +1155,8 @@ declare class Message extends Model {
     get sources(): SearchSource[] | undefined;
     /** Response time in seconds */
     get responseDuration(): number | undefined;
+    /** Whether the message generation was stopped by the user */
+    get wasStopped(): boolean;
 }
 /**
  * Conversation model representing conversation metadata
@@ -1210,4 +1226,4 @@ declare class Memory extends Model {
     get isDeleted(): boolean;
 }
 
-export { Conversation as ChatConversation, Message as ChatMessage, type ChatRole, type CreateConversationOptions, type CreateMemoryOptions, type CreateMessageOptions, type FileMetadata, type MemoryItem, type MemoryType, type SearchSource, type SendMessageWithStorageArgs, type SendMessageWithStorageResult, type ChatCompletionUsage as StoredChatCompletionUsage, type StoredConversation, type StoredMemory, Memory as StoredMemoryModel, type StoredMemoryWithSimilarity, type StoredMessage, type StoredMessageWithSimilarity, type UpdateMemoryOptions, type UseChatStorageOptions, type UseChatStorageResult, type UseMemoryStorageOptions, type UseMemoryStorageResult, type UseModelsOptions, type UseModelsResult, chatStorageSchema, generateCompositeKey, generateConversationId, generateUniqueKey, memoryStorageSchema, useChat, useChatStorage, useImageGeneration, useMemoryStorage, useModels };
+export { Conversation as ChatConversation, Message as ChatMessage, type ChatRole, type CreateConversationOptions, type CreateMemoryOptions, type CreateMessageOptions, type FileMetadata, type MemoryItem, type MemoryType, type SearchSource, type SendMessageWithStorageArgs, type SendMessageWithStorageResult, type ChatCompletionUsage as StoredChatCompletionUsage, type StoredConversation, type StoredMemory, Memory as StoredMemoryModel, type StoredMemoryWithSimilarity, type StoredMessage, type StoredMessageWithSimilarity, type UpdateMemoryOptions, type UseChatStorageOptions, type UseChatStorageResult, type UseMemoryStorageOptions, type UseMemoryStorageResult, type UseModelsOptions, type UseModelsResult, chatStorageMigrations, chatStorageSchema, generateCompositeKey, generateConversationId, generateUniqueKey, memoryStorageSchema, useChat, useChatStorage, useImageGeneration, useMemoryStorage, useModels };

package/dist/expo/index.d.ts

@@ -1,4 +1,5 @@
 import { Database, Model } from '@nozbe/watermelondb';
+import * as _nozbe_watermelondb_Schema_migrations from '@nozbe/watermelondb/Schema/migrations';
 import * as _nozbe_watermelondb_Schema from '@nozbe/watermelondb/Schema';
 import { Associations } from '@nozbe/watermelondb/Model';
 
@@ -480,6 +481,8 @@ interface StoredMessage {
     sources?: SearchSource[];
     /** Response time in seconds */
     responseDuration?: number;
+    /** Whether the message generation was stopped by the user */
+    wasStopped?: boolean;
 }
 /**
  * Stored conversation record
@@ -529,6 +532,8 @@ interface CreateMessageOptions {
     vector?: number[];
     /** Model used to generate the embedding */
     embeddingModel?: string;
+    /** Whether the message generation was stopped by the user */
+    wasStopped?: boolean;
 }
 /**
  * Options for creating a new conversation
@@ -1105,6 +1110,15 @@ declare const chatStorageSchema: Readonly<{
     tables: _nozbe_watermelondb_Schema.TableMap;
     unsafeSql?: (_: string, __: _nozbe_watermelondb_Schema.AppSchemaUnsafeSqlKind) => string;
 }>;
+/**
+ * Schema migrations
+ */
+declare const chatStorageMigrations: Readonly<{
+    validated: true;
+    minVersion: _nozbe_watermelondb_Schema.SchemaVersion;
+    maxVersion: _nozbe_watermelondb_Schema.SchemaVersion;
+    sortedMigrations: _nozbe_watermelondb_Schema_migrations.Migration[];
+}>;
 
 /**
  * Message model representing a single chat message
@@ -1141,6 +1155,8 @@ declare class Message extends Model {
     get sources(): SearchSource[] | undefined;
     /** Response time in seconds */
     get responseDuration(): number | undefined;
+    /** Whether the message generation was stopped by the user */
+    get wasStopped(): boolean;
 }
 /**
  * Conversation model representing conversation metadata
@@ -1210,4 +1226,4 @@ declare class Memory extends Model {
     get isDeleted(): boolean;
 }
 
-export { Conversation as ChatConversation, Message as ChatMessage, type ChatRole, type CreateConversationOptions, type CreateMemoryOptions, type CreateMessageOptions, type FileMetadata, type MemoryItem, type MemoryType, type SearchSource, type SendMessageWithStorageArgs, type SendMessageWithStorageResult, type ChatCompletionUsage as StoredChatCompletionUsage, type StoredConversation, type StoredMemory, Memory as StoredMemoryModel, type StoredMemoryWithSimilarity, type StoredMessage, type StoredMessageWithSimilarity, type UpdateMemoryOptions, type UseChatStorageOptions, type UseChatStorageResult, type UseMemoryStorageOptions, type UseMemoryStorageResult, type UseModelsOptions, type UseModelsResult, chatStorageSchema, generateCompositeKey, generateConversationId, generateUniqueKey, memoryStorageSchema, useChat, useChatStorage, useImageGeneration, useMemoryStorage, useModels };
+export { Conversation as ChatConversation, Message as ChatMessage, type ChatRole, type CreateConversationOptions, type CreateMemoryOptions, type CreateMessageOptions, type FileMetadata, type MemoryItem, type MemoryType, type SearchSource, type SendMessageWithStorageArgs, type SendMessageWithStorageResult, type ChatCompletionUsage as StoredChatCompletionUsage, type StoredConversation, type StoredMemory, Memory as StoredMemoryModel, type StoredMemoryWithSimilarity, type StoredMessage, type StoredMessageWithSimilarity, type UpdateMemoryOptions, type UseChatStorageOptions, type UseChatStorageResult, type UseMemoryStorageOptions, type UseMemoryStorageResult, type UseModelsOptions, type UseModelsResult, chatStorageMigrations, chatStorageSchema, generateCompositeKey, generateConversationId, generateUniqueKey, memoryStorageSchema, useChat, useChatStorage, useImageGeneration, useMemoryStorage, useModels };

package/dist/expo/index.mjs

@@ -314,7 +314,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) {
@@ -403,6 +404,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);
@@ -625,6 +627,44 @@ 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
+              userMessage: storedUserMessage,
+              assistantMessage: storedAssistantMessage2
+            };
+          } catch {
+          }
+        }
         return {
           data: null,
           error: result.error || "No response data received",
@@ -2654,8 +2694,9 @@ function useMemoryStorage(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",
@@ -2678,7 +2719,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({
@@ -2693,6 +2735,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";
@@ -2776,6 +2833,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 = {
@@ -2918,6 +2979,7 @@ export {
   Conversation as ChatConversation,
   Message as ChatMessage,
   Memory as StoredMemoryModel,
+  chatStorageMigrations,
   chatStorageSchema,
   generateCompositeKey,
   generateConversationId,