feat(rag): on-device with citations

__mocks__/@gorhom/bottom-sheet.tsx

@@ -1,14 +1,53 @@
-import React from 'react';
-import { View, FlatList } from 'react-native';
+import React, { forwardRef, type PropsWithChildren } from 'react';
+import {
+  View,
+  FlatList,
+  type FlatListProps,
+  type ScrollViewProps,
+  type ViewProps,
+} from 'react-native';
 
-export const BottomSheetModal = React.forwardRef(
-  ({ children }: any, _ref: any) => <>{children}</>
-);
-export const BottomSheetView = ({ children, style }: any) => (
+export interface BottomSheetModalRef {
+  present: () => void;
+  dismiss: () => void;
+}
+
+export type SheetAnimationConfig = Record<string, number | boolean>;
+
+export const BottomSheetModal = forwardRef<
+  BottomSheetModalRef,
+  PropsWithChildren
+>(({ children }, _ref) => <>{children}</>);
+BottomSheetModal.displayName = 'BottomSheetModal';
+
+export const BottomSheetView = ({ children, style }: ViewProps) => (
   <View style={style}>{children}</View>
 );
-export const BottomSheetFlatList = (props: any) => <FlatList {...props} />;
+
+export const BottomSheetScrollView = ({
+  children,
+  contentContainerStyle,
+  testID,
+}: ScrollViewProps) => (
+  <View style={contentContainerStyle} testID={testID}>
+    {children}
+  </View>
+);
+
+export const BottomSheetFlatList = <ItemT,>(props: FlatListProps<ItemT>) => (
+  <FlatList {...props} />
+);
+
 export const BottomSheetBackdrop = () => null;
-export const BottomSheetModalProvider = ({ children }: any) => <>{children}</>;
-export const useBottomSheetTimingConfigs = (_config: any) => ({});
-export const useBottomSheetSpringConfigs = (_config: any) => ({});
+
+export const BottomSheetModalProvider = ({ children }: PropsWithChildren) => (
+  <>{children}</>
+);
+
+export const useBottomSheet = () => ({ close: () => {} });
+
+export const useBottomSheetTimingConfigs = (config: SheetAnimationConfig) =>
+  config;
+
+export const useBottomSheetSpringConfigs = (config: SheetAnimationConfig) =>
+  config;

__mocks__/react-native-executorch-expo-resource-fetcher.ts

@@ -1,4 +1,6 @@
 export const ExpoResourceFetcher = {
   cancelFetching: jest.fn(),
   deleteResources: jest.fn(),
+  listDownloadedFiles: jest.fn(async () => [] as string[]),
+  getFilesTotalSize: jest.fn(async () => 0),
 };

__mocks__/react-native-reanimated.ts

@@ -76,6 +76,7 @@ module.exports = {
     inOut: (fn: any) => fn,
     out: (fn: any) => fn,
     in: (fn: any) => fn,
+    bezier: () => (t: any) => t,
   },
   interpolate: (val: any, inputRange: any, outputRange: any) => {
     if (val <= inputRange[0]) return outputRange[0];

__tests__/ChatBar.test.tsx

@@ -1,5 +1,8 @@
 import React from 'react';
 import { render, screen, fireEvent, act } from '@testing-library/react-native';
+import type { LLMStore } from '../store/llmStore';
+import type { Attachment } from '../hooks/useAttachment';
+import type { PermissionStatus } from 'react-native-audio-api';
 
 // ── mocks ─────────────────────────────────────────────────────────────────────
 
@@ -13,17 +16,20 @@ jest.mock('../context/ThemeContext', () => ({
 }));
 
 jest.mock('../store/llmStore', () => ({
-  useLLMStore: jest.fn(() => ({
-    isGenerating: false,
-    isProcessingPrompt: false,
-    interrupt: jest.fn(),
-    loadModel: jest.fn(),
-    model: null,
-  })),
+  useLLMStore: jest.fn((selector?: (state: Partial<LLMStore>) => unknown) => {
+    const state = {
+      isGenerating: false,
+      isProcessingPrompt: false,
+      interrupt: jest.fn(),
+      loadModel: jest.fn(),
+      model: null,
+    };
+    return selector ? selector(state) : state;
+  }),
 }));
 
 const mockUseAttachment = {
-  attachments: [] as any[],
+  attachments: [] as Attachment[],
   sheetRef: { current: null },
   pickFromLibrary: jest.fn(),
   pickFromCamera: jest.fn(),
@@ -45,7 +51,12 @@ jest.mock('../components/bottomSheets/AttachmentSheet', () => {
     onPickFromCamera,
     onPickDocument,
     isVisionModel,
-  }: any) => (
+  }: {
+    onPickFromLibrary: () => void;
+    onPickFromCamera: () => void;
+    onPickDocument: () => void;
+    isVisionModel: boolean;
+  }) => (
     <View testID="attachment-sheet">
       <Text>{`vision:${isVisionModel}`}</Text>
       <TouchableOpacity testID="pick-library-btn" onPress={onPickFromLibrary}>
@@ -63,7 +74,13 @@ jest.mock('../components/bottomSheets/AttachmentSheet', () => {
 
 jest.mock('../components/chat-screen/AttachmentThumbnail', () => {
   const { View, TouchableOpacity, Text } = require('react-native');
-  return ({ attachment, onRemove }: any) => (
+  return ({
+    attachment,
+    onRemove,
+  }: {
+    attachment: Attachment;
+    onRemove: () => void;
+  }) => (
     <View testID={`attachment-thumb-${attachment.id}`}>
       <Text>{attachment.name || attachment.uri}</Text>
       <TouchableOpacity
@@ -77,8 +94,14 @@ jest.mock('../components/chat-screen/AttachmentThumbnail', () => {
 });
 
 jest.mock('../components/chat-screen/ChatSpeechInput', () => {
-  const { View, TouchableOpacity, Text } = require('react-native');
-  return ({ onSubmit, onCancel }: any) => (
+  const { View, TouchableOpacity } = require('react-native');
+  return ({
+    onSubmit,
+    onCancel,
+  }: {
+    onSubmit: (transcript: string) => void;
+    onCancel: () => void;
+  }) => (
     <View testID="speech-input">
       <TouchableOpacity
         testID="speech-submit"
@@ -91,7 +114,7 @@ jest.mock('../components/chat-screen/ChatSpeechInput', () => {
 
 jest.mock('../components/chat-screen/PromptSuggestions', () => {
   const { TouchableOpacity, Text } = require('react-native');
-  return ({ onSelectPrompt }: any) => (
+  return ({ onSelectPrompt }: { onSelectPrompt: (prompt: string) => void }) => (
     <TouchableOpacity
       testID="prompt-suggestion"
       onPress={() => onSelectPrompt('Suggested prompt')}
@@ -114,7 +137,18 @@ jest.mock('../components/chat-screen/ChatBarActions', () => {
     onThinkingToggle,
     thinkingEnabled,
     onAttach,
-  }: any) => (
+  }: {
+    userInput: string;
+    hasAttachments: boolean;
+    onSend: () => void;
+    isGenerating: boolean;
+    isProcessingPrompt: boolean;
+    onInterrupt: () => void;
+    onSpeechInput: () => void;
+    onThinkingToggle: () => void;
+    thinkingEnabled: boolean;
+    onAttach: () => void;
+  }) => (
     <View testID="chat-bar-actions">
       <TouchableOpacity testID="attach-btn" onPress={onAttach}>
         <Text>+</Text>
@@ -175,7 +209,7 @@ const defaultProps = {
   onSelectModel: jest.fn(),
   onSelectPrompt: jest.fn(),
   model: downloadedModel,
-  scrollRef: { current: null } as any,
+  scrollRef: { current: null },
   isAtBottom: true,
   isVisionModel: false,
   thinkingEnabled: false,
@@ -187,13 +221,18 @@ const renderBar = (props: Partial<typeof defaultProps> = {}) =>
   render(<ChatBar {...defaultProps} {...props} />);
 
 beforeEach(() => {
-  mockUseLLMStore.mockReturnValue({
-    isGenerating: false,
-    isProcessingPrompt: false,
-    interrupt: jest.fn(),
-    loadModel: jest.fn(),
-    model: null,
-  });
+  mockUseLLMStore.mockImplementation(
+    (selector?: (state: Partial<LLMStore>) => unknown) => {
+      const state = {
+        isGenerating: false,
+        isProcessingPrompt: false,
+        interrupt: jest.fn(),
+        loadModel: jest.fn(),
+        model: null,
+      };
+      return selector ? selector(state) : state;
+    }
+  );
   mockUseAttachment.attachments = [];
   mockUseAttachment.openSheet.mockClear();
   mockUseAttachment.clearAll.mockClear();
@@ -203,7 +242,7 @@ beforeEach(() => {
   jest.spyOn(console, 'warn').mockImplementation(() => {});
   // Default: permission granted
   mockAudioManager.requestRecordingPermissions.mockResolvedValue(
-    'Granted' as any
+    'Granted' as PermissionStatus
   );
 });
 
@@ -293,26 +332,36 @@ describe('downloaded model — text input', () => {
 
 describe('generating state', () => {
   it('shows interrupt button when isGenerating', () => {
-    mockUseLLMStore.mockReturnValue({
-      isGenerating: true,
-      isProcessingPrompt: false,
-      interrupt: jest.fn(),
-      loadModel: jest.fn(),
-      model: null,
-    });
+    mockUseLLMStore.mockImplementation(
+      (selector?: (state: Partial<LLMStore>) => unknown) => {
+        const state = {
+          isGenerating: true,
+          isProcessingPrompt: false,
+          interrupt: jest.fn(),
+          loadModel: jest.fn(),
+          model: null,
+        };
+        return selector ? selector(state) : state;
+      }
+    );
     renderBar();
     expect(screen.getByTestId('interrupt-btn')).toBeTruthy();
   });
 
   it('calls interrupt when interrupt button is pressed', () => {
     const interrupt = jest.fn();
-    mockUseLLMStore.mockReturnValue({
-      isGenerating: true,
-      isProcessingPrompt: false,
-      interrupt,
-      loadModel: jest.fn(),
-      model: null,
-    });
+    mockUseLLMStore.mockImplementation(
+      (selector?: (state: Partial<LLMStore>) => unknown) => {
+        const state = {
+          isGenerating: true,
+          isProcessingPrompt: false,
+          interrupt,
+          loadModel: jest.fn(),
+          model: null,
+        };
+        return selector ? selector(state) : state;
+      }
+    );
     renderBar();
     fireEvent.press(screen.getByTestId('interrupt-btn'));
     expect(interrupt).toHaveBeenCalled();
@@ -332,7 +381,7 @@ describe('speech input', () => {
 
   it('shows toast and stays in text mode when microphone permission is denied', async () => {
     mockAudioManager.requestRecordingPermissions.mockResolvedValue(
-      'Denied' as any
+      'Denied' as PermissionStatus
     );
     renderBar();
     await act(async () => {
@@ -405,7 +454,7 @@ describe('speech input', () => {
     // Override speech mock to submit empty string
     jest.mock('../components/chat-screen/ChatSpeechInput', () => {
       const { View, TouchableOpacity } = require('react-native');
-      return ({ onSubmit }: any) => (
+      return ({ onSubmit }: { onSubmit: (transcript: string) => void }) => (
         <View testID="speech-input">
           <TouchableOpacity
             testID="speech-submit"