( overrideContext codersdk.ChatModelOverrideContext, )
| 577 | } |
| 578 | |
| 579 | func (api *API) chatModelOverrideSiteConfig( |
| 580 | overrideContext codersdk.ChatModelOverrideContext, |
| 581 | ) (chatModelOverrideSiteConfig, error) { |
| 582 | switch overrideContext { |
| 583 | case codersdk.ChatModelOverrideContextGeneral: |
| 584 | return chatModelOverrideSiteConfig{ |
| 585 | label: "general", |
| 586 | getter: api.Database.GetChatGeneralModelOverride, |
| 587 | upsert: api.Database.UpsertChatGeneralModelOverride, |
| 588 | }, nil |
| 589 | case codersdk.ChatModelOverrideContextExplore: |
| 590 | return chatModelOverrideSiteConfig{ |
| 591 | label: "explore", |
| 592 | getter: api.Database.GetChatExploreModelOverride, |
| 593 | upsert: api.Database.UpsertChatExploreModelOverride, |
| 594 | }, nil |
| 595 | case codersdk.ChatModelOverrideContextTitleGeneration: |
| 596 | return chatModelOverrideSiteConfig{ |
| 597 | label: "title generation", |
| 598 | getter: api.Database.GetChatTitleGenerationModelOverride, |
| 599 | upsert: api.Database.UpsertChatTitleGenerationModelOverride, |
| 600 | }, nil |
| 601 | default: |
| 602 | return chatModelOverrideSiteConfig{}, xerrors.Errorf( |
| 603 | "unknown chat model override context %q", |
| 604 | overrideContext, |
| 605 | ) |
| 606 | } |
| 607 | } |
| 608 | |
| 609 | func (api *API) readChatModelOverrideConfig( |
| 610 | ctx context.Context, |
no test coverage detected