expo-ai-kit 0.1.13 → 0.1.14

package/README.md

@@ -18,17 +18,17 @@ On-device AI for Expo apps. Run language models locally—no API keys, no cloud,
 
 | Platform | Fallback Behavior |
 |----------|-------------------|
-| iOS < 26 | Returns mock responses |
+| iOS < 26 | Returns fallback message |
 | Android (unsupported devices) | Returns empty string |
 
 ## Features
 
-- 🔒 **Privacy-first** — All inference happens on-device; no data leaves the user's device
-- ⚡ **Zero latency** — No network round-trips required
-- 🆓 **Free forever** — No API costs, rate limits, or subscriptions
-- 📱 **Native performance** — Built on Apple Foundation Models (iOS) and Google ML Kit Prompt API (Android)
-- 💬 **Multi-turn conversations** — Session-based chat with full conversation context
-- 🎛️ **Configurable** — Temperature and max tokens control for response generation
+- **Privacy-first** — All inference happens on-device; no data leaves the user's device
+- **Zero latency** — No network round-trips required
+- **Free forever** — No API costs, rate limits, or subscriptions
+- **Native performance** — Built on Apple Foundation Models (iOS) and Google ML Kit Prompt API (Android)
+- **Multi-turn conversations** — Full conversation context support
+- **Simple API** — Just 2 functions: `isAvailable()` and `sendMessage()`
 
 ## Requirements
 
@@ -68,25 +68,27 @@ For Android, ensure your `app.json` includes the minimum SDK version:
 ## Quick Start
 
 ```tsx
-import { isAvailable, sendPrompt } from 'expo-ai-kit';
+import { isAvailable, sendMessage } from 'expo-ai-kit';
 
 // Check if on-device AI is available
 const available = await isAvailable();
 
 if (available) {
-  const response = await sendPrompt('Hello! What can you do?');
-  console.log(response);
+  const response = await sendMessage([
+    { role: 'user', content: 'Hello! What can you do?' }
+  ]);
+  console.log(response.text);
 }
 ```
 
 ## Usage
 
-### Simple Prompt (Cross-platform)
+### Simple Prompt
 
 The simplest way to use on-device AI:
 
 ```tsx
-import { isAvailable, sendPrompt } from 'expo-ai-kit';
+import { isAvailable, sendMessage } from 'expo-ai-kit';
 
 async function askAI(question: string) {
   const available = await isAvailable();
@@ -96,52 +98,48 @@ async function askAI(question: string) {
     return null;
   }
 
-  return await sendPrompt(question);
+  const response = await sendMessage([
+    { role: 'user', content: question }
+  ]);
+  return response.text;
 }
 
 const answer = await askAI('What is the capital of France?');
 ```
 
-### Session-based Chat
+### With Custom System Prompt
 
-For multi-turn conversations with context, use sessions:
+Customize the AI's behavior with a system prompt:
 
 ```tsx
-import { createSession, sendMessage } from 'expo-ai-kit';
+import { sendMessage } from 'expo-ai-kit';
 
-// Create a chat session
-const sessionId = await createSession({
-  systemPrompt: 'You are a friendly cooking assistant.',
-});
+const response = await sendMessage(
+  [{ role: 'user', content: 'Tell me a joke' }],
+  { systemPrompt: 'You are a comedian who specializes in dad jokes.' }
+);
 
-// Send messages with conversation history
-const { reply } = await sendMessage(sessionId, [
-  { role: 'user', content: 'What can I make with eggs and cheese?' }
-]);
+console.log(response.text);
 ```
 
 ### Multi-turn Conversations
 
-Keep track of the conversation history for context-aware responses:
+For conversations with context, pass the full conversation history:
 
 ```tsx
-const [messages, setMessages] = useState<LLMMessage[]>([]);
+import { sendMessage, type LLMMessage } from 'expo-ai-kit';
 
-async function chat(userMessage: string) {
-  const newMessages = [
-    ...messages,
-    { role: 'user', content: userMessage }
-  ];
+const conversation: LLMMessage[] = [
+  { role: 'user', content: 'My name is Alice.' },
+  { role: 'assistant', content: 'Nice to meet you, Alice!' },
+  { role: 'user', content: 'What is my name?' },
+];
 
-  const { reply } = await sendMessage(sessionId, newMessages);
-
-  setMessages([
-    ...newMessages,
-    { role: 'assistant', content: reply }
-  ]);
+const response = await sendMessage(conversation, {
+  systemPrompt: 'You are a helpful assistant.',
+});
 
-  return reply;
-}
+console.log(response.text); // "Your name is Alice."
 ```
 
 ### Complete Chat Example
