mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-09-07 00:24:17 +08:00
27fc3169bb
- Added `UsesConfig` to `XAIAutoExecutor` for determining config-based binding. - Refactored `Service` to support thread-safe executor registration, using a new mutex (`executorRegistrationMu`). - Improved executor rebind logic to prevent unnecessary replacements unless forced or required by config updates. - Updated handling of config updates to correctly replace stale XAI executors. - Enhanced the test suite for edge cases in executor binding and replacement for multiple providers. Closes: #4567
60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
package cliproxy
|
|
|
|
import (
|
|
"testing"
|
|
|
|
coreauth "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/auth"
|
|
"github.com/router-for-me/CLIProxyAPI/v7/sdk/config"
|
|
)
|
|
|
|
func TestHasNativeOpenAICompatExecutorConfig(t *testing.T) {
|
|
service := &Service{
|
|
cfg: &config.Config{
|
|
OpenAICompatibility: []config.OpenAICompatibility{
|
|
{Name: "native-provider", BaseURL: "https://native.example.com/v1"},
|
|
},
|
|
},
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
auth *coreauth.Auth
|
|
providerKey string
|
|
want bool
|
|
}{
|
|
{
|
|
name: "config provider",
|
|
auth: &coreauth.Auth{Provider: "native-provider"},
|
|
providerKey: "native-provider",
|
|
want: true,
|
|
},
|
|
{
|
|
name: "inline base url",
|
|
auth: &coreauth.Auth{Provider: "plugin-provider", Attributes: map[string]string{"base_url": "https://compat.example.com/v1"}},
|
|
providerKey: "plugin-provider",
|
|
want: true,
|
|
},
|
|
{
|
|
name: "compat metadata",
|
|
auth: &coreauth.Auth{Provider: "openai-compatibility", Attributes: map[string]string{"compat_name": "compat"}},
|
|
providerKey: "compat",
|
|
want: true,
|
|
},
|
|
{
|
|
name: "plain plugin auth",
|
|
auth: &coreauth.Auth{Provider: "plugin-provider", Attributes: map[string]string{"api_key": "test"}},
|
|
providerKey: "plugin-provider",
|
|
want: false,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got := service.hasNativeOpenAICompatExecutorConfig(tt.auth, tt.providerKey, service.cfg)
|
|
if got != tt.want {
|
|
t.Fatalf("hasNativeOpenAICompatExecutorConfig() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|