File tree Expand file tree Collapse file tree 1 file changed +16
-1
lines changed Expand file tree Collapse file tree 1 file changed +16
-1
lines changed Original file line number Diff line number Diff line change @@ -194,7 +194,6 @@ describe("ProfileValidator", () => {
194194 "chutes" ,
195195 "sambanova" ,
196196 "fireworks" ,
197- "io-intelligence" ,
198197 ]
199198
200199 apiModelProviders . forEach ( ( provider ) => {
@@ -230,6 +229,22 @@ describe("ProfileValidator", () => {
230229 expect ( ProfileValidator . isProfileAllowed ( profile , allowList ) ) . toBe ( true )
231230 } )
232231
232+ // Test for io-intelligence provider which uses ioIntelligenceModelId
233+ it ( `should extract ioIntelligenceModelId for io-intelligence provider` , ( ) => {
234+ const allowList : OrganizationAllowList = {
235+ allowAll : false ,
236+ providers : {
237+ "io-intelligence" : { allowAll : false , models : [ "test-model" ] } ,
238+ } ,
239+ }
240+ const profile : ProviderSettings = {
241+ apiProvider : "io-intelligence" as any ,
242+ ioIntelligenceModelId : "test-model" ,
243+ }
244+
245+ expect ( ProfileValidator . isProfileAllowed ( profile , allowList ) ) . toBe ( true )
246+ } )
247+
233248 it ( "should extract vsCodeLmModelSelector.id for vscode-lm provider" , ( ) => {
234249 const allowList : OrganizationAllowList = {
235250 allowAll : false ,
You can’t perform that action at this time.
0 commit comments