Merge pull request #188 from Portkey-AI/hotfix/extraBody

hotfix: kwargs in extra_body

portkey_ai/api_resources/apis/audio.py

@@ -43,7 +43,7 @@ def create(
             response_format=response_format,
             temperature=temperature,
             timestamp_granularities=timestamp_granularities,
-            **kwargs
+            extra_body=kwargs,
         )
         data = Transcription(**json.loads(response.text))
         data._headers = response.headers
@@ -72,7 +72,7 @@ def create(
             prompt=prompt,
             response_format=response_format,
             temperature=temperature,
-            **kwargs
+            extra_body=kwargs,
         )
         data = Translation(**json.loads(response.text))
         data._headers = response.headers
@@ -102,7 +102,7 @@ def create(
             voice=voice,
             response_format=response_format,
             speed=speed,
-            **kwargs
+            extra_body=kwargs,
         )
 
         return response
@@ -144,7 +144,7 @@ async def create(
                 response_format=response_format,
                 temperature=temperature,
                 timestamp_granularities=timestamp_granularities,
-                **kwargs
+                extra_body=kwargs,
             )
         )
         data = Transcription(**json.loads(response.text))
@@ -174,7 +174,7 @@ async def create(
             prompt=prompt,
             response_format=response_format,
             temperature=temperature,
-            **kwargs
+            extra_body=kwargs,
         )
         data = Translation(**json.loads(response.text))
         data._headers = response.headers
@@ -204,7 +204,7 @@ async def create(
             voice=voice,
             response_format=response_format,
             speed=speed,
-            **kwargs
+            extra_body=kwargs,
         )
 
         data = response

portkey_ai/api_resources/apis/batches.py

@@ -28,7 +28,7 @@ def create(
             endpoint=endpoint,
             input_file_id=input_file_id,
             metadata=metadata,
-            **kwargs
+            extra_body=kwargs,
         )
         data = Batch(**json.loads(response.text))
         data._headers = response.headers
