This epic is to capture work to meet the model and UP configuration of all Duo features. ### Background Custom Models [is building ](https://gitlab.com/groups/gitlab-org/-/epics/17181)the infrastructure for .com, Dedicated, and self-managed Duo customers to select from among available models per feature. In order to enable model selection for each feature, the feature teams need to: 1. check to see if the models used by their feature are already declared in [models.yml](https://gitlab.com/gitlab-org/modelops/applied-ml/code-suggestions/ai-assist/-/blob/main/ai_gateway/model_selection/models.yml?ref_type=heads) 2. if default model for the feature is not on models.yml, add it 3. configure the selectable and default models for each feature's UP functions in [unit_primitives.yml](https://gitlab.com/gitlab-org/modelops/applied-ml/code-suggestions/ai-assist/-/blob/main/ai_gateway/model_selection/unit_primitives.yml?ref_type=heads) 1. note that all feature teams should add Claude 3.5 Sonnet to selectable models to support specific customer requirements **This work must be completed by 12 June 2025 to meet an ongoing customer escalation.** ### Technical Lead - @bcardoso- ### Definition of Done * Each feature team has verified/added their supported models to model.yml * Each feature team has added their default models per feature to unit_primitives.yml * Each feature team has added Claude 3.5 Sonnet to selectable models for each unit primative in unit_primitives.yml