feat: get_headers for complete and chat_complete

portkey_ai/api_resources/apis/chat_complete.py

@@ -67,8 +67,9 @@ def normal_create(self, model, messages, **kwargs) -> ChatCompletions:
         response = self.openai_client.with_raw_response.chat.completions.create(
             model=model, messages=messages, **kwargs
         )
-        json_response = json.loads(response.text)
-        return ChatCompletions(**json_response)
+        data = ChatCompletions(**json.loads(response.text))
+        data._headers = response.headers
+        return data
 
     def create(
         self,
@@ -114,8 +115,9 @@ async def normal_create(self, model, messages, **kwargs) -> ChatCompletions:
         response = await self.openai_client.with_raw_response.chat.completions.create(
             model=model, messages=messages, **kwargs
         )
-        json_response = json.loads(response.text)
-        return ChatCompletions(**json_response)
+        data = ChatCompletions(**json.loads(response.text))
+        data._headers = response.headers
+        return data
 
     async def create(
         self,

portkey_ai/api_resources/apis/complete.py

@@ -39,8 +39,9 @@ def normal_create(self, model, prompt, **kwargs) -> TextCompletion:
         response = self.openai_client.with_raw_response.completions.create(
             model=model, prompt=prompt, **kwargs
         )
-        json_response = json.loads(response.text)
-        return TextCompletion(**json_response)
+        data = TextCompletion(**json.loads(response.text))
+        data._headers = response.headers
+        return data
 
     def create(
         self,
@@ -86,8 +87,9 @@ async def normal_create(self, model, prompt, **kwargs) -> TextCompletion:
         response = await self.openai_client.with_raw_response.completions.create(
             model=model, prompt=prompt, **kwargs
         )
-        json_response = json.loads(response.text)
-        return TextCompletion(**json_response)
+        data = TextCompletion(**json.loads(response.text))
+        data._headers = response.headers
+        return data
 
     async def create(
         self,