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

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

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  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 { deepbricksAPIConfig } from './api';
import {
  deepbricksChatCompleteConfig,
  deepbricksChatCompleteResponseTransform,
} from './chat-complete';
import {
  deepbricksImageGenerateConfig,
  deepbricksImageGenerateResponseTransform,
} from './image-generate';
 
export const deepbricksConfig: AIProviderConfig = {
  api: deepbricksAPIConfig,
  [FunctionName.CHAT_COMPLETE]: deepbricksChatCompleteConfig,
  [FunctionName.GENERATE_IMAGE]: deepbricksImageGenerateConfig,
  responseTransforms: {
    [FunctionName.CHAT_COMPLETE]: deepbricksChatCompleteResponseTransform,
    [FunctionName.GENERATE_IMAGE]: deepbricksImageGenerateResponseTransform,
  },
};