Update Python Client to 3.1.8rc1

Author: vatsrahul1001

CHANGELOG.md

@@ -17,6 +17,15 @@
  under the License.
  -->
 
+# v3.1.8
+
+## Bug Fixes:
+
+ - Fix: 404 queued asset events from API server logs ([#62934](https://github.com/apache/airflow/pull/62934))
+ - Fix executor slots showing negative infinity ([#61140](https://github.com/apache/airflow/pull/61140))
+ - Fix Pool API slots validation ([#61071](https://github.com/apache/airflow/pull/61071))
+ - Add ``run_after`` alias to ``XComResponse`` for backward compatibility ([#61443](https://github.com/apache/airflow/pull/61443))
+
 # v3.1.6
 
 ## Bug Fixes:

README.md

@@ -19,22 +19,6 @@
 
 # Apache Airflow Python Client
 
-> [!NOTE]
-> Code in this repository is automatically generated by the [OpenAPI Generator](https://openapi-generator.tech)
-> project using Open-API specification from the Apache Airflow repository.
->
-> The process of generating the code is described in the
-> [Python Client Readme](https://github.com/apache/airflow/blob/main/clients/README.md).
-> We enabled [GitHub discussions](https://github.com/apache/airflow-client-python/discussions)
-> in the `airflow-client-python` repository, and we encourage you to start discussions if you have any questions
-> or suggestions to improve the client. However, in case the discussions result in a need to create an
-> actionable issuee, the issues in this repo are deliberately not enabled.
->
-> Instead, you should create GitHub issues or even PRs improving the client
-> in the main [Apache Airflow repository](https://github.com/apache/airflow) and test it by generating the
-> client locally following the instructions from the repo.
-
-
 # Overview
 
 To facilitate management, Apache Airflow supports a range of REST API endpoints across its
@@ -49,7 +33,6 @@ Content-type: application/json
 Accept: application/json
 ```
 
-
 ## Resources
 
 The term `resource` refers to a single type of object in the Airflow metadata. An API is broken up by its

airflow_client/client/__init__.py

@@ -14,7 +14,7 @@
 """  # noqa: E501
 
 
-__version__ = "3.1.6"
+__version__ = "3.1.8"
 
 # Define package exports
 __all__ = [
@@ -53,6 +53,10 @@
     "ApiKeyError",
     "ApiAttributeError",
     "ApiException",
+    "ActionsInner",
+    "ActionsInner1",
+    "ActionsInner2",
+    "ActionsInner3",
     "AppBuilderMenuItemResponse",
     "AppBuilderViewResponse",
     "AssetAliasCollectionResponse",
@@ -69,19 +73,14 @@
     "BulkActionOnExistence",
     "BulkActionResponse",
     "BulkBodyBulkTaskInstanceBody",
-    "BulkBodyBulkTaskInstanceBodyActionsInner",
     "BulkBodyConnectionBody",
-    "BulkBodyConnectionBodyActionsInner",
     "BulkBodyPoolBody",
-    "BulkBodyPoolBodyActionsInner",
     "BulkBodyVariableBody",
-    "BulkBodyVariableBodyActionsInner",
     "BulkCreateActionBulkTaskInstanceBody",
     "BulkCreateActionConnectionBody",
     "BulkCreateActionPoolBody",
     "BulkCreateActionVariableBody",
     "BulkDeleteActionBulkTaskInstanceBody",
-    "BulkDeleteActionBulkTaskInstanceBodyEntitiesInner",
     "BulkDeleteActionConnectionBody",
     "BulkDeleteActionPoolBody",
     "BulkDeleteActionVariableBody",
@@ -132,6 +131,7 @@
     "Detail",
     "DryRunBackfillCollectionResponse",
     "DryRunBackfillResponse",
+    "EntitiesInner",
     "EventLogCollectionResponse",
     "EventLogResponse",
     "ExternalLogUrlResponse",
@@ -151,6 +151,7 @@
     "JobCollectionResponse",
     "JobResponse",
     "LastAssetEventResponse",
+    "LocationInner",
     "PatchTaskInstanceBody",
     "PluginCollectionResponse",
     "PluginImportErrorCollectionResponse",
@@ -189,7 +190,6 @@
     "TriggererInfoResponse",
     "UpdateHITLDetailPayload",
     "ValidationError",
-    "ValidationErrorLocInner",
     "Value",
     "VariableBody",
     "VariableCollectionResponse",
@@ -243,6 +243,10 @@
 from airflow_client.client.exceptions import ApiException as ApiException
 
 # import models into sdk package
+from airflow_client.client.models.actions_inner import ActionsInner as ActionsInner
+from airflow_client.client.models.actions_inner1 import ActionsInner1 as ActionsInner1
+from airflow_client.client.models.actions_inner2 import ActionsInner2 as ActionsInner2
+from airflow_client.client.models.actions_inner3 import ActionsInner3 as ActionsInner3
 from airflow_client.client.models.app_builder_menu_item_response import AppBuilderMenuItemResponse as AppBuilderMenuItemResponse
 from airflow_client.client.models.app_builder_view_response import AppBuilderViewResponse as AppBuilderViewResponse
 from airflow_client.client.models.asset_alias_collection_response import AssetAliasCollectionResponse as AssetAliasCollectionResponse
@@ -259,19 +263,14 @@
 from airflow_client.client.models.bulk_action_on_existence import BulkActionOnExistence as BulkActionOnExistence
 from airflow_client.client.models.bulk_action_response import BulkActionResponse as BulkActionResponse
 from airflow_client.client.models.bulk_body_bulk_task_instance_body import BulkBodyBulkTaskInstanceBody as BulkBodyBulkTaskInstanceBody
-from airflow_client.client.models.bulk_body_bulk_task_instance_body_actions_inner import BulkBodyBulkTaskInstanceBodyActionsInner as BulkBodyBulkTaskInstanceBodyActionsInner
 from airflow_client.client.models.bulk_body_connection_body import BulkBodyConnectionBody as BulkBodyConnectionBody
-from airflow_client.client.models.bulk_body_connection_body_actions_inner import BulkBodyConnectionBodyActionsInner as BulkBodyConnectionBodyActionsInner
 from airflow_client.client.models.bulk_body_pool_body import BulkBodyPoolBody as BulkBodyPoolBody
-from airflow_client.client.models.bulk_body_pool_body_actions_inner import BulkBodyPoolBodyActionsInner as BulkBodyPoolBodyActionsInner
 from airflow_client.client.models.bulk_body_variable_body import BulkBodyVariableBody as BulkBodyVariableBody
-from airflow_client.client.models.bulk_body_variable_body_actions_inner import BulkBodyVariableBodyActionsInner as BulkBodyVariableBodyActionsInner
 from airflow_client.client.models.bulk_create_action_bulk_task_instance_body import BulkCreateActionBulkTaskInstanceBody as BulkCreateActionBulkTaskInstanceBody
 from airflow_client.client.models.bulk_create_action_connection_body import BulkCreateActionConnectionBody as BulkCreateActionConnectionBody
 from airflow_client.client.models.bulk_create_action_pool_body import BulkCreateActionPoolBody as BulkCreateActionPoolBody
 from airflow_client.client.models.bulk_create_action_variable_body import BulkCreateActionVariableBody as BulkCreateActionVariableBody
 from airflow_client.client.models.bulk_delete_action_bulk_task_instance_body import BulkDeleteActionBulkTaskInstanceBody as BulkDeleteActionBulkTaskInstanceBody
-from airflow_client.client.models.bulk_delete_action_bulk_task_instance_body_entities_inner import BulkDeleteActionBulkTaskInstanceBodyEntitiesInner as BulkDeleteActionBulkTaskInstanceBodyEntitiesInner
 from airflow_client.client.models.bulk_delete_action_connection_body import BulkDeleteActionConnectionBody as BulkDeleteActionConnectionBody
 from airflow_client.client.models.bulk_delete_action_pool_body import BulkDeleteActionPoolBody as BulkDeleteActionPoolBody
 from airflow_client.client.models.bulk_delete_action_variable_body import BulkDeleteActionVariableBody as BulkDeleteActionVariableBody
@@ -322,6 +321,7 @@
 from airflow_client.client.models.detail import Detail as Detail
 from airflow_client.client.models.dry_run_backfill_collection_response import DryRunBackfillCollectionResponse as DryRunBackfillCollectionResponse
 from airflow_client.client.models.dry_run_backfill_response import DryRunBackfillResponse as DryRunBackfillResponse
+from airflow_client.client.models.entities_inner import EntitiesInner as EntitiesInner
 from airflow_client.client.models.event_log_collection_response import EventLogCollectionResponse as EventLogCollectionResponse
 from airflow_client.client.models.event_log_response import EventLogResponse as EventLogResponse
 from airflow_client.client.models.external_log_url_response import ExternalLogUrlResponse as ExternalLogUrlResponse
@@ -341,6 +341,7 @@
 from airflow_client.client.models.job_collection_response import JobCollectionResponse as JobCollectionResponse
 from airflow_client.client.models.job_response import JobResponse as JobResponse
 from airflow_client.client.models.last_asset_event_response import LastAssetEventResponse as LastAssetEventResponse
+from airflow_client.client.models.location_inner import LocationInner as LocationInner
 from airflow_client.client.models.patch_task_instance_body import PatchTaskInstanceBody as PatchTaskInstanceBody
 from airflow_client.client.models.plugin_collection_response import PluginCollectionResponse as PluginCollectionResponse
 from airflow_client.client.models.plugin_import_error_collection_response import PluginImportErrorCollectionResponse as PluginImportErrorCollectionResponse
@@ -379,7 +380,6 @@
 from airflow_client.client.models.triggerer_info_response import TriggererInfoResponse as TriggererInfoResponse
 from airflow_client.client.models.update_hitl_detail_payload import UpdateHITLDetailPayload as UpdateHITLDetailPayload
 from airflow_client.client.models.validation_error import ValidationError as ValidationError
-from airflow_client.client.models.validation_error_loc_inner import ValidationErrorLocInner as ValidationErrorLocInner
 from airflow_client.client.models.value import Value as Value
 from airflow_client.client.models.variable_body import VariableBody as VariableBody
 from airflow_client.client.models.variable_collection_response import VariableCollectionResponse as VariableCollectionResponse

airflow_client/client/api/asset_api.py

@@ -1,5 +1,3 @@
-# coding: utf-8
-
 """
     Airflow API
 
@@ -11,6 +9,7 @@
     Do not edit the class manually.
 """  # noqa: E501
 
+
 import warnings
 from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union
@@ -2663,7 +2662,6 @@ def get_asset_queued_events(
             '200': "QueuedEventCollectionResponse",
             '401': "HTTPExceptionResponse",
             '403': "HTTPExceptionResponse",
-            '404': "HTTPExceptionResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -2738,7 +2736,6 @@ def get_asset_queued_events_with_http_info(
             '200': "QueuedEventCollectionResponse",
             '401': "HTTPExceptionResponse",
             '403': "HTTPExceptionResponse",
-            '404': "HTTPExceptionResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -2813,7 +2810,6 @@ def get_asset_queued_events_without_preload_content(
             '200': "QueuedEventCollectionResponse",
             '401': "HTTPExceptionResponse",
             '403': "HTTPExceptionResponse",
-            '404': "HTTPExceptionResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -3640,7 +3636,6 @@ def get_dag_asset_queued_events(
             '200': "QueuedEventCollectionResponse",
             '401': "HTTPExceptionResponse",
             '403': "HTTPExceptionResponse",
-            '404': "HTTPExceptionResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -3715,7 +3710,6 @@ def get_dag_asset_queued_events_with_http_info(
             '200': "QueuedEventCollectionResponse",
             '401': "HTTPExceptionResponse",
             '403': "HTTPExceptionResponse",
-            '404': "HTTPExceptionResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -3790,7 +3784,6 @@ def get_dag_asset_queued_events_without_preload_content(
             '200': "QueuedEventCollectionResponse",
             '401': "HTTPExceptionResponse",
             '403': "HTTPExceptionResponse",
-            '404': "HTTPExceptionResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(

airflow_client/client/api/backfill_api.py

@@ -1,5 +1,3 @@
-# coding: utf-8
-
 """
     Airflow API
 
@@ -11,6 +9,7 @@
     Do not edit the class manually.
 """  # noqa: E501
 
+
 import warnings
 from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union

airflow_client/client/api/config_api.py

@@ -1,5 +1,3 @@
-# coding: utf-8
-
 """
     Airflow API
 
@@ -11,6 +9,7 @@
     Do not edit the class manually.
 """  # noqa: E501
 
+
 import warnings
 from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union

airflow_client/client/api/connection_api.py

@@ -1,5 +1,3 @@
-# coding: utf-8
-
 """
     Airflow API
 
@@ -11,6 +9,7 @@
     Do not edit the class manually.
 """  # noqa: E501
 
+
 import warnings
 from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union

airflow_client/client/api/dag_api.py

@@ -1,5 +1,3 @@
-# coding: utf-8
-
 """
     Airflow API
 
@@ -11,6 +9,7 @@
     Do not edit the class manually.
 """  # noqa: E501
 
+
 import warnings
 from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union

airflow_client/client/api/dag_parsing_api.py

@@ -1,5 +1,3 @@
-# coding: utf-8
-
 """
     Airflow API
 
@@ -11,6 +9,7 @@
     Do not edit the class manually.
 """  # noqa: E501
 
+
 import warnings
 from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union

airflow_client/client/api/dag_run_api.py

@@ -1,5 +1,3 @@
-# coding: utf-8
-
 """
     Airflow API
 
@@ -11,13 +9,14 @@
     Do not edit the class manually.
 """  # noqa: E501
 
+
 import warnings
 from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union
 from typing_extensions import Annotated
 
 from datetime import datetime
-from pydantic import Field, StrictFloat, StrictInt, StrictStr, field_validator
+from pydantic import Field, StrictInt, StrictStr, field_validator
 from typing import Any, List, Optional, Union
 from typing_extensions import Annotated
 from airflow_client.client.models.asset_event_collection_response import AssetEventCollectionResponse
@@ -3115,7 +3114,7 @@ def wait_dag_run_until_finished(
         self,
         dag_id: StrictStr,
         dag_run_id: StrictStr,
-        interval: Annotated[Union[StrictFloat, StrictInt], Field(description="Seconds to wait between dag run state checks")],
+        interval: Annotated[Union[Annotated[float, Field(strict=True, gt=0.0)], Annotated[int, Field(strict=True, gt=0)]], Field(description="Seconds to wait between dag run state checks")],
         result: Annotated[Optional[List[StrictStr]], Field(description="Collect result XCom from task. Can be set multiple times.")] = None,
         _request_timeout: Union[
             None,
@@ -3198,7 +3197,7 @@ def wait_dag_run_until_finished_with_http_info(
         self,
         dag_id: StrictStr,
         dag_run_id: StrictStr,
-        interval: Annotated[Union[StrictFloat, StrictInt], Field(description="Seconds to wait between dag run state checks")],
+        interval: Annotated[Union[Annotated[float, Field(strict=True, gt=0.0)], Annotated[int, Field(strict=True, gt=0)]], Field(description="Seconds to wait between dag run state checks")],
         result: Annotated[Optional[List[StrictStr]], Field(description="Collect result XCom from task. Can be set multiple times.")] = None,
         _request_timeout: Union[
             None,
@@ -3281,7 +3280,7 @@ def wait_dag_run_until_finished_without_preload_content(
         self,
         dag_id: StrictStr,
         dag_run_id: StrictStr,
-        interval: Annotated[Union[StrictFloat, StrictInt], Field(description="Seconds to wait between dag run state checks")],
+        interval: Annotated[Union[Annotated[float, Field(strict=True, gt=0.0)], Annotated[int, Field(strict=True, gt=0)]], Field(description="Seconds to wait between dag run state checks")],
         result: Annotated[Optional[List[StrictStr]], Field(description="Collect result XCom from task. Can be set multiple times.")] = None,
         _request_timeout: Union[
             None,