From 5f28834588dba5ddd1f8466bab598886d5b6664f Mon Sep 17 00:00:00 2001
From: Nico <105345946+nicougou@users.noreply.github.com>
Date: Wed, 19 Mar 2025 14:49:51 +0100
Subject: [PATCH] feature: add custom headers to AsyncOpenAI

---
 src/pipecat/services/openai.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/pipecat/services/openai.py b/src/pipecat/services/openai.py
index 5a3a993aa..feddc8c44 100644
--- a/src/pipecat/services/openai.py
+++ b/src/pipecat/services/openai.py
@@ -116,6 +116,7 @@ class BaseOpenAILLMService(LLMService):
         base_url=None,
         organization=None,
         project=None,
+        default_headers: Mapping[str, str] | None = None,
         params: InputParams = InputParams(),
         **kwargs,
     ):
@@ -132,10 +133,10 @@ class BaseOpenAILLMService(LLMService):
         }
         self.set_model_name(model)
         self._client = self.create_client(
-            api_key=api_key, base_url=base_url, organization=organization, project=project, **kwargs
+            api_key=api_key, base_url=base_url, organization=organization, project=project, default_headers=default_headers, **kwargs
         )
 
-    def create_client(self, api_key=None, base_url=None, organization=None, project=None, **kwargs):
+    def create_client(self, api_key=None, base_url=None, organization=None, project=None, default_headers=None, **kwargs):
         return AsyncOpenAI(
             api_key=api_key,
             base_url=base_url,
@@ -146,6 +147,7 @@ class BaseOpenAILLMService(LLMService):
                     max_keepalive_connections=100, max_connections=1000, keepalive_expiry=None
                 )
             ),
+            default_headers=default_headers
         )
 
     def can_generate_metrics(self) -> bool: