fireworks-ai index.ts

0% Statements 0/27
100% Branches 1/1
100% Functions 1/1
0% Lines 0/27
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
36
37
38
39
40
41
42
43
44
45  
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  import type { AIProviderConfig } from '@shared/types/ai-providers/config';
import { FunctionName } from '@shared/types/api/request';
import FireworksAIAPIConfig from './api';
import {
  FireworksAIChatCompleteConfig,
  FireworksAIChatCompleteResponseTransform,
  FireworksAIChatCompleteStreamChunkTransform,
} from './chat-complete';
import {
  FireworksAICompleteConfig,
  FireworksAICompleteResponseTransform,
  FireworksAICompleteStreamChunkTransform,
} from './complete';
import {
  FireworksAIEmbedConfig,
  FireworksAIEmbedResponseTransform,
} from './embed';
import {
  FireworksAIImageGenerateConfig,
  FireworksAIImageGenerateResponseTransform,
} from './image-generate';
import { FireworksFileListResponseTransform } from './list-files';
import { fireworksFileRetrieveResponseTransform } from './retrieve-file';
 
const FireworksAIConfig: AIProviderConfig = {
  api: FireworksAIAPIConfig,
  [FunctionName.COMPLETE]: FireworksAICompleteConfig,
  [FunctionName.CHAT_COMPLETE]: FireworksAIChatCompleteConfig,
  [FunctionName.EMBED]: FireworksAIEmbedConfig,
  [FunctionName.GENERATE_IMAGE]: FireworksAIImageGenerateConfig,
  responseTransforms: {
    [FunctionName.COMPLETE]: FireworksAICompleteResponseTransform,
    [FunctionName.STREAM_COMPLETE]: FireworksAICompleteStreamChunkTransform,
    [FunctionName.CHAT_COMPLETE]: FireworksAIChatCompleteResponseTransform,
    [FunctionName.STREAM_CHAT_COMPLETE]:
      FireworksAIChatCompleteStreamChunkTransform,
    [FunctionName.EMBED]: FireworksAIEmbedResponseTransform,
    [FunctionName.GENERATE_IMAGE]: FireworksAIImageGenerateResponseTransform,
    [FunctionName.LIST_FILES]: FireworksFileListResponseTransform,
    [FunctionName.RETRIEVE_FILE]: fireworksFileRetrieveResponseTransform,
  },
};
 
export default FireworksAIConfig;