truefoundry 0.11.1rc1__py3-none-any.whl → 0.11.3__py3-none-any.whl

truefoundry/deploy/builder/builders/tfy_task_pyspark_buildpack/__init__.py

@@ -0,0 +1,52 @@
+import os
+from tempfile import TemporaryDirectory
+from typing import List, Optional
+
+from truefoundry.deploy._autogen.models import DockerFileBuild, TaskPySparkBuild
+from truefoundry.deploy.builder.builders import dockerfile
+from truefoundry.deploy.builder.builders.tfy_task_pyspark_buildpack.dockerfile_template import (
+    generate_dockerfile_content,
+)
+from truefoundry.deploy.builder.utils import has_python_package_manager_conf_secret
+
+__all__ = ["generate_dockerfile_content", "build"]
+
+
+def _convert_to_dockerfile_build_config(
+    build_configuration: TaskPySparkBuild,
+    dockerfile_path: str,
+    mount_python_package_manager_conf_secret: bool = False,
+) -> DockerFileBuild:
+    dockerfile_content = generate_dockerfile_content(
+        build_configuration=build_configuration,
+        mount_python_package_manager_conf_secret=mount_python_package_manager_conf_secret,
+    )
+    with open(dockerfile_path, "w", encoding="utf8") as fp:
+        fp.write(dockerfile_content)
+
+    return DockerFileBuild(
+        type="dockerfile",
+        dockerfile_path=dockerfile_path,
+    )
+
+
+def build(
+    tag: str,
+    build_configuration: TaskPySparkBuild,
+    extra_opts: Optional[List[str]] = None,
+):
+    mount_python_package_manager_conf_secret = (
+        has_python_package_manager_conf_secret(extra_opts) if extra_opts else False
+    )
+
+    with TemporaryDirectory() as local_dir:
+        docker_build_configuration = _convert_to_dockerfile_build_config(
+            build_configuration,
+            dockerfile_path=os.path.join(local_dir, "Dockerfile"),
+            mount_python_package_manager_conf_secret=mount_python_package_manager_conf_secret,
+        )
+        dockerfile.build(
+            tag=tag,
+            build_configuration=docker_build_configuration,
+            extra_opts=extra_opts,
+        )

truefoundry/deploy/builder/builders/tfy_task_pyspark_buildpack/dockerfile_template.py

