template / macro: return_type / correct duplicate

Author: p1-ra

openapi_python_client/templates/endpoint_macros.py.jinja

@@ -71,15 +71,22 @@ params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
 {% endif %}
 {% endmacro %}
 
+{% macro reponses_type_csv(responses) %}
+{% for response in responses %}
+{{ response.prop.get_type_string() }}{{"," if not loop.last }}
+{% endfor %}
+{% endmacro %}
+
 {% macro return_type(endpoint) %}
 {% if endpoint.responses | length == 0 %}
 None
 {%- elif endpoint.responses | length == 1 %}
 {{ endpoint.responses[0].prop.get_type_string() }}
 {%- else %}
 Union[
-    {% for response in endpoint.responses %}
-    {{ response.prop.get_type_string() }}{{ "," if not loop.last }}
+    {% set response_types = reponses_type_csv(endpoint.responses) | trim  %}
+    {% for type_string in (response_types.split(',') | unique | list) %}
+    {{ type_string }}{{ "," if not loop.last }}
     {% endfor %}
 ]
 {%- endif %}