diff --git a/ai-provider/model-runtime/provider.go b/ai-provider/model-runtime/provider.go index 435f44fd..5183810f 100644 --- a/ai-provider/model-runtime/provider.go +++ b/ai-provider/model-runtime/provider.go @@ -50,7 +50,7 @@ type IProviderInfo interface { } func GetCustomizeLogo() string { - logo, _ := providerDir.ReadFile("customize/assets/icon_s_en.svg") + logo, _ := providerDir.ReadFile("model-providers/customize/assets/icon_s_en.svg") return string(logo) } diff --git a/module/ai/dto/output.go b/module/ai/dto/output.go index d768f5bd..d87aca1c 100644 --- a/module/ai/dto/output.go +++ b/module/ai/dto/output.go @@ -37,15 +37,16 @@ type ModelConfig struct { } type ConfiguredProviderItem struct { - Id string `json:"id"` - Name string `json:"name"` - Logo string `json:"logo"` - DefaultLLM string `json:"default_llm"` - Status ProviderStatus `json:"status"` - APICount int64 `json:"api_count"` - KeyCount int64 `json:"key_count"` - ModelCount int64 `json:"model_count"` - CanDelete bool `json:"can_delete"` + Id string `json:"id"` + Name string `json:"name"` + Logo string `json:"logo"` + DefaultLLM string `json:"default_llm"` + DefaultLLMName string `json:"default_llm_name"` + Status ProviderStatus `json:"status"` + APICount int64 `json:"api_count"` + KeyCount int64 `json:"key_count"` + ModelCount int64 `json:"model_count"` + CanDelete bool `json:"can_delete"` } type KeyStatus struct { diff --git a/module/ai/iml.go b/module/ai/iml.go index 595c7f73..26e0f1c0 100644 --- a/module/ai/iml.go +++ b/module/ai/iml.go @@ -219,7 +219,7 @@ func (i *imlProviderModule) AddProvider(ctx context.Context, input *ai_dto.NewPr Id: input.Name, Name: input.Name, DefaultConfig: config, - Logo: model_runtime.GetCustomizeLogo(), + Logo: iProvider.Logo(), }, nil } @@ -280,17 +280,21 @@ func (i *imlProviderModule) ConfiguredProviders(ctx context.Context, keyword str continue } apiCount := aiAPIMap[l.Id] - + defaultLLMName := "" + if defaultModel, has := p.GetModel(l.DefaultLLM); has { + defaultLLMName = defaultModel.Name() + } providers = append(providers, &ai_dto.ConfiguredProviderItem{ - Id: l.Id, - Name: l.Name, - Logo: p.Logo(), - DefaultLLM: l.DefaultLLM, - Status: ai_dto.ToProviderStatus(l.Status), - APICount: apiCount, - KeyCount: keyMap[l.Id], - CanDelete: apiCount < 1, - ModelCount: int64(len(p.Models())), + Id: l.Id, + Name: l.Name, + Logo: p.Logo(), + DefaultLLM: l.DefaultLLM, + DefaultLLMName: defaultLLMName, + Status: ai_dto.ToProviderStatus(l.Status), + APICount: apiCount, + KeyCount: keyMap[l.Id], + CanDelete: apiCount < 1, + ModelCount: int64(len(p.Models())), }) }