truefoundry 0.7.3rc1__tar.gz → 0.9.0rc1__tar.gz

{truefoundry-0.7.3rc1 → truefoundry-0.9.0rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truefoundry
-Version: 0.7.3rc1
+Version: 0.9.0rc1
 Summary: TrueFoundry CLI
 Author-email: TrueFoundry Team <abhishek@truefoundry.com>
 Requires-Python: <3.14,>=3.8.1
@@ -31,7 +31,7 @@ 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.1.0
+Requires-Dist: truefoundry-sdk<0.2.0,>=0.1.1
 Requires-Dist: typing-extensions>=4.0
 Requires-Dist: urllib3<3,>=1.26.18
 Requires-Dist: yq<4.0.0,>=3.1.0

{truefoundry-0.7.3rc1 → truefoundry-0.9.0rc1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "truefoundry"
-version = "0.7.3rc1"  # This is a placeholder version, it will be replaced when releasing
+version = "0.9.0rc1"  # 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" }]
@@ -32,7 +32,7 @@ dependencies = [
   "rich-click>=1.2.1,<2.0.0",
   "rich>=13.7.1,<14.0.0",
   "tqdm>=4.0.0,<5.0.0",
-  "truefoundry-sdk==0.1.0",
+  "truefoundry-sdk>=0.1.1,<0.2.0",
   "typing-extensions>=4.0",
   "urllib3>=1.26.18,<3",
   "yq>=3.1.0,<4.0.0",

{truefoundry-0.7.3rc1 → truefoundry-0.9.0rc1}/truefoundry/cli/util.py

@@ -51,6 +51,9 @@ def handle_exception(exception):
             title="Command Failed",
             border_style="red",
         )
+    elif isinstance(exception, KeyError):
+        # KeyError messages in Python just suck - they tell you the key that was missing, but not the context
+        console.print_exception(show_locals=False, max_frames=1)
     else:
         print_dict_as_table_panel(
             {"Error": str(exception)},

{truefoundry-0.7.3rc1 → truefoundry-0.9.0rc1}/truefoundry/common/entities.py

@@ -109,6 +109,9 @@ class PythonSDKConfig(BaseModel):
     use_sfy_server_auth_apis: Optional[bool] = Field(
         alias="useSFYServerAuthAPIs", default=False
     )
+    python_build_default_image_tag: str = Field(
+        alias="pythonBuildDefaultImageTag", default="3.11"
+    )
 
 
 class DeviceCode(BaseModel):

{truefoundry-0.7.3rc1 → truefoundry-0.9.0rc1}/truefoundry/deploy/_autogen/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2025-05-03T01:45:46+00:00
+#   timestamp: 2025-05-15T05:28:11+00:00
 
 from __future__ import annotations
 
@@ -360,13 +360,6 @@ class Image(BaseModel):
     )
 
 
-class JobEvent(str, Enum):
-    START = "START"
-    SUCCEEDED = "SUCCEEDED"
-    FAILED = "FAILED"
-    TERMINATED = "TERMINATED"
-
-
 class Claim(BaseModel):
     key: str
     values: List[str]
@@ -623,9 +616,11 @@ class PythonBuild(BaseModel):
     """
 
     type: Literal["tfy-python-buildpack"] = Field(..., description="")
-    python_version: constr(regex=r"^\d+(\.\d+){1,2}([\-\.a-z0-9]+)?$") = Field(
-        "3.11",
-        description="Python version to run your application. Should be one of the tags listed on [Official Python Docker Page](https://hub.docker.com/_/python)",
+    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."
@@ -963,9 +958,11 @@ class TaskPythonBuild(BaseModel):
         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",
     )
-    python_version: constr(regex=r"^\d+(\.\d+){1,2}([\-\.a-z0-9]+)?$") = Field(
-        "3.11",
-        description="Python version to run your application. Should be one of the tags listed on [Official Python Docker Page](https://hub.docker.com/_/python)",
+    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)",
+        )
     )
     requirements_path: Optional[str] = Field(
         None,
@@ -1060,13 +1057,6 @@ 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)
@@ -1485,10 +1475,6 @@ class WorkflowAlert(BaseModel):
     """
 
     notification_target: Optional[Union[Email, SlackWebhook, SlackBot]] = None
-    events: List[WorkflowEvent] = Field(
-        ...,
-        description="Specify the events to send alerts for, it should be one of the following: SUCCEEDED, FAILED, ABORTED, TIMED_OUT",
-    )
     on_completion: bool = Field(
         False, description="Send an alert when the job completes"
     )
@@ -1556,10 +1542,6 @@ class JobAlert(BaseModel):
         description="List of recipients' email addresses if the notification channel is Email.",
     )
     notification_target: Optional[Union[Email, SlackWebhook, SlackBot]] = None
-    events: List[JobEvent] = Field(
-        ...,
-        description="Specify the events to send alerts for, it should be one of the following: START, SUCCEEDED, FAILED, TERMINATED",
-    )
     on_start: bool = Field(False, description="Send an alert when the job starts")
     on_completion: bool = False
     on_failure: bool = Field(True, description="Send an alert when the job fails")

{truefoundry-0.7.3rc1 → truefoundry-0.9.0rc1}/truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py

@@ -36,6 +36,10 @@ def build(
     build_configuration: PythonBuild,
     extra_opts: Optional[List[str]] = None,
 ):
+    if not build_configuration.python_version:
+        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
     )

