truefoundry 0.11.11rc1__py3-none-any.whl → 0.11.12__py3-none-any.whl

truefoundry/common/constants.py

@@ -74,8 +74,7 @@ class TrueFoundrySdkEnv(BaseSettings):
     TFY_CLI_LOCAL_DEV_MODE: bool = False
 
     TFY_PYTHON_BUILD_PACKAGE_MANAGER: PythonPackageManager = PythonPackageManager.UV
-    TFY_PYTHON_BUILD_UV_IMAGE_REPO: str = "ghcr.io/astral-sh/uv"
-    TFY_PYTHON_BUILD_UV_IMAGE_TAG: str = "latest"
+    TFY_PYTHON_BUILD_UV_IMAGE_URI: str = "ghcr.io/astral-sh/uv:latest"
 
     # Global Constants for OpenAI Integration
     OPENAI_API_KEY: Optional[str] = Field(default=None, env=OPENAI_API_KEY_KEY)
@@ -94,12 +93,6 @@ class TrueFoundrySdkEnv(BaseSettings):
     )
     TFY_ASK_SYSTEM_PROMPT_NAME: str = Field(default="tfy-ask-k8s-prompt")
     TFY_INTERNAL_ASK_CONFIG_OVERRIDE_FILE: Optional[str] = Field(default=None)
-    TFY_PYTHON_BUILD_POETRY_VERSION: str = Field(
-        default="2.0", env="TFY_PYTHON_BUILD_POETRY_VERSION"
-    )
-    TFY_PYTHON_BUILD_LATEST_POETRY_MAJOR_VERSION: int = Field(
-        default=2, env="TFY_PYTHON_BUILD_LATEST_POETRY_MAJOR_VERSION"
-    )
 
     # This is a hack to fresh read the env vars because people can end up importing this file
     # before setting the correct env vars. E.g. in notebook environments.

truefoundry/common/storage_provider_utils.py

@@ -22,6 +22,23 @@ from truefoundry.pydantic_v1 import BaseModel
 logger = logging.getLogger("truefoundry")
 
 
+def truncate_path_for_progress(
+    path: str, max_length: int, relpath: bool = False
+) -> str:
+    if relpath:
+        path = os.path.relpath(path)
+    if len(path) <= max_length:
+        return path
+    parts = path.split(os.sep)
+    result = parts[-1]  # start with filename
+    i = len(parts) - 2
+    # keep prepending directories until adding more would exceed max_len - 3
+    while i >= 0 and len(result) + len(parts[i]) + 1 <= max_length - 3:
+        result = parts[i] + os.sep + result
+        i -= 1
+    return "..." + os.sep + result
+
+
 class MultiPartUploadStorageProvider(str, Enum):
     S3_COMPATIBLE = "S3_COMPATIBLE"
     AZURE_BLOB = "AZURE_BLOB"
