All files / api/src/ai-providers/stability-ai image-generate-v2.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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import {
  generateErrorResponse,
  generateInvalidProviderResponseError,
} from '@api/utils/ai-provider';
import type { AIProviderFunctionConfig } from '@shared/types/ai-providers/config';
import type { ErrorResponseBody } from '@shared/types/api/response';
import type { GenerateImageResponseBody } from '@shared/types/api/routes/images-api';
import { AIProvider } from '@shared/types/constants';
 
export const StabilityAIImageGenerateV2Config: AIProviderFunctionConfig = {
  prompt: {
    param: 'prompt',
    required: true,
  },
  negative_prompt: {
    param: 'negative_prompt',
    required: false,
  },
  aspect_ratio: {
    param: 'aspect_ratio',
    required: false,
  },
  seed: {
    param: 'seed',
    required: false,
  },
  style_preset: {
    param: 'style_preset',
    required: false,
  },
  mode: {
    param: 'mode',
    required: false,
  },
  output_format: {
    param: 'output_format',
    required: false,
  },
  image: {
    param: 'image',
    required: false,
  },
  strength: {
    param: 'strength',
    required: false,
  },
};
 
enum StabilityAIIMageGenerateV2FinishReason {
  CONTENT_FILTERED = 'CONTENT_FILTERED',
  SUCCESS = 'SUCCESS',
}
 
interface StabilityAIIMageGenerateV2Response {
  image: string;
  finish_reason: StabilityAIIMageGenerateV2FinishReason;
  seed: number;
}
 
interface StabilityAIIMageGenerateV2ErrorResponse {
  id: string;
  name: string;
  errors: string[];
}
 
export const StabilityAIImageGenerateV2ResponseTransform: (
  response:
    | StabilityAIIMageGenerateV2Response
    | StabilityAIIMageGenerateV2ErrorResponse,
  responseStatus: number,
) => GenerateImageResponseBody | ErrorResponseBody = (
  response,
  responseStatus,
) => {
  if (responseStatus !== 200 && 'errors' in response) {
    return generateErrorResponse(
      {
        message: response.errors.join(', '),
        type: response.name,
        param: response.id,
        code: undefined,
      },
      AIProvider.STABILITY_AI,
    );
  }
 
  if ('image' in response) {
    return {
      created: Math.floor(Date.now() / 1000),
      data: [
        {
          b64_json: response.image,
        },
      ],
      provider: AIProvider.STABILITY_AI,
    };
  }
 
  return generateInvalidProviderResponseError(
    response as unknown as Record<string, unknown>,
    AIProvider.STABILITY_AI,
  );
};