From 66fc6dfaeebb5bab4d88e9ec7fa3521caeb8fa7d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=92=B0=E8=AF=9A?= Date: Tue, 25 Feb 2025 17:47:00 +0800 Subject: [PATCH] update NeedExectue --- plugins/wasm-go/extensions/ai-search/engine/bing/bing.go | 2 +- plugins/wasm-go/extensions/ai-search/engine/google/google.go | 2 +- plugins/wasm-go/extensions/ai-search/engine/quark/quark.go | 2 +- plugins/wasm-go/extensions/ai-search/main.go | 4 ++-- 4 files changed, 5 insertions(+), 5 deletions(-) diff --git a/plugins/wasm-go/extensions/ai-search/engine/bing/bing.go b/plugins/wasm-go/extensions/ai-search/engine/bing/bing.go index 71d39883ee..b24fe33464 100644 --- a/plugins/wasm-go/extensions/ai-search/engine/bing/bing.go +++ b/plugins/wasm-go/extensions/ai-search/engine/bing/bing.go @@ -60,7 +60,7 @@ func NewBingSearch(config *gjson.Result) (*BingSearch, error) { } func (b BingSearch) NeedExectue(ctx engine.SearchContext) bool { - return ctx.EngineType == "internet" + return ctx.EngineType == "" || ctx.EngineType == "internet" } func (b BingSearch) Client() wrapper.HttpClient { diff --git a/plugins/wasm-go/extensions/ai-search/engine/google/google.go b/plugins/wasm-go/extensions/ai-search/engine/google/google.go index c13cd0c9d2..e189646b99 100644 --- a/plugins/wasm-go/extensions/ai-search/engine/google/google.go +++ b/plugins/wasm-go/extensions/ai-search/engine/google/google.go @@ -68,7 +68,7 @@ func NewGoogleSearch(config *gjson.Result) (*GoogleSearch, error) { } func (g GoogleSearch) NeedExectue(ctx engine.SearchContext) bool { - return ctx.EngineType == "internet" + return ctx.EngineType == "" || ctx.EngineType == "internet" } func (g GoogleSearch) Client() wrapper.HttpClient { diff --git a/plugins/wasm-go/extensions/ai-search/engine/quark/quark.go b/plugins/wasm-go/extensions/ai-search/engine/quark/quark.go index 9fd098b7eb..57d5db1e58 100644 --- a/plugins/wasm-go/extensions/ai-search/engine/quark/quark.go +++ b/plugins/wasm-go/extensions/ai-search/engine/quark/quark.go @@ -119,7 +119,7 @@ func NewQuarkSearch(config *gjson.Result) (*QuarkSearch, error) { } func (g QuarkSearch) NeedExectue(ctx engine.SearchContext) bool { - return ctx.EngineType == "internet" + return ctx.EngineType == "" || ctx.EngineType == "internet" } func (g QuarkSearch) Client() wrapper.HttpClient { diff --git a/plugins/wasm-go/extensions/ai-search/main.go b/plugins/wasm-go/extensions/ai-search/main.go index 235a566899..73e1935209 100644 --- a/plugins/wasm-go/extensions/ai-search/main.go +++ b/plugins/wasm-go/extensions/ai-search/main.go @@ -382,7 +382,7 @@ func executeSearch(ctx wrapper.HttpContext, config Config, queryIndex int, body // Check if engine needs to execute for any of the search contexts var needsExecute bool for _, searchCtx := range searchContexts { - if searchCtx.EngineType == "" || configEngine.NeedExectue(searchCtx) { + if configEngine.NeedExectue(searchCtx) { needsExecute = true break } @@ -393,7 +393,7 @@ func executeSearch(ctx wrapper.HttpContext, config Config, queryIndex int, body // Process all search contexts for this engine for _, searchCtx := range searchContexts { - if searchCtx.EngineType != "" && !configEngine.NeedExectue(searchCtx) { + if !configEngine.NeedExectue(searchCtx) { continue } args := configEngine.CallArgs(searchCtx)