openai_url_test.go RawHistoryFolder

1package provider
2
3import "testing"
4
5// TestChatCompletionsURL pins the endpoint built for OpenAI-compatible
6// providers. The regression that motivated it: Z.AI's coding-plan base
7// carries a "/v4" version segment, so blindly appending
8// "/v1/chat/completions" produced ".../paas/v4/v1/chat/completions"
9// and a 404. Any base that already ends in a version segment must get
10// "/chat/completions" appended directly.
11func TestChatCompletionsURL(t *testing.T) {
12	cases := []struct {
13		name    string
14		baseURL string
15		want    string
16	}{
17		{
18			name:    "bare host gets conventional /v1 path",
19			baseURL: "https://api.openai.com",
20			want:    "https://api.openai.com/v1/chat/completions",
21		},
22		{
23			name:    "v1 base is not doubled",
24			baseURL: "https://api.moonshot.ai/v1",
25			want:    "https://api.moonshot.ai/v1/chat/completions",
26		},
27		{
28			name:    "zai coding plan v4 base is not given a spurious /v1",
29			baseURL: "https://api.z.ai/api/coding/paas/v4",
30			want:    "https://api.z.ai/api/coding/paas/v4/chat/completions",
31		},
32	}
33	for _, tc := range cases {
34		t.Run(tc.name, func(t *testing.T) {
35			if got := chatCompletionsURL(tc.baseURL); got != tc.want {
36				t.Errorf("chatCompletionsURL(%q) = %q, want %q", tc.baseURL, got, tc.want)
37			}
38		})
39	}
40}
41
42func TestGithubCopilotChatCompletionsURL(t *testing.T) {
43	copilot, ok := NewGithubCopilotClient("test-token").(*copilotClient)
44	if !ok {
45		t.Fatal("NewGithubCopilotClient did not return a copilotClient")
46	}
47	client, ok := copilot.router.fallback.(*openaiClient)
48	if !ok {
49		t.Fatal("GitHub Copilot router fallback is not an openaiClient")
50	}
51
52	const want = "https://api.individual.githubcopilot.com/chat/completions"
53	if got := client.chatCompletionsURL(); got != want {
54		t.Errorf("GitHub Copilot chat completions URL = %q, want %q", got, want)
55	}
56}