EnableModel 启用指定模型
(provider, model string)
| 237 | |
| 238 | // EnableModel 启用指定模型 |
| 239 | func (m *ModelFallbackManager) EnableModel(provider, model string) error { |
| 240 | modelKey := fmt.Sprintf("%s/%s", provider, model) |
| 241 | |
| 242 | for _, fb := range m.fallbacks { |
| 243 | fbKey := fmt.Sprintf("%s/%s", fb.Config.Provider, fb.Config.Model) |
| 244 | if fbKey == modelKey { |
| 245 | if !fb.Enabled && fb.provider == nil { |
| 246 | // 需要重新创建 Provider |
| 247 | prov, err := m.deps.ProviderFactory.Create(fb.Config) |
| 248 | if err != nil { |
| 249 | return fmt.Errorf("failed to create provider: %w", err) |
| 250 | } |
| 251 | fb.provider = prov |
| 252 | } |
| 253 | fb.Enabled = true |
| 254 | fallbackLog.Info(context.Background(), "enabled model", map[string]any{"model": modelKey}) |
| 255 | return nil |
| 256 | } |
| 257 | } |
| 258 | |
| 259 | return fmt.Errorf("model not found: %s", modelKey) |
| 260 | } |
| 261 | |
| 262 | // DisableModel 禁用指定模型 |
| 263 | func (m *ModelFallbackManager) DisableModel(provider, model string) error { |