ai21 index.ts

100% Statements 17/17

100% Branches 0/0

100% Functions 0/0

100% Lines 17/17

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  
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 { aI21APIConfig } from './api';
import {
  aI21ChatCompleteConfig,
  aI21ChatCompleteResponseTransform,
} from './chat-complete';
import { aI21CompleteConfig, aI21CompleteResponseTransform } from './complete';
import { aI21EmbedConfig, aI21EmbedResponseTransform } from './embed';
 
export const aI21Config: AIProviderConfig = {
  api: aI21APIConfig,
  [FunctionName.COMPLETE]: aI21CompleteConfig,
  [FunctionName.EMBED]: aI21EmbedConfig,
  [FunctionName.CHAT_COMPLETE]: aI21ChatCompleteConfig,
  requestTransforms: {},
  responseTransforms: {
    [FunctionName.COMPLETE]: aI21CompleteResponseTransform,
    [FunctionName.CHAT_COMPLETE]: aI21ChatCompleteResponseTransform,
    [FunctionName.EMBED]: aI21EmbedResponseTransform,
  },
};