feat: defaults in setup()

geomgle · Dec 4, 2023 · fe744c9 · fe744c9
1 parent 2f548ce
commit fe744c9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/lua/model/init.lua b/lua/model/init.lua
@@ -407,10 +407,12 @@ end
 ---@field hl_group? string default = 'Comment'. Set the default highlight group of in-progress responses
 ---@field join_undo? boolean default = true. Join streaming response text as a single `u` undo. Edits during streaming will also be undone.
 
----@param opts SetupOptions
+---@param opts? SetupOptions
 function M.setup(opts)
   M.opts = vim.tbl_extend('force', {
-    default_prompt = require('model.providers.openai').default_prompt
+    default_prompt = require('model.providers.openai').default_prompt,
+    prompts = require('model.prompts.starters'),
+    chats = require('model.prompts.chats')
   }, opts or {})
 
   if M.opts.prompts then

diff --git a/plugin/init.lua b/plugin/init.lua
@@ -8,7 +8,4 @@ vim.filetype.add({
   }
 })
 
-require('model').setup({
-  prompts = require('model.prompts.starters'),
-  chats = require('model.prompts.chats')
-})
+require('model').setup()