FilesExpand file tree

main

/

pi-model-options.test.ts

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

58 lines (49 loc) · 1.97 KB

main

/

pi-model-options.test.ts

Copy path

Top

File metadata and controls

• Code
• Blame

58 lines (49 loc) · 1.97 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

import { afterAll, beforeAll, describe, expect, it } from 'vitest'

import { getPiModelOptions } from '@/blocks/utils'

import { resolvePiModelId } from '@/providers/pi-providers'

import { getProviderFromModel } from '@/providers/utils'

import { useProvidersStore } from '@/stores/providers/store'

const originalBaseModels = useProvidersStore.getState().providers.base.models

const originalOpenRouterModels = useProvidersStore.getState().providers.openrouter.models

describe('Pi model options', () => {

beforeAll(() => {

const store = useProvidersStore.getState()

store.setProviderModels('base', [

'claude-sonnet-4-6',

'claude-opus-4-1',

'claude-sonnet-4-0',

'gpt-5.4',

'cerebras/zai-glm-4.7',

'glm-5.1',

'glm-4.5-air',

])

store.setProviderModels('openrouter', [

'openrouter/openai/gpt-5',

'openrouter/openrouter/fusion',

])

})

afterAll(() => {

const store = useProvidersStore.getState()

store.setProviderModels('base', originalBaseModels)

store.setProviderModels('openrouter', originalOpenRouterModels)

})

it("only exposes models present in Pi's pinned catalog", () => {

const options = getPiModelOptions()

expect(options.length).toBeGreaterThan(0)

for (const option of options) {

const providerId = getProviderFromModel(option.id)

expect(resolvePiModelId(providerId, option.id), option.id).toBeDefined()

}

})

it('keeps current models and excludes retired catalog entries', () => {

const modelIds = getPiModelOptions().map(({ id }) => id)

expect(modelIds).toContain('claude-sonnet-4-6')

expect(modelIds).not.toContain('claude-opus-4-1')

expect(modelIds).not.toContain('claude-sonnet-4-0')

})

it("does not apply OpenRouter capability filters beyond Pi's catalog", () => {

const modelIds = getPiModelOptions().map(({ id }) => id)

expect(modelIds).toContain('openrouter/openai/gpt-5')

expect(modelIds).toContain('openrouter/openrouter/fusion')

})

})