kestrapy 2.1.0.dev541__tar.gz → 2.1.0.dev568__tar.gz

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kestrapy
-Version: 2.1.0.dev541
+Version: 2.1.0.dev568
 Summary: Kestra EE
 Home-page: https://github.com/kestra-io/client-sdk
 Author: Kestra

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/kestrapy/__init__.py

@@ -9,7 +9,7 @@
 """  # noqa: E501
 
 
-__version__ = "2.1.0.dev541"
+__version__ = "2.1.0.dev568"
 
 # Define package exports
 __all__ = [
@@ -310,6 +310,7 @@ __all__ = [
     "FlowUsageTaskUsage",
     "FlowUsageTriggerUsage",
     "FlowWithSource",
+    "FlowWorkerSelector",
     "FollowLogEvent",
     "ForwardSupportTicketRequest",
     "GateOutcome",
@@ -1012,6 +1013,7 @@ from kestrapy.models.flow_usage import FlowUsage as FlowUsage
 from kestrapy.models.flow_usage_task_usage import FlowUsageTaskUsage as FlowUsageTaskUsage
 from kestrapy.models.flow_usage_trigger_usage import FlowUsageTriggerUsage as FlowUsageTriggerUsage
 from kestrapy.models.flow_with_source import FlowWithSource as FlowWithSource
+from kestrapy.models.flow_worker_selector import FlowWorkerSelector as FlowWorkerSelector
 from kestrapy.models.follow_log_event import FollowLogEvent as FollowLogEvent
 from kestrapy.models.forward_support_ticket_request import ForwardSupportTicketRequest as ForwardSupportTicketRequest
 from kestrapy.models.gate_outcome import GateOutcome as GateOutcome

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/kestrapy/configuration.py

@@ -530,7 +530,7 @@ conf = kestrapy.Configuration(
                "OS: {env}\n"\
                "Python Version: {pyversion}\n"\
                "Version of the API: 2.0.0-SNAPSHOT\n"\
-               "SDK Package Version: 2.1.0.dev541".\
+               "SDK Package Version: 2.1.0.dev568".\
                format(env=sys.platform, pyversion=sys.version)
 
     def get_host_settings(self) -> List[HostSetting]:

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/kestrapy/models/__init__.py

@@ -285,6 +285,7 @@ from kestrapy.models.flow_usage import FlowUsage
 from kestrapy.models.flow_usage_task_usage import FlowUsageTaskUsage
 from kestrapy.models.flow_usage_trigger_usage import FlowUsageTriggerUsage
 from kestrapy.models.flow_with_source import FlowWithSource
+from kestrapy.models.flow_worker_selector import FlowWorkerSelector
 from kestrapy.models.follow_log_event import FollowLogEvent
 from kestrapy.models.forward_support_ticket_request import ForwardSupportTicketRequest
 from kestrapy.models.gate_outcome import GateOutcome

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/kestrapy/models/abstract_flow.py

@@ -19,7 +19,7 @@ from typing_extensions import Annotated
 from kestrapy.models.input_object import InputObject
 from kestrapy.models.label import Label
 from kestrapy.models.output import Output
-from kestrapy.models.worker_selector import WorkerSelector
+from kestrapy.models.flow_worker_selector import FlowWorkerSelector
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -38,7 +38,7 @@ class AbstractFlow(BaseModel):
     draft: StrictBool = Field(description="Whether this flow revision is a draft. Draft revisions are skipped when an execution starts without an explicit revision (webhooks, schedules, subflows, manual triggers). Executions can still target a draft by passing the revision explicitly.")
     labels: Optional[List[Label]] = None
     variables: Optional[Dict[str, Any]] = None
-    worker_selector: Optional[WorkerSelector] = Field(default=None, description="Routing requirements (tags + fallback) for this flow.", alias="workerSelector")
+    worker_selector: Optional[FlowWorkerSelector] = Field(default=None, description="Routing requirements (tags + fallback) for this flow.", alias="workerSelector")
     deleted: StrictBool
     additional_properties: Dict[str, Any] = {}
     __properties: ClassVar[List[str]] = ["id", "namespace", "revision", "updated", "description", "inputs", "outputs", "disabled", "draft", "labels", "variables", "workerSelector", "deleted"]
@@ -150,7 +150,7 @@ class AbstractFlow(BaseModel):
             "draft": obj.get("draft"),
             "labels": [Label.from_dict(_item) for _item in obj["labels"]] if obj.get("labels") is not None else None,
             "variables": obj.get("variables"),
-            "workerSelector": WorkerSelector.from_dict(obj["workerSelector"]) if obj.get("workerSelector") is not None else None,
+            "workerSelector": FlowWorkerSelector.from_dict(obj["workerSelector"]) if obj.get("workerSelector") is not None else None,
             "deleted": obj.get("deleted")
         })
         # store additional fields in additional_properties

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/kestrapy/models/abstract_trigger.py

@@ -19,7 +19,7 @@ from kestrapy.models.assets_declaration import AssetsDeclaration
 from kestrapy.models.level import Level
 from kestrapy.models.state_type import StateType
 from kestrapy.models.the_labels_to_pass_to_the_execution_created import TheLabelsToPassToTheExecutionCreated
-from kestrapy.models.worker_selector import WorkerSelector
+from kestrapy.models.flow_worker_selector import FlowWorkerSelector
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -36,7 +36,7 @@ class AbstractTrigger(BaseModel):
     # every real trigger payload (dropping plugin props like `cron` downstream).
     when: Optional[StrictStr] = Field(default=None, description="A Pebble expression evaluated at trigger time. The trigger fires only when the expression evaluates to a truthy value (`true`, a non-empty string, a non-zero number). Use this to gate trigger execution on dynamic runtime values such as execution labels, flow variables, or environment conditions.")
     disabled: Optional[StrictBool] = False
-    worker_selector: Optional[WorkerSelector] = Field(default=None, description="Routing requirements (tags + fallback) for this trigger.", alias="workerSelector")
+    worker_selector: Optional[FlowWorkerSelector] = Field(default=None, description="Routing requirements (tags + fallback) for this trigger.", alias="workerSelector")
     policy_refs: Optional[List[StrictStr]] = Field(default=None, description="Identifiers of `enforcement: REFERENCE` governance policies to attach to this trigger and everything nested under it (Enterprise Edition; ignored in the open-source edition).", alias="policyRefs")
     log_level: Optional[Level] = Field(default=None, alias="logLevel")
     labels: Optional[TheLabelsToPassToTheExecutionCreated] = None
@@ -135,7 +135,7 @@ class AbstractTrigger(BaseModel):
             "description": obj.get("description"),
             "when": obj.get("when"),
             "disabled": obj.get("disabled") if obj.get("disabled") is not None else False,
-            "workerSelector": WorkerSelector.from_dict(obj["workerSelector"]) if obj.get("workerSelector") is not None else None,
+            "workerSelector": FlowWorkerSelector.from_dict(obj["workerSelector"]) if obj.get("workerSelector") is not None else None,
             "policyRefs": obj.get("policyRefs"),
             "logLevel": obj.get("logLevel"),
             "labels": TheLabelsToPassToTheExecutionCreated.from_dict(obj["labels"]) if obj.get("labels") is not None else None,

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/kestrapy/models/flow.py

@@ -25,7 +25,7 @@ from kestrapy.models.output import Output
 from kestrapy.models.quota import Quota
 from kestrapy.models.sla import SLA
 from kestrapy.models.task import Task
-from kestrapy.models.worker_selector import WorkerSelector
+from kestrapy.models.flow_worker_selector import FlowWorkerSelector
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -44,7 +44,7 @@ class Flow(BaseModel):
     draft: StrictBool = Field(description="Whether this flow revision is a draft. Draft revisions are skipped when an execution starts without an explicit revision (webhooks, schedules, subflows, manual triggers). Executions can still target a draft by passing the revision explicitly.")
     labels: Optional[List[Label]] = None
     variables: Optional[Dict[str, Any]] = None
-    worker_selector: Optional[WorkerSelector] = Field(default=None, description="Routing requirements (tags + fallback) for this flow.", alias="workerSelector")
+    worker_selector: Optional[FlowWorkerSelector] = Field(default=None, description="Routing requirements (tags + fallback) for this flow.", alias="workerSelector")
     deleted: StrictBool
     var_finally: Optional[List[Task]] = Field(default=None, alias="finally")
     tasks: Annotated[List[Task], Field(min_length=1)]
@@ -226,7 +226,7 @@ class Flow(BaseModel):
             "draft": obj.get("draft"),
             "labels": [Label.from_dict(_item) for _item in obj["labels"]] if obj.get("labels") is not None else None,
             "variables": obj.get("variables"),
-            "workerSelector": WorkerSelector.from_dict(obj["workerSelector"]) if obj.get("workerSelector") is not None else None,
+            "workerSelector": FlowWorkerSelector.from_dict(obj["workerSelector"]) if obj.get("workerSelector") is not None else None,
             "deleted": obj.get("deleted"),
             "finally": [Task.from_dict(_item) for _item in obj["finally"]] if obj.get("finally") is not None else None,
             "tasks": [Task.from_dict(_item) for _item in obj["tasks"]] if obj.get("tasks") is not None else None,

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/kestrapy/models/flow_for_execution.py

@@ -21,7 +21,7 @@ from kestrapy.models.input_object import InputObject
 from kestrapy.models.map_object_object import MapObjectObject
 from kestrapy.models.output import Output
 from kestrapy.models.task_for_execution import TaskForExecution
-from kestrapy.models.worker_selector import WorkerSelector
+from kestrapy.models.flow_worker_selector import FlowWorkerSelector
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -40,7 +40,7 @@ class FlowForExecution(BaseModel):
     draft: StrictBool = Field(description="Whether this flow revision is a draft. Draft revisions are skipped when an execution starts without an explicit revision (webhooks, schedules, subflows, manual triggers). Executions can still target a draft by passing the revision explicitly.")
     labels: Optional[MapObjectObject] = Field(default=None, description="Labels as a list of Label (key/value pairs) or as a map of string to string.")
     variables: Optional[Dict[str, Any]] = None
-    worker_selector: Optional[WorkerSelector] = Field(default=None, description="Routing requirements (tags + fallback) for this flow.", alias="workerSelector")
+    worker_selector: Optional[FlowWorkerSelector] = Field(default=None, description="Routing requirements (tags + fallback) for this flow.", alias="workerSelector")
     deleted: StrictBool
     tasks: Annotated[List[TaskForExecution], Field(min_length=1)]
     errors: Optional[List[TaskForExecution]] = None
@@ -188,7 +188,7 @@ class FlowForExecution(BaseModel):
             "draft": obj.get("draft"),
             "labels": MapObjectObject.from_dict(obj["labels"]) if obj.get("labels") is not None else None,
             "variables": obj.get("variables"),
-            "workerSelector": WorkerSelector.from_dict(obj["workerSelector"]) if obj.get("workerSelector") is not None else None,
+            "workerSelector": FlowWorkerSelector.from_dict(obj["workerSelector"]) if obj.get("workerSelector") is not None else None,
             "deleted": obj.get("deleted"),
             "tasks": [TaskForExecution.from_dict(_item) for _item in obj["tasks"]] if obj.get("tasks") is not None else None,
             "errors": [TaskForExecution.from_dict(_item) for _item in obj["errors"]] if obj.get("errors") is not None else None,

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/kestrapy/models/flow_interface.py

@@ -20,7 +20,7 @@ from kestrapy.models.label import Label
 from kestrapy.models.output import Output
 from kestrapy.models.quota import Quota
 from kestrapy.models.sla import SLA
-from kestrapy.models.worker_selector import WorkerSelector
+from kestrapy.models.flow_worker_selector import FlowWorkerSelector
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -40,7 +40,7 @@ class FlowInterface(BaseModel):
     inputs: Optional[List[InputObject]] = None
     outputs: Optional[List[Output]] = None
     variables: Optional[Dict[str, Dict[str, Any]]] = None
-    worker_selector: Optional[WorkerSelector] = Field(default=None, alias="workerSelector")
+    worker_selector: Optional[FlowWorkerSelector] = Field(default=None, alias="workerSelector")
     concurrency: Optional[Concurrency] = None
     quotas: Optional[List[Quota]] = None
     sla: Optional[List[SLA]] = None
@@ -159,7 +159,7 @@ class FlowInterface(BaseModel):
             "inputs": [InputObject.from_dict(_item) for _item in obj["inputs"]] if obj.get("inputs") is not None else None,
             "outputs": [Output.from_dict(_item) for _item in obj["outputs"]] if obj.get("outputs") is not None else None,
             "variables": obj.get("variables"),
-            "workerSelector": WorkerSelector.from_dict(obj["workerSelector"]) if obj.get("workerSelector") is not None else None,
+            "workerSelector": FlowWorkerSelector.from_dict(obj["workerSelector"]) if obj.get("workerSelector") is not None else None,
             "concurrency": Concurrency.from_dict(obj["concurrency"]) if obj.get("concurrency") is not None else None,
             "quotas": [Quota.from_dict(_item) for _item in obj["quotas"]] if obj.get("quotas") is not None else None,
             "sla": [SLA.from_dict(_item) for _item in obj["sla"]] if obj.get("sla") is not None else None,

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/kestrapy/models/flow_with_source.py

@@ -25,7 +25,7 @@ from kestrapy.models.output import Output
 from kestrapy.models.quota import Quota
 from kestrapy.models.sla import SLA
 from kestrapy.models.task import Task
-from kestrapy.models.worker_selector import WorkerSelector
+from kestrapy.models.flow_worker_selector import FlowWorkerSelector
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -44,7 +44,7 @@ class FlowWithSource(BaseModel):
     draft: StrictBool = Field(description="Whether this flow revision is a draft. Draft revisions are skipped when an execution starts without an explicit revision (webhooks, schedules, subflows, manual triggers). Executions can still target a draft by passing the revision explicitly.")
     labels: Optional[List[Label]] = None
     variables: Optional[Dict[str, Any]] = None
-    worker_selector: Optional[WorkerSelector] = Field(default=None, description="Routing requirements (tags + fallback) for this flow.", alias="workerSelector")
+    worker_selector: Optional[FlowWorkerSelector] = Field(default=None, description="Routing requirements (tags + fallback) for this flow.", alias="workerSelector")
     deleted: StrictBool
     var_finally: Optional[List[Task]] = Field(default=None, alias="finally")
     tasks: Annotated[List[Task], Field(min_length=1)]
@@ -227,7 +227,7 @@ class FlowWithSource(BaseModel):
             "draft": obj.get("draft"),
             "labels": [Label.from_dict(_item) for _item in obj["labels"]] if obj.get("labels") is not None else None,
             "variables": obj.get("variables"),
-            "workerSelector": WorkerSelector.from_dict(obj["workerSelector"]) if obj.get("workerSelector") is not None else None,
+            "workerSelector": FlowWorkerSelector.from_dict(obj["workerSelector"]) if obj.get("workerSelector") is not None else None,
             "deleted": obj.get("deleted"),
             "finally": [Task.from_dict(_item) for _item in obj["finally"]] if obj.get("finally") is not None else None,
             "tasks": [Task.from_dict(_item) for _item in obj["tasks"]] if obj.get("tasks") is not None else None,

kestrapy-2.1.0.dev568/kestrapy/models/flow_worker_selector.py

@@ -0,0 +1,104 @@
+# coding: utf-8
+
+"""
+    Kestra EE
+
+    All API operations, except for Instance-owner-only endpoints, require a tenant identifier in the HTTP path.<br/> Endpoints designated as Instance-owner-only are not tenant-scoped.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import regex as re
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from kestrapy.models.worker_queue_fallback import WorkerQueueFallback
+from kestrapy.models.worker_selector_match import WorkerSelectorMatch
+from typing import Optional, Set
+from typing_extensions import Self
+
+class FlowWorkerSelector(BaseModel):
+    """
+    Routing requirements for a Flow, Task or Trigger. Unlike WorkerSelector (namespace and
+    tenant defaults), ``tags`` is a Kestra property: a JSON array literal such as ``["gpu"]``
+    or a Pebble expression rendered when the job is routed. The server returns it as a
+    string in both cases.
+    """ # noqa: E501
+    tags: Optional[StrictStr] = Field(default=None, description="Required tags used to route to a matching Worker Queue, as a JSON array literal (e.g. [\"gpu\"]) or a Pebble expression rendering to the tags. Each tag is an RFC 1123 label, max 63 chars per tag, max 20 tags; a tag rendering to an empty value is ignored.")
+    match: Optional[WorkerSelectorMatch] = Field(default=None, description="How selector tags are matched against a Worker Queue's tag set. ALL: queue tags must be a superset of the selector tags. ANY: queue tags must intersect the selector tags. Defaults to ALL when null.")
+    fallback: Optional[WorkerQueueFallback] = Field(default=None, description="Strategy when no worker is available. Defaults to FAIL when null.")
+    additional_properties: Dict[str, Any] = {}
+    __properties: ClassVar[List[str]] = ["tags", "match", "fallback"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of FlowWorkerSelector from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        * Fields in `self.additional_properties` are added to the output dict.
+        """
+        excluded_fields: Set[str] = set([
+            "additional_properties",
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # puts key-value pairs in additional_properties in the top level
+        if self.additional_properties is not None:
+            for _key, _value in self.additional_properties.items():
+                _dict[_key] = _value
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of FlowWorkerSelector from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "tags": obj.get("tags"),
+            "match": obj.get("match"),
+            "fallback": obj.get("fallback")
+        })
+        # store additional fields in additional_properties
+        for _key in obj.keys():
+            if _key not in cls.__properties:
+                _obj.additional_properties[_key] = obj.get(_key)
+
+        return _obj
+
+

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/kestrapy/models/task.py

@@ -19,7 +19,7 @@ from kestrapy.models.assets_declaration import AssetsDeclaration
 from kestrapy.models.cache import Cache
 from kestrapy.models.level import Level
 from kestrapy.models.property_duration import PropertyDuration
-from kestrapy.models.worker_selector import WorkerSelector
+from kestrapy.models.flow_worker_selector import FlowWorkerSelector
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -34,7 +34,7 @@ class Task(BaseModel):
     retry: Optional[Dict[str, Any]] = Field(default=None, description="Retry policy applied when the task fails.")
     timeout: Optional[PropertyDuration] = None
     disabled: Optional[StrictBool] = None
-    worker_selector: Optional[WorkerSelector] = Field(default=None, description="Routing requirements (tags + fallback) for this task.", alias="workerSelector")
+    worker_selector: Optional[FlowWorkerSelector] = Field(default=None, description="Routing requirements (tags + fallback) for this task.", alias="workerSelector")
     policy_refs: Optional[List[StrictStr]] = Field(default=None, description="Identifiers of `enforcement: REFERENCE` governance policies to attach to this task and everything nested under it (Enterprise Edition; ignored in the open-source edition).", alias="policyRefs")
     log_level: Optional[Level] = Field(default=None, alias="logLevel")
     allow_failure: Optional[StrictBool] = Field(default=None, alias="allowFailure")
@@ -142,7 +142,7 @@ class Task(BaseModel):
             "retry": obj.get("retry"),
             "timeout": PropertyDuration.from_dict(obj["timeout"]) if obj.get("timeout") is not None else None,
             "disabled": obj.get("disabled"),
-            "workerSelector": WorkerSelector.from_dict(obj["workerSelector"]) if obj.get("workerSelector") is not None else None,
+            "workerSelector": FlowWorkerSelector.from_dict(obj["workerSelector"]) if obj.get("workerSelector") is not None else None,
             "policyRefs": obj.get("policyRefs"),
             "logLevel": obj.get("logLevel"),
             "allowFailure": obj.get("allowFailure"),

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/kestrapy.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kestrapy
-Version: 2.1.0.dev541
+Version: 2.1.0.dev568
 Summary: Kestra EE
 Home-page: https://github.com/kestra-io/client-sdk
 Author: Kestra

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/kestrapy.egg-info/SOURCES.txt

@@ -341,6 +341,7 @@ kestrapy/models/flow_usage.py
 kestrapy/models/flow_usage_task_usage.py
 kestrapy/models/flow_usage_trigger_usage.py
 kestrapy/models/flow_with_source.py
+kestrapy/models/flow_worker_selector.py
 kestrapy/models/follow_log_event.py
 kestrapy/models/forward_support_ticket_request.py
 kestrapy/models/gate_outcome.py
@@ -784,7 +785,9 @@ testApis/test_test_suites_api.py
 testApis/test_triggers_api.py
 testApis/test_users_api.py
 tests/test_asset_relation_requests.py
+tests/test_flow_worker_selector.py
 tests/test_flow_yaml.py
+tests/test_kv_api.py
 tests/test_model_types.py
 tests/test_query_filter_golden.py
 tests/test_replay_with_inputs_empty_body.py

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/pyproject.toml

@@ -1,6 +1,6 @@
     [project]
 name = "kestrapy"
-version = "2.1.0.dev541"
+version = "2.1.0.dev568"
 description = "Kestra EE"
     authors = [
     {name = "Kestra",email = "hello@kestra.io"},

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/setup.py

@@ -19,7 +19,7 @@ from setuptools import setup, find_packages  # noqa: H301
 # prerequisite: setuptools
 # http://pypi.python.org/pypi/setuptools
 NAME = "kestrapy"
-VERSION = "2.1.0.dev541"
+VERSION = "2.1.0.dev568"
 PYTHON_REQUIRES = ">= 3.9"
 REQUIRES = [
 "requests (>= 2.32.5)",

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/testApis/test_flows_api.py

@@ -13,6 +13,7 @@ from kestrapy import (
     QueryFilterField,
     QueryFilterOp,
     StateType,
+    WorkerSelectorMatch,
 )
 from test_helpers import (
     TENANT,
@@ -53,6 +54,31 @@ class TestCreateFlow:
         assert result.deleted is False
         assert result.tasks is not None and len(result.tasks) > 0
 
+    def test_worker_selector(self, client):
+        fid = random_id()
+        ns = random_namespace()
+        body = f"""\
+id: {fid}
+namespace: {ns}
+workerSelector:
+  tags:
+    - gpu
+  match: ALL
+tasks:
+  - id: h
+    type: io.kestra.plugin.core.log.Log
+    message: hi
+"""
+        created = client.flows.create_flow(TENANT, body)
+
+        # tags is a Pebble-capable property on flows: a literal list comes back as its JSON text
+        assert created.worker_selector.tags == '["gpu"]'
+        assert created.worker_selector.match == WorkerSelectorMatch.ALL
+        assert created.draft is False
+
+        fetched = client.flows.flow(ns, fid, TENANT)
+        assert fetched.worker_selector.tags == '["gpu"]'
+
     @pytest.mark.xfail(
         reason="kestra-ee 2.0 validates Schedule-trigger inputs more strictly and "
         "rejects the shared complete flow (422 / 'Missing inputs for Schedule "

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/testApis/test_instance_family_api.py

@@ -231,7 +231,8 @@ def test_kill_switch_create_search_delete_roundtrip(client):
     ks_name = "ks-" + random_id()[:8]
     with _tolerate_gating("create_kill_switch"):
         created = client.kill_switches.create_kill_switch(
-            KillSwitch(name=ks_name, startDate=start, evaluationType=EvaluationType.KILL)
+            # Kestra requires enabled (422 when null); False keeps the switch inert for other tests.
+            KillSwitch(name=ks_name, startDate=start, evaluationType=EvaluationType.KILL, enabled=False)
         )
     ks_id = created.id
     try:

kestrapy-2.1.0.dev568/tests/test_flow_worker_selector.py

@@ -0,0 +1,28 @@
+"""FlowWorkerSelector decode tests: no live Kestra server required.
+
+Since kestra f497a8372b, a flow/task/trigger workerSelector.tags is a Pebble-capable
+property and Kestra returns it as a string (the JSON text of a literal list, or the
+expression), so a List[str] field rejected every flow with a worker selector.
+"""
+
+from kestrapy.models.flow_with_source import FlowWithSource
+from kestrapy.models.worker_selector_match import WorkerSelectorMatch
+
+
+def _flow(worker_selector):
+    return FlowWithSource.from_dict({
+        "id": "f",
+        "namespace": "ns",
+        "disabled": False,
+        "deleted": False,
+        "draft": False,
+        "tasks": [{"id": "t", "type": "io.kestra.plugin.core.log.Log"}],
+        "workerSelector": worker_selector,
+    })
+
+
+def test_flow_reads_the_string_tags_kestra_returns_for_a_literal_list():
+    flow = _flow({"tags": '["gpu"]', "match": "ALL"})
+    assert flow.worker_selector.tags == '["gpu"]'
+    assert flow.worker_selector.match == WorkerSelectorMatch.ALL
+

{kestrapy-2.1.0.dev541 → kestrapy-2.1.0.dev568}/tests/test_flow_yaml.py

@@ -423,3 +423,25 @@ def test_construct_model_fallback_keeps_undeclared_keys():
     flow = BaseApi._construct_model(payload, FlowWithSource)
     assert flow.updated == datetime(2026, 9, 23, 10, 11, 12, 345000, tzinfo=timezone.utc)
     assert flow.triggers[0].additional_properties["timezone"] == "Europe/Paris"
+
+
+def test_worker_selector_tags_stay_strings():
+    # Pins that the emitted workerSelector.tags stays a string: a literal list must not be
+    # re-read as a YAML sequence.
+    flow = Flow.from_dict({
+        "id": "p",
+        "namespace": "ns",
+        "disabled": False,
+        "deleted": False,
+        "draft": False,
+        "workerSelector": {"tags": '["gpu"]', "match": "ANY"},
+        "tasks": [{
+            "id": "t",
+            "type": "io.kestra.plugin.core.log.Log",
+            "message": "hi",
+        }],
+    })
+
+    parsed = yaml.safe_load(flow_to_yaml(flow))
+
+    assert parsed["workerSelector"] == {"tags": '["gpu"]', "match": "ANY"}

kestrapy-2.1.0.dev568/tests/test_kv_api.py

@@ -0,0 +1,100 @@
+from unittest.mock import MagicMock, patch
+import pytest
+
+from kestrapy.kestra_client import KestraClient
+from kestrapy.api.kv_api import KVApi
+
+
+@pytest.fixture
+def client_and_mock_session():
+    """Fixture providing a KestraClient instance with a mocked HTTP session."""
+    with patch("requests.Session") as mock_session_cls:
+        mock_session = MagicMock()
+        mock_session_cls.return_value = mock_session
+        c = KestraClient(host="http://localhost:8080")
+        c._session = mock_session
+        yield c, mock_session
+
+
+def test_kv_property_returns_kv_api(client_and_mock_session):
+    """Test that accessing client.kv returns an instance of KVApi."""
+    client, _ = client_and_mock_session
+    kv = client.kv
+    assert isinstance(kv, KVApi)
+
+
+def test_set_key_value(client_and_mock_session):
+    """Test that set_key_value sends a PUT request with correct path and body."""
+    client, mock_session = client_and_mock_session
+
+    mock_resp = MagicMock()
+    mock_resp.status_code = 200
+    mock_session.request.return_value = mock_resp
+
+    client.kv.set_key_value(
+        namespace="test-namespace",
+        key="test-key",
+        tenant="test-tenant",
+        body="test-value",
+    )
+
+    assert mock_session.request.called
+    call_args, call_kwargs = mock_session.request.call_args
+
+    # Verify HTTP method and endpoint path
+    assert call_args[0].upper() == "PUT"
+    assert "namespaces" in call_args[1]
+    assert "test-namespace" in call_args[1]
+    assert "test-key" in call_args[1]
+    assert call_kwargs["data"] == "test-value"
+    assert call_kwargs["headers"]["Content-Type"] == "text/plain"
+
+
+def test_get_key_value(client_and_mock_session):
+    """Test that key_value sends a GET request to the KV endpoint."""
+    client, mock_session = client_and_mock_session
+
+    mock_resp = MagicMock()
+    mock_resp.status_code = 200
+    mock_resp.json.return_value = {"key": "test-key", "value": "test-val"}
+    mock_session.request.return_value = mock_resp
+
+    client.kv.key_value(
+        namespace="test-namespace",
+        key="test-key",
+        tenant="test-tenant",
+    )
+
+    assert mock_session.request.called
+    call_args, _ = mock_session.request.call_args
+
+    assert call_args[0].upper() == "GET"
+    assert "test-namespace" in call_args[1]
+    assert "test-key" in call_args[1]
+
+
+def test_delete_key_value(client_and_mock_session):
+    """Test that delete_key_value sends a DELETE request to the KV endpoint."""
+    client, mock_session = client_and_mock_session
+
+    # 1. Arrange: Setup mock response
+    mock_resp = MagicMock()
+    mock_resp.status_code = 200
+    mock_resp.json.return_value = True
+    mock_session.request.return_value = mock_resp
+
+    # 2. Act: Call delete_key_value method
+    result = client.kv.delete_key_value(
+        namespace="test-namespace",
+        key="test-key",
+        tenant="test-tenant",
+    )
+
+    # 3. Assert: Verify result and HTTP request parameters
+    assert result is True
+    assert mock_session.request.called
+
+    call_args, _ = mock_session.request.call_args
+    assert call_args[0].upper() == "DELETE"
+    assert "test-namespace" in call_args[1]
+    assert "test-key" in call_args[1]