[Azure] Use model in request body for audio operations

Author: deyaaeldeen

src/azure/audio.ts

@@ -0,0 +1,43 @@
+import { Audio, Transcriptions, Translations } from '../resources/audio';
+import type { OpenAI } from '../index';
+
+class AzureTranslation extends Translations {
+  constructor(
+    client: OpenAI,
+    private deployments: string[],
+  ) {
+    super(client);
+  }
+  // @ts-ignore we don't want to redeclare all the method signatures
+  override create(...args: Parameters<Translations['create']>): ReturnType<Translations['create']> {
+    this.deployments.push(args[0].model);
+    return super.create(args[0], args[1]);
+  }
+}
+
+class AzureTranscription extends Transcriptions {
+  constructor(
+    client: OpenAI,
+    private deployments: string[],
+  ) {
+    super(client);
+  }
+  // @ts-ignore we don't want to redeclare all the method signatures
+  override create(...args: Parameters<Transcriptions['create']>): ReturnType<Transcriptions['create']> {
+    this.deployments.push(args[0].model);
+    return super.create(args[0], args[1]);
+  }
+}
+
+export class AzureAudio extends Audio {
+  constructor(
+    client: OpenAI,
+    private _deployments: string[],
+  ) {
+    super(client);
+  }
+  // @ts-ignore we don't want to redeclare all the method signatures
+  override translations: Audio.Translations = new AzureTranslation(this._client, this._deployments);
+  // @ts-ignore we don't want to redeclare all the method signatures
+  override transcriptions: Audio.Transcriptions = new AzureTranscription(this._client, this._deployments);
+}

src/index.ts

@@ -106,6 +106,7 @@ import {
   UploadCreateParams,
   Uploads as UploadsAPIUploads,
 } from './resources/uploads/uploads';
+import { AzureAudio } from './azure/audio';
 
 export interface ClientOptions {
   /**
@@ -492,6 +493,7 @@ export interface AzureClientOptions extends ClientOptions {
 /** API Client for interfacing with the Azure OpenAI API. */
 export class AzureOpenAI extends OpenAI {
   private _azureADTokenProvider: (() => Promise<string>) | undefined;
+  private _audioDeployments: string[] = [];
   deploymentName: string | undefined;
   apiVersion: string = '';
   /**
@@ -578,6 +580,8 @@ export class AzureOpenAI extends OpenAI {
     this.deploymentName = deployment;
   }
 
+  override audio: API.Audio = new AzureAudio(this, this._audioDeployments);
+
   override buildRequest(
     options: Core.FinalRequestOptions<unknown>,
     props: { retryCount?: number } = {},
@@ -590,7 +594,7 @@ export class AzureOpenAI extends OpenAI {
       if (!Core.isObj(options.body)) {
         throw new Error('Expected request body to be an object');
       }
-      const model = this.deploymentName || options.body['model'];
+      const model = this.deploymentName || options.body['model'] || this._audioDeployments.shift();
       if (model !== undefined && !this.baseURL.includes('/deployments')) {
         options.path = `/deployments/${model}${options.path}`;
       }

tests/lib/azure.test.ts

@@ -495,21 +495,23 @@ describe('azure request building', () => {
         );
       });
 
-      test('Audio translations is not handled', async () => {
+      test('handles audio translations', async () => {
         const { url } = (await client.audio.translations.create({
           model: deployment,
           file: { url: 'https://example.com', blob: () => 0 as any },
         })) as any;
-        expect(url).toStrictEqual(`https://example.com/openai/audio/translations?api-version=${apiVersion}`);
+        expect(url).toStrictEqual(
+          `https://example.com/openai/deployments/deployment/audio/translations?api-version=${apiVersion}`,
+        );
       });
 
-      test('Audio transcriptions is not handled', async () => {
+      test('handles audio transcriptions', async () => {
         const { url } = (await client.audio.transcriptions.create({
           model: deployment,
           file: { url: 'https://example.com', blob: () => 0 as any },
         })) as any;
         expect(url).toStrictEqual(
-          `https://example.com/openai/audio/transcriptions?api-version=${apiVersion}`,
+          `https://example.com/openai/deployments/deployment/audio/transcriptions?api-version=${apiVersion}`,
         );
       });