From 3b86549485d10c67817b4e7e1976460c791525eb Mon Sep 17 00:00:00 2001
From: Florian Engelhardt <flo@dotbox.org>
Date: Mon, 17 Mar 2025 10:51:08 +0100
Subject: [PATCH] Add extra headers for OpenAI (#1615)

---
 lua/avante/config.lua           | 1 +
 lua/avante/providers/openai.lua | 6 ++++++
 lua/avante/types.lua            | 1 +
 3 files changed, 8 insertions(+)

diff --git a/lua/avante/config.lua b/lua/avante/config.lua
index 4570b98..aca3c0c 100644
--- a/lua/avante/config.lua
+++ b/lua/avante/config.lua
@@ -557,6 +557,7 @@ end
 
 M.BASE_PROVIDER_KEYS = {
   "endpoint",
+  "extra_headers",
   "model",
   "deployment",
   "api_version",
diff --git a/lua/avante/providers/openai.lua b/lua/avante/providers/openai.lua
index 39624a0..acd9941 100644
--- a/lua/avante/providers/openai.lua
+++ b/lua/avante/providers/openai.lua
@@ -278,6 +278,12 @@ function M:parse_curl_args(prompt_opts)
     ["Content-Type"] = "application/json",
   }
 
+  if provider_conf.extra_headers then
+    for key, value in pairs(provider_conf.extra_headers) do
+      headers[key] = value
+    end
+  end
+
   if P.env.require_api_key(provider_conf) then
     local api_key = self.parse_api_key()
     if api_key == nil then
diff --git a/lua/avante/types.lua b/lua/avante/types.lua
index 5fc7054..036e845 100644
--- a/lua/avante/types.lua
+++ b/lua/avante/types.lua
@@ -200,6 +200,7 @@ vim.g.avante_login = vim.g.avante_login
 ---
 ---@class AvanteDefaultBaseProvider: table<string, any>
 ---@field endpoint? string
+---@field extra_headers? table<string, any>
 ---@field model? string
 ---@field local? boolean
 ---@field proxy? string