Update Python Client to 3.0.2 (#136)

Author: kaxil

CHANGELOG.md

@@ -17,6 +17,24 @@
  under the License.
  -->
 
+# v3.0.2
+
+## Major changes:
+
+- Add `owner_links` field to DAGDetailsResponse ([#50557](https://github.com/apache/airflow/pull/50557))
+- Allow non-string valid JSON values in Variable import ([#49844](https://github.com/apache/airflow/pull/49844))
+- Add `bundle_version` to DagRun response ([#49726](https://github.com/apache/airflow/pull/49726))
+- Use `NonNegativeInt` for `backfill_id` ([#49691](https://github.com/apache/airflow/pull/49691))
+- Rename operation IDs for task instance endpoints to include map indexes ([#49608](https://github.com/apache/airflow/pull/49608))
+- Remove filtering by last dag run state in patch dags endpoint ([#51176](https://github.com/apache/airflow/pull/51176))
+- Make `dag_run` nullable in Details page ([#50719](https://github.com/apache/airflow/pull/50719))
+
+## Bug Fixes
+
+- Fix OpenAPI schema for `get_log` API ([#50547](https://github.com/apache/airflow/pull/50547))
+- Fix bulk action annotation ([#50852](https://github.com/apache/airflow/pull/50852))
+- Fix `patch_task_instance` endpoint ([#50550](https://github.com/apache/airflow/pull/50550))
+
 # v3.0.0
 
 This is the first release of the **Airflow 3.0.0** Python client. It introduces compatibility with the new [Airflow 3.0 REST API](https://airflow.apache.org/docs/apache-airflow/3.0.0/stable-rest-api-ref.html), and includes several **breaking changes** and behavior updates.

airflow_client/client/__init__.py

@@ -14,7 +14,7 @@
 """  # noqa: E501
 
 
-__version__ = "3.0.0"
+__version__ = "3.0.2"
 
 # import apis into sdk package
 from airflow_client.client.api.asset_api import AssetApi
@@ -68,7 +68,6 @@
 from airflow_client.client.models.backfill_post_body import BackfillPostBody
 from airflow_client.client.models.backfill_response import BackfillResponse
 from airflow_client.client.models.base_info_response import BaseInfoResponse
-from airflow_client.client.models.bulk_action import BulkAction
 from airflow_client.client.models.bulk_action_not_on_existence import BulkActionNotOnExistence
 from airflow_client.client.models.bulk_action_on_existence import BulkActionOnExistence
 from airflow_client.client.models.bulk_action_response import BulkActionResponse

airflow_client/client/api/dag_api.py

@@ -2097,7 +2097,6 @@ def patch_dags(
         dag_id_pattern: Optional[StrictStr] = None,
         exclude_stale: Optional[StrictBool] = None,
         paused: Optional[StrictBool] = None,
-        last_dag_run_state: Optional[DagRunState] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -2135,8 +2134,6 @@ def patch_dags(
         :type exclude_stale: bool
         :param paused:
         :type paused: bool
-        :param last_dag_run_state:
-        :type last_dag_run_state: DagRunState
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -2170,7 +2167,6 @@ def patch_dags(
             dag_id_pattern=dag_id_pattern,
             exclude_stale=exclude_stale,
             paused=paused,
-            last_dag_run_state=last_dag_run_state,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -2209,7 +2205,6 @@ def patch_dags_with_http_info(
         dag_id_pattern: Optional[StrictStr] = None,
         exclude_stale: Optional[StrictBool] = None,
         paused: Optional[StrictBool] = None,
-        last_dag_run_state: Optional[DagRunState] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -2247,8 +2242,6 @@ def patch_dags_with_http_info(
         :type exclude_stale: bool
         :param paused:
         :type paused: bool
-        :param last_dag_run_state:
-        :type last_dag_run_state: DagRunState
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -2282,7 +2275,6 @@ def patch_dags_with_http_info(
             dag_id_pattern=dag_id_pattern,
             exclude_stale=exclude_stale,
             paused=paused,
-            last_dag_run_state=last_dag_run_state,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -2321,7 +2313,6 @@ def patch_dags_without_preload_content(
         dag_id_pattern: Optional[StrictStr] = None,
         exclude_stale: Optional[StrictBool] = None,
         paused: Optional[StrictBool] = None,
-        last_dag_run_state: Optional[DagRunState] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -2359,8 +2350,6 @@ def patch_dags_without_preload_content(
         :type exclude_stale: bool
         :param paused:
         :type paused: bool
-        :param last_dag_run_state:
-        :type last_dag_run_state: DagRunState
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -2394,7 +2383,6 @@ def patch_dags_without_preload_content(
             dag_id_pattern=dag_id_pattern,
             exclude_stale=exclude_stale,
             paused=paused,
-            last_dag_run_state=last_dag_run_state,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -2428,7 +2416,6 @@ def _patch_dags_serialize(
         dag_id_pattern,
         exclude_stale,
         paused,
-        last_dag_run_state,
         _request_auth,
         _content_type,
         _headers,
@@ -2490,10 +2477,6 @@ def _patch_dags_serialize(
 
             _query_params.append(('paused', paused))
 
-        if last_dag_run_state is not None:
-            
-            _query_params.append(('last_dag_run_state', last_dag_run_state.value))
-            
         # process the header parameters
         # process the form parameters
         # process the body parameter

airflow_client/client/api/task_instance_api.py

@@ -726,7 +726,7 @@ def _get_log_serialize(
             _header_params['Accept'] = self.api_client.select_header_accept(
                 [
                     'application/json', 
-                    'text/plain'
+                    'application/x-ndjson'
                 ]
             )
 

airflow_client/client/api_client.py

@@ -90,7 +90,7 @@ def __init__(
             self.default_headers[header_name] = header_value
         self.cookie = cookie
         # Set default User-Agent.
-        self.user_agent = 'OpenAPI-Generator/3.0.0/python'
+        self.user_agent = 'OpenAPI-Generator/3.0.2/python'
         self.client_side_validation = configuration.client_side_validation
 
     def __enter__(self):

airflow_client/client/configuration.py

@@ -510,7 +510,7 @@ def to_debug_report(self) -> str:
                "OS: {env}\n"\
                "Python Version: {pyversion}\n"\
                "Version of the API: 2\n"\
-               "SDK Package Version: 3.0.0".\
+               "SDK Package Version: 3.0.2".\
                format(env=sys.platform, pyversion=sys.version)
 
     def get_host_settings(self) -> List[HostSetting]:

airflow_client/client/models/__init__.py

@@ -26,7 +26,6 @@
 from airflow_client.client.models.backfill_post_body import BackfillPostBody
 from airflow_client.client.models.backfill_response import BackfillResponse
 from airflow_client.client.models.base_info_response import BaseInfoResponse
-from airflow_client.client.models.bulk_action import BulkAction
 from airflow_client.client.models.bulk_action_not_on_existence import BulkActionNotOnExistence
 from airflow_client.client.models.bulk_action_on_existence import BulkActionOnExistence
 from airflow_client.client.models.bulk_action_response import BulkActionResponse

airflow_client/client/models/bulk_action.py

@@ -17,9 +17,8 @@
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field
+from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
-from airflow_client.client.models.bulk_action import BulkAction
 from airflow_client.client.models.bulk_action_on_existence import BulkActionOnExistence
 from airflow_client.client.models.connection_body import ConnectionBody
 from typing import Optional, Set
@@ -29,11 +28,18 @@ class BulkCreateActionConnectionBody(BaseModel):
     """
     BulkCreateActionConnectionBody
     """ # noqa: E501
-    action: BulkAction = Field(description="The action to be performed on the entities.")
+    action: StrictStr = Field(description="The action to be performed on the entities.")
     action_on_existence: Optional[BulkActionOnExistence] = None
     entities: List[ConnectionBody] = Field(description="A list of entities to be created.")
     __properties: ClassVar[List[str]] = ["action", "action_on_existence", "entities"]
 
+    @field_validator('action')
+    def action_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['create']):
+            raise ValueError("must be one of enum values ('create')")
+        return value
+
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

airflow_client/client/models/bulk_create_action_connection_body.py

@@ -17,9 +17,8 @@
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field
+from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
-from airflow_client.client.models.bulk_action import BulkAction
 from airflow_client.client.models.bulk_action_on_existence import BulkActionOnExistence
 from airflow_client.client.models.pool_body import PoolBody
 from typing import Optional, Set
@@ -29,11 +28,18 @@ class BulkCreateActionPoolBody(BaseModel):
     """
     BulkCreateActionPoolBody
     """ # noqa: E501
-    action: BulkAction = Field(description="The action to be performed on the entities.")
+    action: StrictStr = Field(description="The action to be performed on the entities.")
     action_on_existence: Optional[BulkActionOnExistence] = None
     entities: List[PoolBody] = Field(description="A list of entities to be created.")
     __properties: ClassVar[List[str]] = ["action", "action_on_existence", "entities"]
 
+    @field_validator('action')
+    def action_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['create']):
+            raise ValueError("must be one of enum values ('create')")
+        return value
+
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,