All files / api/src/ai-providers/segmind 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
25
26
27
28
29
30
31
32
33
34
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import type { InternalProviderAPIConfig } from '@shared/types/ai-providers/config';
import { FunctionName } from '@shared/types/api/request';
 
const segmindAPIConfig: InternalProviderAPIConfig = {
  getBaseURL: () => 'https://api.segmind.com/v1',
  headers: ({ saTarget }) => {
    return { 'x-api-key': `${saTarget.api_key}` };
  },
  getEndpoint: ({ saRequestData }) => {
    switch (saRequestData.functionName) {
      case FunctionName.CHAT_COMPLETE:
        return '/chat/completions';
      case FunctionName.GENERATE_IMAGE: {
        const model = saRequestData.requestBody.model;
 
        // Validate model parameter to prevent invalid endpoints
        if (!model || typeof model !== 'string' || model.trim().length === 0) {
          throw new Error(
            'Model parameter is required for Segmind image generation',
          );
        }
 
        // Segmind uses model-specific endpoints for image generation
        // e.g., /v1/sdxl1.0-txt2img, /v1/fast-flux-schnell
        return `/${model.trim()}`;
      }
      default:
        return '/chat/completions';
    }
  },
};
 
export default segmindAPIConfig;