@@ -151,10 +149,10 @@ Here's a full cross-platform chat component:
 ```tsx
 import React, { useState, useEffect } from 'react';
 import { View, TextInput, Button, Text, FlatList } from 'react-native';
-import { isAvailable, sendPrompt } from 'expo-ai-kit';
+import { isAvailable, sendMessage, type LLMMessage } from 'expo-ai-kit';
 
 export default function ChatScreen() {
-  const [messages, setMessages] = useState<Array<{ role: string; content: string }>>([]);
+  const [messages, setMessages] = useState<LLMMessage[]>([]);
   const [input, setInput] = useState('');
   const [loading, setLoading] = useState(false);
   const [available, setAvailable] = useState(false);
@@ -166,14 +164,17 @@ export default function ChatScreen() {
   const handleSend = async () => {
     if (!input.trim() || loading || !available) return;
 
-    const userMessage = { role: 'user', content: input.trim() };
-    setMessages(prev => [...prev, userMessage]);
+    const userMessage: LLMMessage = { role: 'user', content: input.trim() };
+    const newMessages = [...messages, userMessage];
+    setMessages(newMessages);
     setInput('');
     setLoading(true);
 
     try {
-      const reply = await sendPrompt(input.trim());
-      setMessages(prev => [...prev, { role: 'assistant', content: reply }]);
+      const response = await sendMessage(newMessages, {
+        systemPrompt: 'You are a helpful assistant.',
+      });
+      setMessages(prev => [...prev, { role: 'assistant', content: response.text }]);
     } catch (error) {
       console.error('Error:', error);
     } finally {
@@ -225,62 +226,41 @@ export default function ChatScreen() {
 
 ## API Reference
 
-### `isAvailable()` — iOS, Android
+### `isAvailable()`
 
 Checks if on-device AI is available on the current device.
 
-**Returns:** `Promise<boolean>` — `true` if on-device AI is supported and ready
-
----
-
-### `sendPrompt(prompt)` — iOS, Android
-
-Sends a prompt and gets a response from the on-device model.
-
-| Parameter | Type | Description |
-|-----------|------|-------------|
-| `prompt` | `string` | The text prompt to send |
+```typescript
+function isAvailable(): Promise<boolean>
+```
 
-**Returns:** `Promise<string>` — The AI's response (empty string if unavailable)
+**Returns:** `Promise<boolean>` — `true` if on-device AI is supported and ready
 
 ---
 
-### `createSession(options?)` — iOS, Android
+### `sendMessage(messages, options?)`
 
-Creates a new chat session for multi-turn conversations.
-
-| Parameter | Type | Description |
-|-----------|------|-------------|
-| `options.systemPrompt` | `string` | Optional system prompt to guide the AI's behavior |
-
-**Returns:** `Promise<string>` — A unique session ID
-
----
+Sends a conversation and gets a response from the on-device model.
 
-### `sendMessage(sessionId, messages, options?)` — iOS, Android
-
-Sends messages and gets a response from the on-device model with conversation context.
+```typescript
+function sendMessage(messages: LLMMessage[], options?: LLMSendOptions): Promise<LLMResponse>
+```
 
 | Parameter | Type | Description |
 |-----------|------|-------------|
-| `sessionId` | `string` | The session ID from `createSession` |
 | `messages` | `LLMMessage[]` | Array of conversation messages |
-| `options.temperature` | `number` | Controls randomness (0-1) |
-| `options.maxTokens` | `number` | Maximum response length |
-
-**Returns:** `Promise<{ reply: string }>` — The AI's response
-
----
+| `options.systemPrompt` | `string` | Fallback system prompt (ignored if messages contain a system message) |
 
-### `prepareModel(options?)` — iOS, Android
+**Returns:** `Promise<LLMResponse>` — Object with `text` property containing the response
 
-Pre-loads the model for faster first response.
-
-| Parameter | Type | Description |
-|-----------|------|-------------|
-| `options.model` | `string` | Model identifier (optional) |
-
-**Returns:** `Promise<void>`
+**Example:**
+```tsx
+const response = await sendMessage([
+  { role: 'system', content: 'You are a pirate.' },
+  { role: 'user', content: 'Hello!' },
+]);
+console.log(response.text); // "Ahoy, matey!"
+```
 
 ---
 
@@ -294,10 +274,14 @@ type LLMMessage = {
   content: string;
 };
 
-type LLMOptions = {
-  temperature?: number;
-  maxTokens?: number;
-  model?: string;
+type LLMSendOptions = {
+  /** Fallback system prompt if no system message in messages array */
+  systemPrompt?: string;
+};
+
+type LLMResponse = {
+  /** The generated response text */
+  text: string;
 };
 ```
 
