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

0% Statements 0/12
100% Branches 1/1
100% Functions 1/1
0% Lines 0/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                                     
import type { AIProviderConfig } from '@shared/types/ai-providers/config';
import { FunctionName } from '@shared/types/api/request';
import replicateAPIConfig from './api';
import {
  replicateChatCompleteConfig,
  replicateChatCompleteResponseTransform,
  replicateChatCompleteStreamChunkTransform,
} from './chat-complete';
 
export const replicateConfig: AIProviderConfig = {
  api: replicateAPIConfig,
  [FunctionName.CHAT_COMPLETE]: replicateChatCompleteConfig,
  responseTransforms: {
    [FunctionName.CHAT_COMPLETE]: replicateChatCompleteResponseTransform,
    [FunctionName.STREAM_CHAT_COMPLETE]:
      replicateChatCompleteStreamChunkTransform,
  },
};