@@ -0,0 +1,121 @@
+from mako.template import Template
+
+from truefoundry.common.constants import ENV_VARS, PythonPackageManager
+from truefoundry.deploy._autogen.models import TaskPySparkBuild
+from truefoundry.deploy.builder.constants import (
+    PIP_CONF_BUILDKIT_SECRET_MOUNT,
+    UV_CONF_BUILDKIT_SECRET_MOUNT,
+)
+from truefoundry.deploy.builder.utils import (
+    generate_apt_install_command,
+    generate_pip_install_command,
+    generate_uv_pip_install_command,
+)
+from truefoundry.deploy.v2.lib.patched_models import (
+    _resolve_requirements_path,
+)
+
+# TODO[GW]: Switch to a non-root user inside the container
+_POST_PYTHON_INSTALL_TEMPLATE = """
+% if apt_install_command is not None:
+RUN ${apt_install_command}
+% endif
+% if requirements_path is not None:
+COPY ${requirements_path} ${requirements_destination_path}
+% endif
+% if python_packages_install_command is not None:
+RUN ${package_manager_config_secret_mount} ${python_packages_install_command}
+% endif
+COPY . /app
+WORKDIR /app
+"""
+
+# TODO[GW]: Check if the entrypoint for the image needs to change
+# Using /opt/venv/ because flyte seems to be using it and this doesn't look configurable
+# TODO[GW]: Double check this^
+DOCKERFILE_TEMPLATE = Template(
+    """
+FROM ${spark_image_repo}:${spark_version}
+ENV PATH=/opt/venv/bin:$PATH
+USER root
+RUN mkdir -p /var/lib/apt/lists/partial && \
+    apt update && \
+    DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends git && \
+    python -m venv /opt/venv/ && \
+    rm -rf /var/lib/apt/lists/*
+"""
+    + _POST_PYTHON_INSTALL_TEMPLATE
+)
+
+
+def get_additional_pip_packages(build_configuration: TaskPySparkBuild):
+    return [
+        f"pyspark=={build_configuration.spark_version}",
+    ]
+
+
+def generate_dockerfile_content(
+    build_configuration: TaskPySparkBuild,
+    package_manager: str = ENV_VARS.TFY_PYTHON_BUILD_PACKAGE_MANAGER,
+    mount_python_package_manager_conf_secret: bool = False,
+) -> str:
+    # TODO (chiragjn): Handle recursive references to other requirements files e.g. `-r requirements-gpu.txt`
+    requirements_path = _resolve_requirements_path(
+        build_context_path="",
+        requirements_path=build_configuration.requirements_path,
+    )
+    requirements_destination_path = (
+        "/tmp/requirements.txt" if requirements_path else None
+    )
+    # if not build_configuration.python_version:
+    #     raise ValueError(
+    #         "`python_version` is required for `tfy-python-buildpack` builder"
+    #     )
+    pip_packages = get_additional_pip_packages(build_configuration) + (
+        build_configuration.pip_packages or []
+    )
+    if package_manager == PythonPackageManager.PIP.value:
+        python_packages_install_command = generate_pip_install_command(
+            requirements_path=requirements_destination_path,
+            pip_packages=pip_packages,
+            mount_pip_conf_secret=mount_python_package_manager_conf_secret,
+        )
+    elif package_manager == PythonPackageManager.UV.value:
+        python_packages_install_command = generate_uv_pip_install_command(
+            requirements_path=requirements_destination_path,
+            pip_packages=pip_packages,
+            mount_uv_conf_secret=mount_python_package_manager_conf_secret,
+        )
+    else:
+        raise ValueError(f"Unsupported package manager: {package_manager}")
+
+    apt_install_command = generate_apt_install_command(
+        apt_packages=build_configuration.apt_packages
+    )
+    template_args = {
+        "spark_image_repo": ENV_VARS.TFY_TASK_PYSPARK_BUILD_SPARK_IMAGE_REPO,
+        "spark_version": build_configuration.spark_version,
+        "apt_install_command": apt_install_command,
+        "requirements_path": requirements_path,
+        "requirements_destination_path": requirements_destination_path,
+        "python_packages_install_command": python_packages_install_command,
+    }
+
+    if mount_python_package_manager_conf_secret:
+        if package_manager == PythonPackageManager.PIP.value:
+            template_args["package_manager_config_secret_mount"] = (
+                PIP_CONF_BUILDKIT_SECRET_MOUNT
+            )
+        elif package_manager == PythonPackageManager.UV.value:
+            template_args["package_manager_config_secret_mount"] = (
+                UV_CONF_BUILDKIT_SECRET_MOUNT
+            )
+        else:
+            raise ValueError(f"Unsupported package manager: {package_manager}")
+    else:
+        template_args["package_manager_config_secret_mount"] = ""
+
+    template = DOCKERFILE_TEMPLATE
+
+    dockerfile_content = template.render(**template_args)
+    return dockerfile_content

truefoundry/deploy/builder/utils.py

@@ -1,8 +1,12 @@
+import shlex
 from typing import List, Optional
 
+from truefoundry.common.constants import ENV_VARS
 from truefoundry.deploy.builder.constants import (
     BUILDKIT_SECRET_MOUNT_PIP_CONF_ID,
     BUILDKIT_SECRET_MOUNT_UV_CONF_ID,
+    PIP_CONF_SECRET_MOUNT_AS_ENV,
+    UV_CONF_SECRET_MOUNT_AS_ENV,
 )
 
 
@@ -35,3 +39,82 @@ def has_python_package_manager_conf_secret(docker_build_extra_args: List[str]) -
         ):
             return True
     return False