{truefoundry-0.7.3rc1 → truefoundry-0.9.0rc1}/truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py

@@ -174,6 +174,10 @@ def generate_dockerfile_content(
     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"
+        )
     if package_manager == PythonPackageManager.PIP.value:
         python_packages_install_command = generate_pip_install_command(
             requirements_path=requirements_destination_path,

{truefoundry-0.7.3rc1 → truefoundry-0.9.0rc1}/truefoundry/deploy/v2/lib/deploy.py

@@ -25,6 +25,19 @@ from truefoundry.pydantic_v1 import BaseModel
 Component = TypeVar("Component", bound=BaseModel)
 
 
+def _set_python_version_if_missing(build_spec: autogen_models.PythonBuild) -> None:
+    if build_spec.python_version is None:
+        client = ServiceFoundryServiceClient()
+        server_default_python_version = (
+            client.python_sdk_config.python_build_default_image_tag
+        )
+        logger.warning(
+            f"No python version was provided in the spec, "
+            f"using the default python version ({server_default_python_version}) from the server."
+        )
+        build_spec.python_version = server_default_python_version
+
+
 def _handle_if_local_source(component: Component, workspace_fqn: str) -> Component:
     if (
         hasattr(component, "image")
@@ -33,7 +46,7 @@ def _handle_if_local_source(component: Component, workspace_fqn: str) -> Compone
     ):
         new_component = component.copy(deep=True)
 
-        if component.image.build_source.local_build:
+        if new_component.image.build_source.local_build:
             if not env_has_docker():
                 logger.warning(
                     "Did not find Docker locally installed on this system, image will be built remotely. "
@@ -64,25 +77,33 @@ def _handle_if_local_source(component: Component, workspace_fqn: str) -> Compone
             local_build = False
 
         if local_build:
+            if isinstance(new_component.image.build_spec, autogen_models.PythonBuild):
+                _set_python_version_if_missing(new_component.image.build_spec)
             # We are to build the image locally, push and update `image` in spec
-            logger.info("Building image for %s '%s'", component.type, component.name)
+            logger.info(
+                "Building image for %s '%s'", new_component.type, new_component.name
+            )
             new_component.image = local_source_to_image(
-                build=component.image,
-                docker_registry_fqn=component.image.docker_registry,
+                build=new_component.image,
+                docker_registry_fqn=new_component.image.docker_registry,
                 workspace_fqn=workspace_fqn,
-                component_name=component.name,
+                component_name=new_component.name,
             )
         else:
             # We'll build image on TrueFoundry servers, upload the source and update image.build_source
-            logger.info("Uploading code for %s '%s'", component.type, component.name)
+            logger.info(
+                "Uploading code for %s '%s'", new_component.type, new_component.name
+            )
             client = ServiceFoundryServiceClient()
             new_component.image.build_source = local_source_to_remote_source(
-                local_source=component.image.build_source,
+                local_source=new_component.image.build_source,
                 workspace_fqn=workspace_fqn,
-                component_name=component.name,
+                component_name=new_component.name,
                 upload_code_package=client.upload_code_package,
             )
-            logger.debug("Uploaded code for %s '%s'", component.type, component.name)
+            logger.debug(
+                "Uploaded code for %s '%s'", new_component.type, new_component.name
+            )
         return new_component
     return component