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

100% Statements 40/40
100% Branches 0/0
100% Functions 0/0
100% Lines 40/40

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 48 49 50 51 52 53  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 1x 1x 1x 1x 1x 1x 1x  
import type { AIProviderConfig } from '@shared/types/ai-providers/config';
import { FunctionName } from '@shared/types/api/request';
import { AIProvider } from '@shared/types/constants';
import { azureAIInferenceAPI } from './api';
import {
  azureAIInferenceChatCompleteConfig,
  azureAIInferenceChatCompleteResponseTransform,
} from './chat-complete';
import {
  azureAIInferenceCompleteConfig,
  azureAIInferenceCompleteResponseTransform,
} from './complete';
import {
  azureAIInferenceEmbedConfig,
  azureAIInferenceEmbedResponseTransform,
} from './embed';
 
export const azureAIInferenceConfig: AIProviderConfig = {
  api: azureAIInferenceAPI,
  [FunctionName.COMPLETE]: azureAIInferenceCompleteConfig,
  [FunctionName.EMBED]: azureAIInferenceEmbedConfig,
  [FunctionName.CHAT_COMPLETE]: azureAIInferenceChatCompleteConfig,
  responseTransforms: {
    [FunctionName.COMPLETE]: azureAIInferenceCompleteResponseTransform(
      AIProvider.AZURE_AI_FOUNDRY,
    ),
    [FunctionName.CHAT_COMPLETE]: azureAIInferenceChatCompleteResponseTransform(
      AIProvider.AZURE_AI_FOUNDRY,
    ),
    [FunctionName.EMBED]: azureAIInferenceEmbedResponseTransform(
      AIProvider.AZURE_AI_FOUNDRY,
    ),
  },
};
 
export const githubModelAPiConfig: AIProviderConfig = {
  api: azureAIInferenceAPI,
  [FunctionName.COMPLETE]: azureAIInferenceCompleteConfig,
  [FunctionName.EMBED]: azureAIInferenceEmbedConfig,
  [FunctionName.CHAT_COMPLETE]: azureAIInferenceChatCompleteConfig,
  responseTransforms: {
    [FunctionName.COMPLETE]: azureAIInferenceCompleteResponseTransform(
      AIProvider.GITHUB,
    ),
    [FunctionName.CHAT_COMPLETE]: azureAIInferenceChatCompleteResponseTransform(
      AIProvider.GITHUB,
    ),
    [FunctionName.EMBED]: azureAIInferenceEmbedResponseTransform(
      AIProvider.GITHUB,
    ),
  },
};