@@ -37,7 +37,7 @@ def create(
 
     def retrieve(self, batch_id, **kwargs) -> Batch:
         response = self.openai_client.with_raw_response.batches.retrieve(
-            batch_id=batch_id, **kwargs
+            batch_id=batch_id, extra_body=kwargs
         )
         data = Batch(**json.loads(response.text))
         data._headers = response.headers
@@ -52,7 +52,7 @@ def list(
         **kwargs
     ) -> BatchList:
         response = self.openai_client.with_raw_response.batches.list(
-            after=after, limit=limit, **kwargs
+            after=after, limit=limit, extra_body=kwargs
         )
         data = BatchList(**json.loads(response.text))
         data._headers = response.headers
@@ -61,7 +61,7 @@ def list(
 
     def cancel(self, batch_id: str, **kwargs) -> Batch:
         response = self.openai_client.with_raw_response.batches.cancel(
-            batch_id=batch_id, **kwargs
+            batch_id=batch_id, extra_body=kwargs
         )
         data = Batch(**json.loads(response.text))
         data._headers = response.headers
@@ -89,7 +89,7 @@ async def create(
             endpoint=endpoint,
             input_file_id=input_file_id,
             metadata=metadata,
-            **kwargs
+            extra_body=kwargs,
         )
         data = Batch(**json.loads(response.text))
         data._headers = response.headers
@@ -98,7 +98,7 @@ async def create(
 
     async def retrieve(self, batch_id, **kwargs) -> Batch:
         response = await self.openai_client.with_raw_response.batches.retrieve(
-            batch_id=batch_id, **kwargs
+            batch_id=batch_id, extra_body=kwargs
         )
         data = Batch(**json.loads(response.text))
         data._headers = response.headers
@@ -113,7 +113,7 @@ async def list(
         **kwargs
     ) -> BatchList:
         response = await self.openai_client.with_raw_response.batches.list(
-            after=after, limit=limit, **kwargs
+            after=after, limit=limit, extra_body=kwargs
         )
         data = BatchList(**json.loads(response.text))
         data._headers = response.headers
@@ -122,7 +122,7 @@ async def list(
 
     async def cancel(self, batch_id: str, **kwargs) -> Batch:
         response = await self.openai_client.with_raw_response.batches.cancel(
-            batch_id=batch_id, **kwargs
+            batch_id=batch_id, extra_body=kwargs
         )
         data = Batch(**json.loads(response.text))
         data._headers = response.headers

portkey_ai/api_resources/apis/chat_complete.py

@@ -53,7 +53,7 @@ def stream_create(  # type: ignore[return]
             temperature=temperature,
             max_tokens=max_tokens,
             top_p=top_p,
-            **kwargs,
+            extra_body=kwargs,
         ) as response:
             for line in response.iter_lines():
                 json_string = line.replace("data: ", "")
@@ -79,7 +79,7 @@ def normal_create(
             temperature=temperature,
             max_tokens=max_tokens,
             top_p=top_p,
-            **kwargs,
+            extra_body=kwargs,
         )
         data = ChatCompletions(**json.loads(response.text))
         data._headers = response.headers
@@ -133,7 +133,7 @@ async def stream_create(
             temperature=temperature,
             max_tokens=max_tokens,
             top_p=top_p,
-            **kwargs,
+            extra_body=kwargs,
         ) as response:
             async for line in response.iter_lines():
                 json_string = line.replace("data: ", "")
@@ -159,7 +159,7 @@ async def normal_create(
             temperature=temperature,
             max_tokens=max_tokens,
             top_p=top_p,
-            **kwargs,
+            extra_body=kwargs,
         )
         data = ChatCompletions(**json.loads(response.text))
         data._headers = response.headers

portkey_ai/api_resources/apis/complete.py

@@ -27,7 +27,7 @@ def stream_create(  # type: ignore[return]
             temperature=temperature,
             max_tokens=max_tokens,
             top_p=top_p,
-            **kwargs,
+            extra_body=kwargs,
         ) as response:
             for line in response.iter_lines():
                 json_string = line.replace("data: ", "")
@@ -53,7 +53,7 @@ def normal_create(
             temperature=temperature,
             max_tokens=max_tokens,
             top_p=top_p,
-            **kwargs,
+            extra_body=kwargs,
         )
         data = TextCompletion(**json.loads(response.text))
         data._headers = response.headers
@@ -107,7 +107,7 @@ async def stream_create(
             temperature=temperature,
             max_tokens=max_tokens,
             top_p=top_p,
-            **kwargs,
+            extra_body=kwargs,
         ) as response:
             async for line in response.iter_lines():
                 json_string = line.replace("data: ", "")
@@ -133,7 +133,7 @@ async def normal_create(
             temperature=temperature,
             max_tokens=max_tokens,
             top_p=top_p,
-            **kwargs,
+            extra_body=kwargs,
         )
         data = TextCompletion(**json.loads(response.text))
         data._headers = response.headers

portkey_ai/api_resources/apis/embeddings.py

@@ -29,7 +29,7 @@ def create(
             dimensions=dimensions,
             encoding_format=encoding_format,
             user=user,
-            **kwargs
+            extra_body=kwargs,
         )
 
         data = CreateEmbeddingResponse(**json.loads(response.text))
@@ -60,7 +60,7 @@ async def create(
             dimensions=dimensions,
             encoding_format=encoding_format,
             user=user,
-            **kwargs
+            extra_body=kwargs,
         )
         data = CreateEmbeddingResponse(**json.loads(response.text))
         data._headers = response.headers

portkey_ai/api_resources/apis/fine_tuning.py

@@ -48,7 +48,7 @@ def create(
             seed=seed,
             suffix=suffix,
             validation_file=validation_file,
-            **kwargs,
+            extra_body=kwargs,
         )
         data = FineTuningJob(**json.loads(response.text))
         data._headers = response.headers
@@ -57,7 +57,7 @@ def create(
 
     def retrieve(self, fine_tuning_job_id: str, **kwargs) -> FineTuningJob:
         response = self.openai_client.with_raw_response.fine_tuning.jobs.retrieve(
-            fine_tuning_job_id=fine_tuning_job_id, **kwargs
+            fine_tuning_job_id=fine_tuning_job_id, extra_body=kwargs
         )
         data = FineTuningJob(**json.loads(response.text))
         data._headers = response.headers
@@ -72,7 +72,7 @@ def list(
         **kwargs,
     ) -> FineTuningJobList:
         response = self.openai_client.with_raw_response.fine_tuning.jobs.list(
-            after=after, limit=limit, **kwargs
+            after=after, limit=limit, extra_body=kwargs
         )
         data = FineTuningJobList(**json.loads(response.text))
         data._headers = response.headers
@@ -81,7 +81,7 @@ def list(
 
     def cancel(self, fine_tuning_job_id: str, **kwargs) -> FineTuningJob:
         response = self.openai_client.with_raw_response.fine_tuning.jobs.cancel(
-            fine_tuning_job_id=fine_tuning_job_id, **kwargs
+            fine_tuning_job_id=fine_tuning_job_id, extra_body=kwargs
         )
         data = FineTuningJob(**json.loads(response.text))
         data._headers = response.headers
@@ -97,7 +97,10 @@ def list_events(
         **kwargs,
     ) -> FineTuningJobEventList:
         response = self.openai_client.with_raw_response.fine_tuning.jobs.list_events(
-            fine_tuning_job_id=fine_tuning_job_id, after=after, limit=limit, **kwargs
+            fine_tuning_job_id=fine_tuning_job_id,
+            after=after,
+            limit=limit,
+            extra_body=kwargs,
         )
         data = FineTuningJobEventList(**json.loads(response.text))
         data._headers = response.headers
@@ -123,7 +126,7 @@ def list(
                 fine_tuning_job_id=fine_tuning_job_id,
                 after=after,
                 limit=limit,
-                **kwargs,
+                extra_body=kwargs,
             )
         )
 
@@ -168,7 +171,7 @@ async def create(
             seed=seed,
             suffix=suffix,
             validation_file=validation_file,
-            **kwargs,
+            extra_body=kwargs,
         )
         data = FineTuningJob(**json.loads(response.text))
         data._headers = response.headers
@@ -177,7 +180,7 @@ async def create(
 
     async def retrieve(self, fine_tuning_job_id: str, **kwargs) -> FineTuningJob:
         response = await self.openai_client.with_raw_response.fine_tuning.jobs.retrieve(
-            fine_tuning_job_id=fine_tuning_job_id, **kwargs
+            fine_tuning_job_id=fine_tuning_job_id, extra_body=kwargs
         )
         data = FineTuningJob(**json.loads(response.text))
         data._headers = response.headers
@@ -192,7 +195,7 @@ async def list(
         **kwargs,
     ) -> FineTuningJobList:
         response = await self.openai_client.with_raw_response.fine_tuning.jobs.list(
-            after=after, limit=limit, **kwargs
+            after=after, limit=limit, extra_body=kwargs
         )
         data = FineTuningJobList(**json.loads(response.text))
         data._headers = response.headers
@@ -201,7 +204,7 @@ async def list(
 
     async def cancel(self, fine_tuning_job_id: str, **kwargs) -> FineTuningJob:
         response = await self.openai_client.with_raw_response.fine_tuning.jobs.cancel(
-            fine_tuning_job_id, **kwargs
+            fine_tuning_job_id, extra_body=kwargs
         )
         data = FineTuningJob(**json.loads(response.text))
         data._headers = response.headers
@@ -221,7 +224,7 @@ async def list_events(
                 fine_tuning_job_id=fine_tuning_job_id,
                 after=after,
                 limit=limit,
-                **kwargs,
+                extra_body=kwargs,
             )
         )
         data = FineTuningJobEventList(**json.loads(response.text))
@@ -247,7 +250,7 @@ async def list(
             fine_tuning_job_id=fine_tuning_job_id,
             after=after,
             limit=limit,
-            **kwargs,
+            extra_body=kwargs,
         )
 
         data = FineTuningJobCheckpointList(**json.loads(response.text))

portkey_ai/api_resources/apis/images.py

@@ -33,9 +33,9 @@ def generate(
             quality=quality,
             response_format=response_format,
             size=size,
-            user=user,
             style=style,
-            **kwargs
+            user=user,
+            extra_body=kwargs,
         )
         data = ImagesResponse(**json.loads(response.text))
         data._headers = response.headers
@@ -65,7 +65,7 @@ def edit(
             response_format=response_format,
             size=size,
             user=user,
-            **kwargs
+            extra_body=kwargs,
         )
         data = ImagesResponse(**json.loads(response.text))
         data._headers = response.headers
@@ -89,7 +89,7 @@ def create_variation(
             response_format=response_format,
             size=size,
             user=user,
-            **kwargs
+            extra_body=kwargs,
         )
         data = ImagesResponse(**json.loads(response.text))
         data._headers = response.headers
@@ -123,9 +123,9 @@ async def generate(
             quality=quality,
             response_format=response_format,
             size=size,
-            user=user,
             style=style,
-            **kwargs
+            user=user,
+            extra_body=kwargs,
         )
         data = ImagesResponse(**json.loads(response.text))
         data._headers = response.headers
@@ -155,7 +155,7 @@ async def edit(
             response_format=response_format,
             size=size,
             user=user,
-            **kwargs
+            extra_body=kwargs,
         )
         data = ImagesResponse(**json.loads(response.text))
         data._headers = response.headers
@@ -179,7 +179,7 @@ async def create_variation(
             response_format=response_format,
             size=size,
             user=user,
-            **kwargs
+            extra_body=kwargs,
         )
         data = ImagesResponse(**json.loads(response.text))
         data._headers = response.headers