together-ai index.ts

0% Statements 0/20

100% Branches 1/1

100% Functions 1/1

0% Lines 0/20

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
31
32
33
34
35  
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  import type { AIProviderConfig } from '@shared/types/ai-providers/config';
import { FunctionName } from '@shared/types/api/request';
import togetherAIAPIConfig from './api';
import {
  togetherAIChatCompleteConfig,
  togetherAIChatCompleteResponseTransform,
  togetherAIChatCompleteStreamChunkTransform,
} from './chat-complete';
import {
  togetherAICompleteConfig,
  togetherAICompleteResponseTransform,
  togetherAICompleteStreamChunkTransform,
} from './complete';
import {
  togetherAIEmbedConfig,
  togetherAIEmbedResponseTransform,
} from './embed';
 
const togetherAIConfig: AIProviderConfig = {
  api: togetherAIAPIConfig,
  [FunctionName.COMPLETE]: togetherAICompleteConfig,
  [FunctionName.CHAT_COMPLETE]: togetherAIChatCompleteConfig,
  [FunctionName.EMBED]: togetherAIEmbedConfig,
  responseTransforms: {
    [FunctionName.COMPLETE]: togetherAICompleteResponseTransform,
    [FunctionName.STREAM_COMPLETE]: togetherAICompleteStreamChunkTransform,
    [FunctionName.CHAT_COMPLETE]: togetherAIChatCompleteResponseTransform,
    [FunctionName.STREAM_CHAT_COMPLETE]:
      togetherAIChatCompleteStreamChunkTransform,
    [FunctionName.EMBED]: togetherAIEmbedResponseTransform,
  },
};
 
export default togetherAIConfig;