diff --git a/pkg/config/config.go b/pkg/config/config.go
index e1554e23d3755637178a033c25aca2fee75525b5..b2dd6e2ed1b625106e8c6b3727853e2e2d7d6bd5 100644
--- a/pkg/config/config.go
+++ b/pkg/config/config.go
@@ -206,3 +206,19 @@ func (c *Config) GetModel(provider, model string) *provider.Model {
 	}
 	return nil
 }
+
+func (c *Config) LargeModel() *provider.Model {
+	model, ok := c.Models[SelectedModelTypeLarge]
+	if !ok {
+		return nil
+	}
+	return c.GetModel(model.Provider, model.Model)
+}
+
+func (c *Config) SmallModel() *provider.Model {
+	model, ok := c.Models[SelectedModelTypeSmall]
+	if !ok {
+		return nil
+	}
+	return c.GetModel(model.Provider, model.Model)
+}