expo-ai-kit 0.1.20 → 0.1.22

package/README.md

@@ -32,6 +32,7 @@ On-device AI for Expo apps. Run language models locally—no API keys, no cloud,
 - **Simple API** — Core functions plus prompt helpers for common tasks
 - **Prompt helpers** — Built-in `summarize()`, `translate()`, `rewrite()`, and more
 - **Smart suggestions** — `suggest()`, `smartReply()`, and `autocomplete()` for predictive text
+- **React Hooks** — `useChat`, `useCompletion`, and `useOnDeviceAI` for plug-and-play integration
 - **Chat memory** — Built-in `ChatMemoryManager` for managing conversation history
 
 ## Requirements
@@ -290,6 +291,95 @@ const { promise, stop } = streamSummarize(
 );
 ```
 
+### React Hooks
+
+expo-ai-kit provides React hooks that handle state management, streaming, and conversation memory automatically.
+
+#### `useChat` — Full Chat Interface
+
+The easiest way to build a chat UI. Manages messages, input, streaming, and memory for you:
+
+```tsx
+import { useChat } from 'expo-ai-kit';
+
+function ChatScreen() {
+  const { messages, input, setInput, sendMessage, isStreaming, stop, clear, error } = useChat({
+    systemPrompt: 'You are a helpful assistant.',
+    maxTurns: 10,
+  });
+
+  return (
+    <View style={{ flex: 1 }}>
+      <FlatList
+        data={messages}
+        keyExtractor={(_, i) => i.toString()}
+        renderItem={({ item }) => (
+          <Text>{item.role}: {item.content}</Text>
+        )}
+      />
+      <TextInput value={input} onChangeText={setInput} />
+      {isStreaming ? (
+        <Button title="Stop" onPress={stop} />
+      ) : (
+        <Button title="Send" onPress={() => sendMessage()} />
+      )}
+      <Button title="Clear" onPress={clear} />
+    </View>
+  );
+}
+```
+
+You can also send a message programmatically:
+
+```tsx
+// Send custom text instead of current input
+sendMessage('What is the weather today?');
+```
+
+#### `useCompletion` — Single-shot Completions
+
+For one-off tasks like summarization, translation, or content generation (no conversation history):
+
+```tsx
+import { useCompletion } from 'expo-ai-kit';
+
+function Summarizer() {
+  const { completion, isLoading, complete, stop, error } = useCompletion({
+    systemPrompt: 'Summarize the given text concisely.',
+  });
+
+  return (
+    <View>
+      <Button
+        title="Summarize"
+        onPress={() => complete('Long article text here...')}
+      />
+      {isLoading && <Button title="Stop" onPress={stop} />}
+      <Text>{completion}</Text>
+    </View>
+  );
+}
+```
+
+#### `useOnDeviceAI` — Availability Check
+
+A simple hook to check if on-device AI is available, with caching across components:
+
+```tsx
+import { useOnDeviceAI } from 'expo-ai-kit';
+
+function App() {
+  const { isAvailable, isChecking } = useOnDeviceAI();
+
+  if (isChecking) return <Text>Checking AI availability...</Text>;
+  if (!isAvailable) return <Text>On-device AI not available</Text>;
+
+  return <ChatScreen />;
+}
+```
+
+---
+
 ### Streaming with Cancel Button
 
 ```tsx
@@ -668,6 +758,80 @@ function parseSuggestResponse(raw: string): LLMSuggestResponse
 
 ---
 
+### `useChat(options?)`
+
+React hook for building chat interfaces. Manages messages, input, streaming, and conversation memory automatically.
+
+```typescript
+function useChat(options?: UseChatOptions): UseChatReturn
+```
+
+| Option | Type | Description |
+|--------|------|-------------|
+| `systemPrompt` | `string` | System prompt for the AI assistant |
+| `maxTurns` | `number` | Maximum conversation turns to keep in memory (default: `10`) |
+| `initialMessages` | `LLMMessage[]` | Initial messages to populate the chat |
+| `onFinish` | `(response: LLMResponse) => void` | Callback when a response is complete |
+| `onError` | `(error: Error) => void` | Callback when an error occurs |
+
+**Returns:**
+
+| Property | Type | Description |
+|----------|------|-------------|
+| `messages` | `LLMMessage[]` | All messages in the conversation |
+| `input` | `string` | Current input text value |
+| `setInput` | `(input: string) => void` | Set the input text value |
+| `sendMessage` | `(text?: string) => Promise<void>` | Send the current input (or provided text) |
+| `isStreaming` | `boolean` | Whether the AI is currently streaming |
+| `stop` | `() => void` | Stop the current streaming response |
+| `clear` | `() => void` | Clear all messages and reset |
+| `error` | `Error \| null` | The most recent error, if any |
+
+---
+
+### `useCompletion(options?)`
+
+React hook for single-shot AI completions (summarization, translation, etc.).
+
+```typescript
+function useCompletion(options?: UseCompletionOptions): UseCompletionReturn
+```
+
+| Option | Type | Description |
+|--------|------|-------------|
+| `systemPrompt` | `string` | System prompt for the AI |
+| `onFinish` | `(response: LLMResponse) => void` | Callback when completion is done |
+| `onError` | `(error: Error) => void` | Callback when an error occurs |
+
+**Returns:**
+
+| Property | Type | Description |
+|----------|------|-------------|
+| `completion` | `string` | The current completion text |
+| `isLoading` | `boolean` | Whether a completion is in progress |
+| `complete` | `(prompt: string) => Promise<string>` | Request a completion |
+| `stop` | `() => void` | Stop the current completion |
+| `error` | `Error \| null` | The most recent error, if any |
+
+---
+
+### `useOnDeviceAI()`
+
+React hook to check if on-device AI is available. Caches the result across components.
+
+```typescript
+function useOnDeviceAI(): UseOnDeviceAIReturn
+```
+
+**Returns:**
+
+| Property | Type | Description |
+|----------|------|-------------|
+| `isAvailable` | `boolean` | Whether on-device AI is available |
+| `isChecking` | `boolean` | Whether the check is still in progress |
+
+---
+
 ### `ChatMemoryManager`
 
 Manages conversation history for stateless on-device AI models. Automatically handles turn limits and provides the full message array for each request.