+
+
+def generate_pip_install_command(
+    requirements_path: Optional[str],
+    pip_packages: Optional[List[str]],
+    mount_pip_conf_secret: bool = False,
+) -> Optional[str]:
+    upgrade_pip_command = "python -m pip install -U pip setuptools wheel"
+    envs = []
+    if mount_pip_conf_secret:
+        envs.append(PIP_CONF_SECRET_MOUNT_AS_ENV)
+
+    command = ["python", "-m", "pip", "install", "--use-pep517", "--no-cache-dir"]
+    args = []
+    if requirements_path:
+        args.append("-r")
+        args.append(requirements_path)
+
+    if pip_packages:
+        args.extend(pip_packages)
+
+    if not args:
+        return None
+
+    final_pip_install_command = shlex.join(envs + command + args)
+    final_docker_run_command = " && ".join(
+        [upgrade_pip_command, final_pip_install_command]
+    )
+    return final_docker_run_command
+
+
+def generate_uv_pip_install_command(
+    requirements_path: Optional[str],
+    pip_packages: Optional[List[str]],
+    mount_uv_conf_secret: bool = False,
+) -> Optional[str]:
+    upgrade_pip_command = "python -m pip install -U pip setuptools wheel"
+    uv_mount = f"--mount=from={ENV_VARS.TFY_PYTHON_BUILD_UV_IMAGE_URI},source=/uv,target=/usr/local/bin/uv"
+    envs = [
+        "UV_LINK_MODE=copy",
+        "UV_PYTHON_DOWNLOADS=never",
+        "UV_INDEX_STRATEGY=unsafe-best-match",
+    ]
+    if mount_uv_conf_secret:
+        envs.append(UV_CONF_SECRET_MOUNT_AS_ENV)
+
+    command = ["uv", "pip", "install", "--no-cache-dir"]
+
+    args = []
+
+    if requirements_path:
+        args.append("-r")
+        args.append(requirements_path)
+
+    if pip_packages:
+        args.extend(pip_packages)
+
+    if not args:
+        return None
+
+    uv_pip_install_command = shlex.join(envs + command + args)
+    shell_commands = " && ".join([upgrade_pip_command, uv_pip_install_command])
+    final_docker_run_command = " ".join([uv_mount, shell_commands])
+
+    return final_docker_run_command
+
+
+def generate_apt_install_command(apt_packages: Optional[List[str]]) -> Optional[str]:
+    packages_list = None
+    if apt_packages:
+        packages_list = " ".join(p.strip() for p in apt_packages if p.strip())
+    if not packages_list:
+        return None
+    apt_update_command = "apt update"
+    apt_install_command = f"DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends {packages_list}"
+    clear_apt_lists_command = "rm -rf /var/lib/apt/lists/*"
+    return " && ".join(
+        [apt_update_command, apt_install_command, clear_apt_lists_command]
+    )

truefoundry/deploy/cli/commands/trigger_command.py

@@ -35,8 +35,20 @@ def trigger_command():
     nargs=-1,
     required=False,
 )
+@click.option(
+    "--run-name-alias",
+    "--run_name_alias",
+    type=click.STRING,
+    required=False,
+    help="Alias for the job run name.",
+)
 @handle_exception_wrapper
-def trigger_job(application_fqn: str, command: Optional[Sequence[str]], params):
+def trigger_job(
+    application_fqn: str,
+    params,
+    command: Optional[Sequence[str]],
+    run_name_alias: Optional[str],
+):
     """
     Trigger a Job on TrueFoundry asynchronously
 
@@ -54,6 +66,10 @@ def trigger_job(application_fqn: str, command: Optional[Sequence[str]], params):
     Passing params:
 
         [b]tfy trigger job --application-fqn "my-cluster:my-workspace:my-job" -- --param1_name param1_value --param2_name param2_value ...[/]
+    \n
+
+    passing run_name_alias:
+        [b]tfy trigger job --application-fqn "my-cluster:my-workspace:my-job" --run_name_alias "my_run_alias"[/]
     """
     if params:
         params_dict = {}
@@ -78,7 +94,10 @@ def trigger_job(application_fqn: str, command: Optional[Sequence[str]], params):
             params_dict[key] = value
 
     application.trigger_job(
-        application_fqn=application_fqn, command=command, params=params
+        application_fqn=application_fqn,
+        command=command,
+        params=params,
+        run_name_alias=run_name_alias,
     )
 
 

truefoundry/deploy/lib/clients/servicefoundry_client.py

@@ -578,6 +578,7 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
     def trigger_job(
         self,
         deployment_id: str,
+        run_name_alias: Optional[str] = None,
         command: Optional[str] = None,
         params: Optional[Dict[str, str]] = None,
     ) -> TriggerJobResult:
