phyer/avante.nvim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: support azure (#2)

Browse Source
This commit is contained in:
Hanchin Hsieh 2024-08-15 11:17:03 +08:00 committed by GitHub
parent ba10a2ac3e
commit ea53082844
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lua/avante/init.lua
View File

@ -266,6 +266,9 @@ local function call_openai_api_stream(prompt, original_content, on_chunk, on_com
local user_prompt = user_prompt_tpl:gsub("${{question}}", prompt):gsub("${{code}}", original_content) local user_prompt = user_prompt_tpl:gsub("${{question}}", prompt):gsub("${{code}}", original_content)
local url = utils.trim_suffix(M.config.openai.endpoint, "/") .. "/v1/chat/completions" local url = utils.trim_suffix(M.config.openai.endpoint, "/") .. "/v1/chat/completions"
if M.config.provider == "azure" then
url = M.config.openai.endpoint
end
print("Sending request to OpenAI API...") print("Sending request to OpenAI API...")
@ -295,6 +298,7 @@ local function call_openai_api_stream(prompt, original_content, on_chunk, on_com
headers = { headers = {
["Content-Type"] = "application/json", ["Content-Type"] = "application/json",
["Authorization"] = "Bearer " .. api_key, ["Authorization"] = "Bearer " .. api_key,
["api_key"] = api_key,
}, },
body = fn.json_encode({ body = fn.json_encode({
model = M.config.openai.model, model = M.config.openai.model,
@ -310,7 +314,7 @@ local function call_openai_api_stream(prompt, original_content, on_chunk, on_com
end end
local function call_ai_api_stream(prompt, original_content, on_chunk, on_complete) local function call_ai_api_stream(prompt, original_content, on_chunk, on_complete)
if M.config.provider == "openai" then if M.config.provider == "openai" or M.config.provider == "azure" then
call_openai_api_stream(prompt, original_content, on_chunk, on_complete) call_openai_api_stream(prompt, original_content, on_chunk, on_complete)
elseif M.config.provider == "claude" then elseif M.config.provider == "claude" then
call_claude_api_stream(prompt, original_content, on_chunk, on_complete) call_claude_api_stream(prompt, original_content, on_chunk, on_complete)
@ -643,7 +647,7 @@ function M.render_sidebar()
end end
M.config = { M.config = {
provider = "claude", -- "claude" or "openai" provider = "claude", -- "claude" or "openai" or "azure"
openai = { openai = {
endpoint = "https://api.openai.com", endpoint = "https://api.openai.com",
model = "gpt-4o", model = "gpt-4o",