| import { describe, expect, it } from 'vitest' |
| import { |
| DEFAULT_IMAGE_OUTPUT_FORMAT, |
| IMAGE_BACKGROUND_LABELS, |
| IMAGE_BACKGROUNDS, |
| IMAGE_INPUT_FIDELITY_LABELS, |
| IMAGE_INPUT_FIDELITIES, |
| IMAGE_MODERATION_LABELS, |
| IMAGE_MODERATIONS, |
| IMAGE_OUTPUT_FORMAT_LABELS, |
| IMAGE_OUTPUT_FORMATS, |
| IMAGE_QUALITIES, |
| IMAGE_QUALITY_LABELS, |
| IMAGE_RATIOS, |
| buildImageEditEndpoint, |
| buildImageEndpoint, |
| buildImageRequestBody, |
| formatImageQuality, |
| getDefaultImageSize, |
| getImageSizeOptions, |
| ratioToSize, |
| supportsImageInputFidelity |
| } from './image-options' |
|
|
| describe('image options', () => { |
| it('maps expanded ratios to API size strings', () => { |
| expect(IMAGE_RATIOS).toEqual(['1:1', '3:2', '2:3', '4:3', '3:4', '16:9', '9:16', '21:9', '9:21']) |
| expect(ratioToSize('1:1')).toBe('1024x1024') |
| expect(ratioToSize('3:2')).toBe('1536x1024') |
| expect(ratioToSize('2:3')).toBe('1024x1536') |
| expect(ratioToSize('4:3')).toBe('1024x768') |
| expect(ratioToSize('3:4')).toBe('768x1024') |
| expect(ratioToSize('16:9')).toBe('1792x1008') |
| expect(ratioToSize('9:16')).toBe('1008x1792') |
| expect(ratioToSize('21:9')).toBe('1344x576') |
| expect(ratioToSize('9:21')).toBe('576x1344') |
| expect(getDefaultImageSize('16:9')).toBe('1792x1008') |
| expect(getDefaultImageSize('9:16')).toBe('1008x1792') |
| expect(getImageSizeOptions('16:9').at(-1)?.value).toBe('3840x2160') |
| }) |
|
|
| it('exposes GPT and compatibility quality options', () => { |
| expect(IMAGE_QUALITIES).toEqual(['auto', 'low', 'medium', 'high']) |
| expect(IMAGE_QUALITY_LABELS).toEqual({ |
| auto: '自动', |
| low: '低', |
| medium: '中', |
| high: '高' |
| }) |
| expect(formatImageQuality('high')).toBe('高') |
| expect(IMAGE_OUTPUT_FORMATS).toEqual(['jpeg', 'png', 'webp']) |
| expect(DEFAULT_IMAGE_OUTPUT_FORMAT).toBe('png') |
| expect(IMAGE_OUTPUT_FORMAT_LABELS).toEqual({ jpeg: 'JPEG', png: 'PNG', webp: 'WebP' }) |
| expect(IMAGE_BACKGROUNDS).toEqual(['auto', 'opaque']) |
| expect(IMAGE_BACKGROUND_LABELS).toEqual({ auto: '自动', opaque: '不透明' }) |
| expect(IMAGE_MODERATIONS).toEqual(['auto', 'low']) |
| expect(IMAGE_MODERATION_LABELS).toEqual({ auto: '自动', low: '低' }) |
| expect(IMAGE_INPUT_FIDELITIES).toEqual(['low', 'high']) |
| expect(IMAGE_INPUT_FIDELITY_LABELS).toEqual({ low: '低', high: '高' }) |
| }) |
|
|
| it('normalizes baseURL into the generations endpoint', () => { |
| expect(buildImageEndpoint('https://example.test///')).toBe('https://example.test/v1/images/generations') |
| }) |
|
|
| it('normalizes baseURL into the edits endpoint', () => { |
| expect(buildImageEditEndpoint('https://example.test///')).toBe('https://example.test/v1/images/edits') |
| }) |
|
|
| it('treats gpt-image-2 as not supporting input fidelity overrides', () => { |
| expect(supportsImageInputFidelity('gpt-image-2')).toBe(false) |
| expect(supportsImageInputFidelity('gpt-image-1')).toBe(true) |
| }) |
|
|
| it('builds a request body without leaking empty optional fields', () => { |
| expect( |
| buildImageRequestBody({ |
| conversationId: 'c1', |
| prompt: ' mint glasshouse ', |
| model: 'gpt-image-2', |
| ratio: '3:2', |
| size: '1536x1024', |
| quality: 'auto', |
| n: 2 |
| }) |
| ).toEqual({ |
| prompt: 'mint glasshouse', |
| model: 'gpt-image-2', |
| size: '1536x1024', |
| quality: 'auto', |
| n: 2 |
| }) |
| }) |
|
|
| it('maps advanced request fields when they are enabled', () => { |
| expect( |
| buildImageRequestBody({ |
| conversationId: 'c1', |
| prompt: 'scene', |
| model: 'gpt-image-2', |
| ratio: '1:1', |
| size: '1024x1024', |
| quality: 'high', |
| n: 1, |
| outputFormat: 'webp', |
| outputCompression: 85, |
| background: 'opaque', |
| moderation: 'low', |
| stream: true, |
| partialImages: 3 |
| }) |
| ).toEqual({ |
| prompt: 'scene', |
| model: 'gpt-image-2', |
| size: '1024x1024', |
| quality: 'high', |
| n: 1, |
| output_format: 'webp', |
| output_compression: 85, |
| background: 'opaque', |
| moderation: 'low', |
| stream: true, |
| partial_images: 3 |
| }) |
| }) |
| }) |
|
|
