upstage index.ts

0% Statements 0/21

100% Branches 1/1

100% Functions 1/1

0% Lines 0/21

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  
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  import type { AIProviderConfig } from '@shared/types/ai-providers/config';
import { FunctionName } from '@shared/types/api/request';
import { AIProvider } from '@shared/types/constants';
import {
  createModelResponseParams,
  embedParams,
  responseTransformers,
} from '../open-ai-base';
import { upstageAPIConfig } from './api';
import {
  upstageChatCompleteConfig,
  upstageChatCompleteResponseTransform,
  upstageChatCompleteStreamChunkTransform,
} from './chat-complete';
 
export const upstageConfig: AIProviderConfig = {
  api: upstageAPIConfig,
  [FunctionName.CHAT_COMPLETE]: upstageChatCompleteConfig,
  [FunctionName.EMBED]: embedParams([], {
    model: 'solar-embedding-1-large-query',
  }),
  [FunctionName.CREATE_MODEL_RESPONSE]: createModelResponseParams([]),
  responseTransforms: {
    [FunctionName.CHAT_COMPLETE]: upstageChatCompleteResponseTransform,
    [FunctionName.STREAM_CHAT_COMPLETE]:
      upstageChatCompleteStreamChunkTransform,
    ...responseTransformers(AIProvider.UPSTAGE, {
      embed: true,
    }),
  },
};