|
1 | 1 | import { describe, it, expect } from 'bun:test'; |
2 | | -import { musicGenerateModel, musicCoverModel, isCodingPlan } from '../../../src/commands/music/models'; |
| 2 | +import { musicGenerateModel, musicCoverModel } from '../../../src/commands/music/models'; |
3 | 3 | import type { Config } from '../../../src/config/schema'; |
4 | 4 |
|
5 | 5 | describe('music models', () => { |
6 | | - it('isCodingPlan returns true for sk-cp- key', () => { |
7 | | - expect(isCodingPlan({ apiKey: 'sk-cp-abc' } as Config)).toBe(true); |
8 | | - }); |
9 | | - |
10 | | - it('isCodingPlan returns false for sk-api- key', () => { |
11 | | - expect(isCodingPlan({ apiKey: 'sk-api-xyz' } as Config)).toBe(false); |
12 | | - }); |
13 | | - |
14 | 6 | it('musicGenerateModel uses defaultMusicModel when set', () => { |
15 | | - const config = { apiKey: 'sk-api-xyz', defaultMusicModel: 'music-2.6' } as Config; |
16 | | - expect(musicGenerateModel(config)).toBe('music-2.6'); |
| 7 | + const config = { defaultMusicModel: 'music-2.5+' } as Config; |
| 8 | + expect(musicGenerateModel(config)).toBe('music-2.5+'); |
17 | 9 | }); |
18 | 10 |
|
19 | | - it('musicGenerateModel falls back to key-type default when no defaultMusicModel', () => { |
20 | | - const cpConfig = { apiKey: 'sk-cp-abc' } as Config; |
21 | | - expect(musicGenerateModel(cpConfig)).toBe('music-2.6'); |
22 | | - |
23 | | - const apiConfig = { apiKey: 'sk-api-xyz' } as Config; |
24 | | - expect(musicGenerateModel(apiConfig)).toBe('music-2.6-free'); |
| 11 | + it('musicGenerateModel defaults to music-2.6', () => { |
| 12 | + expect(musicGenerateModel({} as Config)).toBe('music-2.6'); |
25 | 13 | }); |
26 | 14 |
|
27 | 15 | it('musicCoverModel ignores defaultMusicModel for non-cover models', () => { |
28 | | - const config = { apiKey: 'sk-api-xyz', defaultMusicModel: 'music-2.6' } as Config; |
29 | | - expect(musicCoverModel(config)).toBe('music-cover-free'); |
| 16 | + const config = { defaultMusicModel: 'music-2.6' } as Config; |
| 17 | + expect(musicCoverModel(config)).toBe('music-cover'); |
30 | 18 | }); |
31 | 19 |
|
32 | | - it('musicCoverModel uses key-type default when no defaultMusicModel', () => { |
33 | | - const cpConfig = { apiKey: 'sk-cp-abc' } as Config; |
34 | | - expect(musicCoverModel(cpConfig)).toBe('music-cover'); |
| 20 | + it('musicCoverModel uses defaultMusicModel when it is a cover model', () => { |
| 21 | + const config = { defaultMusicModel: 'music-cover' } as Config; |
| 22 | + expect(musicCoverModel(config)).toBe('music-cover'); |
| 23 | + }); |
35 | 24 |
|
36 | | - const apiConfig = { apiKey: 'sk-api-xyz' } as Config; |
37 | | - expect(musicCoverModel(apiConfig)).toBe('music-cover-free'); |
| 25 | + it('musicCoverModel defaults to music-cover', () => { |
| 26 | + expect(musicCoverModel({} as Config)).toBe('music-cover'); |
38 | 27 | }); |
39 | 28 | }); |
0 commit comments