fix: add provider checker

Author: qingjing.cai

backend/internal/model/repo/model.go

@@ -18,6 +18,7 @@ import (
 	"github.com/chaitin/MonkeyCode/backend/ent/types"
 	"github.com/chaitin/MonkeyCode/backend/pkg/cvt"
 	"github.com/chaitin/MonkeyCode/backend/pkg/entx"
+	"github.com/chaitin/MonkeyCode/backend/pkg/tools"
 )
 
 type ModelRepo struct {
@@ -69,6 +70,9 @@ func (r *ModelRepo) Create(ctx context.Context, m *domain.CreateModelReq) (*db.M
 		status = consts.ModelStatusDefault
 	}
 
+	// 尝试修正用户不清楚 provider 时的情况
+	m.Provider = tools.CheckProvider(ctx, m.Provider, m.ModelName)
+
 	r.cache.Delete(string(m.ModelType))
 	create := r.db.Model.Create().
 		SetUserID(m.AdminID).
@@ -184,7 +188,6 @@ func (r *ModelRepo) GetTokenUsage(ctx context.Context, modelType consts.ModelTyp
 				OrderBy("date")
 		}).
 		Scan(ctx, &dailyUsages)
-
 	if err != nil {
 		return nil, err
 	}

backend/internal/proxy/proxy.go

@@ -19,6 +19,7 @@ import (
 	"github.com/chaitin/MonkeyCode/backend/internal/middleware"
 	"github.com/chaitin/MonkeyCode/backend/pkg/logger"
 	"github.com/chaitin/MonkeyCode/backend/pkg/tee"
+	"github.com/chaitin/MonkeyCode/backend/pkg/tools"
 )
 
 type CtxKey struct{}
@@ -114,7 +115,7 @@ func (l *LLMProxy) rewrite(r *httputil.ProxyRequest) {
 	}
 
 	metadata := make(map[string]string)
-	if m.Provider == consts.ModelProviderZhiPu {
+	if tools.CheckProvider(r.In.Context(), m.Provider, m.ModelName) == consts.ModelProviderZhiPu {
 		body, err := io.ReadAll(r.In.Body)
 		if err != nil {
 			l.logger.ErrorContext(r.In.Context(), "read request body failed", slog.String("path", r.In.URL.Path), slog.Any("err", err))

backend/pkg/tools/provider_checker.go

@@ -0,0 +1,16 @@
+package tools
+
+import (
+	"context"
+	"strings"
+
+	"github.com/chaitin/MonkeyCode/backend/consts"
+)
+
+// CheckProvider 根据 model 的前缀来判断 provider
+func CheckProvider(ctx context.Context, provider consts.ModelProvider, modelName string) consts.ModelProvider {
+	if strings.HasPrefix(modelName, "glm") {
+		provider = consts.ModelProviderZhiPu
+	}
+	return provider
+}