All files / api/src/ai-providers/anyscale index.ts

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

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  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 { anyscaleAPIConfig } from './api';
import {
  anyscaleChatCompleteConfig,
  anyscaleChatCompleteResponseTransform,
  anyscaleChatCompleteStreamChunkTransform,
} from './chat-complete';
import {
  anyscaleCompleteConfig,
  anyscaleCompleteResponseTransform,
  anyscaleCompleteStreamChunkTransform,
} from './complete';
import { anyscaleEmbedConfig, anyscaleEmbedResponseTransform } from './embed';
 
export const anyscaleConfig: AIProviderConfig = {
  api: anyscaleAPIConfig,
  [FunctionName.COMPLETE]: anyscaleCompleteConfig,
  [FunctionName.CHAT_COMPLETE]: anyscaleChatCompleteConfig,
  [FunctionName.EMBED]: anyscaleEmbedConfig,
  responseTransforms: {
    [FunctionName.STREAM_COMPLETE]: anyscaleCompleteStreamChunkTransform,
    [FunctionName.COMPLETE]: anyscaleCompleteResponseTransform,
    [FunctionName.CHAT_COMPLETE]: anyscaleChatCompleteResponseTransform,
    [FunctionName.STREAM_CHAT_COMPLETE]:
      anyscaleChatCompleteStreamChunkTransform,
    [FunctionName.EMBED]: anyscaleEmbedResponseTransform,
  },
};