feat: allow model definition inheritance

Signed-off-by: blob42 <contact@blob42.xyz>

Author: blob42

README.md

@@ -15,6 +15,7 @@ This fork does the following:
 - **UI Query and select** local or remote model
 - **Strips thinking tokens** from replies if the model forgets to use codeblocks
 - **New callback types**: `insert_lines` and `prepend_lines`
+- **Model definition inheritance**: Define models that inherit other model parameters
 - **Refactored for idiomatic Lua** and neovim plugin style
 - **Simplified command system** with explicit configuration
 - **Tests with plenary library**

lua/codegpt/config.lua

@@ -150,6 +150,7 @@ M.persistent_override = nil
 ---@field callback_type? codegpt.CallbackType Controls what the plugin does with the response
 ---@field extra_params? table Custom parameters to include with this model query
 ---@field append_string? string String to append to prompt -- ex: /no_think
+---@field from? string (optional) Name of parent model to inherit params from
 
 ---@alias ModelDef { [string] : codegpt.Model | string }
 

lua/codegpt/models.lua

@@ -31,6 +31,15 @@ function M.get_model_by_name(name)
 		end
 	end
 
+	-- (optional) inherit a parent model
+	if result ~= nil and result.from ~= nil and #result.from > 0 then
+		local parent_name, parent = M.get_model_by_name(result.from)
+		if parent ~= nil then
+			result = vim.tbl_deep_extend("force", parent, result)
+			selected = parent_name
+		end
+	end
+
 	return selected, result
 end
 --- default model selection order from highest to lowest priority
@@ -74,6 +83,15 @@ function M.get_model()
 		end
 	end
 
+	-- (optional) inherit a parent model
+	if result ~= nil and result.from ~= nil and #result.from > 0 then
+		local parent_name, parent = M.get_model_by_name(result.from)
+		if parent ~= nil then
+			result = vim.tbl_deep_extend("force", parent, result)
+			selected = parent_name
+		end
+	end
+
 	return selected, result
 end
 

tests/models_spec.lua

@@ -164,4 +164,58 @@ describe("models selection", function()
 		local name, model = models.get_model()
 		assert(name == "gpt4-o")
 	end)
+
+	it("should inherit a model definition", function()
+		codegpt.setup({
+			connection = {
+				api_provider = "openai",
+			},
+			models = {
+				openai = {
+					default = "fbar",
+					foo = {
+						temperature = 0.5,
+						max_tokens = 420,
+					},
+					foobar = {
+						alias = "fbar",
+						from = "foo",
+						temperature = 1,
+						append_string = "/no_think",
+					},
+					bar = {
+						from = "fbar",
+						temperature = 0,
+					},
+				},
+			},
+		})
+		local _, model = models.get_model()
+		assert(vim.deep_equal(model, {
+			temperature = 1,
+			alias = "fbar",
+			from = "foo",
+			max_tokens = 420,
+			append_string = "/no_think",
+		}))
+
+		local name, model = models.get_model_by_name("fbar")
+		assert(vim.deep_equal(model, {
+			temperature = 1,
+			alias = "fbar",
+			from = "foo",
+			max_tokens = 420,
+			append_string = "/no_think",
+		}))
+		assert(name == "foo")
+		local name, model = models.get_model_by_name("bar")
+		assert(vim.deep_equal(model, {
+			temperature = 0,
+			alias = "fbar",
+			from = "fbar",
+			max_tokens = 420,
+			append_string = "/no_think",
+		}))
+		assert(name == "foo")
+	end)
 end)