fix: Add support for input audio length header

Portkey-AI · Sep 16, 2024 · 63a29e2 · 63a29e2
1 parent 1ba5d89
commit 63a29e2
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/src/_types/generalTypes.ts b/src/_types/generalTypes.ts
@@ -14,6 +14,7 @@ export interface ApiClientInterface {
     customHost?: string | null | undefined;
     openaiProject?: string | null | undefined;
     openaiOrganization?: string | null | undefined;
+    inputAudioLength?: number | null | undefined;
     awsSecretAccessKey?: string | null | undefined;
     awsAccessKeyId?: string | null | undefined;
     awsSessionToken?: string | null | undefined;

diff --git a/src/baseClient.ts b/src/baseClient.ts
@@ -120,10 +120,10 @@ export abstract class ApiClient {
     portkeyHeaders: Record<string, string>
 
     private fetch: Fetch;
-    constructor({ apiKey, baseURL, config, virtualKey, traceID, metadata, provider, Authorization, cacheForceRefresh, debug, customHost, openaiProject, openaiOrganization, awsSecretAccessKey, awsAccessKeyId, awsSessionToken, awsRegion, vertexProjectId, vertexRegion, workersAiAccountId, azureResourceName, azureDeploymentId, azureApiVersion, huggingfaceBaseUrl, forwardHeaders, cacheNamespace, requestTimeout, strictOpenAiCompliance }: ApiClientInterface) {
+    constructor({ apiKey, baseURL, config, virtualKey, traceID, metadata, provider, Authorization, cacheForceRefresh, debug, customHost, openaiProject, openaiOrganization, inputAudioLength, awsSecretAccessKey, awsAccessKeyId, awsSessionToken, awsRegion, vertexProjectId, vertexRegion, workersAiAccountId, azureResourceName, azureDeploymentId, azureApiVersion, huggingfaceBaseUrl, forwardHeaders, cacheNamespace, requestTimeout, strictOpenAiCompliance }: ApiClientInterface) {
         this.apiKey = apiKey ?? "";
         this.baseURL = baseURL ?? "";
-        this.customHeaders = createHeaders({ apiKey, config, virtualKey, traceID, metadata, provider, Authorization, cacheForceRefresh, debug, customHost, cacheNamespace, openaiProject, openaiOrganization, awsSecretAccessKey, awsAccessKeyId, awsSessionToken, awsRegion, vertexProjectId, vertexRegion, workersAiAccountId, azureResourceName, azureDeploymentId, azureApiVersion, huggingfaceBaseUrl, forwardHeaders, requestTimeout, strictOpenAiCompliance })
+        this.customHeaders = createHeaders({ apiKey, config, virtualKey, traceID, metadata, provider, Authorization, cacheForceRefresh, debug, customHost, cacheNamespace, openaiProject, openaiOrganization, inputAudioLength, awsSecretAccessKey, awsAccessKeyId, awsSessionToken, awsRegion, vertexProjectId, vertexRegion, workersAiAccountId, azureResourceName, azureDeploymentId, azureApiVersion, huggingfaceBaseUrl, forwardHeaders, requestTimeout, strictOpenAiCompliance })
         this.portkeyHeaders = this.defaultHeaders()
         this.fetch = fetch;
         this.responseHeaders = {}

diff --git a/src/client.ts b/src/client.ts
@@ -20,6 +20,7 @@ export class Portkey extends ApiClient {
 	customHost?: string | null | undefined;
 	openaiProject?: string | null | undefined;
 	openaiOrganization?: string | null | undefined;
+	inputAudioLength?: number | null | undefined;
 	awsSecretAccessKey?: string | null | undefined;
 	awsAccessKeyId?: string | null | undefined;
 	awsSessionToken?: string | null | undefined;
@@ -49,6 +50,7 @@ export class Portkey extends ApiClient {
 		customHost,
 		openaiProject, 
 		openaiOrganization,
+		inputAudioLength,
 		awsSecretAccessKey,
 		awsAccessKeyId,
 		awsSessionToken,
@@ -81,6 +83,7 @@ export class Portkey extends ApiClient {
 			cacheNamespace,
 			openaiProject,
 			openaiOrganization,
+			inputAudioLength,
 			awsSecretAccessKey,
 			awsAccessKeyId,
 			awsSessionToken,
@@ -113,6 +116,7 @@ export class Portkey extends ApiClient {
 		this.cacheNamespace = cacheNamespace;;
 		this.openaiProject = openaiProject;
 		this.openaiOrganization = openaiOrganization;
+		this.inputAudioLength = inputAudioLength;
 		this.awsSecretAccessKey = awsSecretAccessKey;
 		this.awsAccessKeyId = awsAccessKeyId;
 		this.awsSessionToken = awsSessionToken;