Fix consecutive-acronym handling for tags like 'Bits AI SRE'

Author: pdv

.generator/conftest.py

@@ -352,9 +352,9 @@ def undo_operations():
 @given(parsers.parse('an instance of "{name}" API'))
 def api(context, api_version, specs, name):
     """Return an API instance."""
-    assert name in {tag["name"].replace(" ", "") for tag in specs[api_version]["tags"]}
+    raw_tag = next(t["name"] for t in specs[api_version]["tags"] if t["name"].replace(" ", "") == name)
     sanitized_name = name.replace("-", "")
-    context["api_instance"] = {"name": sanitized_name}
+    context["api_instance"] = {"name": sanitized_name, "raw_tag": raw_tag}
 
 
 @given(parsers.parse('operation "{name}" enabled'))

.generator/src/generator/templates/apis.j2

@@ -1,6 +1,6 @@
 {%- for api in apis %}
 {%- set classname = api|class_name %}
-from {{ package }}.{{ version }}.api.{{ classname|safe_snake_case }} import {{ classname }}
+from {{ package }}.{{ version }}.api.{{ api|safe_snake_case }} import {{ classname }}
 {%- endfor %}
 
 

.generator/src/generator/templates/example.j2

@@ -14,7 +14,7 @@
 from {{ package }} import {{ values|sort|join(', ') }}
 {%- endfor %}
 from datadog_api_client import ApiClient, Configuration
-from datadog_api_client.{{ version }}.api.{{ context.api_instance.name|safe_snake_case }}_api import {{ context.api_instance.name }}Api
+from datadog_api_client.{{ version }}.api.{{ context.api_instance.raw_tag|safe_snake_case }}_api import {{ context.api_instance.name }}Api
 {%- for package in imports|sort %}
 from {{ package }} import {{ imports[package]|sort|join(', ') }}
 {%- endfor %}