wehub-resource-sync
108 lines
3.8 KiB
Go
108 lines
3.8 KiB
Go
package main
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/router-for-me/CLIProxyAPI/v7/sdk/pluginapi"
|
|
)
|
|
|
|
// defaultWebSearchFallbackChain is the ordered backend try list when route=fallback.
|
|
func defaultWebSearchFallbackChain() []routeBackend {
|
|
return []routeBackend{
|
|
backendAntigravityGoogle,
|
|
backendCodexWebSearch,
|
|
backendXAIWebSearch,
|
|
backendTavily,
|
|
}
|
|
}
|
|
|
|
func isFallbackRoute(route string) bool {
|
|
r := strings.ToLower(strings.TrimSpace(route))
|
|
return r == "" || r == string(backendFallback)
|
|
}
|
|
|
|
// tryRouteBackend returns a handled ModelRouteResponse and true when this backend can serve the request.
|
|
func tryRouteBackend(backend routeBackend, cfg pluginConfig, req pluginapi.ModelRouteRequest) (pluginapi.ModelRouteResponse, bool) {
|
|
switch backend {
|
|
case backendTavily:
|
|
client := newTavilyClient(cfg.TavilyAPIKeys)
|
|
if !client.available() {
|
|
return pluginapi.ModelRouteResponse{Handled: false, Reason: "tavily_unavailable"}, false
|
|
}
|
|
return pluginapi.ModelRouteResponse{
|
|
Handled: true,
|
|
TargetKind: pluginapi.ModelRouteTargetSelf,
|
|
Reason: "claude_code_web_search_tavily",
|
|
}, true
|
|
case backendAntigravityGoogle:
|
|
if !hasProvider(req.AvailableProviders, "antigravity") {
|
|
return pluginapi.ModelRouteResponse{Handled: false, Reason: "antigravity_unavailable"}, false
|
|
}
|
|
targetModel := resolveAntigravityWebSearchTargetModel(cfg.AntigravityModel, req.RequestedModel)
|
|
if targetModel == "" {
|
|
return pluginapi.ModelRouteResponse{Handled: false, Reason: "antigravity_web_search_model_unresolved"}, false
|
|
}
|
|
return pluginapi.ModelRouteResponse{
|
|
Handled: true,
|
|
TargetKind: pluginapi.ModelRouteTargetProvider,
|
|
Target: "antigravity",
|
|
TargetModel: targetModel,
|
|
Reason: "claude_code_web_search_antigravity_google",
|
|
}, true
|
|
case backendCodexWebSearch:
|
|
if !hasProvider(req.AvailableProviders, "codex") {
|
|
return pluginapi.ModelRouteResponse{Handled: false, Reason: "codex_unavailable"}, false
|
|
}
|
|
targetModel := resolveCodexWebSearchTargetModel(cfg.CodexModel)
|
|
return pluginapi.ModelRouteResponse{
|
|
Handled: true,
|
|
TargetKind: pluginapi.ModelRouteTargetProvider,
|
|
Target: "codex",
|
|
TargetModel: targetModel,
|
|
Reason: "claude_code_web_search_codex",
|
|
}, true
|
|
case backendXAIWebSearch:
|
|
if !hasProvider(req.AvailableProviders, "xai") {
|
|
return pluginapi.ModelRouteResponse{Handled: false, Reason: "xai_unavailable"}, false
|
|
}
|
|
targetModel := resolveXAIWebSearchTargetModel(cfg.XAIModel)
|
|
return pluginapi.ModelRouteResponse{
|
|
Handled: true,
|
|
TargetKind: pluginapi.ModelRouteTargetProvider,
|
|
Target: "xai",
|
|
TargetModel: targetModel,
|
|
Reason: "claude_code_web_search_xai",
|
|
}, true
|
|
case backendDefaultProvider:
|
|
provider := cfg.DefaultProvider
|
|
if provider == "" || !hasProvider(req.AvailableProviders, provider) {
|
|
return pluginapi.ModelRouteResponse{Handled: false, Reason: "default_provider_unavailable"}, false
|
|
}
|
|
return pluginapi.ModelRouteResponse{
|
|
Handled: true,
|
|
TargetKind: pluginapi.ModelRouteTargetProvider,
|
|
Target: provider,
|
|
TargetModel: cfg.DefaultProviderModel,
|
|
Reason: "claude_code_web_search_default_provider",
|
|
}, true
|
|
default:
|
|
return pluginapi.ModelRouteResponse{Handled: false}, false
|
|
}
|
|
}
|
|
|
|
func routeWithFallback(cfg pluginConfig, req pluginapi.ModelRouteRequest) pluginapi.ModelRouteResponse {
|
|
return routeWithExecutionOrchestration(cfg, req, string(backendFallback))
|
|
}
|
|
|
|
func routeWithExecutionOrchestration(cfg pluginConfig, req pluginapi.ModelRouteRequest, route string) pluginapi.ModelRouteResponse {
|
|
plans := executionPlansForRoute(cfg, req, route)
|
|
if len(plans) == 0 {
|
|
return pluginapi.ModelRouteResponse{Handled: false, Reason: "web_search_fallback_exhausted"}
|
|
}
|
|
return pluginapi.ModelRouteResponse{
|
|
Handled: true,
|
|
TargetKind: pluginapi.ModelRouteTargetSelf,
|
|
Reason: "claude_code_web_search_orchestrated",
|
|
}
|
|
}
|