From 3fbf2176dfa59ba376684c371ffb81ccacf29480 Mon Sep 17 00:00:00 2001
From: Aleksandras Kostarevas <aleks076@protonmail.com>
Date: Sat, 24 Aug 2024 10:51:18 +0300
Subject: [PATCH] Fix bug in LanguageModel native code

---
 native/jni/org_futo_inputmethod_latin_xlm_LanguageModel.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/native/jni/org_futo_inputmethod_latin_xlm_LanguageModel.cpp b/native/jni/org_futo_inputmethod_latin_xlm_LanguageModel.cpp
index 0655afb17..b9674c2d4 100644
--- a/native/jni/org_futo_inputmethod_latin_xlm_LanguageModel.cpp
+++ b/native/jni/org_futo_inputmethod_latin_xlm_LanguageModel.cpp
@@ -1060,6 +1060,8 @@ namespace latinime {
 
             sortProbabilityPairVectorDescending(index_value, NUM_TOKEN_MIX);
 
+            if(index_value.size() < NUM_TOKEN_MIX) break;
+
             bool needs_resorting = false;
             int num_symbols = 0;
             for(int s=0; s<4; s++) {