@@ -585,11 +586,14 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
         body = {
             "deploymentId": deployment_id,
             "input": {},
+            "metadata": {},
         }
         if command:
             body["input"]["command"] = command
         if params:
             body["input"]["params"] = params
+        if run_name_alias:
+            body["metadata"]["job_run_name_alias"] = run_name_alias
         response = session_with_retries().post(
             url, json=body, headers=self._get_headers()
         )

truefoundry/deploy/lib/dao/application.py

@@ -117,6 +117,7 @@ def trigger_job(
     application_fqn: str,
     command: Optional[Union[str, Sequence[str]]] = None,
     params: Optional[Dict[str, str]] = None,
+    run_name_alias: Optional[str] = None,
 ) -> TriggerJobResult:
     """
     Trigger a Job on TrueFoundry platform
@@ -178,6 +179,7 @@ def trigger_job(
         deployment_id=application_info.activeDeploymentId,
         command=command_str if command_str else None,
         params=params if params else None,
+        run_name_alias=run_name_alias.strip() if run_name_alias else None,
     )
     jobRunName = result.jobRunName
     previous_runs_url = f"{client.tfy_host.strip('/')}/deployments/{application_info.id}?tab=previousRuns"

truefoundry/deploy/v2/lib/patched_models.py

@@ -526,3 +526,42 @@ class SparkJobPythonNotebookEntrypoint(
     models.SparkJobPythonNotebookEntrypoint, PatchedModelBase
 ):
     type: Literal["python-notebook"] = "python-notebook"
+
+
+class SparkJobScalaNotebookEntrypoint(
+    models.SparkJobScalaNotebookEntrypoint, PatchedModelBase
+):
+    type: Literal["scala-notebook"] = "scala-notebook"
+
+
+class PySparkTaskConfig(models.PySparkTaskConfig, PatchedModelBase):
+    type: Literal["pyspark-task-config"] = "pyspark-task-config"
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        try:
+            import truefoundry.workflow.spark_task as _  # noqa: F401
+        except ImportError as e:
+            raise ImportError(
+                "truefoundry.workflow.spark_task is not installed. Please install it with `pip install truefoundry[workflow,spark]`"
+            ) from e
+
+
+class SparkDriverConfig(models.SparkDriverConfig, PatchedModelBase):
+    type: Literal["spark-driver-config"] = "spark-driver-config"
+
+
+class SparkExecutorConfig(models.SparkExecutorConfig, PatchedModelBase):
+    type: Literal["spark-executor-config"] = "spark-executor-config"
+
+
+class SparkExecutorFixedInstances(models.SparkExecutorFixedInstances, PatchedModelBase):
+    type: Literal["fixed"] = "fixed"
+
+
+class SparkExecutorDynamicScaling(models.SparkExecutorDynamicScaling, PatchedModelBase):
+    type: Literal["dynamic"] = "dynamic"
+
+
+class TaskPySparkBuild(models.TaskPySparkBuild, PatchedModelBase):
+    type: Literal["task-pyspark-build"] = "task-pyspark-build"

truefoundry/ml/artifact/truefoundry_artifact_repo.py

@@ -6,17 +6,7 @@ import uuid
 from concurrent.futures import FIRST_EXCEPTION, Future, ThreadPoolExecutor, wait
 from shutil import rmtree
 from threading import Event
-from typing import (
-    Any,
-    Callable,
-    Dict,
-    Iterator,
-    List,
-    Optional,
-    Sequence,
-    Tuple,
-    Union,
-)
+from typing import Any, Callable, Dict, Iterator, List, Optional, Sequence, Tuple
 from urllib.parse import unquote
 from urllib.request import pathname2url
 
@@ -31,6 +21,7 @@ from rich.progress import (
 )
 from tqdm.utils import CallbackIOWrapper
 from truefoundry_sdk import (
+    FileInfo,
     MultiPartUploadResponse,
     MultiPartUploadStorageProvider,
     Operation,
@@ -52,11 +43,6 @@ from truefoundry.common.storage_provider_utils import (
 )
 from truefoundry.ml._autogen.client import (  # type: ignore[attr-defined]
     ApiClient,
-    FileInfoDto,
-    ListFilesForArtifactVersionRequestDto,
-    ListFilesForArtifactVersionsResponseDto,
-    ListFilesForDatasetRequestDto,
-    ListFilesForDatasetResponseDto,
     MlfoundryArtifactsApi,
     RunArtifactsApi,
 )
@@ -592,44 +578,23 @@ class MlFoundryArtifactsRepository:
             progress=progress,
         )
 
-    def _list_files(
-        self, artifact_identifier: ArtifactIdentifier, path, page_size, page_token
-    ) -> Union[ListFilesForDatasetResponseDto, ListFilesForArtifactVersionsResponseDto]:
-        if artifact_identifier.dataset_fqn:
-            return self._mlfoundry_artifacts_api.list_files_for_dataset_post(
-                list_files_for_dataset_request_dto=ListFilesForDatasetRequestDto(
-                    dataset_fqn=artifact_identifier.dataset_fqn,
-                    path=path,
-                    max_results=page_size,
-                    page_token=page_token,
-                )
-            )
-        else:
-            return self._mlfoundry_artifacts_api.list_files_for_artifact_version_post(
-                list_files_for_artifact_version_request_dto=ListFilesForArtifactVersionRequestDto(
-                    id=str(artifact_identifier.artifact_version_id),
-                    path=path,
-                    max_results=page_size,
-                    page_token=page_token,
-                )
-            )
-
     def list_artifacts(
         self, path=None, page_size=_LIST_FILES_PAGE_SIZE, **kwargs
-    ) -> Iterator[FileInfoDto]:
-        page_token = None
-        started = False
-        while not started or page_token is not None:
-            started = True
-            page = self._list_files(
-                artifact_identifier=self.artifact_identifier,
+    ) -> Iterator[FileInfo]:
+        if self.artifact_identifier.dataset_id:
+            for file_info in client.data_directories.list_files(
+                id=str(self.artifact_identifier.dataset_id),
                 path=path,
-                page_size=page_size,
-                page_token=page_token,
-            )
-            for file_info in page.files:
+                limit=page_size,
+            ):
+                yield file_info
+        else:
+            for file_info in client.artifact_versions.list_files(
+                id=str(self.artifact_identifier.artifact_version_id),
+                path=path,
+                limit=page_size,
+            ):
                 yield file_info
-            page_token = page.next_page_token
 
     def _is_directory(self, artifact_path):
         # TODO: Ideally server should return a flag to indicate if it is a directory

truefoundry/workflow/__init__.py

@@ -15,6 +15,7 @@ from flytekit.types.file import FlyteFile
 from truefoundry.common.constants import ENV_VARS
 from truefoundry.deploy.v2.lib.patched_models import (
     ContainerTaskConfig,
+    PySparkTaskConfig,
     PythonTaskConfig,
     TaskDockerFileBuild,
     TaskPythonBuild,
@@ -41,6 +42,7 @@ __all__ = [
     "ExecutionConfig",
     "FlyteFile",
     "FlyteError",
+    "PySparkTaskConfig",
 ]
 
 

truefoundry/workflow/spark_task.py

@@ -0,0 +1,91 @@
+import os
+import shutil
+from typing import Any, Callable, Dict, Optional
+
+from flytekit import FlyteContextManager, PythonFunctionTask, lazy_module
+from flytekit.configuration import SerializationSettings
+from flytekit.core.context_manager import ExecutionParameters
+from flytekit.extend import ExecutionState, TaskPlugins
+from flytekit.extend.backend.base_agent import AsyncAgentExecutorMixin
+
+from truefoundry.deploy.v2.lib.patched_models import PySparkTaskConfig
+
+pyspark_sql = lazy_module("pyspark.sql")
+SparkSession = pyspark_sql.SparkSession
+
+
+class TfySparkFunctionTask(
+    AsyncAgentExecutorMixin, PythonFunctionTask[PySparkTaskConfig]
+):
+    """
+    Actual Plugin that transforms the local python code for execution within a spark context
+    """
+
+    _SPARK_TASK_TYPE = "spark"
+
+    def __init__(
+        self,
+        task_config: PySparkTaskConfig,
+        task_function: Callable,
+        **kwargs,
+    ):
+        self.sess: Optional[SparkSession] = None  # type: ignore
+
+        task_type = self._SPARK_TASK_TYPE
+
+        super(TfySparkFunctionTask, self).__init__(
+            task_config=task_config,
+            task_type=task_type,
+            task_function=task_function,
+            **kwargs,
+        )
+
+    def get_custom(self, settings: SerializationSettings) -> Dict[str, Any]:
+        return {"truefoundry": self._task_config.dict()}
+
+    def pre_execute(self, user_params: ExecutionParameters) -> ExecutionParameters:
+        import pyspark as _pyspark
+
+        ctx = FlyteContextManager.current_context()
+        sess_builder = _pyspark.sql.SparkSession.builder.appName(
+            f"FlyteSpark: {user_params.execution_id}"
+        )
+        if not (
+            ctx.execution_state
+            and ctx.execution_state.mode == ExecutionState.Mode.TASK_EXECUTION
+        ):
+            # If either of above cases is not true, then we are in local execution of this task
+            # Add system spark-conf for local/notebook based execution.
+            spark_conf = _pyspark.SparkConf()
+            spark_conf.set("spark.driver.bindAddress", "127.0.0.1")
+            for k, v in self.task_config.spark_conf.items():
+                spark_conf.set(k, v)
+            # In local execution, propagate PYTHONPATH to executors too. This makes the spark
+            # execution hermetic to the execution environment. For example, it allows running
+            # Spark applications using Bazel, without major changes.
+            if "PYTHONPATH" in os.environ:
+                spark_conf.setExecutorEnv("PYTHONPATH", os.environ["PYTHONPATH"])
+            sess_builder = sess_builder.config(conf=spark_conf)
+
+        self.sess = sess_builder.getOrCreate()
+
+        if (
+            ctx.serialization_settings
+            and ctx.serialization_settings.fast_serialization_settings
+            and ctx.serialization_settings.fast_serialization_settings.enabled
+            and ctx.execution_state
+            and ctx.execution_state.mode == ExecutionState.Mode.TASK_EXECUTION
+        ):
+            file_name = "flyte_wf"
+            file_format = "zip"
+            shutil.make_archive(file_name, file_format, os.getcwd())
+            self.sess.sparkContext.addPyFile(f"{file_name}.{file_format}")
+
+        return user_params.builder().add_attr("SPARK_SESSION", self.sess).build()
+
+    def execute(self, **kwargs) -> Any:
+        return PythonFunctionTask.execute(self, **kwargs)
+
+
+# Inject the Spark plugin into flytekits dynamic plugin loading system
+TaskPlugins.register_pythontask_plugin(PySparkTaskConfig, TfySparkFunctionTask)

{truefoundry-0.11.1rc1.dist-info → truefoundry-0.11.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truefoundry
-Version: 0.11.1rc1
+Version: 0.11.3
 Summary: TrueFoundry CLI
 Author-email: TrueFoundry Team <abhishek@truefoundry.com>
 Requires-Python: <3.14,>=3.8.1
@@ -30,14 +30,17 @@ Requires-Dist: requirements-parser<0.12.0,>=0.11.0
 Requires-Dist: rich-click<2.0.0,>=1.2.1
 Requires-Dist: rich<14.0.0,>=13.7.1
 Requires-Dist: tqdm<5.0.0,>=4.0.0
-Requires-Dist: truefoundry-sdk<0.2.0,>=0.1.7
+Requires-Dist: truefoundry-sdk<0.2.0,>=0.1.9
 Requires-Dist: typing-extensions>=4.0
 Requires-Dist: urllib3<3,>=1.26.18
 Requires-Dist: yq<4.0.0,>=3.1.0
 Provides-Extra: ai
 Requires-Dist: mcp==1.9.4; (python_version >= '3.10') and extra == 'ai'
+Provides-Extra: spark
+Requires-Dist: flytekit==1.15.3; (python_version >= '3.9' and python_version <= '3.12') and extra == 'spark'
+Requires-Dist: flytekitplugins-spark==1.15.3; (python_version >= '3.9' and python_version <= '3.12') and extra == 'spark'
 Provides-Extra: workflow
-Requires-Dist: flytekit==1.15.3; (python_version >= '3.9' and python_version < '3.13') and extra == 'workflow'
+Requires-Dist: flytekit==1.15.3; (python_version >= '3.9' and python_version <= '3.12') and extra == 'workflow'
 Description-Content-Type: text/markdown
 
 # TrueFoundry