Skip to content

Commit d9075f2

Browse files
cosminachoclaude
andcommitted
fix: update BYOM tests to use availableOperationCodes
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent 3f2f154 commit d9075f2

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

tests/core/test_base_client.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)