truefoundry 0.3.2__py3-none-any.whl → 0.3.3__py3-none-any.whl

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/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 (

{truefoundry-0.3.2.dist-info → truefoundry-0.3.3.dist-info}/METADATA

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

{truefoundry-0.3.2.dist-info → truefoundry-0.3.3.dist-info}/RECORD

@@ -26,7 +26,7 @@ truefoundry/autodeploy/utils/pydantic_compat.py,sha256=hEAUy5kLjhPdzw7yGZ2iXGMXb
 truefoundry/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/cli/__main__.py,sha256=Jap_IddZ9zNyMIyIkCw75xHQCN0WtV2dPZJ_pzdLsVc,916
 truefoundry/deploy/__init__.py,sha256=ugawKF2G02EmEXX35oZ2tec12d9oWN28Sf6mtGGIERY,2281
-truefoundry/deploy/auto_gen/models.py,sha256=cWY3KE-_NTPrDf1ywgUuOqqV31QQglUM1qeUGMj9mVU,78972
+truefoundry/deploy/auto_gen/models.py,sha256=YhvSvYruchLu4jx9UgKUUWJRekgbdEfVHLhdAkbvHMw,79990
 truefoundry/deploy/builder/__init__.py,sha256=a1qR6nicHGcxRaeNTxWRsmDs8zsmXc7j13-I8q0qqVk,4938
 truefoundry/deploy/builder/builders/__init__.py,sha256=tlFLXqyDaKLd4iZbo4Hcu_8gOmgtL6drnXpbmQ6x1P8,636
 truefoundry/deploy/builder/builders/dockerfile.py,sha256=AXXTziCkaqIhuM_bwyD1vT1znOwemN1TKgU7eyo-KuM,1522
@@ -117,7 +117,7 @@ truefoundry/langchain/utils.py,sha256=XIDpvz3zXpRLeyxLizacqLVDakUJYQZUoXRS1m-tMf
 truefoundry/logger.py,sha256=7dLqW_Q2rEgo-_z1WZnQbGHaoy1L1MP3NqGgssaSS6o,685
 truefoundry/ml/__init__.py,sha256=yFjvF-e1RW488vLHgn5M7TXoajqww6grkKHb3mhqDEw,179
 truefoundry/pydantic_v1.py,sha256=IJy5FtaPg-_H5XLRd9nueTuyijEJPdCa1vZ7-pinmRI,158
-truefoundry/python_deploy_codegen.py,sha256=hd8XRytuT-U6AXSzYSjISbtYY4HLB_B1EeLB5TRDfNw,3947
+truefoundry/python_deploy_codegen.py,sha256=SoXIyPKLCA-VxUlUKt9ZmSrNPVTzHwZISOSZQNf-nUc,4022
 truefoundry/version.py,sha256=bqiT4Q-VWrTC6P4qfK43mez-Ppf-smWfrl6DcwV7mrw,137
 truefoundry/workflow/__init__.py,sha256=m0puY8RJ3GBvdAXW0A9PLuCpmDdLqh6eF0t3DCEpZkg,659
 truefoundry/workflow/container_task.py,sha256=8arieePsX4__OnG337hOtCiNgJwtKJJCsZcmFmCBJtk,402
@@ -130,7 +130,7 @@ truefoundry/workflow/map_task.py,sha256=2m3qGXQ90k9LdS45q8dqCCECc3qr8t2m_LMCVd1m
 truefoundry/workflow/python_task.py,sha256=SRXRLC4vdBqGjhkwuaY39LEWN6iPCpJAuW17URRdWTY,1128
 truefoundry/workflow/task.py,sha256=ToitYiKcNzFCtOVQwz1W8sRjbR97eVS7vQBdbgUQtKg,1779
 truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.3.2.dist-info/METADATA,sha256=gUrAlvlOkQE9AwfRsYRDRTOTH9N6hDS1pgRGemzQbEk,2690
-truefoundry-0.3.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-truefoundry-0.3.2.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.3.2.dist-info/RECORD,,
+truefoundry-0.3.3.dist-info/METADATA,sha256=7d016o0q4MKuad7mbPMg2ZKzPlXGtk4GuffjyEE13K0,2690
+truefoundry-0.3.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+truefoundry-0.3.3.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.3.3.dist-info/RECORD,,