@@ -206,3 +206,19 @@ func (c *Config) GetModel(provider, model string) *provider.Model {
206206 }
207207 return nil
208208}
209+
210+func (c *Config) LargeModel() *provider.Model {
211+ model, ok := c.Models[SelectedModelTypeLarge]
212+ if !ok {
213+ return nil
214+ }
215+ return c.GetModel(model.Provider, model.Model)
216+}
217+
218+func (c *Config) SmallModel() *provider.Model {
219+ model, ok := c.Models[SelectedModelTypeSmall]
220+ if !ok {
221+ return nil
222+ }
223+ return c.GetModel(model.Provider, model.Model)
224+}