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

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

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  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 { predibaseAPIConfig } from './api';
import {
  predibaseChatCompleteConfig,
  predibaseChatCompleteResponseTransform,
  predibaseChatCompleteStreamChunkTransform,
} from './chat-complete';
 
export const predibaseConfig: AIProviderConfig = {
  api: predibaseAPIConfig,
  [FunctionName.CHAT_COMPLETE]: predibaseChatCompleteConfig,
  responseTransforms: {
    [FunctionName.CHAT_COMPLETE]: predibaseChatCompleteResponseTransform,
    [FunctionName.STREAM_CHAT_COMPLETE]:
      predibaseChatCompleteStreamChunkTransform,
  },
};