feat(assistant): add base_url option

S1M0N38 · Jan 29, 2024 · 7ab924b · 7ab924b
1 parent 749b486
commit 7ab924b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/lua/dante/assistant.lua b/lua/dante/assistant.lua
@@ -28,7 +28,7 @@ local function command(req)
 		"--no-buffer",
 		'--header "Authorization: Bearer $OPENAI_API_KEY"',
 		'--header "content-type: application/json"',
-		"--url https://api.openai.com/v1/chat/completions",
+		"--url " .. config.options.base_url .. "/chat/completions",
 		"--data " .. vim.fn.shellescape(req),
 	}
 	return "curl " .. table.concat(args, " ")

diff --git a/lua/dante/config.lua b/lua/dante/config.lua
@@ -10,6 +10,7 @@ local defaults = {
 		},
 	},
 	openai_api_key = "OPENAI_API_KEY",
+	base_url = "https://api.openai.com/v1",
 	diffopt = { "internal", "filler", "closeoff", "algorithm:patience", "followwrap", "linematch:120" },
 }