mirror of
https://github.com/chaitin/MonkeyCode.git
synced 2026-02-06 00:33:57 +08:00
17 lines
371 B
Go
17 lines
371 B
Go
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
|
|
}
|