truefoundry 0.6.5__tar.gz → 0.7.0rc2__tar.gz

{truefoundry-0.6.5 → truefoundry-0.7.0rc2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truefoundry
-Version: 0.6.5
+Version: 0.7.0rc2
 Summary: TrueFoundry CLI
 Author-email: TrueFoundry Team <abhishek@truefoundry.com>
 Requires-Python: <3.14,>=3.8.1
@@ -35,7 +35,7 @@ Requires-Dist: typing-extensions>=4.0
 Requires-Dist: urllib3<3,>=1.26.18
 Requires-Dist: yq<4.0.0,>=3.1.0
 Provides-Extra: workflow
-Requires-Dist: flytekit==1.13.13; extra == 'workflow'
+Requires-Dist: flytekit==1.15.3; (python_version >= '3.9' and python_version < '3.13') and extra == 'workflow'
 Description-Content-Type: text/markdown
 
 # TrueFoundry

{truefoundry-0.6.5 → truefoundry-0.7.0rc2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "truefoundry"
-version = "0.6.5"  # This is a placeholder version, it will be replaced when releasing
+version = "0.7.0rc2"  # This is a placeholder version, it will be replaced when releasing
 requires-python = ">=3.8.1,<3.14"
 description = "TrueFoundry CLI"
 authors = [{ name = "TrueFoundry Team", email = "abhishek@truefoundry.com" }]
@@ -40,7 +40,7 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
-workflow = ["flytekit==1.13.13"]
+workflow = ["flytekit==1.15.3; python_version >= '3.9' and python_version < '3.13'"]
 
 [project.scripts]
 tfy = "truefoundry.cli.__main__:main"

{truefoundry-0.6.5 → truefoundry-0.7.0rc2}/truefoundry/__init__.py

@@ -1,10 +1,15 @@
 from truefoundry_sdk import (
     AssistantMessage,
     ChatPromptManifest,
+    DeveloperMessage,
+    FunctionSchema,
     ModelConfiguration,
     Parameters,
     PromptVersion,
     SystemMessage,
+    ToolCall,
+    ToolMessage,
+    ToolSchema,
     UserMessage,
 )
 
@@ -17,10 +22,13 @@ from truefoundry.deploy.core import login, logout
 from truefoundry.ml.prompt_utils import render_prompt
 
 surface_truefoundry_deprecation_warnings()
+
 __all__ = [
     "AssistantMessage",
     "ChatPromptManifest",
     "client",
+    "DeveloperMessage",
+    "FunctionSchema",
     "login",
     "logout",
     "ModelConfiguration",
@@ -29,5 +37,8 @@ __all__ = [
     "render_prompt",
     "suppress_truefoundry_deprecation_warnings",
     "SystemMessage",
+    "ToolCall",
+    "ToolMessage",
+    "ToolSchema",
     "UserMessage",
 ]

{truefoundry-0.6.5 → truefoundry-0.7.0rc2}/truefoundry/deploy/__init__.py

@@ -111,6 +111,7 @@ from truefoundry.deploy.v2.lib.patched_models import (
     Schedule,
     SecretMount,
     ServiceAutoscaling,
+    SlackBot,
     SlackWebhook,
     SparkJobJavaEntrypoint,
     SparkJobPythonEntrypoint,

{truefoundry-0.6.5 → truefoundry-0.7.0rc2}/truefoundry/deploy/_autogen/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2025-04-07T11:22:45+00:00
+#   timestamp: 2025-04-21T06:26:10+00:00
 
 from __future__ import annotations
 
@@ -360,6 +360,13 @@ class Image(BaseModel):
     )
 
 
+class JobEvent(str, Enum):
+    START = "START"
+    SUCCEEDED = "SUCCEEDED"
+    FAILED = "FAILED"
+    TERMINATED = "TERMINATED"
+
+
 class Claim(BaseModel):
     key: str
     values: List[str]
@@ -466,10 +473,6 @@ class NodeSelector(BaseModel):
     """
 
     type: Literal["node_selector"] = Field(..., description="")
-    gpu_type: Optional[str] = Field(
-        None,
-        description="Name of the Nvidia GPU. One of [P4, P100, V100, T4, A10G, A100_40GB, A100_80GB]\nOne instance of the card contains the following amount of memory -\nP4: 8 GB, P100: 16 GB, V100: 16 GB, T4: 16 GB, A10G: 24 GB, A100_40GB: 40GB, A100_80GB: 80 GB",
-    )
     instance_families: Optional[List[str]] = Field(
         None,
         description="Instance family of the underlying machine to use. Multiple instance families can be supplied.\nThe workload is guaranteed to be scheduled on one of them.",
@@ -699,10 +702,6 @@ class Resources(BaseModel):
         2000,
         description="Disk storage limit. The unit of memory is in megabytes(MB). Exceeding this limit will result in eviction.\nIt should be greater than the request. This is ephemeral storage and will be wiped out on pod restarts or eviction",
     )
-    gpu_count: Optional[conint(ge=0, le=16)] = Field(
-        None,
-        description="Count of GPUs to provide to the application\nNote the exact count and max count available for a given GPU type depends on cloud provider and cluster type.",
-    )
     shared_memory_size: Optional[conint(ge=64, le=2000000)] = Field(
         None,
         description="Define the shared memory requirements for your workload. Machine learning libraries like Pytorch can use Shared Memory\nfor inter-process communication. If you use this, we will mount a `tmpfs` backed volume at the `/dev/shm` directory.\nAny usage will also count against the workload's memory limit (`resources.memory_limit`) along with your workload's memory usage.\nIf the overall usage goes above `resources.memory_limit` the user process may get killed.\nShared Memory Size cannot be more than the defined Memory Limit for the workload.",
@@ -829,6 +828,16 @@ class ServiceAutoscaling(BaseAutoscaling):
     )
 
 
+class SlackBot(BaseModel):
+    type: Literal["slack-bot"] = Field(..., description="")
+    notification_channel: constr(min_length=1) = Field(
+        ..., description="Specify the notification channel to send alerts to"
+    )
+    channels: List[constr(regex=r"^#[a-z0-9\-_]{2,80}$")] = Field(
+        ..., description="List of channels to send messages to."
+    )
+
+
 class SlackWebhook(BaseModel):
     type: Literal["slack-webhook"] = Field(..., description="")
     notification_channel: constr(min_length=1) = Field(
@@ -856,6 +865,26 @@ class SparkExecutorFixedInstances(BaseModel):
     count: conint(ge=0, le=500) = Field(1, description="Number of instances to start")
 
 
+class SparkImage(BaseModel):
+    """
+    Describes that we are using a pre-built image stored in a Docker Image registry
+    """
+
+    type: Literal["image"] = Field(..., description="")
+    spark_version: str = Field(
+        "3.5.2",
+        description="Spark version should match the spark version installed in the image.",
+    )
+    image_uri: constr(regex=r"^\S*$") = Field(
+        ...,
+        description="The image URI. Specify the name of the image and the tag.\nIf the image is in Dockerhub, you can skip registry-url (for e.g. `tensorflow/tensorflow`).\nYou can use an image from a private registry using Advanced fields",
+    )
+    docker_registry: Optional[str] = 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",
+    )
+
+
 class SparkJobJavaEntrypoint(BaseModel):
     type: Literal["java"] = Field(..., description="")
     main_application_file: str = Field(
@@ -1031,6 +1060,13 @@ class WorkbenchImage(BaseModel):
     )
 
 
+class WorkflowEvent(str, Enum):
+    SUCCEEDED = "SUCCEEDED"
+    FAILED = "FAILED"
+    ABORTED = "ABORTED"
+    TIMED_OUT = "TIMED_OUT"
+
+
 class ArtifactsDownload(BaseModel):
     """
     Download and cache models in a volume to enhance loading speeds and reduce costs by avoiding repeated downloads. [Docs](https://docs.truefoundry.com/docs/download-and-cache-models)
@@ -1385,14 +1421,10 @@ class SparkJob(BaseModel):
     name: constr(regex=r"^[a-z](?:[a-z0-9]|-(?!-)){1,30}[a-z0-9]$") = Field(
         ..., description="Name of the job"
     )
+    image: SparkImage
     entrypoint: Union[
         SparkJobPythonEntrypoint, SparkJobScalaEntrypoint, SparkJobJavaEntrypoint
     ] = Field(..., description="")
-    image: Image
-    spark_version: str = Field(
-        "3.5.2",
-        description="Spark version should match the spark version installed in the image.",
-    )
     driver_config: SparkDriverConfig
     executor_config: SparkExecutorConfig
     env: Optional[Dict[str, Any]] = Field(
@@ -1452,7 +1484,7 @@ class WorkflowAlert(BaseModel):
     Describes the configuration for the workflow alerts
     """
 
-    notification_target: Optional[Union[Email, SlackWebhook]] = None
+    notification_target: Optional[Union[Email, SlackWebhook, SlackBot]] = None
     on_completion: bool = Field(
         False, description="Send an alert when the job completes"
     )
@@ -1519,7 +1551,7 @@ class JobAlert(BaseModel):
         None,
         description="List of recipients' email addresses if the notification channel is Email.",
     )
-    notification_target: Optional[Union[Email, SlackWebhook]] = None
+    notification_target: Optional[Union[Email, SlackWebhook, SlackBot]] = None
     on_start: bool = Field(False, description="Send an alert when the job starts")
     on_completion: bool = Field(
         False, description="Send an alert when the job completes"

{truefoundry-0.6.5 → truefoundry-0.7.0rc2}/truefoundry/deploy/v2/lib/deploy.py

@@ -189,30 +189,6 @@ def _deploy_wait_handler(  # noqa: C901
     return last_status_printed
 
 
-def _warn_when_gpu_selected_without_cuda(component: Component):
-    is_python_build_without_cuda = (
-        hasattr(component, "image")
-        and isinstance(component.image, autogen_models.Build)
-        and isinstance(component.image.build_spec, autogen_models.PythonBuild)
-        and not component.image.build_spec.cuda_version
-    )
-    uses_gpu = (
-        hasattr(component, "resources")
-        and isinstance(component, autogen_models.Resources)
-        and component.resources.gpu_count > 0
-    )
-    if is_python_build_without_cuda and uses_gpu:
-        logger.warning(
-            "Warning: `gpu_count` is greater than 0 in `Resources` (i.e. `resources.gpu_count`) "
-            "but no `cuda_version` was passed to `PythonBuild` "
-            "(i.e. `image.build_spec.cuda_version`). "
-            "Your application might optionally need CUDA toolkit installed "
-            "to utilize the GPU. You can choose one by passing one of "
-            "`truefoundry.deploy.CUDAVersion` in `PythonBuild` instance. "
-            "\n\nE.g.\n```\nPythonBuild(..., cuda_version=CUDAVersion.CUDA_11_3_CUDNN8)\n```"
-        )
-
-
 def _resolve_workspace_fqn(
     component: Component, workspace_fqn: Optional[str] = None
 ) -> str:
@@ -266,7 +242,6 @@ def deploy_component(
     wait: bool = True,
     force: bool = False,
 ) -> Deployment:
-    _warn_when_gpu_selected_without_cuda(component=component)
     workspace_fqn = _resolve_workspace_fqn(
         component=component, workspace_fqn=workspace_fqn
     )

{truefoundry-0.6.5 → truefoundry-0.7.0rc2}/truefoundry/deploy/v2/lib/patched_models.py

@@ -2,7 +2,6 @@ import enum
 import os
 import re
 import warnings
-from collections.abc import Mapping
 from typing import Literal, Optional, Union
 
 from truefoundry.common.warnings import TrueFoundryDeprecationWarning
@@ -15,63 +14,6 @@ from truefoundry.pydantic_v1 import (
     validator,
 )
 
-LEGACY_GPU_TYPE_COUNT_WARNING_MESSAGE_TEMPLATE = """
----------
-The `gpu_count` and `gpu_type` fields are deprecated. Please remove these fields
-from your deployment Python script or YAML Spec.
-
-If you are using Python SDK, add GPUs in the following way:
-
-```
-from truefoundry.deploy import NvidiaGPU, Resources
-...
-
-  resources=Resources(
-    ...
-    devices=[NvidiaGPU(name="{gpu_type}", count={gpu_count})],
-  )
-```
-
-If you are using YAML Spec to deploy, add GPUs in the following way:
-
-```
-resources:
-  devices:
-    - type: nvidia_gpu
-      name: {gpu_type}
-      count: {gpu_count}
-```
----------
-"""
-
-LEGACY_GPU_COUNT_WARNING_MESSAGE_TEMPLATE = """
----------
-The `gpu_count` field is deprecated. Please remove this field from your
-deployment Python script or YAML Spec.
-
-If you are using Python SDK, add GPUs in the following way:
-
-```
-from truefoundry.deploy import NvidiaGPU, Resources
-...
-
-  resources=Resources(
-    ...
-    devices=[NvidiaGPU(count={gpu_count})],
-  )
-```
-
-If you are using YAML Spec to deploy, add GPUs in the following way:
-
-```
-resources:
-  devices:
-    - type: nvidia_gpu
-      count: {gpu_count}
-```
----------
-"""
-
 AUTO_DISCOVERED_REQUIREMENTS_TXT_WARNING_MESSAGE_TEMPLATE = """\
 Using automatically discovered {requirements_txt_path} as the requirements file.
 This auto discovery behavior is deprecated and will be removed in a future release.
@@ -304,35 +246,7 @@ class Port(models.Port, PatchedModelBase):
 
 
 class Resources(models.Resources, PatchedModelBase):
-    @root_validator(pre=False)
-    def warn_gpu_count_type_depreciation(cls, values):
-        gpu_count = values.get("gpu_count")
-        gpu_type = None
-        node = values.get("node")
-        if node:
-            if isinstance(node, NodeSelector):
-                gpu_type = node.gpu_type
-            elif isinstance(node, Mapping):
-                gpu_count = node.get("gpu_type")
-
-        if gpu_count and gpu_type:
-            warnings.warn(
-                LEGACY_GPU_TYPE_COUNT_WARNING_MESSAGE_TEMPLATE.format(
-                    gpu_type=gpu_type,
-                    gpu_count=gpu_count,
-                ),
-                category=TrueFoundryDeprecationWarning,
-                stacklevel=2,
-            )
-        elif gpu_count:
-            warnings.warn(
-                LEGACY_GPU_COUNT_WARNING_MESSAGE_TEMPLATE.format(
-                    gpu_count=gpu_count,
-                ),
-                category=TrueFoundryDeprecationWarning,
-                stacklevel=2,
-            )
-        return values
+    pass
 
 
 class Param(models.Param, PatchedModelBase):
@@ -397,7 +311,6 @@ class VolumeMount(models.VolumeMount, PatchedModelBase):
 
 class NodeSelector(models.NodeSelector, PatchedModelBase):
     type: Literal["node_selector"] = "node_selector"
-    gpu_type: Optional[Union[GPUType, str]] = None
 
 
 class NodepoolSelector(models.NodepoolSelector, PatchedModelBase):
@@ -573,6 +486,10 @@ class SlackWebhook(models.SlackWebhook, PatchedModelBase):
     type: Literal["slack-webhook"] = "slack-webhook"
 
 
+class SlackBot(models.SlackBot, PatchedModelBase):
+    type: Literal["slack-bot"] = "slack-bot"
+
+
 class SparkJobScalaEntrypoint(models.SparkJobScalaEntrypoint, PatchedModelBase):
     type: Literal["scala"] = "scala"