workers-ai index.ts

0% Statements 0/19

100% Branches 1/1

100% Functions 1/1

0% Lines 0/19

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  
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  import type { AIProviderConfig } from '@shared/types/ai-providers/config';
import { FunctionName } from '@shared/types/api/request';
import { workersAIAPIConfig } from './api';
import {
  workersAIChatCompleteConfig,
  workersAIChatCompleteResponseTransform,
} from './chat-complete';
import {
  workersAICompleteConfig,
  workersAICompleteResponseTransform,
} from './complete';
import { workersAIEmbedConfig, workersAIEmbedResponseTransform } from './embed';
import {
  workersAIImageGenerateConfig,
  workersAIImageGenerateResponseTransform,
} from './image-generate';
 
export const workersAIConfig: AIProviderConfig = {
  api: workersAIAPIConfig,
  [FunctionName.COMPLETE]: workersAICompleteConfig,
  [FunctionName.CHAT_COMPLETE]: workersAIChatCompleteConfig,
  [FunctionName.EMBED]: workersAIEmbedConfig,
  [FunctionName.GENERATE_IMAGE]: workersAIImageGenerateConfig,
  responseTransforms: {
    [FunctionName.COMPLETE]: workersAICompleteResponseTransform,
    [FunctionName.CHAT_COMPLETE]: workersAIChatCompleteResponseTransform,
    [FunctionName.EMBED]: workersAIEmbedResponseTransform,
    [FunctionName.GENERATE_IMAGE]: workersAIImageGenerateResponseTransform,
  },
};