fix: add header for input audio cost calculation

Portkey-AI · Sep 16, 2024 · 4a5016a · 4a5016a
1 parent 7023486
commit 4a5016a
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/portkey_ai/api_resources/base_client.py b/portkey_ai/api_resources/base_client.py
@@ -62,6 +62,7 @@ def __init__(
         forward_headers: Optional[List[str]] = None,
         openai_project: Optional[str] = None,
         openai_organization: Optional[str] = None,
+        input_audio_length: Optional[int] = None,
         aws_secret_access_key: Optional[str] = None,
         aws_access_key_id: Optional[str] = None,
         aws_session_token: Optional[str] = None,
@@ -91,6 +92,7 @@ def __init__(
         self.forward_headers = forward_headers
         self.openai_project = openai_project
         self.openai_organization = openai_organization
+        self.input_audio_length = input_audio_length
         self.aws_secret_access_key = aws_secret_access_key
         self.aws_access_key_id = aws_access_key_id
         self.aws_session_token = aws_session_token
@@ -120,6 +122,7 @@ def __init__(
             forward_headers=forward_headers,
             openai_project=openai_project,
             openai_organization=openai_organization,
+            input_audio_length=input_audio_length,
             aws_secret_access_key=aws_secret_access_key,
             aws_access_key_id=aws_access_key_id,
             aws_session_token=aws_session_token,
@@ -552,6 +555,7 @@ def __init__(
         forward_headers: Optional[List[str]] = None,
         openai_project: Optional[str] = None,
         openai_organization: Optional[str] = None,
+        input_audio_length: Optional[int] = None,
         aws_secret_access_key: Optional[str] = None,
         aws_access_key_id: Optional[str] = None,
         aws_session_token: Optional[str] = None,
@@ -581,6 +585,7 @@ def __init__(
         self.forward_headers = forward_headers
         self.openai_project = openai_project
         self.openai_organization = openai_organization
+        self.input_audio_length = input_audio_length
         self.aws_secret_access_key = aws_secret_access_key
         self.aws_access_key_id = aws_access_key_id
         self.aws_session_token = aws_session_token
@@ -610,6 +615,7 @@ def __init__(
             forward_headers=forward_headers,
             openai_project=openai_project,
             openai_organization=openai_organization,
+            input_audio_length=input_audio_length,
             aws_secret_access_key=aws_secret_access_key,
             aws_access_key_id=aws_access_key_id,
             aws_session_token=aws_session_token,

diff --git a/portkey_ai/api_resources/client.py b/portkey_ai/api_resources/client.py
@@ -57,6 +57,7 @@ def __init__(
         forward_headers: Optional[List[str]] = None,
         openai_project: Optional[str] = None,
         openai_organization: Optional[str] = None,
+        input_audio_length: Optional[int] = None,
         aws_secret_access_key: Optional[str] = None,
         aws_access_key_id: Optional[str] = None,
         aws_session_token: Optional[str] = None,
@@ -87,6 +88,7 @@ def __init__(
             forward_headers=forward_headers,
             openai_project=openai_project,
             openai_organization=openai_organization,
+            input_audio_length=input_audio_length,
             aws_secret_access_key=aws_secret_access_key,
             aws_access_key_id=aws_access_key_id,
             aws_session_token=aws_session_token,
@@ -146,6 +148,7 @@ def copy(
         forward_headers: Optional[List[str]] = None,
         openai_project: Optional[str] = None,
         openai_organization: Optional[str] = None,
+        input_audio_length: Optional[int] = None,
         aws_secret_access_key: Optional[str] = None,
         aws_access_key_id: Optional[str] = None,
         aws_session_token: Optional[str] = None,
@@ -176,6 +179,7 @@ def copy(
             forward_headers=forward_headers or self.forward_headers,
             openai_project=openai_project or self.openai_project,
             openai_organization=openai_organization or self.openai_organization,
+            input_audio_length=input_audio_length or self.input_audio_length,
             aws_secret_access_key=aws_secret_access_key or self.aws_secret_access_key,
             aws_access_key_id=aws_access_key_id or self.aws_access_key_id,
             aws_session_token=aws_session_token or self.aws_session_token,
@@ -247,6 +251,7 @@ def __init__(
         forward_headers: Optional[List[str]] = None,
         openai_project: Optional[str] = None,
         openai_organization: Optional[str] = None,
+        input_audio_length: Optional[int] = None,
         aws_secret_access_key: Optional[str] = None,
         aws_access_key_id: Optional[str] = None,
         aws_session_token: Optional[str] = None,
@@ -277,6 +282,7 @@ def __init__(
             forward_headers=forward_headers,
             openai_project=openai_project,
             openai_organization=openai_organization,
+            input_audio_length=input_audio_length,
             aws_secret_access_key=aws_secret_access_key,
             aws_access_key_id=aws_access_key_id,
             aws_session_token=aws_session_token,
@@ -336,6 +342,7 @@ def copy(
         forward_headers: Optional[List[str]] = None,
         openai_project: Optional[str] = None,
         openai_organization: Optional[str] = None,
+        input_audio_length: Optional[int] = None,
         aws_secret_access_key: Optional[str] = None,
         aws_access_key_id: Optional[str] = None,
         aws_session_token: Optional[str] = None,
@@ -366,6 +373,7 @@ def copy(
             forward_headers=forward_headers or self.forward_headers,
             openai_project=openai_project or self.openai_project,
             openai_organization=openai_organization or self.openai_organization,
+            input_audio_length=input_audio_length or self.input_audio_length,
             aws_secret_access_key=aws_secret_access_key or self.aws_secret_access_key,
             aws_access_key_id=aws_access_key_id or self.aws_access_key_id,
             aws_session_token=aws_session_token or self.aws_session_token,

diff --git a/portkey_ai/api_resources/utils.py b/portkey_ai/api_resources/utils.py
@@ -255,6 +255,7 @@ class Constructs(BaseModel):
     api_version: Optional[str] = None
     openai_project: Optional[str] = None
     openai_organization: Optional[str] = None
+    input_audio_length: Optional[int] = None
     aws_secret_access_key: Optional[str] = None
     aws_access_key_id: Optional[str] = None
     aws_session_token: Optional[str] = None