[fix] Fix LiteRT and ObjectBox proguard-rules.pro

SkyD666 · Jan 14, 2025 · 1855abb · 1855abb
1 parent ed9e7fc
commit 1855abb
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/app/proguard-rules.pro b/app/proguard-rules.pro
@@ -127,3 +127,14 @@
 -dontwarn org.openjsse.javax.net.ssl.SSLParameters
 -dontwarn org.openjsse.javax.net.ssl.SSLSocket
 -dontwarn org.openjsse.net.ssl.OpenJSSE
+
+# com.google.ai.edge.litert
+-keep class javax.lang.model.** { *; }
+-dontwarn javax.lang.model.**
+-keep class com.google.mediapipe.proto.** { *; }
+-keepclassmembers class * extends com.google.protobuf.GeneratedMessageLite { *; }
+-keep class com.google.common.flogger.** { *; }
+
+# ObjectBox
+-keep class io.objectbox.** { *; }
+-dontwarn io.objectbox.**
diff --git a/app/src/main/java/com/skyd/rays/model/db/objectbox/entity/StickerEmbedding.kt b/app/src/main/java/com/skyd/rays/model/db/objectbox/entity/StickerEmbedding.kt
@@ -9,7 +9,11 @@ import io.objectbox.annotation.Index
 data class StickerEmbedding(
     @Id var id: Long = 0,
     @Index var uuid: String,
-    @HnswIndex(dimensions = 1024)
+    @HnswIndex(
+        dimensions = 1024,
+        neighborsPerNode = 128,
+        indexingSearchCount = 400,
+    )
     var embedding: FloatArray? = null,
 ) {
     override fun equals(other: Any?): Boolean {