diff --git a/codegenerator/rust_cli.py b/codegenerator/rust_cli.py index 97cdc8c..90d7cff 100644 --- a/codegenerator/rust_cli.py +++ b/codegenerator/rust_cli.py @@ -901,7 +901,7 @@ class RustCliGenerator(BaseGenerator): ) if response: - response_key: str + response_key: str | None if args.response_key: response_key = ( args.response_key @@ -910,7 +910,7 @@ class RustCliGenerator(BaseGenerator): ) else: response_key = resource_name - response_def, _ = common.find_resource_schema( + response_def, response_key = common.find_resource_schema( response, None, response_key ) @@ -960,9 +960,15 @@ class RustCliGenerator(BaseGenerator): ) response_props = response.get("properties", {}) - if ( - response_props - and response_props[ + if response_props and ( + ( + response_key + and response_props.get(response_key, {}).get( + "type" + ) + == "array" + ) + or response_props[ list(response_props.keys())[0] ].get("type") == "array"