Allow configuration of the openAI url and the displayed name

README.md

@@ -195,8 +195,12 @@ require("neoai").setup({
         cutoff_width = 75,
     },
     prompts = {
+        default_prompt = function()
+            return "Please only follow instructions or answer to questions. Be concise."
+        end,
         context_prompt = function(context)
-            return "Hey, I'd like to provide some context for future "
+            return "Please only follow instructions or answer to questions. Be concise. "
+                .. "I'd like to provide some context for future "
                 .. "messages. Here is the code/text that I want to refer "
                 .. "to in our upcoming conversations:\n\n"
                 .. context

lua/neoai/chat/history.lua

@@ -24,6 +24,8 @@ function ChatHistory:new(model, params, context)
     if context ~= nil then
         local context_prompt = config.options.prompts.context_prompt(context)
         self:set_prompt(context_prompt)
+    else
+        self:set_prompt(config.options.prompts.default_prompt())
     end
     return obj
 end

lua/neoai/chat/models/openai.lua

@@ -1,10 +1,13 @@
 local utils = require("neoai.utils")
 local config = require("neoai.config")
 
+--- This model definition supports OpenAI - and compatible APIs. The
+--- configuration defaults to OpenAI
+
 ---@type ModelModule
 local M = {}
 
-M.name = "OpenAI"
+M.name = config.options.open_ai.display_name
 
 M._chunks = {}
 local raw_chunks = {}
@@ -53,6 +56,7 @@ end
 ---@param on_complete fun(err?: string, output?: string) Function to call when model has finished
 M.send_to_model = function(chat_history, on_stdout_chunk, on_complete)
     local api_key = config.options.open_ai.api_key.get()
+    local url = config.options.open_ai.url
 
     local data = {
         model = chat_history.model,
@@ -67,7 +71,7 @@ M.send_to_model = function(chat_history, on_stdout_chunk, on_complete)
         "--silent",
         "--show-error",
         "--no-buffer",
-        "https://api.openai.com/v1/chat/completions",
+        url,
         "-H",
         "Content-Type: application/json",
         "-H",

lua/neoai/config.lua

@@ -12,7 +12,7 @@ M.get_defaults = function()
         ui = {
             output_popup_text = "NeoAI",
             input_popup_text = "Prompt",
-            width = 30, -- As percentage eg. 30%
+            width = 30,               -- As percentage eg. 30%
             output_popup_height = 80, -- As percentage eg. 80%
             submit = "<Enter>",
         },
@@ -34,17 +34,22 @@ M.get_defaults = function()
         },
         prompts = {
             context_prompt = function(context)
-                return "Hey, I'd like to provide some context for future "
+                return "Please only follow instructions or answer to questions. Be concise. "
+                    .. "I'd like to provide some context for future "
                     .. "messages. Here is the code/text that I want to refer "
-                    .. "to in our upcoming conversations (TEXT/CODE ONLY):\n\n"
+                    .. "to in our upcoming conversations:\n\n"
                     .. context
             end,
+            default_prompt = function()
+                return "Please only follow instructions or answer to questions. Be concise."
+            end,
         },
         mappings = {
             ["select_up"] = "<C-k>",
             ["select_down"] = "<C-j>",
         },
         open_ai = {
+            display_name = "OpenAI",
             api_key = {
                 env = "OPENAI_API_KEY",
                 value = nil,
@@ -72,6 +77,7 @@ M.get_defaults = function()
                     error(msg)
                 end,
             },
+            url = "https://api.openai.com/v1/chat/completions",
         },
         shortcuts = {
             {
@@ -122,7 +128,8 @@ end
 ---@field cutoff_width integer | nil When injecting if the text becomes longer than this then it should go to a new line, if nil then ignore
 
 ---@class Prompt_Options
----@field context_prompt fun(context: string) string Prompt to generate the prompt that should be used when using Context modes
+---@field context_prompt fun(context: string): string Prompt to generate the prompt that should be used when using Context modes
+---@field default_prompt fun(): string string Prompt to generate the prompt that should be used as default prompt
 
 ---@class Shortcut
 ---@field name string The name of the shortcut, can trigger using :NeoAIShortcut <name>