feat(android): avoid warning by use DeviceEventEmitter

mybigday · Sep 7, 2023 · b9487dd · b9487dd
1 parent 710dc71
commit b9487dd
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -1,4 +1,4 @@
-import { NativeEventEmitter } from 'react-native'
+import { NativeEventEmitter, DeviceEventEmitter, Platform } from 'react-native'
 import type { DeviceEventEmitterStatic } from 'react-native'
 import RNLlama from './NativeRNLlama'
 import type {
@@ -15,9 +15,14 @@ export { SchemaGrammarConverter, convertJsonSchemaToGrammar }
 
 const EVENT_ON_TOKEN = '@RNLlama_onToken'
 
-const EventEmitter: NativeEventEmitter | DeviceEventEmitterStatic =
+let EventEmitter: NativeEventEmitter | DeviceEventEmitterStatic
+if (Platform.OS === 'ios') {
   // @ts-ignore
-  new NativeEventEmitter(RNLlama)
+  EventEmitter = new NativeEventEmitter(RNLlama)
+}
+if (Platform.OS === 'android') {
+  EventEmitter = DeviceEventEmitter
+}
 
 export type TokenData = {
   token: string