lanyuanxiaoyao
395887667d
实现统一模型 ID 格式 (provider_id/model_name)，支持跨协议模型标识和 Smart Passthrough。 核心变更： - 新增 pkg/modelid 包：解析、格式化、校验统一模型 ID - 数据库迁移：models 表使用 UUID 主键 + UNIQUE(provider_id, model_name) 约束 - Repository 层：FindByProviderAndModelName、ListEnabled 方法 - Service 层：联合唯一校验、provider ID 字符集校验 - Conversion 层：ExtractModelName、RewriteRequestModelName/RewriteResponseModelName 方法 - Handler 层：统一模型 ID 路由、Smart Passthrough、Models API 本地聚合 - 新增 error-responses、unified-model-id 规范 测试覆盖： - 单元测试：modelid、conversion、handler、service、repository - 集成测试：统一模型 ID 路由、Smart Passthrough 保真性、跨协议转换 - 迁移测试：UUID 主键、UNIQUE 约束、级联删除 OpenSpec： - 归档 unified-model-id 变更到 archive/2026-04-21-unified-model-id - 同步 11 个 delta specs 到 main specs - 新增 error-responses、unified-model-id 规范文件
86 lines
2.3 KiB
Go
86 lines
2.3 KiB
Go
package service
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"nex/backend/pkg/modelid"
|
|
|
|
"nex/backend/internal/domain"
|
|
"nex/backend/internal/repository"
|
|
appErrors "nex/backend/pkg/errors"
|
|
)
|
|
|
|
type providerService struct {
|
|
providerRepo repository.ProviderRepository
|
|
modelRepo repository.ModelRepository
|
|
}
|
|
|
|
func NewProviderService(providerRepo repository.ProviderRepository, modelRepo repository.ModelRepository) ProviderService {
|
|
return &providerService{providerRepo: providerRepo, modelRepo: modelRepo}
|
|
}
|
|
|
|
func (s *providerService) Create(provider *domain.Provider) error {
|
|
// 校验 provider_id 字符集
|
|
if err := modelid.ValidateProviderID(provider.ID); err != nil {
|
|
return appErrors.ErrInvalidProviderID
|
|
}
|
|
provider.Enabled = true
|
|
err := s.providerRepo.Create(provider)
|
|
if err != nil && isUniqueConstraintError(err) {
|
|
return appErrors.ErrConflict
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (s *providerService) Get(id string, maskKey bool) (*domain.Provider, error) {
|
|
provider, err := s.providerRepo.GetByID(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if maskKey {
|
|
provider.MaskAPIKey()
|
|
}
|
|
return provider, nil
|
|
}
|
|
|
|
func (s *providerService) List() ([]domain.Provider, error) {
|
|
providers, err := s.providerRepo.List()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for i := range providers {
|
|
providers[i].MaskAPIKey()
|
|
}
|
|
return providers, nil
|
|
}
|
|
|
|
func (s *providerService) Update(id string, updates map[string]interface{}) error {
|
|
if _, ok := updates["id"]; ok {
|
|
return appErrors.ErrImmutableField
|
|
}
|
|
return s.providerRepo.Update(id, updates)
|
|
}
|
|
|
|
func (s *providerService) Delete(id string) error {
|
|
return s.providerRepo.Delete(id)
|
|
}
|
|
|
|
// ListEnabledModels 返回所有启用的模型（用于 Models 接口本地聚合）
|
|
func (s *providerService) ListEnabledModels() ([]domain.Model, error) {
|
|
return s.modelRepo.ListEnabled()
|
|
}
|
|
|
|
// GetModelByProviderAndName 按 provider_id 和 model_name 查询模型（用于 ModelInfo 接口本地查询）
|
|
func (s *providerService) GetModelByProviderAndName(providerID, modelName string) (*domain.Model, error) {
|
|
return s.modelRepo.FindByProviderAndModelName(providerID, modelName)
|
|
}
|
|
|
|
// isUniqueConstraintError 判断是否为数据库唯一约束冲突错误
|
|
func isUniqueConstraintError(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
msg := strings.ToLower(err.Error())
|
|
return strings.Contains(msg, "unique constraint") || strings.Contains(msg, "duplicate")
|
|
}
|