@@ -1933,7 +1933,7 @@ def test_valid_operation_code(self, llmgw_env_vars):
19331933 settings .operation_code = "code1"
19341934 model_info = {
19351935 "modelName" : "custom-model" ,
1936- "byomDetails" : {"operationCodes" : ["code1" , "code2" ]},
1936+ "byomDetails" : {"availableOperationCodes" : ["code1" , "code2" ]},
19371937 }
19381938 settings .validate_byo_model (model_info ) # Should not raise
19391939
@@ -1943,7 +1943,7 @@ def test_invalid_operation_code_raises(self, llmgw_env_vars):
19431943 settings .operation_code = "invalid-code"
19441944 model_info = {
19451945 "modelName" : "custom-model" ,
1946- "byomDetails" : {"operationCodes" : ["code1" , "code2" ]},
1946+ "byomDetails" : {"availableOperationCodes" : ["code1" , "code2" ]},
19471947 }
19481948 with pytest .raises (ValueError , match = "operation code" ):
19491949 settings .validate_byo_model (model_info )
@@ -1954,7 +1954,7 @@ def test_auto_picks_first_operation_code(self, llmgw_env_vars):
19541954 settings .operation_code = None
19551955 model_info = {
19561956 "modelName" : "custom-model" ,
1957- "byomDetails" : {"operationCodes" : ["auto-code" ]},
1957+ "byomDetails" : {"availableOperationCodes" : ["auto-code" ]},
19581958 }
19591959 settings .validate_byo_model (model_info )
19601960 assert settings .operation_code == "auto-code"
@@ -1965,7 +1965,7 @@ def test_auto_picks_first_with_warning_for_multiple(self, llmgw_env_vars):
19651965 settings .operation_code = None
19661966 model_info = {
19671967 "modelName" : "custom-model" ,
1968- "byomDetails" : {"operationCodes" : ["code1" , "code2" ]},
1968+ "byomDetails" : {"availableOperationCodes" : ["code1" , "code2" ]},
19691969 }
19701970 with patch ("uipath.llm_client.settings.llmgateway.settings.logging" ) as mock_logging :
19711971 settings .validate_byo_model (model_info )
@@ -1978,7 +1978,7 @@ def test_no_operation_codes_no_change(self, llmgw_env_vars):
19781978 settings .operation_code = None
19791979 model_info = {
19801980 "modelName" : "custom-model" ,
1981- "byomDetails" : {"operationCodes" : []},
1981+ "byomDetails" : {"availableOperationCodes" : []},
19821982 }
19831983 settings .validate_byo_model (model_info )
19841984 assert settings .operation_code is None
0 commit comments