All files / api/src/ai-providers/azure-ai-foundry embed.ts

100% Statements 32/32
100% Branches 4/4
100% Functions 2/2
100% Lines 32/32

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          1x   1x   1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x   1x 46x 46x 46x 2x 2x 2x 2x 1x 1x   1x 1x 1x 1x 2x   46x 46x  
import type {
  AIProviderFunctionConfig,
  ResponseTransformFunction,
} from '@shared/types/ai-providers/config';
 
import { CreateEmbeddingsResponseBody } from '@shared/types/api/routes/embeddings-api';
import type { AIProvider } from '@shared/types/constants';
import { openAIErrorResponseTransform } from '../openai/utils';
 
export const azureAIInferenceEmbedConfig: AIProviderFunctionConfig = {
  model: {
    param: 'model',
    required: false,
  },
  input: {
    param: 'input',
    required: true,
  },
  user: {
    param: 'user',
  },
};
 
export const azureAIInferenceEmbedResponseTransform = (
  provider: AIProvider,
): ResponseTransformFunction => {
  const transformer: ResponseTransformFunction = (
    aiProviderResponseBody,
    aiProviderResponseStatus,
  ) => {
    if (aiProviderResponseStatus !== 200 && 'error' in aiProviderResponseBody) {
      return openAIErrorResponseTransform(aiProviderResponseBody, provider);
    }
 
    const createEmbeddingsResponseBody = CreateEmbeddingsResponseBody.parse(
      aiProviderResponseBody,
    );
    return createEmbeddingsResponseBody;
  };
 
  return transformer;
};