All files / api/src/ai-providers/together-ai api.ts

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import type { InternalProviderAPIConfig } from '@shared/types/ai-providers/config';
import { FunctionName } from '@shared/types/api/request';
 
const togetherAIAPIConfig: InternalProviderAPIConfig = {
  getBaseURL: () => 'https://api.together.xyz',
  headers: ({ saTarget }) => {
    return { Authorization: `Bearer ${saTarget.api_key}` };
  },
  getEndpoint: ({ saRequestData }) => {
    switch (saRequestData.functionName) {
      case FunctionName.COMPLETE:
        return '/v1/completions';
      case FunctionName.CHAT_COMPLETE:
        return '/v1/chat/completions';
      case FunctionName.EMBED:
        return '/v1/embeddings';
      default:
        return '';
    }
  },
};
 
export default togetherAIAPIConfig;