Update mobile/examples/phi-3/android/app/src/main/java/ai/onnxruntime/genai/demo/MainActivity.java

vraspar · edgchen1 · web-flow · commit 6ecefb59400e · 2025-06-05T15:15:41.000-07:00
Co-authored-by: Edward Chen &lt;18449977+edgchen1@users.noreply.github.com&gt;
diff --git a/mobile/examples/phi-3/android/app/src/main/java/ai/onnxruntime/genai/demo/MainActivity.java b/mobile/examples/phi-3/android/app/src/main/java/ai/onnxruntime/genai/demo/MainActivity.java
@@ -155,8 +155,7 @@ public void run() {
                             
                             // Token listener for streaming tokens
                             Consumer<String> tokenListener = token -> {
-                                if (numTokens.get() == 0) { // first token
-                                    firstTokenTime.set(System.currentTimeMillis());
+                                firstTokenTime.compareAndSet(startTime, System.currentTimeMillis());
                                 }
                                 
                                 // Update UI with new token

Original file line number	Diff line number	Diff line change
`@@ -155,8 +155,7 @@ public void run() {`
`155`	`155`
`156`	`156`	`// Token listener for streaming tokens`
`157`	`157`	`Consumer<String> tokenListener = token -> {`
`158`		`- if (numTokens.get() == 0) { // first token`
`159`		`- firstTokenTime.set(System.currentTimeMillis());`
	`158`	`+ firstTokenTime.compareAndSet(startTime, System.currentTimeMillis());`
`160`	`159`	`}`
`161`	`160`
`162`	`161`	`// Update UI with new token`