|
- package model
-
- func GetModelEnableGroups(modelName string) []string {
- // 确保缓存最新
- GetPricing()
-
- if modelName == "" {
- return make([]string, 0)
- }
-
- modelEnableGroupsLock.RLock()
- groups, ok := modelEnableGroups[modelName]
- modelEnableGroupsLock.RUnlock()
- if !ok {
- return make([]string, 0)
- }
- return groups
- }
-
- // GetModelQuotaTypes 返回指定模型的计费类型集合(来自缓存)
- func GetModelQuotaTypes(modelName string) []int {
- GetPricing()
-
- modelEnableGroupsLock.RLock()
- quota, ok := modelQuotaTypeMap[modelName]
- modelEnableGroupsLock.RUnlock()
- if !ok {
- return []int{}
- }
- return []int{quota}
- }
|