1
- import { getModel , getEndpoint , models } from "../src/models.ts" ;
1
+ import { getModelSettings , getImageModelSettings , getEndpoint , models } from "../src/models.ts" ;
2
2
import { ModelProviderName , ModelClass } from "../src/types.ts" ;
3
3
import { describe , test , expect , vi } from "vitest" ;
4
4
@@ -129,22 +129,22 @@ describe("Model Provider Configuration", () => {
129
129
} ) ;
130
130
describe ( "Livepeer Provider" , ( ) => {
131
131
test ( "should have correct endpoint configuration" , ( ) => {
132
- expect ( models [ ModelProviderName . LIVEPEER ] . endpoint ) . toBe ( "http://gateway.test-gateway" ) ;
132
+ expect ( getEndpoint ( ModelProviderName . LIVEPEER ) ) . toBe ( "http://gateway.test-gateway" ) ;
133
133
} ) ;
134
134
135
135
test ( "should have correct model mappings" , ( ) => {
136
136
const livepeerModels = models [ ModelProviderName . LIVEPEER ] . model ;
137
- expect ( livepeerModels [ ModelClass . SMALL ] ) . toBe ( "meta-llama/Meta-Llama-3.1-8B-Instruct" ) ;
138
- expect ( livepeerModels [ ModelClass . MEDIUM ] ) . toBe ( "meta-llama/Meta-Llama-3.1-8B-Instruct" ) ;
139
- expect ( livepeerModels [ ModelClass . LARGE ] ) . toBe ( "meta-llama/Meta-Llama-3.1-8B-Instruct" ) ;
140
- expect ( livepeerModels [ ModelClass . IMAGE ] ) . toBe ( "ByteDance/SDXL-Lightning" ) ;
137
+ expect ( livepeerModels [ ModelClass . SMALL ] ?. name ) . toBe ( "meta-llama/Meta-Llama-3.1-8B-Instruct" ) ;
138
+ expect ( livepeerModels [ ModelClass . MEDIUM ] ?. name ) . toBe ( "meta-llama/Meta-Llama-3.1-8B-Instruct" ) ;
139
+ expect ( livepeerModels [ ModelClass . LARGE ] ?. name ) . toBe ( "meta-llama/Meta-Llama-3.1-8B-Instruct" ) ;
140
+ expect ( livepeerModels [ ModelClass . IMAGE ] ?. name ) . toBe ( "ByteDance/SDXL-Lightning" ) ;
141
141
} ) ;
142
142
143
143
test ( "should have correct settings configuration" , ( ) => {
144
- const settings = models [ ModelProviderName . LIVEPEER ] . settings ;
145
- expect ( settings . maxInputTokens ) . toBe ( 128000 ) ;
146
- expect ( settings . maxOutputTokens ) . toBe ( 8192 ) ;
147
- expect ( settings . temperature ) . toBe ( 0 ) ;
144
+ const settings = getModelSettings ( ModelProviderName . LIVEPEER , ModelClass . LARGE ) ;
145
+ expect ( settings ? .maxInputTokens ) . toBe ( 8000 ) ;
146
+ expect ( settings ? .maxOutputTokens ) . toBe ( 8192 ) ;
147
+ expect ( settings ? .temperature ) . toBe ( 0 ) ;
148
148
} ) ;
149
149
} ) ;
150
150
} ) ;
@@ -169,10 +169,10 @@ describe("Model Retrieval Functions", () => {
169
169
) . toBe ( "nousresearch/hermes-3-llama-3.1-405b" ) ;
170
170
} ) ;
171
171
172
- test ( "should throw error for invalid model provider" , ( ) => {
173
- expect ( ( ) =>
174
- getModel ( "INVALID_PROVIDER" as any , ModelClass . SMALL )
175
- ) . toThrow ( ) ;
172
+ test ( "Test to ensure an invalid model provider returns undefined " , ( ) => {
173
+ expect (
174
+ getModelSettings ( "INVALID_PROVIDER" as any , ModelClass . SMALL )
175
+ ) . toBe ( undefined ) ;
176
176
} ) ;
177
177
} ) ;
178
178
@@ -250,12 +250,12 @@ describe("Environment Variable Integration", () => {
250
250
describe ( "Generation with Livepeer" , ( ) => {
251
251
test ( "should have correct image generation settings" , ( ) => {
252
252
const livepeerConfig = models [ ModelProviderName . LIVEPEER ] ;
253
- expect ( livepeerConfig . model [ ModelClass . IMAGE ] ) . toBe ( "ByteDance/SDXL-Lightning" ) ;
254
- expect ( livepeerConfig . settings . temperature ) . toBe ( 0 ) ;
253
+ expect ( livepeerConfig . model [ ModelClass . IMAGE ] ?. name ) . toBe ( "ByteDance/SDXL-Lightning" ) ;
254
+ expect ( getModelSettings ( ModelProviderName . LIVEPEER , ModelClass . SMALL ) ? .temperature ) . toBe ( 0 ) ;
255
255
} ) ;
256
256
257
257
test ( "should use default image model" , ( ) => {
258
258
delete process . env . IMAGE_LIVEPEER_MODEL ;
259
- expect ( models [ ModelProviderName . LIVEPEER ] . model [ ModelClass . IMAGE ] ) . toBe ( "ByteDance/SDXL-Lightning" ) ;
259
+ expect ( getImageModelSettings ( ModelProviderName . LIVEPEER ) ?. name ) . toBe ( "ByteDance/SDXL-Lightning" ) ;
260
260
} ) ;
261
261
} ) ;
0 commit comments