Add projects

backend/sdk/cliproxy/config_model_max_context_length_test.go

@@ -0,0 +1,92 @@
+package cliproxy
+
+import (
+	"testing"
+
+	"github.com/router-for-me/CLIProxyAPI/v7/internal/config"
+)
+
+func TestBuildConfigModelsPropagatesMaxContextLength(t *testing.T) {
+	const want = 1048576
+
+	tests := []struct {
+		name string
+		got  func() *ModelInfo
+	}{
+		{
+			name: "codex",
+			got: func() *ModelInfo {
+				return buildCodexConfigModels(&config.CodexKey{
+					Models: []config.CodexModel{{
+						Name: "codex-upstream", Alias: "codex-alias", MaxContextLength: want,
+					}},
+				})[0]
+			},
+		},
+		{
+			name: "claude",
+			got: func() *ModelInfo {
+				return buildClaudeConfigModels(&config.ClaudeKey{
+					Models: []config.ClaudeModel{{
+						Name: "claude-upstream", Alias: "claude-alias", MaxContextLength: want,
+					}},
+				})[0]
+			},
+		},
+		{
+			name: "gemini",
+			got: func() *ModelInfo {
+				return buildGeminiConfigModels(&config.GeminiKey{
+					Models: []config.GeminiModel{{
+						Name: "gemini-upstream", Alias: "gemini-alias", MaxContextLength: want,
+					}},
+				})[0]
+			},
+		},
+		{
+			name: "interactions",
+			got: func() *ModelInfo {
+				return buildGeminiConfigModels(&config.GeminiKey{
+					Models: []config.GeminiModel{{
+						Name: "interactions-upstream", Alias: "interactions-alias", MaxContextLength: want,
+					}},
+				})[0]
+			},
+		},
+		{
+			name: "xai",
+			got: func() *ModelInfo {
+				return buildXAIConfigModels(&config.XAIKey{
+					Models: []config.XAIModel{{
+						Name: "xai-upstream", Alias: "xai-alias", MaxContextLength: want,
+					}},
+				})[0]
+			},
+		},
+		{
+			name: "openai compatibility",
+			got: func() *ModelInfo {
+				return buildOpenAICompatibilityConfigModels(&config.OpenAICompatibility{
+					Models: []config.OpenAICompatibilityModel{{
+						Name: "compat-upstream", Alias: "compat-alias", MaxContextLength: want,
+					}},
+				})[0]
+			},
+		},
+	}
+
+	for _, testCase := range tests {
+		t.Run(testCase.name, func(t *testing.T) {
+			model := testCase.got()
+			if model == nil {
+				t.Fatal("model = nil")
+			}
+			if model.ContextLength != want {
+				t.Errorf("context length = %d, want %d", model.ContextLength, want)
+			}
+			if model.MaxContextLength != want {
+				t.Errorf("max context length = %d, want %d", model.MaxContextLength, want)
+			}
+		})
+	}
+}