git.iko.soy/crush
on main help sign in
chore: add helper funcs
d51853cKujtim Hoxha
pkg/config/config.go+16
compare with main
unifiedsplit
@@ -206,3 +206,19 @@ func (c *Config) GetModel(provider, model string) *provider.Model {
206 }
206 }
207 return nil
207 return nil
208}
208}
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+}