@@ -95,16 +95,17 @@ describe("FireworksHandler", () => {
9595 } )
9696
9797 it . each ( [
98- "accounts/fireworks/models/glm-5p1" ,
99- "accounts/fireworks/models/kimi-k2p6" ,
100- "accounts/fireworks/models/deepseek-v4-pro" ,
101- ] as const satisfies readonly FireworksModelId [ ] ) ( "should expose newly added model %s " , ( modelId ) => {
98+ { modelId : "accounts/fireworks/models/glm-5p1" as const , contextWindow : 202752 , inputPrice : 1.4 , outputPrice : 4.4 , cacheReadsPrice : 0.26 } ,
99+ { modelId : "accounts/fireworks/models/kimi-k2p6" as const , contextWindow : 262144 , inputPrice : 0.95 , outputPrice : 4.0 , cacheReadsPrice : 0.16 } ,
100+ { modelId : "accounts/fireworks/models/deepseek-v4-pro" as const , contextWindow : 1048576 , inputPrice : 1.74 , outputPrice : 3.48 , cacheReadsPrice : 0.14 } ,
101+ ] ) ( "should expose newly added model $modelId " , ( { modelId, contextWindow , inputPrice , outputPrice , cacheReadsPrice } ) => {
102102 expect ( fireworksModels [ modelId ] ) . toBeDefined ( )
103103 const info = fireworksModels [ modelId ]
104104 expect ( info . maxTokens ) . toBeGreaterThan ( 0 )
105- expect ( info . contextWindow ) . toBeGreaterThan ( 0 )
106- expect ( info . inputPrice ) . toBeGreaterThanOrEqual ( 0 )
107- expect ( info . outputPrice ) . toBeGreaterThanOrEqual ( 0 )
105+ expect ( info . contextWindow ) . toBe ( contextWindow )
106+ expect ( info . inputPrice ) . toBe ( inputPrice )
107+ expect ( info . outputPrice ) . toBe ( outputPrice )
108+ expect ( info . cacheReadsPrice ) . toBe ( cacheReadsPrice )
108109 expect ( info . description ) . toBeTruthy ( )
109110
110111 const handlerWithModel = new FireworksHandler ( {
0 commit comments