All files / api/src/ai-providers/azure-openai create-speech.ts

86.84% Statements 33/38
33.33% Branches 1/3
100% Functions 1/1
86.84% Lines 33/38

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47          1x 1x   1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x   1x 1x 1x             1x 1x  
import type {
  AIProviderFunctionConfig,
  ResponseTransformFunction,
} from '@shared/types/ai-providers/config';
import type { CreateSpeechResponseBody } from '@shared/types/api/routes/audio-api';
import { AIProvider } from '@shared/types/constants';
import { openAIErrorResponseTransform } from '../openai/utils';
 
export const azureOpenAICreateSpeechConfig: AIProviderFunctionConfig = {
  model: {
    param: 'model',
    required: true,
    default: 'tts-1',
  },
  input: {
    param: 'input',
    required: true,
  },
  voice: {
    param: 'voice',
    required: true,
    default: 'alloy',
  },
  response_format: {
    param: 'response_format',
    required: false,
    default: 'mp3',
  },
  speed: {
    param: 'speed',
    required: false,
    default: 1,
  },
};
 
export const azureOpenAICreateSpeechResponseTransform: ResponseTransformFunction =
  (aiProviderResponseBody, aiProviderResponseStatus) => {
    if (aiProviderResponseStatus !== 200 && 'error' in aiProviderResponseBody) {
      return openAIErrorResponseTransform(
        aiProviderResponseBody,
        AIProvider.AZURE_OPENAI,
      );
    }
 
    return aiProviderResponseBody as unknown as CreateSpeechResponseBody;
  };