@@ -182,7 +199,7 @@ def _file_part_upload(
                 return response
 
 
-def s3_compatible_multipart_upload(
+def s3_compatible_multipart_upload(  # noqa: C901
     multipart_upload: MultiPartUpload,
     local_file: str,
     multipart_info: _FileMultiPartInfo,
@@ -190,13 +207,15 @@ def s3_compatible_multipart_upload(
     progress_bar: Optional[Progress] = None,
     abort_event: Optional[Event] = None,
     exception_class=HttpRequestException,
-):
+) -> None:
     abort_event = abort_event or Event()
     parts = []
 
-    if progress_bar:
+    if progress_bar is not None:
         multi_part_upload_progress = progress_bar.add_task(
-            f"[green]Uploading {local_file}:", start=True
+            f"[green]⬆ {truncate_path_for_progress(local_file, 64, relpath=True)}",
+            start=True,
+            visible=True,
         )
 
     def upload(part_number: int, seek: int) -> None:
@@ -221,7 +240,7 @@ def s3_compatible_multipart_upload(
             multipart_info.num_parts,
             local_file,
         )
-        if progress_bar:
+        if progress_bar is not None:
             progress_bar.update(
                 multi_part_upload_progress,
                 advance=multipart_info.part_size,
@@ -254,10 +273,12 @@ def s3_compatible_multipart_upload(
         timeout=2 * 60,
     )
     response.raise_for_status()
+    if progress_bar is not None:
+        progress_bar.refresh()
     logger.debug("Multipart upload of %s completed", local_file)
 
 
-def azure_multi_part_upload(
+def azure_multi_part_upload(  # noqa: C901
     multipart_upload: MultiPartUpload,
     local_file: str,
     multipart_info: _FileMultiPartInfo,
@@ -265,12 +286,14 @@ def azure_multi_part_upload(
     progress_bar: Optional[Progress] = None,
     abort_event: Optional[Event] = None,
     exception_class=HttpRequestException,
-):
+) -> None:
     abort_event = abort_event or Event()
 
-    if progress_bar:
+    if progress_bar is not None:
         multi_part_upload_progress = progress_bar.add_task(
-            f"[green]Uploading {local_file}:", start=True
+            f"[green]⬆ {truncate_path_for_progress(local_file, 64, relpath=True)}",
+            start=True,
+            visible=True,
         )
 
     def upload(part_number: int, seek: int):
@@ -289,7 +312,7 @@ def azure_multi_part_upload(
             abort_event=abort_event,
             exception_class=exception_class,
         )
-        if progress_bar:
+        if progress_bar is not None:
             progress_bar.update(
                 multi_part_upload_progress,
                 advance=multipart_info.part_size,
@@ -328,4 +351,6 @@ def azure_multi_part_upload(
             timeout=2 * 60,
         )
         response.raise_for_status()
+    if progress_bar is not None:
+        progress_bar.refresh()
     logger.debug("Multipart upload of %s completed", local_file)

truefoundry/deploy/__init__.py

@@ -52,7 +52,6 @@ from truefoundry.deploy.v2.lib.deployable_patched_models import (
     Workflow,
 )
 from truefoundry.deploy.v2.lib.patched_models import (
-    UV,
     AMQPInputConfig,
     AMQPMetricConfig,
     AMQPOutputConfig,
@@ -103,8 +102,6 @@ from truefoundry.deploy.v2.lib.patched_models import (
     NvidiaTimeslicingGPU,
     OCIRepo,
     Param,
-    Pip,
-    Poetry,
     Port,
     PythonBuild,
     RemoteSource,

truefoundry/deploy/_autogen/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2025-09-09T14:09:56+00:00
+#   timestamp: 2025-09-09T12:09:08+00:00
 
 from __future__ import annotations
 
@@ -603,37 +603,6 @@ class Param(BaseModel):
     param_type: ParamType = "string"
 
 
-class Pip(BaseModel):
-    """
-    Use pip to install requirements files and packages
-    """
-
-    type: Literal["pip"] = Field(..., description="")
-    requirements_path: Optional[str] = Field(
-        None,
-        description="Path to `requirements.txt` relative to `Path to build context`",
-    )
-    pip_packages: Optional[List[str]] = Field(
-        None,
-        description='Define pip package requirements.\nIn Python/YAML E.g. ["fastapi>=0.90,<1.0", "uvicorn"]',
-    )
-
-
-class Poetry(BaseModel):
-    """
-    Use `poetry` to setup environment and sync `poetry.lock` file
-    Your build context root must contain a `poetry.lock` file
-    """
-
-    type: Literal["poetry"] = Field(..., description="")
-    poetry_version: Union[
-        Literal["latest"], constr(regex=r"^\d+(\.\d+){1,2}([\-\.a-z0-9]+)?$")
-    ] = Field("latest", description="Poetry version to use")
-    install_options: Optional[str] = Field(
-        None, description="install options to pass to poetry command"
-    )
-
-
 class Protocol(str, Enum):
     """
     Protocol for the port.
@@ -656,6 +625,47 @@ class AppProtocol(str, Enum):
     tcp = "tcp"
 
 
+class PythonBuild(BaseModel):
+    """
+    Describes that we are using python to build a container image with a specific python version and pip packages installed.
+    """
+
+    type: Literal["tfy-python-buildpack"] = Field(..., description="")
+    python_version: Optional[constr(regex=r"^\d+(\.\d+){1,2}([\-\.a-z0-9]+)?$")] = (
+        Field(
+            None,
+            description="Python version to run your application. Should be one of the tags listed on [Official Python Docker Page](https://hub.docker.com/_/python)",
+        )
+    )
+    build_context_path: str = Field(
+        "./", description="Build path relative to project root path."
+    )
+    requirements_path: Optional[str] = Field(
+        None,
+        description="Path to `requirements.txt` relative to\n`Path to build context`",
+    )
+    pip_packages: Optional[List[str]] = Field(
+        None,
+        description='Define pip package requirements.\nIn Python/YAML E.g. ["fastapi>=0.90,<1.0", "uvicorn"]',
+    )
+    apt_packages: Optional[List[str]] = Field(
+        None,
+        description='Debian packages to install via `apt get`.\nIn Python/YAML E.g. ["git", "ffmpeg", "htop"]',
+    )
+    command: Union[str, List[str]] = Field(
+        ...,
+        description="Command to run when the container starts.\nCommand will be set as the Entrypoint of the generated image.\nWhen deploying a Job, the command can be templatized by defining `params` and referencing them in command\nE.g. `python main.py --learning_rate {{learning_rate}}`",
+    )
+    cuda_version: Optional[
+        constr(
+            regex=r"^((\d+\.\d+(\.\d+)?-cudnn\d+-(runtime|devel)-ubuntu\d+\.\d+)|11\.0-cudnn8|11\.1-cudnn8|11\.2-cudnn8|11\.3-cudnn8|11\.4-cudnn8|11\.5-cudnn8|11\.6-cudnn8|11\.7-cudnn8|11\.8-cudnn8|12\.0-cudnn8|12\.1-cudnn8|12\.2-cudnn8|12\.3-cudnn9|12\.4-cudnn9|12\.5-cudnn9|12\.6-cudnn9)$"
+        )
+    ] = Field(
+        None,
+        description="Version of CUDA Toolkit and CUDNN to install in the image\nThese combinations are based off of publically available docker images on docker hub\nYou can also specify a valid tag of the form {cuda_version_number}-cudnn{cudnn_version_number}-{runtime|devel}-ubuntu{ubuntu_version}\nRefer https://hub.docker.com/r/nvidia/cuda/tags for valid set of values\nNote: We use deadsnakes ubuntu ppa to add Python that currently supports only Ubuntu 18.04, 20.04 and 22.04",
+    )
+
+
 class RPSMetric(BaseModel):
     type: Literal["rps"] = Field(..., description="")
     value: PositiveFloat = Field(
@@ -1092,19 +1102,6 @@ class TrueFoundryInteractiveLogin(BaseModel):
     )
 
 
-class UV(BaseModel):
-    """
-    Use `uv` to setup environment and sync `uv.lock` file
-    Your build context root must contain a `uv.lock` file
-    """
-
-    type: Literal["uv"] = Field(..., description="")
-    uv_version: str = Field("latest", description="UV version to use")
-    sync_options: Optional[str] = Field(
-        None, description="Sync options to pass to uv command"
-    )
-
-
 class VolumeBrowser(BaseModel):
     username: Optional[constr(regex=r"^[a-z][a-z0-9]{1,8}[a-z0-9]$")] = Field(
         None, description="Username for logging in the volume browser."
@@ -1202,6 +1199,25 @@ class BaseWorkbenchInput(BaseModel):
     )
 
 
+class Build(BaseModel):
+    """
+    Describes how we build our code into a Docker image.
+    """
+
+    type: Literal["build"] = Field(..., description="")
+    docker_registry: Optional[constr(regex=r"^\S+$")] = Field(
+        None,
+        description="FQN of the container registry. If you can't find your registry here,\nadd it through the [Integrations](/integrations?tab=docker-registry) page",
+    )
+    build_source: Union[RemoteSource, GitSource, LocalSource] = Field(
+        ..., description="Source code location."
+    )
+    build_spec: Union[DockerFileBuild, PythonBuild] = Field(
+        ...,
+        description="Instructions to build a container image out of the build source",
+    )
+
+
 class Canary(BaseModel):
     """
     This strategy brings up the new release without bringing the older release down. Traffic is shifted from the older release to the newer release in a staged manner.
@@ -1224,6 +1240,23 @@ class Codeserver(BaseWorkbenchInput):
     image: WorkbenchImage
 
 
+class ContainerTaskConfig(BaseModel):
+    type: Literal["container-task-config"] = Field(..., description="")
+    image: Union[Build, Image] = Field(
+        ...,
+        description="Specify whether you want to deploy a Docker image or build and deploy from source code",
+    )
+    env: Optional[Dict[str, str]] = Field(
+        None,
+        description="Configure environment variables to be injected in the task either as plain text or secrets. [Docs](https://docs.truefoundry.com/docs/env-variables)",
+    )
+    resources: Optional[Resources] = None
+    mounts: Optional[List[Union[SecretMount, StringDataMount, VolumeMount]]] = Field(
+        None, description="Configure data to be mounted to Workflow pod(s) as a volume."
+    )
+    service_account: Optional[str] = Field(None, description="")
+
+
 class CoreNATSOutputConfig(BaseModel):
     """
     Describes the configuration for the output Core NATS worker
@@ -1412,50 +1445,6 @@ class Port(BaseModel):
     ] = Field(None, description="Authentication method for inbound traffic")
 
 
-class PythonBuild(BaseModel):
-    """
-    Describes that we are using python to build a container image with a specific python version and pip packages installed.
-    """
-
-    type: Literal["tfy-python-buildpack"] = Field(..., description="")
-    python_version: Optional[constr(regex=r"^\d+(\.\d+){1,2}([\-\.a-z0-9]+)?$")] = (
-        Field(
-            None,
-            description="Python version to run your application. Should be one of the tags listed on [Official Python Docker Page](https://hub.docker.com/_/python)",
-        )
-    )
-    build_context_path: str = Field(
-        "./", description="Build path relative to project root path."
-    )
-    requirements_path: Optional[str] = Field(
-        None,
-        description="Path to `requirements.txt` relative to\n`Path to build context`",
-    )
-    pip_packages: Optional[List[str]] = Field(
-        None,
-        description='Define pip package requirements.\nIn Python/YAML E.g. ["fastapi>=0.90,<1.0", "uvicorn"]',
-    )
-    python_dependencies: Optional[Union[Pip, UV, Poetry]] = Field(
-        None, description="Python dependencies to install"
-    )
-    apt_packages: Optional[List[str]] = Field(
-        None,
-        description='Debian packages to install via `apt get`.\nIn Python/YAML E.g. ["git", "ffmpeg", "htop"]',
-    )
-    command: Union[str, List[str]] = Field(
-        ...,
-        description="Command to run when the container starts.\nCommand will be set as the Entrypoint of the generated image.\nWhen deploying a Job, the command can be templatized by defining `params` and referencing them in command\nE.g. `python main.py --learning_rate {{learning_rate}}`",
-    )
-    cuda_version: Optional[
-        constr(
-            regex=r"^((\d+\.\d+(\.\d+)?-cudnn\d+-(runtime|devel)-ubuntu\d+\.\d+)|11\.0-cudnn8|11\.1-cudnn8|11\.2-cudnn8|11\.3-cudnn8|11\.4-cudnn8|11\.5-cudnn8|11\.6-cudnn8|11\.7-cudnn8|11\.8-cudnn8|12\.0-cudnn8|12\.1-cudnn8|12\.2-cudnn8|12\.3-cudnn9|12\.4-cudnn9|12\.5-cudnn9|12\.6-cudnn9)$"
-        )
-    ] = Field(
-        None,
-        description="Version of CUDA Toolkit and CUDNN to install in the image\nThese combinations are based off of publically available docker images on docker hub\nYou can also specify a valid tag of the form {cuda_version_number}-cudnn{cudnn_version_number}-{runtime|devel}-ubuntu{ubuntu_version}\nRefer https://hub.docker.com/r/nvidia/cuda/tags for valid set of values\nNote: We use deadsnakes ubuntu ppa to add Python that currently supports only Ubuntu 18.04, 20.04 and 22.04",
-    )
-
-
 class PythonTaskConfig(BaseModel):
     """
     Describes the configuration for the python function task
@@ -1591,40 +1580,38 @@ class WorkflowAlert(BaseModel):
     on_failure: bool = Field(True, description="Send an alert when the job fails")
 
 
-class Build(BaseModel):
-    """
-    Describes how we build our code into a Docker image.
-    """
-
-    type: Literal["build"] = Field(..., description="")
-    docker_registry: Optional[constr(regex=r"^\S+$")] = Field(
-        None,
-        description="FQN of the container registry. If you can't find your registry here,\nadd it through the [Integrations](/integrations?tab=docker-registry) page",
-    )
-    build_source: Union[RemoteSource, GitSource, LocalSource] = Field(
-        ..., description="Source code location."
-    )
-    build_spec: Union[DockerFileBuild, PythonBuild] = Field(
+class BaseService(BaseModel):
+    name: constr(regex=r"^[a-z](?:[a-z0-9]|-(?!-)){1,30}[a-z0-9]$") = Field(
         ...,
-        description="Instructions to build a container image out of the build source",
+        description="Name of the service. This uniquely identifies this service in the workspace.\n> Name can only contain alphanumeric characters and '-' and can be atmost 25 characters long",
     )
-
-
-class ContainerTaskConfig(BaseModel):
-    type: Literal["container-task-config"] = Field(..., description="")
     image: Union[Build, Image] = Field(
         ...,
         description="Specify whether you want to deploy a Docker image or build and deploy from source code",
     )
+    artifacts_download: Optional[ArtifactsDownload] = None
+    resources: Optional[Resources] = None
     env: Optional[Dict[str, str]] = Field(
         None,
-        description="Configure environment variables to be injected in the task either as plain text or secrets. [Docs](https://docs.truefoundry.com/docs/env-variables)",
+        description="Configure environment variables to be injected in the service either as plain text or secrets. [Docs](https://docs.truefoundry.com/docs/env-variables)",
     )
-    resources: Optional[Resources] = None
+    ports: List[Port] = Field(
+        ...,
+        description="Expose the deployment to make it accessible over the internet or keep it private. Implement authentication to restrict access. [Docs](https://docs.truefoundry.com/docs/define-ports-and-domains)",
+    )
+    service_account: Optional[str] = None
     mounts: Optional[List[Union[SecretMount, StringDataMount, VolumeMount]]] = Field(
-        None, description="Configure data to be mounted to Workflow pod(s) as a volume."
+        None,
+        description="Configure data to be mounted to service pod(s) as a string, secret or volume. [Docs](https://docs.truefoundry.com/docs/mounting-volumes-service)",
+    )
+    labels: Optional[Dict[str, str]] = Field(None, description="")
+    kustomize: Optional[Kustomize] = None
+    liveness_probe: Optional[HealthProbe] = None
+    readiness_probe: Optional[HealthProbe] = None
+    startup_probe: Optional[HealthProbe] = None
+    workspace_fqn: Optional[str] = Field(
+        None, description="Fully qualified name of the workspace"
     )
-    service_account: Optional[str] = Field(None, description="")
 
 
 class FlyteLaunchPlan(BaseModel):
@@ -1670,38 +1657,40 @@ class PySparkTaskConfig(BaseModel):
     service_account: Optional[str] = Field(None, description="")
 
 
-class BaseService(BaseModel):
-    name: constr(regex=r"^[a-z](?:[a-z0-9]|-(?!-)){1,30}[a-z0-9]$") = Field(
-        ...,
-        description="Name of the service. This uniquely identifies this service in the workspace.\n> Name can only contain alphanumeric characters and '-' and can be atmost 25 characters long",
-    )
-    image: Union[Build, Image] = Field(
-        ...,
-        description="Specify whether you want to deploy a Docker image or build and deploy from source code",
-    )
-    artifacts_download: Optional[ArtifactsDownload] = None
-    resources: Optional[Resources] = None
-    env: Optional[Dict[str, str]] = Field(
-        None,
-        description="Configure environment variables to be injected in the service either as plain text or secrets. [Docs](https://docs.truefoundry.com/docs/env-variables)",
+class Service(BaseService):
+    """
+    Describes the configuration for the service
+    """
+
+    type: Literal["service"] = Field(..., description="")
+    replicas: Union[confloat(ge=0.0, le=500.0), ServiceAutoscaling] = Field(
+        1,
+        description="Deploy multiple instances of your pods to distribute incoming traffic across them, ensuring effective load balancing.",
     )
-    ports: List[Port] = Field(
-        ...,
-        description="Expose the deployment to make it accessible over the internet or keep it private. Implement authentication to restrict access. [Docs](https://docs.truefoundry.com/docs/define-ports-and-domains)",
+    auto_shutdown: Optional[Autoshutdown] = None
+    allow_interception: bool = Field(
+        False,
+        description="Whether to allow intercepts to be applied for this service.\nThis would inject an additional sidecar in each pod of the service. Not recommended on production",
     )
-    service_account: Optional[str] = None
-    mounts: Optional[List[Union[SecretMount, StringDataMount, VolumeMount]]] = Field(
+    rollout_strategy: Optional[Union[Rolling, Canary, BlueGreen]] = Field(
         None,
-        description="Configure data to be mounted to service pod(s) as a string, secret or volume. [Docs](https://docs.truefoundry.com/docs/mounting-volumes-service)",
+        description="Strategy to dictate how a rollout should happen when a new release for this service is made [Docs](https://docs.truefoundry.com/docs/rollout-strategy)",
     )
-    labels: Optional[Dict[str, str]] = Field(None, description="")
-    kustomize: Optional[Kustomize] = None
-    liveness_probe: Optional[HealthProbe] = None
-    readiness_probe: Optional[HealthProbe] = None
-    startup_probe: Optional[HealthProbe] = None
-    workspace_fqn: Optional[str] = Field(
-        None, description="Fully qualified name of the workspace"
+
+
+class AsyncService(BaseService):
+    """
+    Describes the configuration for the async-service
+    """
+
+    type: Literal["async-service"] = Field(..., description="")
+    replicas: Union[confloat(ge=0.0, le=500.0), AsyncServiceAutoscaling] = Field(
+        1,
+        description="Deploy multiple instances of your pods to distribute incoming traffic across them, ensuring effective load balancing.",
     )
+    rollout_strategy: Optional[Rolling] = None
+    worker_config: WorkerConfig
+    sidecar: Optional[AsyncProcessorSidecar] = None
 
 
 class FlyteTaskCustom(BaseModel):
@@ -1766,40 +1755,30 @@ class Job(BaseModel):
     )
 
 
-class Service(BaseService):
+class ApplicationSet(BaseModel):
     """
-    Describes the configuration for the service
+    Describes the configuration for the application set
     """
 
-    type: Literal["service"] = Field(..., description="")
-    replicas: Union[confloat(ge=0.0, le=500.0), ServiceAutoscaling] = Field(
-        1,
-        description="Deploy multiple instances of your pods to distribute incoming traffic across them, ensuring effective load balancing.",
+    type: Literal["application-set"] = Field(..., description="")
+    name: str = Field(..., description="Name of the application set.")
+    components: Optional[List[Union[Service, AsyncService, Job, Helm]]] = Field(
+        None, description="Array of components with their specifications."
     )
-    auto_shutdown: Optional[Autoshutdown] = None
-    allow_interception: bool = Field(
-        False,
-        description="Whether to allow intercepts to be applied for this service.\nThis would inject an additional sidecar in each pod of the service. Not recommended on production",
+    template: Optional[str] = Field(
+        None, description="Template to be used for the application set."
     )
-    rollout_strategy: Optional[Union[Rolling, Canary, BlueGreen]] = Field(
+    values: Optional[Dict[str, Any]] = Field(
         None,
-        description="Strategy to dictate how a rollout should happen when a new release for this service is made [Docs](https://docs.truefoundry.com/docs/rollout-strategy)",
+        description="Values to be used to render components for the application set.",
     )
-
-
-class AsyncService(BaseService):
-    """
-    Describes the configuration for the async-service
-    """
-
-    type: Literal["async-service"] = Field(..., description="")
-    replicas: Union[confloat(ge=0.0, le=500.0), AsyncServiceAutoscaling] = Field(
-        1,
-        description="Deploy multiple instances of your pods to distribute incoming traffic across them, ensuring effective load balancing.",
+    workspace_fqn: Optional[str] = Field(
+        None, description="Fully qualified name of the workspace"
+    )
+    convert_template_manifest: Optional[bool] = Field(
+        None,
+        description="Flag to indicate if the template manifest should be converted to TrueFoundry manifest",
     )
-    rollout_strategy: Optional[Rolling] = None
-    worker_config: WorkerConfig
-    sidecar: Optional[AsyncProcessorSidecar] = None
 
 
 class FlyteTask(BaseModel):
@@ -1828,32 +1807,6 @@ class Workflow(BaseModel):
     alerts: Optional[List[WorkflowAlert]] = Field(None, description="")
 
 
-class ApplicationSet(BaseModel):
-    """
-    Describes the configuration for the application set
-    """
-
-    type: Literal["application-set"] = Field(..., description="")
-    name: str = Field(..., description="Name of the application set.")
-    components: Optional[List[Union[Service, AsyncService, Job, Helm]]] = Field(
-        None, description="Array of components with their specifications."
-    )
-    template: Optional[str] = Field(
-        None, description="Template to be used for the application set."
-    )
-    values: Optional[Dict[str, Any]] = Field(
-        None,
-        description="Values to be used to render components for the application set.",
-    )
-    workspace_fqn: Optional[str] = Field(
-        None, description="Fully qualified name of the workspace"
-    )
-    convert_template_manifest: Optional[bool] = Field(
-        None,
-        description="Flag to indicate if the template manifest should be converted to TrueFoundry manifest",
-    )
-
-
 class Application(BaseModel):
     __root__: Union[
         Service,

truefoundry/deploy/builder/__init__.py

@@ -2,7 +2,6 @@ from typing import Any, Dict, List, Optional, Union
 
 from truefoundry.deploy._autogen.models import (
     DockerFileBuild,
-    Pip,
     PythonBuild,
     SparkBuild,
     TaskDockerFileBuild,

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

@@ -15,11 +15,11 @@ __all__ = ["generate_dockerfile_content", "build"]
 def _convert_to_dockerfile_build_config(
     build_configuration: PythonBuild,
     dockerfile_path: str,
-    extra_opts: Optional[List[str]] = None,
+    mount_python_package_manager_conf_secret: bool = False,
 ) -> DockerFileBuild:
     dockerfile_content = generate_dockerfile_content(
         build_configuration=build_configuration,
-        docker_build_extra_args=extra_opts,
+        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)
@@ -40,11 +40,14 @@ def build(
         raise ValueError(
             "`python_version` is required for `tfy-python-buildpack` builder"
         )
+    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"),
-            extra_opts=extra_opts,
+            mount_python_package_manager_conf_secret=mount_python_package_manager_conf_secret,
         )
         dockerfile.build(
             tag=tag,