allow for arbitrary headers to be added to requests

cyberchitta · Oct 13, 2024 · 7f20ed7 · 7f20ed7
1 parent e45b158
commit 7f20ed7
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/lib/openai_ex.ex b/lib/openai_ex.ex
@@ -116,6 +116,15 @@ defmodule OpenaiEx do
     openai |> Map.put(:base_url, base_url)
   end
 
+  def with_additional_headers(openai = %OpenaiEx{}, additional_headers) do
+    openai |> Map.get_and_update(
+      :_http_headers,
+      fn headers ->
+        {headers, headers ++ additional_headers}
+      end
+    ) |> elem(1)
+  end
+
   def with_receive_timeout(openai = %OpenaiEx{}, timeout)
       when is_integer(timeout) and timeout > 0 do
     openai |> Map.put(:receive_timeout, timeout)