@@ -306,13 +290,9 @@ type LLMOptions = {
 | Feature | iOS 26+ | Android (Supported) |
 |---------|---------|---------------------|
 | `isAvailable()` | ✅ | ✅ |
-| `sendPrompt()` | ✅ | ✅ |
-| `createSession()` | ✅ Full context | ✅ Basic |
-| `sendMessage()` | ✅ Full context | ✅ Basic |
-| `prepareModel()` | ✅ | ✅ No-op |
-| System prompts | ✅ | ✅ |
-| Temperature control | ✅ | ✅ |
-| Max tokens control | ✅ | ✅ |
+| `sendMessage()` | ✅ | ✅ |
+| System prompts | ✅ Native | ✅ Prepended |
+| Multi-turn context | ✅ | ✅ |
 
 ## How It Works
 
@@ -326,12 +306,35 @@ Uses Google's ML Kit Prompt API. The model may need to be downloaded on first us
 
 ### iOS
 - **AI not available**: Ensure you're running iOS 26.0 or later on a supported device
-- **Mock responses**: On iOS < 26, the module returns mock responses for testing
+- **Fallback responses**: On iOS < 26, the module returns a fallback message
 
 ### Android
 - **Empty responses**: The device may not support ML Kit Prompt API. Check the [supported devices list](https://developers.google.com/ml-kit/genai#prompt-device)
 - **Model downloading**: On first use, the model may need to download. Use `isAvailable()` to check status
 
+## Migration from v0.1.x
+
+If you're upgrading from an earlier version, here are the breaking changes:
+
+| Old API | New API |
+|---------|---------|
+| `sendPrompt(prompt)` | `sendMessage([{ role: 'user', content: prompt }])` |
+| `createSession(options)` | **Removed** — no longer needed |
+| `sendMessage(sessionId, messages, options)` | `sendMessage(messages, options)` — no session ID |
+| `prepareModel(options)` | **Removed** |
+| `{ reply: string }` | `{ text: string }` |
+
+**Before:**
+```tsx
+const sessionId = await createSession({ systemPrompt: '...' });
+const { reply } = await sendMessage(sessionId, messages, {});
+```
+
+**After:**
+```tsx
+const { text } = await sendMessage(messages, { systemPrompt: '...' });
+```
+
 ## License
 
 MIT

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

@@ -11,14 +11,25 @@ class ExpoAiKitModule : Module() {
   override fun definition() = ModuleDefinition {
     Name("ExpoAiKit")
 
-    // Returns true if device supports Prompt API (AVAILABLE/DOWNLOADABLE/DOWNLOADING)
-    // Returns false on unsupported devices, never crashes
-    AsyncFunction("isAvailable") {
+    Function("isAvailable") {
       promptClient.isAvailableBlocking()
     }
 
-    AsyncFunction("sendPrompt") Coroutine { prompt: String ->
-      promptClient.generateText(prompt)
+    AsyncFunction("sendMessage") Coroutine { messages: List<Map<String, Any>>, fallbackSystemPrompt: String ->
+      // Extract system prompt from messages, or use fallback
+      val systemPrompt = messages
+        .firstOrNull { it["role"] == "system" }
+        ?.get("content") as? String
+        ?: fallbackSystemPrompt.ifBlank { "You are a helpful, friendly assistant." }
+
+      // Get the last user message as the prompt
+      val userPrompt = messages
+        .lastOrNull { it["role"] == "user" }
+        ?.get("content") as? String
+        ?: ""
+
+      val text = promptClient.generateText(userPrompt, systemPrompt)
+      mapOf("text" to text)
     }
   }
 }

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

@@ -11,11 +11,12 @@ class PromptApiClient {
 
   private val model by lazy { Generation.getClient() }
 
-  // ✅ suspend version (matches ML Kit's checkStatus usage)
-  // Returns true if device supports Prompt API (AVAILABLE, DOWNLOADABLE, or DOWNLOADING)
-  // Returns false if unsupported or on API < 26
+  /**
+   * Check if on-device AI is available.
+   * Returns true if device supports Prompt API (AVAILABLE, DOWNLOADABLE, or DOWNLOADING).
+   * Returns false if unsupported or on API < 26.
+   */
   suspend fun isAvailable(): Boolean {
-    // Prompt API requires API 26+
     if (Build.VERSION.SDK_INT < Build.VERSION_CODES.O) return false
 
     return try {
@@ -26,31 +27,41 @@ class PromptApiClient {
         else -> false
       }
     } catch (_: Throwable) {
-      // Covers: missing AICore / Gemini Nano not supported / config not fetched / etc.
       false
     }
   }
 
-  // ✅ non-suspend wrapper so Expo Module can call it without Coroutine DSL ambiguity
-  fun isAvailableBlocking(): Boolean = runBlocking {
-    isAvailable()
-  }
+  /**
+   * Non-suspend wrapper for Expo module compatibility.
+   */
+  fun isAvailableBlocking(): Boolean = runBlocking { isAvailable() }
 
-  suspend fun generateText(prompt: String): String = withContext(Dispatchers.IO) {
-    // Prompt API requires API 26+
-    if (Build.VERSION.SDK_INT < Build.VERSION_CODES.O) return@withContext ""
+  /**
+   * Generate text from a prompt.
+   * On Android, we concatenate system prompt + user message since ML Kit
+   * doesn't have a separate system prompt API.
+   */
+  suspend fun generateText(prompt: String, systemPrompt: String): String =
+    withContext(Dispatchers.IO) {
+      if (Build.VERSION.SDK_INT < Build.VERSION_CODES.O) return@withContext ""
 
-    try {
-      val status = model.checkStatus()
-      if (status != FeatureStatus.AVAILABLE) {
-        return@withContext ""
-      }
+      try {
+        val status = model.checkStatus()
+        if (status != FeatureStatus.AVAILABLE) {
+          return@withContext ""
+        }
 
-      // safest + simplest API across ML Kit alpha versions
-      val response = model.generateContent(prompt)
-      response.candidates.firstOrNull()?.text.orEmpty()
-    } catch (_: Throwable) {
-      ""
+        // Prepend system prompt as context if provided
+        val fullPrompt = if (systemPrompt.isNotBlank()) {
+          "$systemPrompt\n\nUser: $prompt"
+        } else {
+          prompt
+        }
+
+        val response = model.generateContent(fullPrompt)
+        response.candidates.firstOrNull()?.text.orEmpty()
+      } catch (_: Throwable) {
+        ""
+      }
     }
-  }
 }

package/build/ExpoAiKitModule.d.ts

@@ -1,16 +1,7 @@
-import type { LLMMessage, LLMOptions } from './types';
+import { LLMMessage, LLMResponse } from './types';
 export type ExpoAiKitNativeModule = {
-    prepareModel(options?: {
-        model?: string;
-    }): Promise<void>;
-    createSession(options?: {
-        systemPrompt?: string;
-    }): Promise<string>;
-    sendMessage(sessionId: string, messages: LLMMessage[], options?: LLMOptions): Promise<{
-        reply: string;
-    }>;
     isAvailable(): boolean;
-    sendPrompt(prompt: string): Promise<string>;
+    sendMessage(messages: LLMMessage[], systemPrompt: string): Promise<LLMResponse>;
 };
 declare const NativeModule: ExpoAiKitNativeModule;
 export default NativeModule;

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,UAAU,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAEtD,MAAM,MAAM,qBAAqB,GAAG;IAClC,YAAY,CAAC,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1D,aAAa,CAAC,OAAO,CAAC,EAAE;QAAE,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IACpE,WAAW,CACT,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,UAAU,EAAE,EACtB,OAAO,CAAC,EAAE,UAAU,GACnB,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC9B,WAAW,IAAI,OAAO,CAAC;IACvB,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;CAC7C,CAAC;AAEF,QAAA,MAAM,YAAY,EAAE,qBACqC,CAAC;AAE1D,eAAe,YAAY,CAAC"}
+{"version":3,"file":"ExpoAiKitModule.d.ts","sourceRoot":"","sources":["../src/ExpoAiKitModule.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAElD,MAAM,MAAM,qBAAqB,GAAG;IAClC,WAAW,IAAI,OAAO,CAAC;IACvB,WAAW,CACT,QAAQ,EAAE,UAAU,EAAE,EACtB,YAAY,EAAE,MAAM,GACnB,OAAO,CAAC,WAAW,CAAC,CAAC;CACzB,CAAC;AAEF,QAAA,MAAM,YAAY,EAAE,qBACqC,CAAC;AAE1D,eAAe,YAAY,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;AAexD,MAAM,YAAY,GAChB,mBAAmB,CAAwB,WAAW,CAAC,CAAC;AAE1D,eAAe,YAAY,CAAC","sourcesContent":["import { requireNativeModule } from 'expo-modules-core';\nimport type { LLMMessage, LLMOptions } from './types';\n\nexport type ExpoAiKitNativeModule = {\n  prepareModel(options?: { model?: string }): Promise<void>;\n  createSession(options?: { systemPrompt?: string }): Promise<string>;\n  sendMessage(\n    sessionId: string,\n    messages: LLMMessage[],\n    options?: LLMOptions\n  ): Promise<{ reply: string }>;\n  isAvailable(): boolean;\n  sendPrompt(prompt: string): Promise<string>;\n};\n\nconst NativeModule: ExpoAiKitNativeModule =\n  requireNativeModule<ExpoAiKitNativeModule>('ExpoAiKit'); \n\nexport default NativeModule;"]}
+{"version":3,"file":"ExpoAiKitModule.js","sourceRoot":"","sources":["../src/ExpoAiKitModule.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAWxD,MAAM,YAAY,GAChB,mBAAmB,CAAwB,WAAW,CAAC,CAAC;AAE1D,eAAe,YAAY,CAAC","sourcesContent":["import { requireNativeModule } from 'expo-modules-core';\nimport { LLMMessage, LLMResponse } from './types';\n\nexport type ExpoAiKitNativeModule = {\n  isAvailable(): boolean;\n  sendMessage(\n    messages: LLMMessage[],\n    systemPrompt: string\n  ): Promise<LLMResponse>;\n};\n\nconst NativeModule: ExpoAiKitNativeModule =\n  requireNativeModule<ExpoAiKitNativeModule>('ExpoAiKit');\n\nexport default NativeModule;\n"]}

package/build/index.d.ts

@@ -1,14 +1,44 @@
+import { LLMMessage, LLMSendOptions, LLMResponse } from './types';
 export * from './types';
-import type { LLMMessage, LLMOptions } from './types';
-export declare function prepareModel(options?: {
-    model?: string;
-}): Promise<void>;
-export declare function createSession(options?: {
-    systemPrompt?: string;
-}): Promise<string>;
-export declare function sendMessage(sessionId: string, messages: LLMMessage[], options?: LLMOptions): Promise<{
-    reply: string;
-}>;
+/**
+ * Check if on-device AI is available on the current device.
+ * Returns false on unsupported platforms (web, etc.).
+ */
 export declare function isAvailable(): Promise<boolean>;
-export declare function sendPrompt(prompt: string): Promise<string>;
+/**
+ * Send messages to the on-device LLM and get a response.
+ *
+ * @param messages - Array of messages representing the conversation
+ * @param options - Optional settings (systemPrompt fallback)
+ * @returns Promise with the generated response
+ *
+ * @example
+ * ```ts
+ * const response = await sendMessage([
+ *   { role: 'user', content: 'What is 2 + 2?' }
+ * ]);
+ * console.log(response.text); // "4"
+ * ```
+ *
+ * @example
+ * ```ts
+ * // With system prompt
+ * const response = await sendMessage(
+ *   [{ role: 'user', content: 'Hello!' }],
+ *   { systemPrompt: 'You are a pirate. Respond in pirate speak.' }
+ * );
+ * ```
+ *
+ * @example
+ * ```ts
+ * // Multi-turn conversation
+ * const response = await sendMessage([
+ *   { role: 'system', content: 'You are a helpful assistant.' },
+ *   { role: 'user', content: 'My name is Alice.' },
+ *   { role: 'assistant', content: 'Nice to meet you, Alice!' },
+ *   { role: 'user', content: 'What is my name?' }
+ * ]);
+ * ```
+ */
+export declare function sendMessage(messages: LLMMessage[], options?: LLMSendOptions): Promise<LLMResponse>;
 //# sourceMappingURL=index.d.ts.map

package/build/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,cAAc,SAAS,CAAC;AACxB,OAAO,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAEtD,wBAAsB,YAAY,CAAC,OAAO,CAAC,EAAE;IAAE,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,iBAE9D;AAED,wBAAsB,aAAa,CAAC,OAAO,CAAC,EAAE;IAAE,YAAY,CAAC,EAAE,MAAM,CAAA;CAAE,mBAEtE;AAED,wBAAsB,WAAW,CAC/B,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,UAAU,EAAE,EACtB,OAAO,CAAC,EAAE,UAAU;;GAGrB;AAED,wBAAsB,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,CAKpD;AAED,wBAAsB,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAKhE"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAElE,cAAc,SAAS,CAAC;AAKxB;;;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"}

package/build/index.js

@@ -1,25 +1,64 @@
 import NativeModule from './ExpoAiKitModule';
 import { Platform } from 'react-native';
 export * from './types';
-export async function prepareModel(options) {
-    return NativeModule.prepareModel(options);
-}
-export async function createSession(options) {
-    return NativeModule.createSession(options);
-}
-export async function sendMessage(sessionId, messages, options) {
-    return NativeModule.sendMessage(sessionId, messages, options);
-}
+const DEFAULT_SYSTEM_PROMPT = 'You are a helpful, friendly assistant. Answer the user directly and concisely.';
+/**
+ * Check if on-device AI is available on the current device.
+ * Returns false on unsupported platforms (web, etc.).
+ */
 export async function isAvailable() {
-    if (Platform.OS === 'ios' || Platform.OS === 'android') {
-        return NativeModule.isAvailable();
+    if (Platform.OS !== 'ios' && Platform.OS !== 'android') {
+        return false;
     }
-    return false;
+    return NativeModule.isAvailable();
 }
-export async function sendPrompt(prompt) {
-    if (Platform.OS === 'ios' || Platform.OS === 'android') {
-        return NativeModule.sendPrompt(prompt);
+/**
+ * Send messages to the on-device LLM and get a response.
+ *
+ * @param messages - Array of messages representing the conversation
+ * @param options - Optional settings (systemPrompt fallback)
+ * @returns Promise with the generated response
+ *
+ * @example
+ * ```ts
+ * const response = await sendMessage([
+ *   { role: 'user', content: 'What is 2 + 2?' }
+ * ]);
+ * console.log(response.text); // "4"
+ * ```
+ *
+ * @example
+ * ```ts
+ * // With system prompt
+ * const response = await sendMessage(
+ *   [{ role: 'user', content: 'Hello!' }],
+ *   { systemPrompt: 'You are a pirate. Respond in pirate speak.' }
+ * );
+ * ```
+ *
+ * @example
+ * ```ts
+ * // Multi-turn conversation
+ * const response = await sendMessage([
+ *   { role: 'system', content: 'You are a helpful assistant.' },
+ *   { role: 'user', content: 'My name is Alice.' },
+ *   { role: 'assistant', content: 'Nice to meet you, Alice!' },
+ *   { role: 'user', content: 'What is my name?' }
+ * ]);
+ * ```
+ */
+export async function sendMessage(messages, options) {
+    if (Platform.OS !== 'ios' && Platform.OS !== 'android') {
+        return { text: '' };
+    }
+    if (!messages || messages.length === 0) {
+        throw new Error('messages array cannot be empty');
     }
-    return '';
+    // Determine system prompt: use from messages array if present, else options, else default
+    const hasSystemMessage = messages.some((m) => m.role === 'system');
+    const systemPrompt = hasSystemMessage
+        ? '' // Native will extract from messages
+        : options?.systemPrompt ?? DEFAULT_SYSTEM_PROMPT;
+    return NativeModule.sendMessage(messages, systemPrompt);
 }
 //# sourceMappingURL=index.js.map

package/build/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,YAAY,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,cAAc,SAAS,CAAC;AAGxB,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,OAA4B;IAC7D,OAAO,YAAY,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;AAC5C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,OAAmC;IACrE,OAAO,YAAY,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;AAC7C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,SAAiB,EACjB,QAAsB,EACtB,OAAoB;IAEpB,OAAO,YAAY,CAAC,WAAW,CAAC,SAAS,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;AAChE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW;IAC/B,IAAI,QAAQ,CAAC,EAAE,KAAK,KAAK,IAAI,QAAQ,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;QACvD,OAAO,YAAY,CAAC,WAAW,EAAE,CAAC;IACpC,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,MAAc;IAC7C,IAAI,QAAQ,CAAC,EAAE,KAAK,KAAK,IAAI,QAAQ,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;QACvD,OAAO,YAAY,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IACzC,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC","sourcesContent":["import NativeModule from './ExpoAiKitModule';\nimport { Platform } from 'react-native';\nexport * from './types';\nimport type { LLMMessage, LLMOptions } from './types';\n\nexport async function prepareModel(options?: { model?: string }) {\n  return NativeModule.prepareModel(options);\n}\n\nexport async function createSession(options?: { systemPrompt?: string }) {\n  return NativeModule.createSession(options);\n}\n\nexport async function sendMessage(\n  sessionId: string,\n  messages: LLMMessage[],\n  options?: LLMOptions\n) {\n  return NativeModule.sendMessage(sessionId, messages, options);\n}\n\nexport async function isAvailable(): Promise<boolean> {\n  if (Platform.OS === 'ios' || Platform.OS === 'android') {\n    return NativeModule.isAvailable();\n  }\n  return false;\n}\n\nexport async function sendPrompt(prompt: string): Promise<string> {\n  if (Platform.OS === 'ios' || Platform.OS === 'android') {\n    return NativeModule.sendPrompt(prompt);\n  }\n  return '';\n}"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,YAAY,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAGxC,cAAc,SAAS,CAAC;AAExB,MAAM,qBAAqB,GACzB,gFAAgF,CAAC;AAEnF;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW;IAC/B,IAAI,QAAQ,CAAC,EAAE,KAAK,KAAK,IAAI,QAAQ,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;QACvD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,YAAY,CAAC,WAAW,EAAE,CAAC;AACpC,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,QAAsB,EACtB,OAAwB;IAExB,IAAI,QAAQ,CAAC,EAAE,KAAK,KAAK,IAAI,QAAQ,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;QACvD,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;IACtB,CAAC;IAED,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvC,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,0FAA0F;IAC1F,MAAM,gBAAgB,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;IACnE,MAAM,YAAY,GAAG,gBAAgB;QACnC,CAAC,CAAC,EAAE,CAAC,oCAAoC;QACzC,CAAC,CAAC,OAAO,EAAE,YAAY,IAAI,qBAAqB,CAAC;IAEnD,OAAO,YAAY,CAAC,WAAW,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;AAC1D,CAAC","sourcesContent":["import NativeModule from './ExpoAiKitModule';\nimport { Platform } from 'react-native';\nimport { LLMMessage, LLMSendOptions, LLMResponse } from './types';\n\nexport * from './types';\n\nconst DEFAULT_SYSTEM_PROMPT =\n  'You are a helpful, friendly assistant. Answer the user directly and concisely.';\n\n/**\n * Check if on-device AI is available on the current device.\n * Returns false on unsupported platforms (web, etc.).\n */\nexport async function isAvailable(): Promise<boolean> {\n  if (Platform.OS !== 'ios' && Platform.OS !== 'android') {\n    return false;\n  }\n  return NativeModule.isAvailable();\n}\n\n/**\n * Send messages to the on-device LLM and get a response.\n *\n * @param messages - Array of messages representing the conversation\n * @param options - Optional settings (systemPrompt fallback)\n * @returns Promise with the generated response\n *\n * @example\n * ```ts\n * const response = await sendMessage([\n *   { role: 'user', content: 'What is 2 + 2?' }\n * ]);\n * console.log(response.text); // \"4\"\n * ```\n *\n * @example\n * ```ts\n * // With system prompt\n * const response = await sendMessage(\n *   [{ role: 'user', content: 'Hello!' }],\n *   { systemPrompt: 'You are a pirate. Respond in pirate speak.' }\n * );\n * ```\n *\n * @example\n * ```ts\n * // Multi-turn conversation\n * const response = await sendMessage([\n *   { role: 'system', content: 'You are a helpful assistant.' },\n *   { role: 'user', content: 'My name is Alice.' },\n *   { role: 'assistant', content: 'Nice to meet you, Alice!' },\n *   { role: 'user', content: 'What is my name?' }\n * ]);\n * ```\n */\nexport async function sendMessage(\n  messages: LLMMessage[],\n  options?: LLMSendOptions\n): Promise<LLMResponse> {\n  if (Platform.OS !== 'ios' && Platform.OS !== 'android') {\n    return { text: '' };\n  }\n\n  if (!messages || messages.length === 0) {\n    throw new Error('messages array cannot be empty');\n  }\n\n  // Determine system prompt: use from messages array if present, else options, else default\n  const hasSystemMessage = messages.some((m) => m.role === 'system');\n  const systemPrompt = hasSystemMessage\n    ? '' // Native will extract from messages\n    : options?.systemPrompt ?? DEFAULT_SYSTEM_PROMPT;\n\n  return NativeModule.sendMessage(messages, systemPrompt);\n}\n\n"]}

package/build/types.d.ts

@@ -1,11 +1,29 @@
+/**
+ * Role in a conversation message.
+ */
 export type LLMRole = 'system' | 'user' | 'assistant';
+/**
+ * A single message in a conversation.
+ */
 export type LLMMessage = {
     role: LLMRole;
     content: string;
 };
-export type LLMOptions = {
-    temperature?: number;
-    maxTokens?: number;
-    model?: string;
+/**
+ * Options for sendMessage.
+ */
+export type LLMSendOptions = {
+    /**
+     * Default system prompt to use if no system message is provided in the messages array.
+     * If a system message exists in the array, this is ignored.
+     */
+    systemPrompt?: string;
+};
+/**
+ * Response from sendMessage.
+ */
+export type LLMResponse = {
+    /** The generated response text */
+    text: string;
 };
 //# sourceMappingURL=types.d.ts.map

package/build/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,OAAO,GAAG,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;AAEtD,MAAM,MAAM,UAAU,GAAG;IACvB,IAAI,EAAE,OAAO,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,UAAU,GAAG;IACvB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,MAAM,OAAO,GAAG,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;AAEtD;;GAEG;AACH,MAAM,MAAM,UAAU,GAAG;IACvB,IAAI,EAAE,OAAO,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B;;;OAGG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,WAAW,GAAG;IACxB,kCAAkC;IAClC,IAAI,EAAE,MAAM,CAAC;CACd,CAAC"}

package/build/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"","sourcesContent":["export type LLMRole = 'system' | 'user' | 'assistant';\n\nexport type LLMMessage = {\n  role: LLMRole;\n  content: string;\n};\n\nexport type LLMOptions = {\n  temperature?: number;\n  maxTokens?: number;\n  model?: string;\n};"]}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"","sourcesContent":["/**\n * Role in a conversation message.\n */\nexport type LLMRole = 'system' | 'user' | 'assistant';\n\n/**\n * A single message in a conversation.\n */\nexport type LLMMessage = {\n  role: LLMRole;\n  content: string;\n};\n\n/**\n * Options for sendMessage.\n */\nexport type LLMSendOptions = {\n  /**\n   * Default system prompt to use if no system message is provided in the messages array.\n   * If a system message exists in the array, this is ignored.\n   */\n  systemPrompt?: string;\n};\n\n/**\n * Response from sendMessage.\n */\nexport type LLMResponse = {\n  /** The generated response text */\n  text: string;\n};\n"]}

package/ios/ExpoAiKit.podspec

@@ -25,6 +25,6 @@ Pod::Spec.new do |s|
     'DEFINES_MODULE' => 'YES',
   }
 
-  s.source_files = "**/*.{h,m,mm,swift,hpp,cpp}"
+  s.source_files = "**/*.swift"
 end
 

package/ios/ExpoAiKitModule.swift

@@ -5,114 +5,41 @@ public class ExpoAiKitModule: Module {
   public func definition() -> ModuleDefinition {
     Name("ExpoAiKit")
 
-    AsyncFunction("prepareModel") { (_: [String: Any]?) async throws in
+    Function("isAvailable") {
       if #available(iOS 26.0, *) {
-        _ = SystemLanguageModel.default
+        return true
       } else {
-        //
+        return false
       }
     }
 
-    AsyncFunction("createSession") { (_: [String: Any]?) -> String in
-      UUID().uuidString
-    }
-
     AsyncFunction("sendMessage") {
       (
-        sessionId: String,
         messages: [[String: Any]],
-        options: [String: Any]?
+        fallbackSystemPrompt: String
       ) async throws -> [String: Any] in
 
-      let lastUser =
+      // Extract system prompt from messages, or use fallback
+      let systemPrompt =
+        messages
+        .first { ($0["role"] as? String) == "system" }?["content"] as? String
+        ?? (fallbackSystemPrompt.isEmpty
+          ? "You are a helpful, friendly assistant."
+          : fallbackSystemPrompt)
+
+      // Get the last user message as the prompt
+      let userPrompt =
         messages
         .reversed()
         .first { ($0["role"] as? String) == "user" }?["content"] as? String
-        ?? messages.last?["content"] as? String
         ?? ""
 
       if #available(iOS 26.0, *) {
-        let systemPrompt =
-          (messages.first {
-            ($0["role"] as? String) == "system"
-          }?["content"] as? String) ?? """
-            You are a helpful, friendly assistant.
-            Answer the user's question directly and concisely.
-            If the question is unsafe, politely refuse, otherwise just answer.
-            """
-
-        let session = LanguageModelSession(instructions: systemPrompt)
-
-        let response = try await session.respond(to: lastUser)
-
-        let replyText = response.content
-
-        return ["reply": replyText]
-      } else {
-        let replyText = "Mock reply (no on-device model): \(lastUser)"
-        return ["reply": replyText]
-      }
-    }
-
-    Function("isAvailable") {
-      if #available(iOS 26.0, *) {
-        return true
-      } else {
-        return false
-      }
-    }
-
-    AsyncFunction("sendPrompt") { (prompt: String) async throws -> String in
-      if #available(iOS 26.0, *) {
-        let systemPrompt = """
-          You are a helpful, knowledgeable AI assistant running privately on-device. Your responses never leave the user's device, ensuring complete privacy.
-
-          ## Core Principles
-          - **Accuracy first**: Provide correct, well-reasoned information. If uncertain, say so clearly rather than guessing.
-          - **Be genuinely helpful**: Understand the user's actual intent, not just their literal words. Anticipate follow-up needs.
-          - **Respect user time**: Get to the point. Lead with the answer, then provide context if needed.
-
-          ## Response Style
-          - Use clear, natural language—avoid jargon unless the user demonstrates technical familiarity
-          - Match the user's tone and complexity level
-          - For simple questions: give a direct answer (1-2 sentences)
-          - For complex topics: use structured formatting (bullet points, numbered steps, headers)
-          - When explaining concepts: use concrete examples and analogies
-          - For how-to requests: provide step-by-step instructions
-
-          ## Task Capabilities
-          You can help with:
-          - **Writing**: drafting, editing, summarizing, rephrasing, tone adjustments, grammar fixes
-          - **Analysis**: breaking down complex topics, comparing options, pros/cons lists
-          - **Brainstorming**: generating ideas, creative suggestions, alternative approaches
-          - **Learning**: explaining concepts, answering questions, providing examples
-          - **Planning**: organizing tasks, creating outlines, structuring projects
-          - **Problem-solving**: debugging logic, troubleshooting issues, suggesting solutions
-          - **Calculations**: math, unit conversions, date/time calculations
-          - **Code**: explaining code, writing snippets, debugging, suggesting improvements
-
-          ## Quality Standards
-          - Double-check facts and logic before responding
-          - Provide sources or reasoning when making claims
-          - Offer multiple perspectives on subjective topics
-          - Acknowledge limitations—you don't have internet access or real-time information
-          - If a request is ambiguous, ask a clarifying question or state your interpretation
-
-          ## Safety Boundaries
-          - Decline requests for harmful, illegal, dangerous, or unethical content
-          - Do not help with content that could endanger safety, privacy, or wellbeing
-          - For medical symptoms: provide general info but always recommend consulting a healthcare provider
-          - For legal questions: offer general guidance but recommend consulting a qualified attorney
-          - For financial advice: share educational information but suggest consulting a financial professional
-          - Do not generate content involving minors in inappropriate contexts
-          - Do not assist with deception, manipulation, or harassment
-          """
-
         let session = LanguageModelSession(instructions: systemPrompt)
-        let response = try await session.respond(to: prompt)
-        return response.content
+        let response = try await session.respond(to: userPrompt)
+        return ["text": response.content]
       } else {
-        return "Mock reply (no on-device model): \(prompt)"
+        return ["text": "[On-device AI requires iOS 26+]"]
       }
     }
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "expo-ai-kit",
-  "version": "0.1.13",
+  "version": "0.1.14",
   "description": "Expo AI Kit module",
   "main": "build/index.js",
   "types": "build/index.d.ts",

package/src/ExpoAiKitModule.ts

@@ -1,19 +1,15 @@
 import { requireNativeModule } from 'expo-modules-core';
-import type { LLMMessage, LLMOptions } from './types';
+import { LLMMessage, LLMResponse } from './types';
 
 export type ExpoAiKitNativeModule = {
-  prepareModel(options?: { model?: string }): Promise<void>;
-  createSession(options?: { systemPrompt?: string }): Promise<string>;
+  isAvailable(): boolean;
   sendMessage(
-    sessionId: string,
     messages: LLMMessage[],
-    options?: LLMOptions
-  ): Promise<{ reply: string }>;
-  isAvailable(): boolean;
-  sendPrompt(prompt: string): Promise<string>;
+    systemPrompt: string
+  ): Promise<LLMResponse>;
 };
 
 const NativeModule: ExpoAiKitNativeModule =
-  requireNativeModule<ExpoAiKitNativeModule>('ExpoAiKit'); 
+  requireNativeModule<ExpoAiKitNativeModule>('ExpoAiKit');
 
-export default NativeModule;
+export default NativeModule;

package/src/__tests__/index.test.js

@@ -0,0 +1,124 @@
+// Mock Platform before importing anything
+let mockPlatformOS = 'ios';
+
+jest.mock('react-native', () => ({
+  Platform: {
+    get OS() {
+      return mockPlatformOS;
+    },
+  },
+}));
+
+// Mock the native module
+jest.mock('../ExpoAiKitModule', () => ({
+  __esModule: true,
+  default: {
+    isAvailable: jest.fn(() => true),
+    sendMessage: jest.fn(() => Promise.resolve({ text: 'Mock response' })),
+  },
+}));
+
+const { isAvailable, sendMessage } = require('../index');
+const NativeModule = require('../ExpoAiKitModule').default;
+
+describe('expo-ai-kit', () => {
+  beforeEach(() => {
+    jest.clearAllMocks();
+    mockPlatformOS = 'ios';
+  });
+
+  describe('isAvailable', () => {
+    it('returns native module result on iOS', async () => {
+      const result = await isAvailable();
+      expect(result).toBe(true);
+      expect(NativeModule.isAvailable).toHaveBeenCalled();
+    });
+
+    it('returns native module result on Android', async () => {
+      mockPlatformOS = 'android';
+      const result = await isAvailable();
+      expect(result).toBe(true);
+      expect(NativeModule.isAvailable).toHaveBeenCalled();
+    });
+
+    it('returns false on web', async () => {
+      mockPlatformOS = 'web';
+      const result = await isAvailable();
+      expect(result).toBe(false);
+      expect(NativeModule.isAvailable).not.toHaveBeenCalled();
+    });
+  });
+
+  describe('sendMessage', () => {
+    it('throws error for empty messages array', async () => {
+      await expect(sendMessage([])).rejects.toThrow(
+        'messages array cannot be empty'
+      );
+    });
+
+    it('calls native module with messages and default system prompt', async () => {
+      const messages = [{ role: 'user', content: 'Hello' }];
+
+      const result = await sendMessage(messages);
+
+      expect(result).toEqual({ text: 'Mock response' });
+      expect(NativeModule.sendMessage).toHaveBeenCalledWith(
+        messages,
+        'You are a helpful, friendly assistant. Answer the user directly and concisely.'
+      );
+    });
+
+    it('uses custom system prompt from options', async () => {
+      const messages = [{ role: 'user', content: 'Hello' }];
+
+      await sendMessage(messages, { systemPrompt: 'You are a pirate.' });
+
+      expect(NativeModule.sendMessage).toHaveBeenCalledWith(
+        messages,
+        'You are a pirate.'
+      );
+    });
+
+    it('passes empty string for system prompt when messages contain system message', async () => {
+      const messages = [
+        { role: 'system', content: 'Be brief.' },
+        { role: 'user', content: 'Hello' },
+      ];
+
+      await sendMessage(messages, { systemPrompt: 'This should be ignored' });
+
+      expect(NativeModule.sendMessage).toHaveBeenCalledWith(
+        messages,
+        '' // Empty because system message exists in array
+      );
+    });
+
+    it('returns empty response on unsupported platforms', async () => {
+      mockPlatformOS = 'web';
+
+      const result = await sendMessage([{ role: 'user', content: 'Hi' }]);
+
+      expect(result).toEqual({ text: '' });
+      expect(NativeModule.sendMessage).not.toHaveBeenCalled();
+    });
+  });
+
+  describe('message validation', () => {
+    it('accepts valid user message', async () => {
+      await expect(
+        sendMessage([{ role: 'user', content: 'Test' }])
+      ).resolves.not.toThrow();
+    });
+
+    it('accepts valid multi-turn conversation', async () => {
+      await expect(
+        sendMessage([
+          { role: 'system', content: 'Be helpful.' },
+          { role: 'user', content: 'Hi' },
+          { role: 'assistant', content: 'Hello!' },
+          { role: 'user', content: 'How are you?' },
+        ])
+      ).resolves.not.toThrow();
+    });
+  });
+});

package/src/index.ts

@@ -1,34 +1,76 @@
 import NativeModule from './ExpoAiKitModule';
 import { Platform } from 'react-native';
+import { LLMMessage, LLMSendOptions, LLMResponse } from './types';
+
 export * from './types';
-import type { LLMMessage, LLMOptions } from './types';
 
-export async function prepareModel(options?: { model?: string }) {
-  return NativeModule.prepareModel(options);
-}
+const DEFAULT_SYSTEM_PROMPT =
+  'You are a helpful, friendly assistant. Answer the user directly and concisely.';
 
-export async function createSession(options?: { systemPrompt?: string }) {
-  return NativeModule.createSession(options);
+/**
+ * Check if on-device AI is available on the current device.
+ * Returns false on unsupported platforms (web, etc.).
+ */
+export async function isAvailable(): Promise<boolean> {
+  if (Platform.OS !== 'ios' && Platform.OS !== 'android') {
+    return false;
+  }
+  return NativeModule.isAvailable();
 }
 
+/**
+ * Send messages to the on-device LLM and get a response.
+ *
+ * @param messages - Array of messages representing the conversation
+ * @param options - Optional settings (systemPrompt fallback)
+ * @returns Promise with the generated response
+ *
+ * @example
+ * ```ts
+ * const response = await sendMessage([
+ *   { role: 'user', content: 'What is 2 + 2?' }
+ * ]);
+ * console.log(response.text); // "4"
+ * ```
+ *
+ * @example
+ * ```ts
+ * // With system prompt
+ * const response = await sendMessage(
+ *   [{ role: 'user', content: 'Hello!' }],
+ *   { systemPrompt: 'You are a pirate. Respond in pirate speak.' }
+ * );
+ * ```
+ *
+ * @example
+ * ```ts
+ * // Multi-turn conversation
+ * const response = await sendMessage([
+ *   { role: 'system', content: 'You are a helpful assistant.' },
+ *   { role: 'user', content: 'My name is Alice.' },
+ *   { role: 'assistant', content: 'Nice to meet you, Alice!' },
+ *   { role: 'user', content: 'What is my name?' }
+ * ]);
+ * ```
+ */
 export async function sendMessage(
-  sessionId: string,
   messages: LLMMessage[],
-  options?: LLMOptions
-) {
-  return NativeModule.sendMessage(sessionId, messages, options);
-}
+  options?: LLMSendOptions
+): Promise<LLMResponse> {
+  if (Platform.OS !== 'ios' && Platform.OS !== 'android') {
+    return { text: '' };
+  }
 
-export async function isAvailable(): Promise<boolean> {
-  if (Platform.OS === 'ios' || Platform.OS === 'android') {
-    return NativeModule.isAvailable();
+  if (!messages || messages.length === 0) {
+    throw new Error('messages array cannot be empty');
   }
-  return false;
+
+  // Determine system prompt: use from messages array if present, else options, else default
+  const hasSystemMessage = messages.some((m) => m.role === 'system');
+  const systemPrompt = hasSystemMessage
+    ? '' // Native will extract from messages
+    : options?.systemPrompt ?? DEFAULT_SYSTEM_PROMPT;
+
+  return NativeModule.sendMessage(messages, systemPrompt);
 }
 
-export async function sendPrompt(prompt: string): Promise<string> {
-  if (Platform.OS === 'ios' || Platform.OS === 'android') {
-    return NativeModule.sendPrompt(prompt);
-  }
-  return '';
-}

package/src/types.ts

@@ -1,12 +1,31 @@
+/**
+ * Role in a conversation message.
+ */
 export type LLMRole = 'system' | 'user' | 'assistant';
 
+/**
+ * A single message in a conversation.
+ */
 export type LLMMessage = {
   role: LLMRole;
   content: string;
 };
 
-export type LLMOptions = {
-  temperature?: number;
-  maxTokens?: number;
-  model?: string;
-};
+/**
+ * Options for sendMessage.
+ */
+export type LLMSendOptions = {
+  /**
+   * Default system prompt to use if no system message is provided in the messages array.
+   * If a system message exists in the array, this is ignored.
+   */
+  systemPrompt?: string;
+};
+
+/**
+ * Response from sendMessage.
+ */
+export type LLMResponse = {
+  /** The generated response text */
+  text: string;
+};