Skip to content

Commit 942387e

Browse files
Christopher ReesChristopher Rees
authored andcommitted
Fix bedrock_stubber to include modelDetails in get_foundation_model response
1 parent dd00e5d commit 942387e

1 file changed

Lines changed: 13 additions & 1 deletion

File tree

python/test_tools/bedrock_stubber.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,19 @@ def stub_list_foundation_models(self, models, error_code=None):
3333

3434
def stub_get_foundation_model(self, model_identifier, error_code=None):
3535
expected_params = {"modelIdentifier": model_identifier}
36-
response = {}
36+
response = {
37+
"modelDetails": {
38+
"modelArn": "arn:aws:test:::test-resource",
39+
"modelId": model_identifier,
40+
"modelName": "testModelName",
41+
"providerName": "testProviderName",
42+
"inputModalities": ["TEXT"],
43+
"outputModalities": ["TEXT"],
44+
"responseStreamingSupported": False,
45+
"customizationsSupported": ["FINE_TUNING"],
46+
"inferenceTypesSupported": ["ON_DEMAND"],
47+
}
48+
}
3749
self._stub_bifurcator(
3850
"get_foundation_model", expected_params, response, error_code=error_code
3951
)

0 commit comments

Comments
 (0)