fix: unittest

This commit is contained in:
lilong.129
2025-07-21 16:53:38 +08:00
parent 88bda7c215
commit 739fb42d27
2 changed files with 4 additions and 2 deletions

View File

@@ -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()

View File

@@ -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")
}
}
}