Merge pull request #109 from 73ai/glm-integration

Add Z.AI (Zhipu AI) GLM provider

Author: priyanshujain

config/profiles.go

@@ -77,6 +77,19 @@ var Profiles = map[string]ModelProfile{
 		},
 		Providers: []string{"openrouter"},
 	},
+	"zai": {
+		Name:        "zai",
+		Label:       "Z.AI (single provider)",
+		Description: "GLM models from Z.AI. Free tier available.",
+		Category:    "single",
+		Tiers: ProfileTiers{
+			Default: "zai/glm-4.5-flash",
+			Complex: "zai/glm-4.7",
+			Fast:    "zai/glm-4.5-flash",
+			Nano:    "zai/glm-4.5-flash",
+		},
+		Providers: []string{"zai"},
+	},
 	"openai": {
 		Name:        "openai",
 		Label:       "OpenAI (single provider)",
@@ -134,7 +147,7 @@ var Profiles = map[string]ModelProfile{
 // ProfileNames returns profile names in display order.
 var ProfileNames = []string{
 	"starter", "standard", "premium",
-	"gemini", "anthropic", "openai", "groq", "openrouter",
+	"gemini", "anthropic", "openai", "groq", "openrouter", "zai",
 }
 
 var profileNameRe = regexp.MustCompile(`^[a-z][a-z0-9-]{1,29}$`)

internal/cli/chat.go

@@ -25,6 +25,7 @@ import (
 	_ "github.com/73ai/openbotkit/provider/groq"
 	_ "github.com/73ai/openbotkit/provider/openai"
 	_ "github.com/73ai/openbotkit/provider/openrouter"
+	_ "github.com/73ai/openbotkit/provider/zai"
 	"github.com/spf13/cobra"
 )
 

internal/cli/settings_cmd.go

@@ -13,6 +13,7 @@ import (
 	_ "github.com/73ai/openbotkit/provider/groq"
 	_ "github.com/73ai/openbotkit/provider/openai"
 	_ "github.com/73ai/openbotkit/provider/openrouter"
+	_ "github.com/73ai/openbotkit/provider/zai"
 	"github.com/73ai/openbotkit/settings"
 	"github.com/spf13/cobra"
 )

internal/cli/setup_models.go

@@ -15,6 +15,7 @@ import (
 	_ "github.com/73ai/openbotkit/provider/groq"
 	_ "github.com/73ai/openbotkit/provider/openai"
 	_ "github.com/73ai/openbotkit/provider/openrouter"
+	_ "github.com/73ai/openbotkit/provider/zai"
 	"github.com/spf13/cobra"
 )
 

provider/context_window.go

@@ -16,6 +16,15 @@ var contextWindows = map[string]int{
 	"llama-3.3-70b-versatile":   131072,
 	"llama-4-scout-17b-16e":     131072,
 	"llama-4-maverick-17b-128e": 131072,
+	"glm-4.5-flash":            128000,
+	"glm-4.5-air":              128000,
+	"glm-4.5":                  128000,
+	"glm-4.6":                  128000,
+	"glm-4.7":                  200000,
+	"glm-4.7-flash":            200000,
+	"glm-4.7-flashx":           200000,
+	"glm-5":                    200000,
+	"glm-5-turbo":              200000,
 }
 
 // DefaultContextWindow returns the context window size for a model.

provider/context_window_test.go

@@ -51,6 +51,25 @@ func TestDefaultContextWindow_NestedModelID(t *testing.T) {
 	}
 }
 
+func TestDefaultContextWindow_GLMModels(t *testing.T) {
+	models := map[string]int{
+		"glm-4.5-flash":  128000,
+		"glm-4.5-air":    128000,
+		"glm-4.5":        128000,
+		"glm-4.6":        128000,
+		"glm-4.7":        200000,
+		"glm-4.7-flash":  200000,
+		"glm-4.7-flashx": 200000,
+		"glm-5":          200000,
+		"glm-5-turbo":    200000,
+	}
+	for model, want := range models {
+		if got := DefaultContextWindow(model); got != want {
+			t.Errorf("%s = %d, want %d", model, got, want)
+		}
+	}
+}
+
 func TestDefaultContextWindow_GroqModels(t *testing.T) {
 	models := map[string]int{
 		"llama-3.1-8b-instant":      131072,

provider/models.go

@@ -31,6 +31,8 @@ func ListModels(ctx context.Context, providerName, apiKey string, cfg config.Mod
 		return listModelsGroq(ctx, apiKey, cfg)
 	case "openrouter":
 		return listModelsOpenRouter(ctx, apiKey, cfg)
+	case "zai":
+		return listModelsZAI(ctx, apiKey, cfg)
 	default:
 		return nil, fmt.Errorf("unknown provider %q", providerName)
 	}
@@ -209,6 +211,43 @@ func listModelsOpenRouter(ctx context.Context, apiKey string, cfg config.ModelPr
 	return listModelsOpenAICompat(ctx, apiKey, cfg, "https://openrouter.ai/api", "openrouter")
 }
 
+// --- Z.AI ---
+
+func listModelsZAI(ctx context.Context, apiKey string, cfg config.ModelProviderConfig) ([]AvailableModel, error) {
+	url := baseURL(cfg, "https://api.z.ai/api/paas/v4") + "/models"
+	req, err := http.NewRequestWithContext(ctx, "GET", url, nil)
+	if err != nil {
+		return nil, err
+	}
+	req.Header.Set("Authorization", "Bearer "+apiKey)
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("zai list models: HTTP %d", resp.StatusCode)
+	}
+
+	var body struct {
+		Data []struct {
+			ID string `json:"id"`
+		} `json:"data"`
+	}
+	if err := json.NewDecoder(resp.Body).Decode(&body); err != nil {
+		return nil, err
+	}
+
+	var models []AvailableModel
+	for _, m := range body.Data {
+		if strings.Contains(m.ID, "glm") {
+			models = append(models, AvailableModel{ID: m.ID, DisplayName: m.ID, Provider: "zai"})
+		}
+	}
+	return models, nil
+}
+
 func listModelsOpenAICompat(ctx context.Context, apiKey string, cfg config.ModelProviderConfig, defaultBase, providerName string) ([]AvailableModel, error) {
 	url := baseURL(cfg, defaultBase) + "/v1/models"
 	req, err := http.NewRequestWithContext(ctx, "GET", url, nil)

provider/registry.go

@@ -14,6 +14,7 @@ var ProviderEnvVars = map[string]string{
 	"gemini":     "GEMINI_API_KEY",
 	"groq":       "GROQ_API_KEY",
 	"openrouter": "OPENROUTER_API_KEY",
+	"zai":        "ZAI_API_KEY",
 }
 
 // Factory creates a Provider from a model provider config and resolved API key.