All files / api/src/ai-providers/stability-ai index.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
35
36
37
38
39
40
41
42
43
44
45
46
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import type {
  AIProviderConfig,
  ResponseTransformFunction,
} from '@shared/types/ai-providers/config';
import { FunctionName } from '@shared/types/api/request';
import type { GenerateImageRequestBody } from '@shared/types/api/routes/images-api';
import StabilityAIAPIConfig from './api';
import {
  StabilityAIImageGenerateV1Config,
  StabilityAIImageGenerateV1ResponseTransform,
} from './image-generate';
import {
  StabilityAIImageGenerateV2Config,
  StabilityAIImageGenerateV2ResponseTransform,
} from './image-generate-v2';
import { isStabilityV1Model } from './utils';
 
const StabilityAIConfig: AIProviderConfig = {
  api: StabilityAIAPIConfig,
  getConfig: (saRequestBody): AIProviderConfig => {
    const generateImageRequestBody = saRequestBody as GenerateImageRequestBody;
    const model = generateImageRequestBody?.model;
    if (typeof model === 'string' && isStabilityV1Model(model)) {
      return {
        api: StabilityAIAPIConfig,
        [FunctionName.GENERATE_IMAGE]: StabilityAIImageGenerateV1Config,
        responseTransforms: {
          [FunctionName.GENERATE_IMAGE]:
            StabilityAIImageGenerateV1ResponseTransform as unknown as ResponseTransformFunction,
        },
      };
    }
    const config: AIProviderConfig = {
      api: StabilityAIAPIConfig,
      [FunctionName.GENERATE_IMAGE]: StabilityAIImageGenerateV2Config,
      responseTransforms: {
        [FunctionName.GENERATE_IMAGE]:
          StabilityAIImageGenerateV2ResponseTransform as unknown as ResponseTransformFunction,
      },
    };
    return config;
  },
};
 
export default StabilityAIConfig;