replace tokens

Author: kirklandsign

examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java

@@ -90,9 +90,7 @@ public void onResult(String result) {
     if (result.equals(PromptFormat.getStopToken(mCurrentSettingsFields.getModelType()))) {
       return;
     }
-    if (PromptFormat.shouldSkipToken(mCurrentSettingsFields.getModelType(), result)) {
-      return;
-    }
+    result = PromptFormat.replaceSpecialToken(mCurrentSettingsFields.getModelType(), result);
     if (result.equals("\n\n") || result.equals("\n")) {
       if (!mResultMessage.getText().isEmpty()) {
         mResultMessage.appendText(result);

examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/PromptFormat.java

@@ -99,12 +99,19 @@ public static String getThinkingModeToken(ModelType modelType, boolean thinkingM
     }
   }
 
-  public static boolean shouldSkipToken(ModelType modelType, String token) {
+  public static String replaceSpecialToken(ModelType modelType, String token) {
     switch (modelType) {
       case QWEN_3:
-        return token.equals("<|im_end|>");
+        switch (token) {
+          case "<|im_end|>":
+            return "";
+          case "<think>":
+            return "Thinking...\n";
+          case "</think>":
+            return "\nDone thinking";
+        }
       default:
-        return false;
+        return token;
     }
   }