mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-10-24 16:45:45 +08:00
34e73c75a3
- Pass configured proxy URL to the npm registry updater client - Validate fetched versions against strict semver and minimum server version floor (`1.0.13`) - Update default fallback client version to `1.0.46` - Manage background updater lifecycle and context via service lifecycle initialization
100 lines
3.1 KiB
Go
100 lines
3.1 KiB
Go
package executor
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/router-for-me/CLIProxyAPI/v8/internal/runtime/executor/helps"
|
|
cliproxyauth "github.com/router-for-me/CLIProxyAPI/v8/sdk/cliproxy/auth"
|
|
)
|
|
|
|
// xaiVersionAtLeast compares dotted numeric version strings.
|
|
func xaiVersionAtLeast(got, floor string) bool {
|
|
gotParts := strings.Split(got, ".")
|
|
floorParts := strings.Split(floor, ".")
|
|
for i := 0; i < len(gotParts) || i < len(floorParts); i++ {
|
|
var g, f int
|
|
if i < len(gotParts) {
|
|
g, _ = strconv.Atoi(gotParts[i])
|
|
}
|
|
if i < len(floorParts) {
|
|
f, _ = strconv.Atoi(floorParts[i])
|
|
}
|
|
if g != f {
|
|
return g > f
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func TestXAIChatProxyClientVersionMeetsServerFloor(t *testing.T) {
|
|
// cli-chat-proxy.grok.com started rejecting client versions older than
|
|
// 1.0.13 with HTTP 426 ("Your Grok CLI version is outdated"); see #6249.
|
|
const serverFloor = "1.0.13"
|
|
if !xaiVersionAtLeast(xaiClientVersionValue, serverFloor) {
|
|
t.Fatalf("xaiClientVersionValue = %q, cli-chat-proxy rejects clients older than %s with 426", xaiClientVersionValue, serverFloor)
|
|
}
|
|
}
|
|
|
|
func TestXAIChatProxyIdentityHeadersDeriveFromOneConstant(t *testing.T) {
|
|
auth := &cliproxyauth.Auth{
|
|
Provider: "xai",
|
|
Attributes: map[string]string{"auth_kind": "oauth"},
|
|
}
|
|
req, err := http.NewRequest(http.MethodPost, xaiChatBaseURL(auth)+"/chat/completions", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
applyXAIChatHeaders(req, auth, "token", false, "")
|
|
|
|
if got := req.Header.Get(xaiClientVersionHeader); got != xaiClientVersionValue {
|
|
t.Fatalf("%s = %q, want %q", xaiClientVersionHeader, got, xaiClientVersionValue)
|
|
}
|
|
if got, want := req.Header.Get("User-Agent"), "xai-grok-workspace/"+xaiClientVersionValue; got != want {
|
|
t.Fatalf("User-Agent = %q, want %q (must derive from xaiClientVersionValue)", got, want)
|
|
}
|
|
}
|
|
|
|
func TestXAIChatProxyCustomHeadersOverridePinnedVersion(t *testing.T) {
|
|
auth := &cliproxyauth.Auth{
|
|
Provider: "xai",
|
|
Attributes: map[string]string{
|
|
"auth_kind": "oauth",
|
|
"header:x-grok-client-version": "9.9.9",
|
|
},
|
|
}
|
|
req, err := http.NewRequest(http.MethodPost, xaiChatBaseURL(auth)+"/chat/completions", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
applyXAIChatHeaders(req, auth, "token", false, "")
|
|
|
|
if got := req.Header.Get(xaiClientVersionHeader); got != "9.9.9" {
|
|
t.Fatalf("%s = %q, want the per-auth custom header to override the pin", xaiClientVersionHeader, got)
|
|
}
|
|
}
|
|
|
|
func TestXAIChatProxyIdentityHeadersFollowDynamicVersion(t *testing.T) {
|
|
restore := helps.SetXAIClientVersionForTest("1.2.34")
|
|
defer restore()
|
|
|
|
auth := &cliproxyauth.Auth{
|
|
Provider: "xai",
|
|
Attributes: map[string]string{"auth_kind": "oauth"},
|
|
}
|
|
req, err := http.NewRequest(http.MethodPost, xaiChatBaseURL(auth)+"/chat/completions", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
applyXAIChatHeaders(req, auth, "token", false, "")
|
|
|
|
if got := req.Header.Get(xaiClientVersionHeader); got != "1.2.34" {
|
|
t.Fatalf("%s = %q, want dynamic version 1.2.34", xaiClientVersionHeader, got)
|
|
}
|
|
if got, want := req.Header.Get("User-Agent"), "xai-grok-workspace/1.2.34"; got != want {
|
|
t.Fatalf("User-Agent = %q, want %q", got, want)
|
|
}
|
|
}
|