fix(ai): Fix the issue where thinking model tags sometimes fail to parse

- Add goquery v1.10.0 as a new dependency
- Upgrade generative-ai-go from v0.18.0 to v0.19.0
- Refactor GoogleProvider model names for clarity
- Enhance processThinking function to handle leading newlines in input
- Improve AI command tests for better coverage of thinking tag processing

Author: qianlifeng

wox.core/ai/provider_google.go

@@ -44,11 +44,7 @@ func (g *GoogleProvider) ChatStream(ctx context.Context, model Model, conversati
 func (g *GoogleProvider) Models(ctx context.Context) ([]Model, error) {
 	return []Model{
 		{
-			Name:     "gemini-exp-1206",
-			Provider: ProviderNameGoogle,
-		},
-		{
-			Name:     "gemini-2.0-flash-exp",
+			Name:     "gemini-2.0-flash",
 			Provider: ProviderNameGoogle,
 		},
 	}, nil

wox.core/go.mod

@@ -4,6 +4,7 @@ go 1.23.0
 
 require (
 	github.com/Masterminds/semver/v3 v3.3.0
+	github.com/PuerkitoBio/goquery v1.10.0
 	github.com/cdfmlr/ellipsis v0.0.1
 	github.com/disintegration/imaging v1.6.2
 	github.com/djherbis/buffer v1.2.0
@@ -12,7 +13,7 @@ require (
 	github.com/fsnotify/fsnotify v1.8.0
 	github.com/go-resty/resty/v2 v2.15.3
 	github.com/godbus/dbus/v5 v5.1.0
-	github.com/google/generative-ai-go v0.18.0
+	github.com/google/generative-ai-go v0.19.0
 	github.com/google/uuid v1.6.0
 	github.com/googleapis/gax-go/v2 v2.13.0
 	github.com/gorilla/websocket v1.5.3
@@ -56,7 +57,6 @@ require (
 	cloud.google.com/go/auth/oauth2adapt v0.2.5 // indirect
 	cloud.google.com/go/compute/metadata v0.5.2 // indirect
 	cloud.google.com/go/longrunning v0.6.2 // indirect
-	github.com/PuerkitoBio/goquery v1.10.0 // indirect
 	github.com/andybalholm/cascadia v1.3.2 // indirect
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/dlclark/regexp2 v1.11.4 // indirect

wox.core/go.sum

@@ -72,6 +72,8 @@ github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek
 github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=
 github.com/google/generative-ai-go v0.18.0 h1:6ybg9vOCLcI/UpBBYXOTVgvKmcUKFRNj+2Cj3GnebSo=
 github.com/google/generative-ai-go v0.18.0/go.mod h1:JYolL13VG7j79kM5BtHz4qwONHkeJQzOCkKXnpqtS/E=
+github.com/google/generative-ai-go v0.19.0 h1:R71szggh8wHMCUlEMsW2A/3T+5LdEIkiaHSYgSpUgdg=
+github.com/google/generative-ai-go v0.19.0/go.mod h1:JYolL13VG7j79kM5BtHz4qwONHkeJQzOCkKXnpqtS/E=
 github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M=
 github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
 github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=

wox.core/plugin/system/ai_command.go

@@ -456,20 +456,26 @@ func (c *Plugin) processThinking(text string) (thinking string, content string)
 	const thinkStart = "<think>"
 	const thinkEnd = "</think>"
 
-	// Check if the text starts with the thinking tag
-	if len(text) >= len(thinkStart) && text[:len(thinkStart)] == thinkStart {
-		// Find the end tag
+	// Trim leading newlines for tag detection
+	trimmedText := strings.TrimLeft(text, "\n")
+
+	// Check if the text starts with the thinking tag after trimming newlines
+	if len(trimmedText) >= len(thinkStart) && trimmedText[:len(thinkStart)] == thinkStart {
+		// Calculate the offset to maintain original indices
+		offset := len(text) - len(trimmedText)
+
+		// Find the end tag in the original text
 		endIndex := strings.Index(text, thinkEnd)
 		if endIndex != -1 {
 			// Extract thinking content (without the tags)
-			thinking = text[len(thinkStart):endIndex]
+			thinking = text[offset+len(thinkStart) : endIndex]
 			// Extract the remaining content after the thinking tag
 			if endIndex+len(thinkEnd) < len(text) {
 				content = text[endIndex+len(thinkEnd):]
 			}
 		} else {
 			// If there's no end tag, the entire text is considered thinking
-			thinking = text[len(thinkStart):]
+			thinking = text[offset+len(thinkStart):]
 		}
 	} else {
 		// If there's no thinking tag at the beginning, the entire text is content

wox.core/plugin/system/ai_command_test.go

@@ -14,6 +14,10 @@ func TestAICommandParseThinking(t *testing.T) {
 	assert.Equal(t, " hello world", thinking)
 	assert.Equal(t, "this is content", content)
 
+	thinking, content = plugin.processThinking("\n<think> hello world</think>this is content")
+	assert.Equal(t, " hello world", thinking)
+	assert.Equal(t, "this is content", content)
+
 	thinking, content = plugin.processThinking("hello world")
 	assert.Equal(t, "", thinking)
 	assert.Equal(t, "hello world", content)