expo-ai-kit 0.2.0 → 0.3.0

package/README.md

@@ -14,6 +14,15 @@ On-device AI for Expo apps. Run language models locally—no API keys, no cloud,
 | iOS 26+ | [Apple Foundation Models](https://developer.apple.com/documentation/FoundationModels) |
 | Android (supported devices) | [ML Kit Prompt API](https://developers.google.com/ml-kit/genai#prompt-device) |
 
+### Downloadable Models (Gemma 4)
+
+| Platform | Status |
+|----------|--------|
+| Android | Gemma 4 E2B (2.3B) and E4B (4.5B) via [LiteRT-LM](https://ai.google.dev/edge/litert-lm) |
+| iOS | Coming soon — waiting for LiteRT-LM Swift APIs from Google |
+
+> **Note:** iOS downloadable model support (Gemma 4 E2B/E4B) is planned for a future release. We are waiting for Google to ship native Swift APIs for LiteRT-LM. Built-in Apple Foundation Models work on iOS 26+ today.
+
 ### Unsupported
 
 | Platform | Fallback Behavior |

package/android/build.gradle

@@ -43,5 +43,6 @@ android {
 }
 
 dependencies {
-  implementation "com.google.mlkit:genai-prompt:1.0.0-alpha1"
+  implementation "com.google.mlkit:genai-prompt:1.0.0-beta2"
+  implementation "com.google.ai.edge.litertlm:litertlm-android:+"
 }

package/android/src/main/java/expo/modules/aikit/ExpoAiKitModule.kt

@@ -1,5 +1,7 @@
 package expo.modules.aikit
 
+import android.app.ActivityManager
+import android.content.Context
 import expo.modules.kotlin.modules.Module
 import expo.modules.kotlin.modules.ModuleDefinition
 import expo.modules.kotlin.functions.Coroutine
@@ -11,15 +13,28 @@ import kotlinx.coroutines.cancel
 
 class ExpoAiKitModule : Module() {
 
+  // Existing ML Kit client -- unchanged
   private val promptClient by lazy { PromptApiClient() }
+
+  // Gemma client -- lazy-initialized with app context
+  private val gemmaClient by lazy {
+    GemmaInferenceClient(appContext.reactContext ?: throw RuntimeException("React context not available"))
+  }
+
   private val activeStreamJobs = mutableMapOf<String, Job>()
   private val streamScope = CoroutineScope(Dispatchers.IO)
 
+  // Active model routing: "mlkit" (default) or a downloadable model ID
+  private var activeModelId: String = "mlkit"
+
   override fun definition() = ModuleDefinition {
     Name("ExpoAiKit")
 
-    // Declare events that can be sent to JavaScript
-    Events("onStreamToken")
+    Events("onStreamToken", "onDownloadProgress", "onModelStateChange")
+
+    // ==================================================================
+    // Existing inference API -- ML Kit path completely untouched
+    // ==================================================================
 
     Function("isAvailable") {
       promptClient.isAvailableBlocking()
@@ -42,7 +57,12 @@ class ExpoAiKitModule : Module() {
           "$role: $content"
         } + "\nASSISTANT:"
 
-      val text = promptClient.generateText(conversationPrompt, systemPrompt)
+      // Route to active model
+      val text = if (activeModelId == "mlkit") {
+        promptClient.generateText(conversationPrompt, systemPrompt)
+      } else {
+        gemmaClient.generateText(conversationPrompt, systemPrompt)
+      }
       mapOf("text" to text)
     }
 
@@ -65,7 +85,7 @@ class ExpoAiKitModule : Module() {
 
       // Launch streaming in a coroutine that can be cancelled
       val job = streamScope.launch {
-        promptClient.generateTextStream(conversationPrompt, systemPrompt) { token, accumulatedText, isDone ->
+        val streamCallback = { token: String, accumulatedText: String, isDone: Boolean ->
           sendEvent("onStreamToken", mapOf(
             "sessionId" to sessionId,
             "token" to token,
@@ -77,6 +97,13 @@ class ExpoAiKitModule : Module() {
             activeStreamJobs.remove(sessionId)
           }
         }
+
+        // Route to active model
+        if (activeModelId == "mlkit") {
+          promptClient.generateTextStream(conversationPrompt, systemPrompt, streamCallback)
+        } else {
+          gemmaClient.generateTextStream(conversationPrompt, systemPrompt, streamCallback)
+        }
       }
 
       activeStreamJobs[sessionId] = job
@@ -86,5 +113,157 @@ class ExpoAiKitModule : Module() {
       activeStreamJobs[sessionId]?.cancel()
       activeStreamJobs.remove(sessionId)
     }
+
+    // ==================================================================
+    // Model discovery
+    // ==================================================================
+
+    Function("getBuiltInModels") {
+      listOf(
+        mapOf(
+          "id" to "mlkit",
+          "name" to "ML Kit Prompt API",
+          "available" to promptClient.isAvailableBlocking(),
+          "platform" to "android",
+          // ML Kit doesn't expose a context window; use a reasonable default
+          "contextWindow" to 4096
+        )
+      )
+    }
+
+    Function("getDownloadableModelStatus") { modelId: String ->
+      // Status reflects runtime state: "ready" if loaded in memory,
+      // "not-downloaded" otherwise (even if file is on disk -- setModel
+      // is the gatekeeper that transitions through loading -> ready).
+      when {
+        gemmaClient.getLoadedModelId() == modelId && gemmaClient.isModelLoaded() -> "ready"
+        else -> "not-downloaded"
+      }
+    }
+
+    Function("getDeviceRamBytes") {
+      val activityManager = appContext.reactContext?.getSystemService(Context.ACTIVITY_SERVICE) as? ActivityManager
+      if (activityManager != null) {
+        val memInfo = ActivityManager.MemoryInfo()
+        activityManager.getMemoryInfo(memInfo)
+        memInfo.totalMem
+      } else {
+        0L
+      }
+    }
+
+    // ==================================================================
+    // Model selection & memory management
+    // ==================================================================
+
+    AsyncFunction("setModel") Coroutine { modelId: String ->
+      if (modelId == "mlkit") {
+        // Switch to built-in: unload any Gemma model
+        if (gemmaClient.isModelLoaded()) {
+          gemmaClient.unloadModel()
+          val previousId = activeModelId
+          if (previousId != "mlkit") {
+            sendEvent("onModelStateChange", mapOf(
+              "modelId" to previousId,
+              "status" to "not-downloaded"
+            ))
+          }
+        }
+        activeModelId = "mlkit"
+        return@Coroutine
+      }
+
+      // Downloadable model: verify file exists
+      if (!gemmaClient.isModelFileDownloaded(modelId)) {
+        throw RuntimeException("MODEL_NOT_DOWNLOADED:$modelId:Model file not found on disk")
+      }
+
+      // Emit loading state
+      sendEvent("onModelStateChange", mapOf(
+        "modelId" to modelId,
+        "status" to "loading"
+      ))
+
+      try {
+        val modelPath = gemmaClient.getModelFilePath(modelId)
+        gemmaClient.loadModel(modelId, modelPath)
+        activeModelId = modelId
+
+        // Emit ready state
+        sendEvent("onModelStateChange", mapOf(
+          "modelId" to modelId,
+          "status" to "ready"
+        ))
+      } catch (e: Exception) {
+        // Emit failure -- revert status
+        sendEvent("onModelStateChange", mapOf(
+          "modelId" to modelId,
+          "status" to "not-downloaded"
+        ))
+        throw e
+      }
+    }
+
+    Function("getActiveModel") {
+      activeModelId
+    }
+
+    AsyncFunction("unloadModel") Coroutine {
+      if (activeModelId != "mlkit" && gemmaClient.isModelLoaded()) {
+        val previousId = activeModelId
+        gemmaClient.unloadModel()
+        activeModelId = "mlkit"
+        sendEvent("onModelStateChange", mapOf(
+          "modelId" to previousId,
+          "status" to "not-downloaded"
+        ))
+      }
+    }
+
+    // ==================================================================
+    // Model lifecycle (downloadable models only)
+    // ==================================================================
+
+    AsyncFunction("downloadModel") Coroutine { modelId: String, url: String, sha256: String ->
+      sendEvent("onModelStateChange", mapOf(
+        "modelId" to modelId,
+        "status" to "downloading"
+      ))
+
+      try {
+        gemmaClient.downloadModelFile(modelId, url, sha256) { bytesRead, totalBytes ->
+          sendEvent("onDownloadProgress", mapOf(
+            "modelId" to modelId,
+            "progress" to if (totalBytes > 0) bytesRead.toDouble() / totalBytes else 0.0
+          ))
+        }
+
+        // Download complete -- file is on disk but not loaded
+        sendEvent("onModelStateChange", mapOf(
+          "modelId" to modelId,
+          "status" to "not-downloaded"
+        ))
+      } catch (e: Exception) {
+        sendEvent("onModelStateChange", mapOf(
+          "modelId" to modelId,
+          "status" to "not-downloaded"
+        ))
+        throw e
+      }
+    }
+
+    AsyncFunction("deleteModel") Coroutine { modelId: String ->
+      // If this model is active, switch back to mlkit first
+      if (activeModelId == modelId) {
+        activeModelId = "mlkit"
+      }
+
+      gemmaClient.deleteModelFile(modelId)
+
+      sendEvent("onModelStateChange", mapOf(
+        "modelId" to modelId,
+        "status" to "not-downloaded"
+      ))
+    }
   }
 }

package/android/src/main/java/expo/modules/aikit/GemmaInferenceClient.kt

@@ -0,0 +1,326 @@
+package expo.modules.aikit
+
+import android.content.Context
+import com.google.ai.edge.litertlm.Engine
+import com.google.ai.edge.litertlm.EngineConfig
+import com.google.ai.edge.litertlm.Conversation
+import com.google.ai.edge.litertlm.Backend
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.sync.Mutex
+import kotlinx.coroutines.sync.withLock
+import kotlinx.coroutines.withContext
+import java.io.File
+import java.io.FileInputStream
+import java.io.FileOutputStream
+import java.io.IOException
+import java.net.HttpURLConnection
+import java.net.URL
+import java.security.MessageDigest
+
+/**
+ * Wrapper around LiteRT-LM Engine for Gemma 4 models.
+ *
+ * Concurrency model:
+ * - A Mutex guards all state transitions (load, unload, inference).
+ * - sendMessage/startStreaming block on the mutex if a load is in progress.
+ * - deleteModel waits for inference to finish, then unloads and deletes.
+ * - A separate isDownloading flag prevents concurrent downloads (checked before
+ *   the long-running download, not inside the mutex).
+ */
+class GemmaInferenceClient(private val context: Context) {
+
+  private val mutex = Mutex()
+  private var engine: Engine? = null
+  private var conversation: Conversation? = null
+  private var loadedModelId: String? = null
+
+  @Volatile
+  private var isDownloading = false
+
+  // -------------------------------------------------------------------------
+  // Model lifecycle
+  // -------------------------------------------------------------------------
+
+  /**
+   * Load a model into memory using LiteRT-LM Engine.
+   * Unloads any previously loaded model first.
+   * Caller is responsible for emitting onModelStateChange events.
+   */
+  suspend fun loadModel(modelId: String, modelPath: String) = mutex.withLock {
+    // Unload previous model if different
+    if (loadedModelId != null && loadedModelId != modelId) {
+      conversation?.close()
+      engine?.close()
+      conversation = null
+      engine = null
+      loadedModelId = null
+    }
+
+    if (loadedModelId == modelId && engine != null) {
+      return@withLock // Already loaded
+    }
+
+    try {
+      withContext(Dispatchers.IO) {
+        val engineConfig = EngineConfig(
+          modelPath = modelPath,
+          backend = Backend.GPU()
+        )
+        val newEngine = Engine(engineConfig)
+        newEngine.initialize()
+        val newConversation = newEngine.createConversation()
+
+        engine = newEngine
+        conversation = newConversation
+        loadedModelId = modelId
+      }
+    } catch (e: OutOfMemoryError) {
+      conversation?.close()
+      engine?.close()
+      conversation = null
+      engine = null
+      loadedModelId = null
+      throw RuntimeException("INFERENCE_OOM:$modelId:Device does not have enough memory to load model")
+    } catch (e: Exception) {
+      conversation?.close()
+      engine?.close()
+      conversation = null
+      engine = null
+      loadedModelId = null
+      throw RuntimeException("MODEL_LOAD_FAILED:$modelId:${e.message}")
+    }
+  }
+
+  /**
+   * Unload the current model from memory.
+   */
+  suspend fun unloadModel() = mutex.withLock {
+    conversation?.close()
+    engine?.close()
+    conversation = null
+    engine = null
+    loadedModelId = null
+  }
+
+  fun getLoadedModelId(): String? = loadedModelId
+
+  fun isModelLoaded(): Boolean = engine != null
+
+  // -------------------------------------------------------------------------
+  // Inference
+  // -------------------------------------------------------------------------
+
+  /**
+   * Generate a complete response. Blocks until done.
+   * The mutex ensures this cannot run concurrently with load/unload.
+   */
+  suspend fun generateText(prompt: String, systemPrompt: String): String = mutex.withLock {
+    val conv = conversation
+      ?: throw RuntimeException("MODEL_NOT_DOWNLOADED:${loadedModelId ?: "unknown"}:No model loaded")
+
+    val fullPrompt = buildFullPrompt(prompt, systemPrompt)
+
+    try {
+      withContext(Dispatchers.IO) {
+        conv.sendMessage(contents = fullPrompt).toString()
+      }
+    } catch (e: OutOfMemoryError) {
+      throw RuntimeException("INFERENCE_OOM:${loadedModelId ?: "unknown"}:Out of memory during inference")
+    } catch (e: Exception) {
+      throw RuntimeException("INFERENCE_FAILED:${loadedModelId ?: "unknown"}:${e.message}")
+    }
+  }
+
+  /**
+   * Generate a streaming response. The onChunk callback receives
+   * (token=delta, accumulatedText=full, isDone) matching the PromptApiClient contract.
+   *
+   * LiteRT-LM's sendMessageAsync() returns a Flow<Message>. Each emission
+   * contains accumulated text, so we diff against previousText to extract
+   * the delta token.
+   */
+  suspend fun generateTextStream(
+    prompt: String,
+    systemPrompt: String,
+    onChunk: (token: String, accumulatedText: String, isDone: Boolean) -> Unit
+  ) = mutex.withLock {
+    val conv = conversation
+      ?: throw RuntimeException("MODEL_NOT_DOWNLOADED:${loadedModelId ?: "unknown"}:No model loaded")
+
+    val fullPrompt = buildFullPrompt(prompt, systemPrompt)
+
+    try {
+      withContext(Dispatchers.IO) {
+        var previousText = ""
+
+        conv.sendMessageAsync(contents = fullPrompt).collect { message ->
+          val accumulated = message.toString()
+          val token = if (accumulated.length > previousText.length) {
+            accumulated.substring(previousText.length)
+          } else {
+            ""
+          }
+          previousText = accumulated
+          onChunk(token, accumulated, false)
+        }
+
+        // Final done event for consistency with PromptApiClient
+        onChunk("", previousText, true)
+      }
+    } catch (e: OutOfMemoryError) {
+      throw RuntimeException("INFERENCE_OOM:${loadedModelId ?: "unknown"}:Out of memory during inference")
+    } catch (e: Exception) {
+      throw RuntimeException("INFERENCE_FAILED:${loadedModelId ?: "unknown"}:${e.message}")
+    }
+  }
+
+  // -------------------------------------------------------------------------
+  // Download
+  // -------------------------------------------------------------------------
+
+  /**
+   * Download a model file with progress reporting.
+   * Prevents concurrent downloads. On failure, deletes partial files.
+   *
+   * Strategy: restart from scratch on failure (no HTTP Range resumption).
+   * Downloads to a .tmp file, atomically renames on success.
+   */
+  suspend fun downloadModelFile(
+    modelId: String,
+    url: String,
+    sha256: String,
+    onProgress: (bytesRead: Long, totalBytes: Long) -> Unit
+  ) {
+    if (isDownloading) {
+      throw RuntimeException("DOWNLOAD_FAILED:$modelId:Download already in progress")
+    }
+    isDownloading = true
+
+    try {
+      withContext(Dispatchers.IO) {
+        val modelsDir = File(context.filesDir, "models")
+        modelsDir.mkdirs()
+
+        val targetFile = File(modelsDir, "$modelId.litertlm")
+        val tempFile = File(modelsDir, "$modelId.litertlm.tmp")
+
+        try {
+          val connection = URL(url).openConnection() as HttpURLConnection
+          connection.connectTimeout = 30_000
+          connection.readTimeout = 30_000
+          connection.connect()
+
+          if (connection.responseCode != HttpURLConnection.HTTP_OK) {
+            throw IOException("HTTP ${connection.responseCode}: ${connection.responseMessage}")
+          }
+
+          val totalBytes = connection.contentLengthLong
+          var bytesRead = 0L
+
+          connection.inputStream.use { input ->
+            FileOutputStream(tempFile).use { output ->
+              val buffer = ByteArray(8192)
+              var read: Int
+              while (input.read(buffer).also { read = it } != -1) {
+                output.write(buffer, 0, read)
+                bytesRead += read
+                if (totalBytes > 0) {
+                  onProgress(bytesRead, totalBytes)
+                }
+              }
+            }
+          }
+
+          // Verify SHA256 if provided
+          if (sha256.isNotEmpty()) {
+            val actualHash = computeSha256(tempFile)
+            if (!actualHash.equals(sha256, ignoreCase = true)) {
+              tempFile.delete()
+              throw RuntimeException("DOWNLOAD_CORRUPT:$modelId:SHA256 mismatch: expected $sha256, got $actualHash")
+            }
+          }
+
+          // Atomic rename
+          if (!tempFile.renameTo(targetFile)) {
+            tempFile.delete()
+            throw IOException("Failed to rename temp file to target")
+          }
+        } catch (e: Exception) {
+          // Always clean up partial file
+          tempFile.delete()
+          when {
+            e is RuntimeException && e.message?.startsWith("DOWNLOAD_CORRUPT") == true -> throw e
+            context.filesDir.freeSpace < 100_000_000 ->
+              throw RuntimeException("DOWNLOAD_STORAGE_FULL:$modelId:Insufficient disk space")
+            else ->
+              throw RuntimeException("DOWNLOAD_FAILED:$modelId:${e.message}")
+          }
+        }
+      }
+    } finally {
+      isDownloading = false
+    }
+  }
+
+  /**
+   * Delete a model file from disk. If the model is loaded, unloads it first.
+   */
+  suspend fun deleteModelFile(modelId: String) = mutex.withLock {
+    // Unload if this model is currently loaded
+    if (loadedModelId == modelId) {
+      conversation?.close()
+      engine?.close()
+      conversation = null
+      engine = null
+      loadedModelId = null
+    }
+
+    val modelFile = File(context.filesDir, "models/$modelId.litertlm")
+    if (modelFile.exists()) {
+      modelFile.delete()
+    }
+    // Also clean up any partial downloads
+    val tempFile = File(context.filesDir, "models/$modelId.litertlm.tmp")
+    if (tempFile.exists()) {
+      tempFile.delete()
+    }
+  }
+
+  /**
+   * Check if a model file exists on disk.
+   */
+  fun isModelFileDownloaded(modelId: String): Boolean {
+    return File(context.filesDir, "models/$modelId.litertlm").exists()
+  }
+
+  /**
+   * Get the file path for a downloaded model.
+   */
+  fun getModelFilePath(modelId: String): String {
+    return File(context.filesDir, "models/$modelId.litertlm").absolutePath
+  }
+
+  // -------------------------------------------------------------------------
+  // Private helpers
+  // -------------------------------------------------------------------------
+
+  private fun buildFullPrompt(prompt: String, systemPrompt: String): String {
+    return if (systemPrompt.isNotBlank()) {
+      "$systemPrompt\n\n$prompt"
+    } else {
+      prompt
+    }
+  }
+
+  private fun computeSha256(file: File): String {
+    val digest = MessageDigest.getInstance("SHA-256")
+    FileInputStream(file).use { fis ->
+      val buffer = ByteArray(8192)
+      var read: Int
+      while (fis.read(buffer).also { read = it } != -1) {
+        digest.update(buffer, 0, read)
+      }
+    }
+    return digest.digest().joinToString("") { "%02x".format(it) }
+  }
+}

package/build/ExpoAiKitModule.d.ts

@@ -12,6 +12,7 @@ export interface ExpoAiKitNativeModule {
     stopStreaming(sessionId: string): Promise<void>;
     getBuiltInModels(): BuiltInModel[];
     getDownloadableModelStatus(modelId: string): DownloadableModelStatus;
+    getDeviceRamBytes(): number;
     setModel(modelId: string): Promise<void>;
     getActiveModel(): string;
     unloadModel(): Promise<void>;

package/build/ExpoAiKitModule.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ExpoAiKitModule.d.ts","sourceRoot":"","sources":["../src/ExpoAiKitModule.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAC3D,OAAO,EACL,UAAU,EACV,WAAW,EACX,cAAc,EACd,YAAY,EACZ,uBAAuB,EACvB,0BAA0B,EAC1B,qBAAqB,EACtB,MAAM,SAAS,CAAC;AAEjB,MAAM,MAAM,qBAAqB,GAAG;IAClC,aAAa,EAAE,CAAC,KAAK,EAAE,cAAc,KAAK,IAAI,CAAC;IAC/C,kBAAkB,EAAE,CAAC,KAAK,EAAE,0BAA0B,KAAK,IAAI,CAAC;IAChE,kBAAkB,EAAE,CAAC,KAAK,EAAE,qBAAqB,KAAK,IAAI,CAAC;CAC5D,CAAC;AAEF,MAAM,WAAW,qBAAqB;IAEpC,WAAW,IAAI,OAAO,CAAC;IACvB,WAAW,CACT,QAAQ,EAAE,UAAU,EAAE,EACtB,YAAY,EAAE,MAAM,GACnB,OAAO,CAAC,WAAW,CAAC,CAAC;IACxB,cAAc,CACZ,QAAQ,EAAE,UAAU,EAAE,EACtB,YAAY,EAAE,MAAM,EACpB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,IAAI,CAAC,CAAC;IACjB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAGhD,gBAAgB,IAAI,YAAY,EAAE,CAAC;IACnC,0BAA0B,CAAC,OAAO,EAAE,MAAM,GAAG,uBAAuB,CAAC;IAKrE,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACzC,cAAc,IAAI,MAAM,CAAC;IAGzB,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAG7B,aAAa,CACX,OAAO,EAAE,MAAM,EACf,GAAG,EAAE,MAAM,EACX,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC,CAAC;IACjB,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAG5C,WAAW,CAAC,CAAC,SAAS,MAAM,qBAAqB,EAC/C,SAAS,EAAE,CAAC,EACZ,QAAQ,EAAE,qBAAqB,CAAC,CAAC,CAAC,GACjC,iBAAiB,CAAC;CACtB;AAED,QAAA,MAAM,eAAe,uBACoC,CAAC;AAE1D,eAAe,eAAe,CAAC"}
+{"version":3,"file":"ExpoAiKitModule.d.ts","sourceRoot":"","sources":["../src/ExpoAiKitModule.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAC3D,OAAO,EACL,UAAU,EACV,WAAW,EACX,cAAc,EACd,YAAY,EACZ,uBAAuB,EACvB,0BAA0B,EAC1B,qBAAqB,EACtB,MAAM,SAAS,CAAC;AAEjB,MAAM,MAAM,qBAAqB,GAAG;IAClC,aAAa,EAAE,CAAC,KAAK,EAAE,cAAc,KAAK,IAAI,CAAC;IAC/C,kBAAkB,EAAE,CAAC,KAAK,EAAE,0BAA0B,KAAK,IAAI,CAAC;IAChE,kBAAkB,EAAE,CAAC,KAAK,EAAE,qBAAqB,KAAK,IAAI,CAAC;CAC5D,CAAC;AAEF,MAAM,WAAW,qBAAqB;IAEpC,WAAW,IAAI,OAAO,CAAC;IACvB,WAAW,CACT,QAAQ,EAAE,UAAU,EAAE,EACtB,YAAY,EAAE,MAAM,GACnB,OAAO,CAAC,WAAW,CAAC,CAAC;IACxB,cAAc,CACZ,QAAQ,EAAE,UAAU,EAAE,EACtB,YAAY,EAAE,MAAM,EACpB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,IAAI,CAAC,CAAC;IACjB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAGhD,gBAAgB,IAAI,YAAY,EAAE,CAAC;IACnC,0BAA0B,CAAC,OAAO,EAAE,MAAM,GAAG,uBAAuB,CAAC;IACrE,iBAAiB,IAAI,MAAM,CAAC;IAK5B,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACzC,cAAc,IAAI,MAAM,CAAC;IAGzB,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAG7B,aAAa,CACX,OAAO,EAAE,MAAM,EACf,GAAG,EAAE,MAAM,EACX,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC,CAAC;IACjB,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAG5C,WAAW,CAAC,CAAC,SAAS,MAAM,qBAAqB,EAC/C,SAAS,EAAE,CAAC,EACZ,QAAQ,EAAE,qBAAqB,CAAC,CAAC,CAAC,GACjC,iBAAiB,CAAC;CACtB;AAED,QAAA,MAAM,eAAe,uBACoC,CAAC;AAE1D,eAAe,eAAe,CAAC"}

package/build/ExpoAiKitModule.js.map

@@ -1 +1 @@
-{"version":3,"file":"ExpoAiKitModule.js","sourceRoot":"","sources":["../src/ExpoAiKitModule.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AA4DxD,MAAM,eAAe,GACnB,mBAAmB,CAAwB,WAAW,CAAC,CAAC;AAE1D,eAAe,eAAe,CAAC","sourcesContent":["import { requireNativeModule } from 'expo-modules-core';\nimport type { EventSubscription } from 'expo-modules-core';\nimport {\n  LLMMessage,\n  LLMResponse,\n  LLMStreamEvent,\n  BuiltInModel,\n  DownloadableModelStatus,\n  ModelDownloadProgressEvent,\n  ModelStateChangeEvent,\n} from './types';\n\nexport type ExpoAiKitModuleEvents = {\n  onStreamToken: (event: LLMStreamEvent) => void;\n  onDownloadProgress: (event: ModelDownloadProgressEvent) => void;\n  onModelStateChange: (event: ModelStateChangeEvent) => void;\n};\n\nexport interface ExpoAiKitNativeModule {\n  // Existing inference API\n  isAvailable(): boolean;\n  sendMessage(\n    messages: LLMMessage[],\n    systemPrompt: string\n  ): Promise<LLMResponse>;\n  startStreaming(\n    messages: LLMMessage[],\n    systemPrompt: string,\n    sessionId: string\n  ): Promise<void>;\n  stopStreaming(sessionId: string): Promise<void>;\n\n  // Model discovery\n  getBuiltInModels(): BuiltInModel[];\n  getDownloadableModelStatus(modelId: string): DownloadableModelStatus;\n\n  // Model selection & memory management\n  // setModel is async: switching to a downloadable model loads it into memory.\n  // Auto-unloads the previous downloadable model (only one loaded at a time).\n  setModel(modelId: string): Promise<void>;\n  getActiveModel(): string;\n  // Explicitly free memory from the loaded downloadable model.\n  // Reverts to the platform built-in model.\n  unloadModel(): Promise<void>;\n\n  // Model lifecycle (downloadable models only)\n  downloadModel(\n    modelId: string,\n    url: string,\n    sha256: string\n  ): Promise<void>;\n  deleteModel(modelId: string): Promise<void>;\n\n  // Event subscription\n  addListener<K extends keyof ExpoAiKitModuleEvents>(\n    eventName: K,\n    listener: ExpoAiKitModuleEvents[K]\n  ): EventSubscription;\n}\n\nconst ExpoAiKitModule =\n  requireNativeModule<ExpoAiKitNativeModule>('ExpoAiKit');\n\nexport default ExpoAiKitModule;\n"]}
+{"version":3,"file":"ExpoAiKitModule.js","sourceRoot":"","sources":["../src/ExpoAiKitModule.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AA6DxD,MAAM,eAAe,GACnB,mBAAmB,CAAwB,WAAW,CAAC,CAAC;AAE1D,eAAe,eAAe,CAAC","sourcesContent":["import { requireNativeModule } from 'expo-modules-core';\nimport type { EventSubscription } from 'expo-modules-core';\nimport {\n  LLMMessage,\n  LLMResponse,\n  LLMStreamEvent,\n  BuiltInModel,\n  DownloadableModelStatus,\n  ModelDownloadProgressEvent,\n  ModelStateChangeEvent,\n} from './types';\n\nexport type ExpoAiKitModuleEvents = {\n  onStreamToken: (event: LLMStreamEvent) => void;\n  onDownloadProgress: (event: ModelDownloadProgressEvent) => void;\n  onModelStateChange: (event: ModelStateChangeEvent) => void;\n};\n\nexport interface ExpoAiKitNativeModule {\n  // Existing inference API\n  isAvailable(): boolean;\n  sendMessage(\n    messages: LLMMessage[],\n    systemPrompt: string\n  ): Promise<LLMResponse>;\n  startStreaming(\n    messages: LLMMessage[],\n    systemPrompt: string,\n    sessionId: string\n  ): Promise<void>;\n  stopStreaming(sessionId: string): Promise<void>;\n\n  // Model discovery\n  getBuiltInModels(): BuiltInModel[];\n  getDownloadableModelStatus(modelId: string): DownloadableModelStatus;\n  getDeviceRamBytes(): number;\n\n  // Model selection & memory management\n  // setModel is async: switching to a downloadable model loads it into memory.\n  // Auto-unloads the previous downloadable model (only one loaded at a time).\n  setModel(modelId: string): Promise<void>;\n  getActiveModel(): string;\n  // Explicitly free memory from the loaded downloadable model.\n  // Reverts to the platform built-in model.\n  unloadModel(): Promise<void>;\n\n  // Model lifecycle (downloadable models only)\n  downloadModel(\n    modelId: string,\n    url: string,\n    sha256: string\n  ): Promise<void>;\n  deleteModel(modelId: string): Promise<void>;\n\n  // Event subscription\n  addListener<K extends keyof ExpoAiKitModuleEvents>(\n    eventName: K,\n    listener: ExpoAiKitModuleEvents[K]\n  ): EventSubscription;\n}\n\nconst ExpoAiKitModule =\n  requireNativeModule<ExpoAiKitNativeModule>('ExpoAiKit');\n\nexport default ExpoAiKitModule;\n"]}

package/build/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,UAAU,EACV,cAAc,EACd,WAAW,EACX,gBAAgB,EAEhB,iBAAiB,EACjB,mBAAmB,EACnB,mBAAmB,EACnB,iBAAiB,EACjB,0BAA0B,EAC1B,wBAAwB,EACxB,iBAAiB,EACjB,kBAAkB,EAClB,oBAAoB,EACpB,sBAAsB,EACtB,YAAY,EACZ,iBAAiB,EAElB,MAAM,SAAS,CAAC;AAGjB,cAAc,SAAS,CAAC;AACxB,cAAc,UAAU,CAAC;AACzB,cAAc,SAAS,CAAC;AACxB,cAAc,UAAU,CAAC;AAiJzB;;;GAGG;AACH,wBAAsB,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,CAKpD;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACH,wBAAsB,WAAW,CAC/B,QAAQ,EAAE,UAAU,EAAE,EACtB,OAAO,CAAC,EAAE,cAAc,GACvB,OAAO,CAAC,WAAW,CAAC,CAgBtB;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2CG;AACH,wBAAgB,aAAa,CAC3B,QAAQ,EAAE,UAAU,EAAE,EACtB,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,gBAAgB,GACzB;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAiErD;AAMD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,wBAAsB,SAAS,CAC7B,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,mBAAmB,GAC5B,OAAO,CAAC,WAAW,CAAC,CActB;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,eAAe,CAC7B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,mBAAmB,GAC5B;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAmBrD;AAED;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,wBAAsB,SAAS,CAC7B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,mBAAmB,GAC3B,OAAO,CAAC,WAAW,CAAC,CAatB;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,eAAe,CAC7B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,iBAAiB,EAC1B,OAAO,EAAE,mBAAmB,GAC3B;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAkBrD;AAED;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,wBAAsB,OAAO,CAC3B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,iBAAiB,GACzB,OAAO,CAAC,WAAW,CAAC,CAatB;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,aAAa,CAC3B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,iBAAiB,EAC1B,OAAO,EAAE,iBAAiB,GACzB;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAkBrD;AAED;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,wBAAsB,gBAAgB,CACpC,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,0BAA0B,GACnC,OAAO,CAAC,WAAW,CAAC,CAatB;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,sBAAsB,CACpC,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,0BAA0B,GACnC;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAkBrD;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AACH,wBAAsB,cAAc,CAClC,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,wBAAwB,GACjC,OAAO,CAAC,WAAW,CAAC,CAkBtB;AAED;;;;;;;;;;;;;;;;;;;GAmBG;AACH,wBAAgB,oBAAoB,CAClC,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,wBAAwB,GACjC;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CA0BrD;AAMD;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,wBAAsB,OAAO,CAC3B,WAAW,EAAE,MAAM,EACnB,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,kBAAkB,CAAC,CAsB7B;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,aAAa,CAC3B,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,iBAAiB,GAC1B;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAmBrD;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACH,wBAAsB,UAAU,CAC9B,QAAQ,EAAE,UAAU,EAAE,EACtB,OAAO,CAAC,EAAE,oBAAoB,GAC7B,OAAO,CAAC,kBAAkB,CAAC,CA0B7B;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,gBAAgB,CAC9B,QAAQ,EAAE,UAAU,EAAE,EACtB,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,oBAAoB,GAC7B;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAyBrD;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,wBAAsB,YAAY,CAChC,WAAW,EAAE,MAAM,EACnB,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC,kBAAkB,CAAC,CAsB7B;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,kBAAkB,CAChC,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,sBAAsB,GAC/B;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAmBrD;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,MAAM,GAAG,kBAAkB,CAKpE;AAMD;;;;;;;GAOG;AACH,wBAAsB,gBAAgB,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC,CAKhE;AAED;;;;;;;GAOG;AACH,wBAAsB,qBAAqB,IAAI,OAAO,CAAC,iBAAiB,EAAE,CAAC,CAqB1E;AAED;;;;;;;;;;;;;;GAcG;AACH,wBAAsB,aAAa,CACjC,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE;IAAE,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAA;CAAE,GACpD,OAAO,CAAC,IAAI,CAAC,CAmCf;AAED;;;;;;;GAOG;AACH,wBAAsB,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAOhE;AAED;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,wBAAsB,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAE7D;AAED;;;;GAIG;AACH,wBAAgB,cAAc,IAAI,MAAM,CAEvC;AAED;;;;;GAKG;AACH,wBAAsB,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC,CAEjD"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,UAAU,EACV,cAAc,EACd,WAAW,EACX,gBAAgB,EAEhB,iBAAiB,EACjB,mBAAmB,EACnB,mBAAmB,EACnB,iBAAiB,EACjB,0BAA0B,EAC1B,wBAAwB,EACxB,iBAAiB,EACjB,kBAAkB,EAClB,oBAAoB,EACpB,sBAAsB,EACtB,YAAY,EACZ,iBAAiB,EAElB,MAAM,SAAS,CAAC;AAGjB,cAAc,SAAS,CAAC;AACxB,cAAc,UAAU,CAAC;AACzB,cAAc,SAAS,CAAC;AACxB,cAAc,UAAU,CAAC;AAiJzB;;;GAGG;AACH,wBAAsB,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,CAKpD;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACH,wBAAsB,WAAW,CAC/B,QAAQ,EAAE,UAAU,EAAE,EACtB,OAAO,CAAC,EAAE,cAAc,GACvB,OAAO,CAAC,WAAW,CAAC,CAgBtB;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2CG;AACH,wBAAgB,aAAa,CAC3B,QAAQ,EAAE,UAAU,EAAE,EACtB,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,gBAAgB,GACzB;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAiErD;AAMD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,wBAAsB,SAAS,CAC7B,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,mBAAmB,GAC5B,OAAO,CAAC,WAAW,CAAC,CActB;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,eAAe,CAC7B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,mBAAmB,GAC5B;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAmBrD;AAED;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,wBAAsB,SAAS,CAC7B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,mBAAmB,GAC3B,OAAO,CAAC,WAAW,CAAC,CAatB;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,eAAe,CAC7B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,iBAAiB,EAC1B,OAAO,EAAE,mBAAmB,GAC3B;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAkBrD;AAED;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,wBAAsB,OAAO,CAC3B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,iBAAiB,GACzB,OAAO,CAAC,WAAW,CAAC,CAatB;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,aAAa,CAC3B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,iBAAiB,EAC1B,OAAO,EAAE,iBAAiB,GACzB;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAkBrD;AAED;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,wBAAsB,gBAAgB,CACpC,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,0BAA0B,GACnC,OAAO,CAAC,WAAW,CAAC,CAatB;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,sBAAsB,CACpC,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,0BAA0B,GACnC;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAkBrD;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AACH,wBAAsB,cAAc,CAClC,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,wBAAwB,GACjC,OAAO,CAAC,WAAW,CAAC,CAkBtB;AAED;;;;;;;;;;;;;;;;;;;GAmBG;AACH,wBAAgB,oBAAoB,CAClC,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,wBAAwB,GACjC;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CA0BrD;AAMD;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,wBAAsB,OAAO,CAC3B,WAAW,EAAE,MAAM,EACnB,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,kBAAkB,CAAC,CAsB7B;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAgB,aAAa,CAC3B,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,iBAAiB,GAC1B;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAmBrD;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACH,wBAAsB,UAAU,CAC9B,QAAQ,EAAE,UAAU,EAAE,EACtB,OAAO,CAAC,EAAE,oBAAoB,GAC7B,OAAO,CAAC,kBAAkB,CAAC,CA0B7B;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,gBAAgB,CAC9B,QAAQ,EAAE,UAAU,EAAE,EACtB,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,oBAAoB,GAC7B;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAyBrD;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,wBAAsB,YAAY,CAChC,WAAW,EAAE,MAAM,EACnB,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC,kBAAkB,CAAC,CAsB7B;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,kBAAkB,CAChC,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,iBAAiB,EAC1B,OAAO,CAAC,EAAE,sBAAsB,GAC/B;IAAE,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAAC,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CAmBrD;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,MAAM,GAAG,kBAAkB,CAKpE;AAMD;;;;;;;GAOG;AACH,wBAAsB,gBAAgB,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC,CAKhE;AAED;;;;;;;GAOG;AACH,wBAAsB,qBAAqB,IAAI,OAAO,CAAC,iBAAiB,EAAE,CAAC,CA6B1E;AAED;;;;;;;;;;;;;;GAcG;AACH,wBAAsB,aAAa,CACjC,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE;IAAE,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAA;CAAE,GACpD,OAAO,CAAC,IAAI,CAAC,CAiDf;AAED;;;;;;;GAOG;AACH,wBAAsB,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAOhE;AAED;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,wBAAsB,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAE7D;AAED;;;;GAIG;AACH,wBAAgB,cAAc,IAAI,MAAM,CAEvC;AAED;;;;;GAKG;AACH,wBAAsB,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC,CAEjD"}

package/build/index.js

@@ -923,6 +923,13 @@ export async function getDownloadableModels() {
         return [];
     }
     const platformModels = MODEL_REGISTRY.filter((entry) => entry.supportedPlatforms.includes(Platform.OS));
+    let deviceRamBytes = 0;
+    try {
+        deviceRamBytes = ExpoAiKitModule.getDeviceRamBytes();
+    }
+    catch {
+        // Native call unavailable -- default to 0 (all models will show meetsRequirements: false)
+    }
     return platformModels.map((entry) => {
         const status = ExpoAiKitModule.getDownloadableModelStatus(entry.id);
         return {
@@ -932,6 +939,7 @@ export async function getDownloadableModels() {
             sizeBytes: entry.sizeBytes,
             contextWindow: entry.contextWindow,
             minRamBytes: entry.minRamBytes,
+            meetsRequirements: deviceRamBytes >= entry.minRamBytes,
             status,
         };
     });
@@ -959,6 +967,17 @@ export async function downloadModel(modelId, options) {
     if (!entry.supportedPlatforms.includes(Platform.OS)) {
         throw new ModelError('DEVICE_NOT_SUPPORTED', modelId, `Model ${modelId} is not supported on ${Platform.OS}`);
     }
+    try {
+        const deviceRamBytes = ExpoAiKitModule.getDeviceRamBytes();
+        if (deviceRamBytes < entry.minRamBytes) {
+            throw new ModelError('DEVICE_NOT_SUPPORTED', modelId, `Device has ${Math.round(deviceRamBytes / 1e9)}GB RAM, model requires ${Math.round(entry.minRamBytes / 1e9)}GB`);
+        }
+    }
+    catch (e) {
+        if (e instanceof ModelError)
+            throw e;
+        // If getDeviceRamBytes is unavailable, skip the check
+    }
     let subscription;
     if (options?.onProgress) {
         subscription = ExpoAiKitModule.addListener('onDownloadProgress', (event) => {