@@ -851,6 +1015,45 @@ type LLMSuggestResponse = {
   raw: string;
 };
 
+// Hook Types
+type UseChatOptions = {
+  systemPrompt?: string;
+  maxTurns?: number;
+  initialMessages?: LLMMessage[];
+  onFinish?: (response: LLMResponse) => void;
+  onError?: (error: Error) => void;
+};
+
+type UseChatReturn = {
+  messages: LLMMessage[];
+  input: string;
+  setInput: (input: string) => void;
+  sendMessage: (text?: string) => Promise<void>;
+  isStreaming: boolean;
+  stop: () => void;
+  clear: () => void;
+  error: Error | null;
+};
+
+type UseCompletionOptions = {
+  systemPrompt?: string;
+  onFinish?: (response: LLMResponse) => void;
+  onError?: (error: Error) => void;
+};
+
+type UseCompletionReturn = {
+  completion: string;
+  isLoading: boolean;
+  complete: (prompt: string) => Promise<string>;
+  stop: () => void;
+  error: Error | null;
+};
+
+type UseOnDeviceAIReturn = {
+  isAvailable: boolean;
+  isChecking: boolean;
+};
+
 // Chat Memory Types
 type ChatMemoryOptions = {
   /** Maximum conversation turns to keep (default: 10) */
@@ -877,6 +1080,7 @@ type ChatMemorySnapshot = {
 | Prompt helpers | ✅ | ✅ |
 | Smart suggestions | ✅ | ✅ |
 | `ChatMemoryManager` | ✅ | ✅ |
+| React Hooks (`useChat`, etc.) | ✅ | ✅ |
 | System prompts | ✅ Native | ✅ Prepended |
 | Multi-turn context | ✅ | ✅ |
 | Cancel streaming | ✅ | ✅ |
@@ -930,10 +1134,11 @@ const { text } = await sendMessage(messages, { systemPrompt: '...' });
 | ✅ Prompt helpers (summarize, translate, etc.) | Done | - |
 | ✅ Chat memory management | Done | - |
 | ✅ Smart suggestions (suggest, smartReply, autocomplete) | Done | - |
+| ✅ React Hooks (useChat, useCompletion, useOnDeviceAI) | Done | - |
 | Web/generic fallback | Idea | Medium |
 | Configurable hyperparameters (temperature, etc.) | Idea | Low |
 
-Have a feature request? [Open an issue](https://github.com/laraelmas/expo-ai-kit/issues)!
+Have a feature request? [Open an issue](https://github.com/saidkaban/expo-ai-kit/issues)!
 
 ## License
 
@@ -947,6 +1152,6 @@ Contributions are welcome! Please refer to guidelines described in the [contribu
 
 - [Documentation](https://expo-ai-kit.com)
 - [npm package](https://www.npmjs.com/package/expo-ai-kit)
-- [GitHub repository](https://github.com/laraelmas/expo-ai-kit)
+- [GitHub repository](https://github.com/saidkaban/expo-ai-kit)
 - [Apple Foundation Models](https://developer.apple.com/documentation/foundationmodels)
 - [Google ML Kit Prompt API](https://developers.google.com/ml-kit/genai)

package/build/hooks.d.ts

@@ -0,0 +1,88 @@
+import type { UseChatOptions, UseChatReturn, UseCompletionOptions, UseCompletionReturn, UseOnDeviceAIReturn } from './types';
+/**
+ * React hook to check if on-device AI is available.
+ *
+ * Caches the result so multiple components don't re-check.
+ *
+ * @returns Object with isAvailable and isChecking states
+ *
+ * @example
+ * ```tsx
+ * function MyComponent() {
+ *   const { isAvailable, isChecking } = useOnDeviceAI();
+ *
+ *   if (isChecking) return <Text>Checking AI availability...</Text>;
+ *   if (!isAvailable) return <Text>On-device AI not available</Text>;
+ *
+ *   return <ChatComponent />;
+ * }
+ * ```
+ */
+export declare function useOnDeviceAI(): UseOnDeviceAIReturn;
+/**
+ * React hook for building chat interfaces with on-device AI.
+ *
+ * Manages messages, input state, streaming, and conversation memory automatically.
+ * Built on top of ChatMemoryManager and streamMessage.
+ *
+ * @param options - Configuration options for the chat
+ * @returns Chat state and control functions
+ *
+ * @example
+ * ```tsx
+ * function ChatScreen() {
+ *   const { messages, input, setInput, sendMessage, isStreaming, stop } = useChat({
+ *     systemPrompt: 'You are a helpful assistant.',
+ *   });
+ *
+ *   return (
+ *     <View>
+ *       <FlatList
+ *         data={messages}
+ *         renderItem={({ item }) => (
+ *           <Text>{item.role}: {item.content}</Text>
+ *         )}
+ *       />
+ *       <TextInput value={input} onChangeText={setInput} />
+ *       {isStreaming ? (
+ *         <Button title="Stop" onPress={stop} />
+ *       ) : (
+ *         <Button title="Send" onPress={() => sendMessage()} />
+ *       )}
+ *     </View>
+ *   );
+ * }
+ * ```
+ */
+export declare function useChat(options?: UseChatOptions): UseChatReturn;
+/**
+ * React hook for single-shot AI completions.
+ *
+ * Unlike useChat (which manages a conversation), useCompletion is for
+ * one-off tasks like summarization, translation, or content generation.
+ *
+ * @param options - Configuration options
+ * @returns Completion state and control functions
+ *
+ * @example
+ * ```tsx
+ * function SummarizerScreen() {
+ *   const { completion, isLoading, complete, stop } = useCompletion({
+ *     systemPrompt: 'You are a summarization assistant. Summarize the given text concisely.',
+ *   });
+ *
+ *   return (
+ *     <View>
+ *       <Button
+ *         title="Summarize"
+ *         onPress={() => complete('Long article text here...')}
+ *       />
+ *       {isLoading && <Button title="Stop" onPress={stop} />}
+ *       <Text>{completion}</Text>
+ *     </View>
+ *   );
+ * }
+ * ```
+ */
+export declare function useCompletion(options?: UseCompletionOptions): UseCompletionReturn;
+//# sourceMappingURL=hooks.d.ts.map

package/build/hooks.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"hooks.d.ts","sourceRoot":"","sources":["../src/hooks.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAEV,cAAc,EACd,aAAa,EACb,oBAAoB,EACpB,mBAAmB,EACnB,mBAAmB,EACpB,MAAM,SAAS,CAAC;AAmBjB;;;;;;;;;;;;;;;;;;GAkBG;AACH,wBAAgB,aAAa,IAAI,mBAAmB,CAwBnD;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACH,wBAAgB,OAAO,CAAC,OAAO,GAAE,cAAmB,GAAG,aAAa,CA8GnE;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,wBAAgB,aAAa,CAAC,OAAO,GAAE,oBAAyB,GAAG,mBAAmB,CAgFrF"}

package/build/hooks.js

@@ -0,0 +1,293 @@
+import { useState, useEffect, useRef, useCallback } from 'react';
+import { isAvailable, streamMessage } from './index';
+import { ChatMemoryManager } from './memory';
+// Cache availability result across all hook instances
+let availabilityCache = null;
+let availabilityPromise = null;
+function checkAvailability() {
+    if (availabilityCache !== null) {
+        return Promise.resolve(availabilityCache);
+    }
+    if (!availabilityPromise) {
+        availabilityPromise = isAvailable().then((result) => {
+            availabilityCache = result;
+            return result;
+        });
+    }
+    return availabilityPromise;
+}
+/**
+ * React hook to check if on-device AI is available.
+ *
+ * Caches the result so multiple components don't re-check.
+ *
+ * @returns Object with isAvailable and isChecking states
+ *
+ * @example
+ * ```tsx
+ * function MyComponent() {
+ *   const { isAvailable, isChecking } = useOnDeviceAI();
+ *
+ *   if (isChecking) return <Text>Checking AI availability...</Text>;
+ *   if (!isAvailable) return <Text>On-device AI not available</Text>;
+ *
+ *   return <ChatComponent />;
+ * }
+ * ```
+ */
+export function useOnDeviceAI() {
+    const [available, setAvailable] = useState(availabilityCache ?? false);
+    const [checking, setChecking] = useState(availabilityCache === null);
+    useEffect(() => {
+        if (availabilityCache !== null) {
+            setAvailable(availabilityCache);
+            setChecking(false);
+            return;
+        }
+        let mounted = true;
+        checkAvailability().then((result) => {
+            if (mounted) {
+                setAvailable(result);
+                setChecking(false);
+            }
+        });
+        return () => {
+            mounted = false;
+        };
+    }, []);
+    return { isAvailable: available, isChecking: checking };
+}
+/**
+ * React hook for building chat interfaces with on-device AI.
+ *
+ * Manages messages, input state, streaming, and conversation memory automatically.
+ * Built on top of ChatMemoryManager and streamMessage.
+ *
+ * @param options - Configuration options for the chat
+ * @returns Chat state and control functions
+ *
+ * @example
+ * ```tsx
+ * function ChatScreen() {
+ *   const { messages, input, setInput, sendMessage, isStreaming, stop } = useChat({
+ *     systemPrompt: 'You are a helpful assistant.',
+ *   });
+ *
+ *   return (
+ *     <View>
+ *       <FlatList
+ *         data={messages}
+ *         renderItem={({ item }) => (
+ *           <Text>{item.role}: {item.content}</Text>
+ *         )}
+ *       />
+ *       <TextInput value={input} onChangeText={setInput} />
+ *       {isStreaming ? (
+ *         <Button title="Stop" onPress={stop} />
+ *       ) : (
+ *         <Button title="Send" onPress={() => sendMessage()} />
+ *       )}
+ *     </View>
+ *   );
+ * }
+ * ```
+ */
+export function useChat(options = {}) {
+    const { systemPrompt, maxTurns, initialMessages, onFinish, onError } = options;
+    const memoryRef = useRef(new ChatMemoryManager({ maxTurns, systemPrompt }));
+    // Initialize with initial messages
+    const initializedRef = useRef(false);
+    if (!initializedRef.current && initialMessages) {
+        for (const msg of initialMessages) {
+            memoryRef.current.addMessage(msg);
+        }
+        initializedRef.current = true;
+    }
+    const [messages, setMessages] = useState(memoryRef.current.getMessages());
+    const [input, setInput] = useState('');
+    const [isStreaming, setIsStreaming] = useState(false);
+    const [error, setError] = useState(null);
+    const stopRef = useRef(null);
+    const mountedRef = useRef(true);
+    const inputRef = useRef(input);
+    inputRef.current = input;
+    const streamingRef = useRef(isStreaming);
+    streamingRef.current = isStreaming;
+    const onFinishRef = useRef(onFinish);
+    onFinishRef.current = onFinish;
+    const onErrorRef = useRef(onError);
+    onErrorRef.current = onError;
+    useEffect(() => {
+        return () => {
+            mountedRef.current = false;
+            stopRef.current?.();
+        };
+    }, []);
+    const send = useCallback(async (text) => {
+        const content = (text ?? inputRef.current).trim();
+        if (!content || streamingRef.current)
+            return;
+        setError(null);
+        // Add user message
+        memoryRef.current.addUserMessage(content);
+        setMessages([...memoryRef.current.getMessages()]);
+        if (!text)
+            setInput('');
+        setIsStreaming(true);
+        try {
+            const allMessages = memoryRef.current.getAllMessages();
+            // Use a temporary message for streaming display
+            let accumulatedText = '';
+            const { promise, stop } = streamMessage(allMessages, (event) => {
+                if (!mountedRef.current)
+                    return;
+                accumulatedText = event.accumulatedText;
+                // Update messages with streaming assistant response
+                const currentMessages = memoryRef.current.getMessages();
+                setMessages([
+                    ...currentMessages,
+                    { role: 'assistant', content: accumulatedText },
+                ]);
+            });
+            stopRef.current = stop;
+            const response = await promise;
+            if (!mountedRef.current)
+                return;
+            // Add the final assistant message to memory
+            memoryRef.current.addAssistantMessage(response.text);
+            setMessages([...memoryRef.current.getMessages()]);
+            onFinishRef.current?.(response);
+        }
+        catch (err) {
+            if (!mountedRef.current)
+                return;
+            const e = err instanceof Error ? err : new Error(String(err));
+            setError(e);
+            onErrorRef.current?.(e);
+        }
+        finally {
+            stopRef.current = null;
+            if (mountedRef.current) {
+                setIsStreaming(false);
+            }
+        }
+    }, []);
+    const stop = useCallback(() => {
+        stopRef.current?.();
+    }, []);
+    const clear = useCallback(() => {
+        memoryRef.current.clear();
+        setMessages([]);
+        setError(null);
+    }, []);
+    return {
+        messages,
+        input,
+        setInput,
+        sendMessage: send,
+        isStreaming,
+        stop,
+        clear,
+        error,
+    };
+}
+/**
+ * React hook for single-shot AI completions.
+ *
+ * Unlike useChat (which manages a conversation), useCompletion is for
+ * one-off tasks like summarization, translation, or content generation.
+ *
+ * @param options - Configuration options
+ * @returns Completion state and control functions
+ *
+ * @example
+ * ```tsx
+ * function SummarizerScreen() {
+ *   const { completion, isLoading, complete, stop } = useCompletion({
+ *     systemPrompt: 'You are a summarization assistant. Summarize the given text concisely.',
+ *   });
+ *
+ *   return (
+ *     <View>
+ *       <Button
+ *         title="Summarize"
+ *         onPress={() => complete('Long article text here...')}
+ *       />
+ *       {isLoading && <Button title="Stop" onPress={stop} />}
+ *       <Text>{completion}</Text>
+ *     </View>
+ *   );
+ * }
+ * ```
+ */
+export function useCompletion(options = {}) {
+    const { systemPrompt, onFinish, onError } = options;
+    const [completion, setCompletion] = useState('');
+    const [isLoading, setIsLoading] = useState(false);
+    const [error, setError] = useState(null);
+    const stopRef = useRef(null);
+    const mountedRef = useRef(true);
+    const loadingRef = useRef(isLoading);
+    loadingRef.current = isLoading;
+    const completionRef = useRef(completion);
+    completionRef.current = completion;
+    const systemPromptRef = useRef(systemPrompt);
+    systemPromptRef.current = systemPrompt;
+    const onFinishRef = useRef(onFinish);
+    onFinishRef.current = onFinish;
+    const onErrorRef = useRef(onError);
+    onErrorRef.current = onError;
+    useEffect(() => {
+        return () => {
+            mountedRef.current = false;
+            stopRef.current?.();
+        };
+    }, []);
+    const complete = useCallback(async (prompt) => {
+        if (loadingRef.current)
+            return completionRef.current;
+        setError(null);
+        setCompletion('');
+        setIsLoading(true);
+        try {
+            const messages = [{ role: 'user', content: prompt }];
+            const { promise, stop } = streamMessage(messages, (event) => {
+                if (!mountedRef.current)
+                    return;
+                setCompletion(event.accumulatedText);
+            }, systemPromptRef.current ? { systemPrompt: systemPromptRef.current } : undefined);
+            stopRef.current = stop;
+            const response = await promise;
+            if (!mountedRef.current)
+                return '';
+            setCompletion(response.text);
+            onFinishRef.current?.(response);
+            return response.text;
+        }
+        catch (err) {
+            if (!mountedRef.current)
+                return '';
+            const e = err instanceof Error ? err : new Error(String(err));
+            setError(e);
+            onErrorRef.current?.(e);
+            return '';
+        }
+        finally {
+            stopRef.current = null;
+            if (mountedRef.current) {
+                setIsLoading(false);
+            }
+        }
+    }, []);
+    const stop = useCallback(() => {
+        stopRef.current?.();
+    }, []);
+    return {
+        completion,
+        isLoading,
+        complete,
+        stop,
+        error,
+    };
+}
+//# sourceMappingURL=hooks.js.map

package/build/hooks.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hooks.js","sourceRoot":"","sources":["../src/hooks.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACjE,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAU7C,sDAAsD;AACtD,IAAI,iBAAiB,GAAmB,IAAI,CAAC;AAC7C,IAAI,mBAAmB,GAA4B,IAAI,CAAC;AAExD,SAAS,iBAAiB;IACxB,IAAI,iBAAiB,KAAK,IAAI,EAAE,CAAC;QAC/B,OAAO,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAC5C,CAAC;IACD,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACzB,mBAAmB,GAAG,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YAClD,iBAAiB,GAAG,MAAM,CAAC;YAC3B,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;IACL,CAAC;IACD,OAAO,mBAAmB,CAAC;AAC7B,CAAC;AAED;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAM,UAAU,aAAa;IAC3B,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAC,iBAAiB,IAAI,KAAK,CAAC,CAAC;IACvE,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAC,iBAAiB,KAAK,IAAI,CAAC,CAAC;IAErE,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,iBAAiB,KAAK,IAAI,EAAE,CAAC;YAC/B,YAAY,CAAC,iBAAiB,CAAC,CAAC;YAChC,WAAW,CAAC,KAAK,CAAC,CAAC;YACnB,OAAO;QACT,CAAC;QAED,IAAI,OAAO,GAAG,IAAI,CAAC;QACnB,iBAAiB,EAAE,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YAClC,IAAI,OAAO,EAAE,CAAC;gBACZ,YAAY,CAAC,MAAM,CAAC,CAAC;gBACrB,WAAW,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;QACH,CAAC,CAAC,CAAC;QACH,OAAO,GAAG,EAAE;YACV,OAAO,GAAG,KAAK,CAAC;QAClB,CAAC,CAAC;IACJ,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC;AAC1D,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkCG;AACH,MAAM,UAAU,OAAO,CAAC,UAA0B,EAAE;IAClD,MAAM,EAAE,YAAY,EAAE,QAAQ,EAAE,eAAe,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;IAE/E,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,iBAAiB,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;IAE5E,mCAAmC;IACnC,MAAM,cAAc,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IACrC,IAAI,CAAC,cAAc,CAAC,OAAO,IAAI,eAAe,EAAE,CAAC;QAC/C,KAAK,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;YAClC,SAAS,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QACpC,CAAC;QACD,cAAc,CAAC,OAAO,GAAG,IAAI,CAAC;IAChC,CAAC;IAED,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAe,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;IACxF,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;IACvC,MAAM,CAAC,WAAW,EAAE,cAAc,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IACtD,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAe,IAAI,CAAC,CAAC;IACvD,MAAM,OAAO,GAAG,MAAM,CAAsB,IAAI,CAAC,CAAC;IAClD,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;IAChC,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC/B,QAAQ,CAAC,OAAO,GAAG,KAAK,CAAC;IACzB,MAAM,YAAY,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC;IACzC,YAAY,CAAC,OAAO,GAAG,WAAW,CAAC;IACnC,MAAM,WAAW,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;IACrC,WAAW,CAAC,OAAO,GAAG,QAAQ,CAAC;IAC/B,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;IACnC,UAAU,CAAC,OAAO,GAAG,OAAO,CAAC;IAE7B,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,GAAG,EAAE;YACV,UAAU,CAAC,OAAO,GAAG,KAAK,CAAC;YAC3B,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;QACtB,CAAC,CAAC;IACJ,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,IAAI,GAAG,WAAW,CACtB,KAAK,EAAE,IAAa,EAAE,EAAE;QACtB,MAAM,OAAO,GAAG,CAAC,IAAI,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC;QAClD,IAAI,CAAC,OAAO,IAAI,YAAY,CAAC,OAAO;YAAE,OAAO;QAE7C,QAAQ,CAAC,IAAI,CAAC,CAAC;QAEf,mBAAmB;QACnB,SAAS,CAAC,OAAO,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAC1C,WAAW,CAAC,CAAC,GAAG,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;QAClD,IAAI,CAAC,IAAI;YAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;QAExB,cAAc,CAAC,IAAI,CAAC,CAAC;QAErB,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,SAAS,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;YAEvD,gDAAgD;YAChD,IAAI,eAAe,GAAG,EAAE,CAAC;YACzB,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC,KAAK,EAAE,EAAE;gBAC7D,IAAI,CAAC,UAAU,CAAC,OAAO;oBAAE,OAAO;gBAChC,eAAe,GAAG,KAAK,CAAC,eAAe,CAAC;gBACxC,oDAAoD;gBACpD,MAAM,eAAe,GAAG,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;gBACxD,WAAW,CAAC;oBACV,GAAG,eAAe;oBAClB,EAAE,IAAI,EAAE,WAAoB,EAAE,OAAO,EAAE,eAAe,EAAE;iBACzD,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC;YACvB,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC;YAE/B,IAAI,CAAC,UAAU,CAAC,OAAO;gBAAE,OAAO;YAEhC,4CAA4C;YAC5C,SAAS,CAAC,OAAO,CAAC,mBAAmB,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACrD,WAAW,CAAC,CAAC,GAAG,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;YAClD,WAAW,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,UAAU,CAAC,OAAO;gBAAE,OAAO;YAChC,MAAM,CAAC,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAC9D,QAAQ,CAAC,CAAC,CAAC,CAAC;YACZ,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1B,CAAC;gBAAS,CAAC;YACT,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC;YACvB,IAAI,UAAU,CAAC,OAAO,EAAE,CAAC;gBACvB,cAAc,CAAC,KAAK,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;IACH,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,IAAI,GAAG,WAAW,CAAC,GAAG,EAAE;QAC5B,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;IACtB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;QAC7B,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAC1B,WAAW,CAAC,EAAE,CAAC,CAAC;QAChB,QAAQ,CAAC,IAAI,CAAC,CAAC;IACjB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO;QACL,QAAQ;QACR,KAAK;QACL,QAAQ;QACR,WAAW,EAAE,IAAI;QACjB,WAAW;QACX,IAAI;QACJ,KAAK;QACL,KAAK;KACN,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,MAAM,UAAU,aAAa,CAAC,UAAgC,EAAE;IAC9D,MAAM,EAAE,YAAY,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;IAEpD,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjD,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAClD,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAe,IAAI,CAAC,CAAC;IACvD,MAAM,OAAO,GAAG,MAAM,CAAsB,IAAI,CAAC,CAAC;IAClD,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;IAChC,MAAM,UAAU,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;IACrC,UAAU,CAAC,OAAO,GAAG,SAAS,CAAC;IAC/B,MAAM,aAAa,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;IACzC,aAAa,CAAC,OAAO,GAAG,UAAU,CAAC;IACnC,MAAM,eAAe,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;IAC7C,eAAe,CAAC,OAAO,GAAG,YAAY,CAAC;IACvC,MAAM,WAAW,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;IACrC,WAAW,CAAC,OAAO,GAAG,QAAQ,CAAC;IAC/B,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;IACnC,UAAU,CAAC,OAAO,GAAG,OAAO,CAAC;IAE7B,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,GAAG,EAAE;YACV,UAAU,CAAC,OAAO,GAAG,KAAK,CAAC;YAC3B,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;QACtB,CAAC,CAAC;IACJ,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,QAAQ,GAAG,WAAW,CAC1B,KAAK,EAAE,MAAc,EAAE,EAAE;QACvB,IAAI,UAAU,CAAC,OAAO;YAAE,OAAO,aAAa,CAAC,OAAO,CAAC;QAErD,QAAQ,CAAC,IAAI,CAAC,CAAC;QACf,aAAa,CAAC,EAAE,CAAC,CAAC;QAClB,YAAY,CAAC,IAAI,CAAC,CAAC;QAEnB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAiB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YACnE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,aAAa,CACrC,QAAQ,EACR,CAAC,KAAK,EAAE,EAAE;gBACR,IAAI,CAAC,UAAU,CAAC,OAAO;oBAAE,OAAO;gBAChC,aAAa,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;YACvC,CAAC,EACD,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,eAAe,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS,CAChF,CAAC;YAEF,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC;YACvB,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC;YAE/B,IAAI,CAAC,UAAU,CAAC,OAAO;gBAAE,OAAO,EAAE,CAAC;YAEnC,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC7B,WAAW,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,CAAC;YAChC,OAAO,QAAQ,CAAC,IAAI,CAAC;QACvB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,UAAU,CAAC,OAAO;gBAAE,OAAO,EAAE,CAAC;YACnC,MAAM,CAAC,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAC9D,QAAQ,CAAC,CAAC,CAAC,CAAC;YACZ,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;YACxB,OAAO,EAAE,CAAC;QACZ,CAAC;gBAAS,CAAC;YACT,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC;YACvB,IAAI,UAAU,CAAC,OAAO,EAAE,CAAC;gBACvB,YAAY,CAAC,KAAK,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;IACH,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,IAAI,GAAG,WAAW,CAAC,GAAG,EAAE;QAC5B,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;IACtB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO;QACL,UAAU;QACV,SAAS;QACT,QAAQ;QACR,IAAI;QACJ,KAAK;KACN,CAAC;AACJ,CAAC","sourcesContent":["import { useState, useEffect, useRef, useCallback } from 'react';\nimport { isAvailable, streamMessage } from './index';\nimport { ChatMemoryManager } from './memory';\nimport type {\n  LLMMessage,\n  UseChatOptions,\n  UseChatReturn,\n  UseCompletionOptions,\n  UseCompletionReturn,\n  UseOnDeviceAIReturn,\n} from './types';\n\n// Cache availability result across all hook instances\nlet availabilityCache: boolean | null = null;\nlet availabilityPromise: Promise<boolean> | null = null;\n\nfunction checkAvailability(): Promise<boolean> {\n  if (availabilityCache !== null) {\n    return Promise.resolve(availabilityCache);\n  }\n  if (!availabilityPromise) {\n    availabilityPromise = isAvailable().then((result) => {\n      availabilityCache = result;\n      return result;\n    });\n  }\n  return availabilityPromise;\n}\n\n/**\n * React hook to check if on-device AI is available.\n *\n * Caches the result so multiple components don't re-check.\n *\n * @returns Object with isAvailable and isChecking states\n *\n * @example\n * ```tsx\n * function MyComponent() {\n *   const { isAvailable, isChecking } = useOnDeviceAI();\n *\n *   if (isChecking) return <Text>Checking AI availability...</Text>;\n *   if (!isAvailable) return <Text>On-device AI not available</Text>;\n *\n *   return <ChatComponent />;\n * }\n * ```\n */\nexport function useOnDeviceAI(): UseOnDeviceAIReturn {\n  const [available, setAvailable] = useState(availabilityCache ?? false);\n  const [checking, setChecking] = useState(availabilityCache === null);\n\n  useEffect(() => {\n    if (availabilityCache !== null) {\n      setAvailable(availabilityCache);\n      setChecking(false);\n      return;\n    }\n\n    let mounted = true;\n    checkAvailability().then((result) => {\n      if (mounted) {\n        setAvailable(result);\n        setChecking(false);\n      }\n    });\n    return () => {\n      mounted = false;\n    };\n  }, []);\n\n  return { isAvailable: available, isChecking: checking };\n}\n\n/**\n * React hook for building chat interfaces with on-device AI.\n *\n * Manages messages, input state, streaming, and conversation memory automatically.\n * Built on top of ChatMemoryManager and streamMessage.\n *\n * @param options - Configuration options for the chat\n * @returns Chat state and control functions\n *\n * @example\n * ```tsx\n * function ChatScreen() {\n *   const { messages, input, setInput, sendMessage, isStreaming, stop } = useChat({\n *     systemPrompt: 'You are a helpful assistant.',\n *   });\n *\n *   return (\n *     <View>\n *       <FlatList\n *         data={messages}\n *         renderItem={({ item }) => (\n *           <Text>{item.role}: {item.content}</Text>\n *         )}\n *       />\n *       <TextInput value={input} onChangeText={setInput} />\n *       {isStreaming ? (\n *         <Button title=\"Stop\" onPress={stop} />\n *       ) : (\n *         <Button title=\"Send\" onPress={() => sendMessage()} />\n *       )}\n *     </View>\n *   );\n * }\n * ```\n */\nexport function useChat(options: UseChatOptions = {}): UseChatReturn {\n  const { systemPrompt, maxTurns, initialMessages, onFinish, onError } = options;\n\n  const memoryRef = useRef(new ChatMemoryManager({ maxTurns, systemPrompt }));\n\n  // Initialize with initial messages\n  const initializedRef = useRef(false);\n  if (!initializedRef.current && initialMessages) {\n    for (const msg of initialMessages) {\n      memoryRef.current.addMessage(msg);\n    }\n    initializedRef.current = true;\n  }\n\n  const [messages, setMessages] = useState<LLMMessage[]>(memoryRef.current.getMessages());\n  const [input, setInput] = useState('');\n  const [isStreaming, setIsStreaming] = useState(false);\n  const [error, setError] = useState<Error | null>(null);\n  const stopRef = useRef<(() => void) | null>(null);\n  const mountedRef = useRef(true);\n  const inputRef = useRef(input);\n  inputRef.current = input;\n  const streamingRef = useRef(isStreaming);\n  streamingRef.current = isStreaming;\n  const onFinishRef = useRef(onFinish);\n  onFinishRef.current = onFinish;\n  const onErrorRef = useRef(onError);\n  onErrorRef.current = onError;\n\n  useEffect(() => {\n    return () => {\n      mountedRef.current = false;\n      stopRef.current?.();\n    };\n  }, []);\n\n  const send = useCallback(\n    async (text?: string) => {\n      const content = (text ?? inputRef.current).trim();\n      if (!content || streamingRef.current) return;\n\n      setError(null);\n\n      // Add user message\n      memoryRef.current.addUserMessage(content);\n      setMessages([...memoryRef.current.getMessages()]);\n      if (!text) setInput('');\n\n      setIsStreaming(true);\n\n      try {\n        const allMessages = memoryRef.current.getAllMessages();\n\n        // Use a temporary message for streaming display\n        let accumulatedText = '';\n        const { promise, stop } = streamMessage(allMessages, (event) => {\n          if (!mountedRef.current) return;\n          accumulatedText = event.accumulatedText;\n          // Update messages with streaming assistant response\n          const currentMessages = memoryRef.current.getMessages();\n          setMessages([\n            ...currentMessages,\n            { role: 'assistant' as const, content: accumulatedText },\n          ]);\n        });\n\n        stopRef.current = stop;\n        const response = await promise;\n\n        if (!mountedRef.current) return;\n\n        // Add the final assistant message to memory\n        memoryRef.current.addAssistantMessage(response.text);\n        setMessages([...memoryRef.current.getMessages()]);\n        onFinishRef.current?.(response);\n      } catch (err) {\n        if (!mountedRef.current) return;\n        const e = err instanceof Error ? err : new Error(String(err));\n        setError(e);\n        onErrorRef.current?.(e);\n      } finally {\n        stopRef.current = null;\n        if (mountedRef.current) {\n          setIsStreaming(false);\n        }\n      }\n    },\n    []\n  );\n\n  const stop = useCallback(() => {\n    stopRef.current?.();\n  }, []);\n\n  const clear = useCallback(() => {\n    memoryRef.current.clear();\n    setMessages([]);\n    setError(null);\n  }, []);\n\n  return {\n    messages,\n    input,\n    setInput,\n    sendMessage: send,\n    isStreaming,\n    stop,\n    clear,\n    error,\n  };\n}\n\n/**\n * React hook for single-shot AI completions.\n *\n * Unlike useChat (which manages a conversation), useCompletion is for\n * one-off tasks like summarization, translation, or content generation.\n *\n * @param options - Configuration options\n * @returns Completion state and control functions\n *\n * @example\n * ```tsx\n * function SummarizerScreen() {\n *   const { completion, isLoading, complete, stop } = useCompletion({\n *     systemPrompt: 'You are a summarization assistant. Summarize the given text concisely.',\n *   });\n *\n *   return (\n *     <View>\n *       <Button\n *         title=\"Summarize\"\n *         onPress={() => complete('Long article text here...')}\n *       />\n *       {isLoading && <Button title=\"Stop\" onPress={stop} />}\n *       <Text>{completion}</Text>\n *     </View>\n *   );\n * }\n * ```\n */\nexport function useCompletion(options: UseCompletionOptions = {}): UseCompletionReturn {\n  const { systemPrompt, onFinish, onError } = options;\n\n  const [completion, setCompletion] = useState('');\n  const [isLoading, setIsLoading] = useState(false);\n  const [error, setError] = useState<Error | null>(null);\n  const stopRef = useRef<(() => void) | null>(null);\n  const mountedRef = useRef(true);\n  const loadingRef = useRef(isLoading);\n  loadingRef.current = isLoading;\n  const completionRef = useRef(completion);\n  completionRef.current = completion;\n  const systemPromptRef = useRef(systemPrompt);\n  systemPromptRef.current = systemPrompt;\n  const onFinishRef = useRef(onFinish);\n  onFinishRef.current = onFinish;\n  const onErrorRef = useRef(onError);\n  onErrorRef.current = onError;\n\n  useEffect(() => {\n    return () => {\n      mountedRef.current = false;\n      stopRef.current?.();\n    };\n  }, []);\n\n  const complete = useCallback(\n    async (prompt: string) => {\n      if (loadingRef.current) return completionRef.current;\n\n      setError(null);\n      setCompletion('');\n      setIsLoading(true);\n\n      try {\n        const messages: LLMMessage[] = [{ role: 'user', content: prompt }];\n        const { promise, stop } = streamMessage(\n          messages,\n          (event) => {\n            if (!mountedRef.current) return;\n            setCompletion(event.accumulatedText);\n          },\n          systemPromptRef.current ? { systemPrompt: systemPromptRef.current } : undefined\n        );\n\n        stopRef.current = stop;\n        const response = await promise;\n\n        if (!mountedRef.current) return '';\n\n        setCompletion(response.text);\n        onFinishRef.current?.(response);\n        return response.text;\n      } catch (err) {\n        if (!mountedRef.current) return '';\n        const e = err instanceof Error ? err : new Error(String(err));\n        setError(e);\n        onErrorRef.current?.(e);\n        return '';\n      } finally {\n        stopRef.current = null;\n        if (mountedRef.current) {\n          setIsLoading(false);\n        }\n      }\n    },\n    []\n  );\n\n  const stop = useCallback(() => {\n    stopRef.current?.();\n  }, []);\n\n  return {\n    completion,\n    isLoading,\n    complete,\n    stop,\n    error,\n  };\n}\n"]}

package/build/index.d.ts

@@ -1,6 +1,7 @@
 import { LLMMessage, LLMSendOptions, LLMResponse, LLMStreamOptions, LLMStreamCallback, LLMSummarizeOptions, LLMTranslateOptions, LLMRewriteOptions, LLMExtractKeyPointsOptions, LLMAnswerQuestionOptions, LLMSuggestOptions, LLMSuggestResponse, LLMSmartReplyOptions, LLMAutocompleteOptions } from './types';
 export * from './types';
 export * from './memory';
+export * from './hooks';
 /**
  * Check if on-device AI is available on the current device.
  * Returns false on unsupported platforms (web, etc.).