bda75aeb0d
- Updated .env.example to include new configuration options for AI gateway and WebUI secret key. - Modified Nginx configuration to support additional API routes for model management and migration. - Implemented new API endpoints for discovering organization-level LLM models and managing hosted mail services. - Enhanced AI gateway logic to support organization-specific model access and permissions. - Improved error handling and response structures in the AI and mail APIs. - Added integration tests for new features and updated existing tests for model access control.
71 lines
1.5 KiB
Go
71 lines
1.5 KiB
Go
package ai
|
|
|
|
import "strings"
|
|
|
|
func ApplyModelCatalog(upstream []map[string]any, catalog []ModelCatalogEntry) []map[string]any {
|
|
if len(catalog) == 0 {
|
|
return upstream
|
|
}
|
|
|
|
allowed := make(map[string]ModelCatalogEntry)
|
|
for _, entry := range catalog {
|
|
id := strings.TrimSpace(entry.ModelID)
|
|
if id == "" || !entry.Enabled {
|
|
continue
|
|
}
|
|
allowed[id] = entry
|
|
}
|
|
if len(allowed) == 0 {
|
|
return []map[string]any{}
|
|
}
|
|
|
|
out := make([]map[string]any, 0, len(allowed))
|
|
seen := make(map[string]struct{}, len(allowed))
|
|
for _, model := range upstream {
|
|
id, _ := model["id"].(string)
|
|
id = strings.TrimSpace(id)
|
|
entry, ok := allowed[id]
|
|
if !ok {
|
|
continue
|
|
}
|
|
seen[id] = struct{}{}
|
|
out = append(out, catalogModel(id, entry.Label, model["owned_by"]))
|
|
}
|
|
for id, entry := range allowed {
|
|
if _, ok := seen[id]; ok {
|
|
continue
|
|
}
|
|
out = append(out, catalogModel(id, entry.Label, nil))
|
|
}
|
|
return out
|
|
}
|
|
|
|
func IsModelAllowed(model string, catalog []ModelCatalogEntry) bool {
|
|
model = strings.TrimSpace(model)
|
|
if model == "" || len(catalog) == 0 {
|
|
return true
|
|
}
|
|
for _, entry := range catalog {
|
|
if strings.TrimSpace(entry.ModelID) == model {
|
|
return entry.Enabled
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func catalogModel(id, label string, ownedBy any) map[string]any {
|
|
display := strings.TrimSpace(label)
|
|
if display == "" {
|
|
display = id
|
|
}
|
|
out := map[string]any{
|
|
"id": id,
|
|
"object": "model",
|
|
"label": display,
|
|
}
|
|
if ownedBy != nil {
|
|
out["owned_by"] = ownedBy
|
|
}
|
|
return out
|
|
}
|