ab-openapi-python-generator 2.2.4__tar.gz → 2.2.6__tar.gz

{ab_openapi_python_generator-2.2.4 → ab_openapi_python_generator-2.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ab-openapi-python-generator
-Version: 2.2.4
+Version: 2.2.6
 Summary: Openapi Python Generator
 Project-URL: Homepage, https://github.com/auth-broker/openapi-python-generator
 Project-URL: Repository, https://github.com/auth-broker/openapi-python-generator

{ab_openapi_python_generator-2.2.4 → ab_openapi_python_generator-2.2.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ab-openapi-python-generator"
-version = "2.2.4"
+version = "2.2.6"
 description = "Openapi Python Generator"
 authors = [
     { name = "Marco Müllner", email = "muellnermarco@gmail.com" },

{ab_openapi_python_generator-2.2.4 → ab_openapi_python_generator-2.2.6}/src/ab_openapi_python_generator/language_converters/python/generator.py

@@ -13,6 +13,7 @@ from ab_openapi_python_generator.language_converters.python.exception_generator
 )
 from ab_openapi_python_generator.language_converters.python.model_generator import (
     generate_models,
+    generate_response_union_alias_models,
 )
 from ab_openapi_python_generator.models import ConversionResult, LibraryConfig
 
@@ -40,6 +41,14 @@ def generator(
     else:
         models = []
 
+    # Generate response union alias models (e.g. AuthorizeResponse) from paths
+    # so client return types that reference those aliases have corresponding
+    # model modules available in the output.
+    if data.paths is not None:
+        resp_alias_models = generate_response_union_alias_models(data.paths, pydantic_version)
+        if resp_alias_models:
+            models.extend(resp_alias_models)
+
     if data.paths is not None:
         clients = generate_clients(data, data.paths, library_config, env_token_name, pydantic_version)
     else:

{ab_openapi_python_generator-2.2.4 → ab_openapi_python_generator-2.2.6}/src/ab_openapi_python_generator/language_converters/python/templates/async_client_httpx_pydantic_2.jinja2

@@ -24,7 +24,7 @@ class AsyncClient(BaseModel):
     access_token: Optional[str] = None
 {% endif %}
 
-    def get_access_token(self) -> Optional[str]:
+    async def get_access_token(self) -> Optional[str]:
 {% if env_token_name is not none %}
         try:
             return os.environ["{{ env_token_name }}"]
@@ -34,7 +34,7 @@ class AsyncClient(BaseModel):
         return self.access_token
 {% endif %}
 
-    def set_access_token(self, value: str) -> None:
+    async def set_access_token(self, value: str) -> None:
 {% if env_token_name is not none %}
         raise Exception(
             "This client was generated with an environment variable for the access token. "
@@ -63,7 +63,7 @@ AsyncGenerator[str | dict[str, Any], None]
 {% else %}
             "Accept": "application/json",
 {% endif %}
-            "Authorization": f"Bearer { self.get_access_token() }",
+            "Authorization": f"Bearer { await self.get_access_token() }",
         }
 
         query_params: Dict[str, Any] = {