From 739fb42d278ecda9303d604f453076253a041c2a Mon Sep 17 00:00:00 2001
From: "lilong.129" <lilong.129@bytedance.com>
Date: Mon, 21 Jul 2025 16:53:38 +0800
Subject: [PATCH] fix: unittest

---
 uixt/driver_ext_ai.go      | 4 +++-
 uixt/driver_ext_ai_test.go | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/uixt/driver_ext_ai.go b/uixt/driver_ext_ai.go
index fa376cba..185f41fc 100644
--- a/uixt/driver_ext_ai.go
+++ b/uixt/driver_ext_ai.go
@@ -217,7 +217,9 @@ func (dExt *XTDriver) AIAction(ctx context.Context, prompt string, opts ...optio
 		ScreenshotElapsed: screenResult.Elapsed,
 		ImagePath:         screenResult.ImagePath,
 		Resolution:        &screenResult.Resolution,
-		PlanningResult:    &planningResult.PlanningResult,
+	}
+	if planningResult != nil {
+		aiExecutionResult.PlanningResult = &planningResult.PlanningResult
 	}
 	if err != nil {
 		aiExecutionResult.Error = err.Error()
diff --git a/uixt/driver_ext_ai_test.go b/uixt/driver_ext_ai_test.go
index 868ef5af..6ca5d2e3 100644
--- a/uixt/driver_ext_ai_test.go
+++ b/uixt/driver_ext_ai_test.go
@@ -314,7 +314,7 @@ func TestDriverExt_AIAction_CompareWithAIAction(t *testing.T) {
 		if aiResult.PlanningResult != nil {
 			t.Logf("AIAction model: %s", aiResult.PlanningResult.ModelName)
 
-			assert.NotEqual(t, "wings-api", aiResult.PlanningResult.ModelName, "AIAction should not use wings-api")
+			assert.Equal(t, "wings-api", aiResult.PlanningResult.ModelName, "AIAction should use wings-api")
 		}
 	}
 }