truefoundry 0.3.2__tar.gz → 0.3.3rc1__tar.gz

{truefoundry-0.3.2 → truefoundry-0.3.3rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.3.2
+Version: 0.3.3rc1
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com

{truefoundry-0.3.2 → truefoundry-0.3.3rc1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truefoundry"
-version = "0.3.2"
+version = "0.3.3rc1"
 description = "Truefoundry CLI"
 authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
 readme = "README.md"

{truefoundry-0.3.2 → truefoundry-0.3.3rc1}/truefoundry/deploy/auto_gen/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2024-09-10T13:19:47+00:00
+#   timestamp: 2024-09-19T11:05:06+00:00
 
 from __future__ import annotations
 
@@ -444,6 +444,30 @@ class Image(BaseModel):
     )
 
 
+class JobAlert(BaseModel):
+    """
+    +docs=Describes the configuration for the job alerts
+    +label=Alert
+    """
+
+    notification_channel: str = Field(
+        ...,
+        description='+label=Notification Channel\n+usage=Specify the notification channel to send alerts to\n+uiType=IntegrationSelect\n+uiProps={"integrationType":"notification-channel"}\n+sort=660',
+    )
+    on_start: bool = Field(
+        False,
+        description="+label=On Start\n+usage=Send an alert when the job starts\n+sort=670",
+    )
+    on_completion: bool = Field(
+        False,
+        description="+label=On Completion\n+usage=Send an alert when the job completes\n+sort=680",
+    )
+    on_failure: bool = Field(
+        True,
+        description="+label=On Failure\n+usage=Send an alert when the job fails\n+sort=690",
+    )
+
+
 class KafkaMetricConfig(BaseModel):
     type: Literal["kafka"] = Field(..., description="+value=kafka")
     lag_threshold: conint(ge=1) = Field(
@@ -1350,6 +1374,10 @@ class Job(BaseModel):
         description="+label=Environment Variables\n+usage=Configure environment variables to be injected in the service either as plain text or secrets. [Docs](https://docs.truefoundry.com/docs/env-variables)\n+icon=fa-globe\n+sort=500",
     )
     resources: Optional[Resources] = None
+    alerts: Optional[List[JobAlert]] = Field(
+        None,
+        description="+label=Alerts\n+usage=Configure alerts to be sent when the job starts/fails/completes\n+icon=fa-bell\n+sort=650",
+    )
     retries: conint(ge=0, le=10) = Field(
         0,
         description="+label=Retries\n+usage=Specify the maximum number of attempts to retry a job before it is marked as failed.\n+icon=fa-repeat\n+sort=700",

{truefoundry-0.3.2 → truefoundry-0.3.3rc1}/truefoundry/deploy/cli/util.py

@@ -36,7 +36,9 @@ def handle_exception(exception):
         )
     elif isinstance(exception, ConnectionError):
         print_dict_as_table_panel(
-            {"Error": "Couldn't connect to TrueFoundry"},
+            {
+                "Error": "Couldn't connect to TrueFoundry. Please make sure that the provided `--host` is correct."
+            },
             title="Command Failed",
             border_style="red",
         )

{truefoundry-0.3.2 → truefoundry-0.3.3rc1}/truefoundry/deploy/v2/lib/deployable_patched_models.py

@@ -3,6 +3,7 @@ from typing import Literal, Union
 from truefoundry.deploy.auto_gen import models
 from truefoundry.deploy.lib.model.entity import Deployment
 from truefoundry.deploy.v2.lib.deploy import deploy_component
+from truefoundry.deploy.v2.lib.patched_models import LocalSource
 from truefoundry.pydantic_v1 import BaseModel, Field, conint
 
 
@@ -79,6 +80,9 @@ class Application(models.Application, DeployablePatchedModelBase):
 
 class Workflow(models.Workflow, DeployablePatchedModelBase):
     type: Literal["workflow"] = "workflow"
+    source: Union[models.RemoteSource, models.LocalSource] = Field(
+        default_factory=LocalSource
+    )
 
     def deploy(self, workspace_fqn: str, wait: bool = True) -> Deployment:
         from truefoundry.deploy.v2.lib.deploy_workflow import deploy_workflow

{truefoundry-0.3.2 → truefoundry-0.3.3rc1}/truefoundry/python_deploy_codegen.py

@@ -16,6 +16,7 @@ def generate_code(
     workspace_fqn: str,
 ):
     symbols = ",".join(symbols_to_import)
+    application_type = application_type.replace(" ", "").replace("-", "_")
     code = f"""\
 import logging
 from truefoundry.deploy import (