truefoundry 0.9.0__py3-none-any.whl → 0.9.1__py3-none-any.whl

truefoundry/deploy/cli/commands/deploy_command.py

@@ -48,7 +48,8 @@ def _get_default_spec_file():
 )
 @click.option(
     "--wait/--no-wait",
-    "--wait/--no_wait",
+    # The leading space is intentional. See: https://github.com/pallets/click/blob/2d610e36a429bfebf0adb0ca90cdc0585f296369/docs/options.md?plain=1#L295
+    " /--no_wait",
     is_flag=True,
     show_default=True,
     default=True,

truefoundry/deploy/cli/commands/patch_application_command.py

@@ -39,7 +39,8 @@ from truefoundry.deploy.lib.dao import application as application_lib
 )
 @click.option(
     "--wait/--no-wait",
-    "--wait/--no_wait",
+    # The leading space is intentional. See: https://github.com/pallets/click/blob/2d610e36a429bfebf0adb0ca90cdc0585f296369/docs/options.md?plain=1#L295
+    " /--no_wait",
     is_flag=True,
     show_default=True,
     default=True,

truefoundry/deploy/v2/lib/deployable_patched_models.py

@@ -79,7 +79,46 @@ class SSHServer(models.SSHServer, DeployablePatchedModelBase):
     resources: models.Resources = Field(default_factory=models.Resources)
 
 
+class Workflow(models.Workflow, DeployablePatchedModelBase):
+    type: Literal["workflow"] = "workflow"
+    source: Union[models.RemoteSource, models.LocalSource] = Field(
+        default_factory=lambda: LocalSource(local_build=False)
+    )
+
+    def deploy(
+        self, workspace_fqn: str, wait: bool = True, force: bool = False
+    ) -> Deployment:
+        from truefoundry.deploy.v2.lib.deploy_workflow import deploy_workflow
+
+        return deploy_workflow(
+            workflow=self, workspace_fqn=workspace_fqn, wait=wait, force=force
+        )
+
+
 class Application(models.Application, DeployablePatchedModelBase):
+    # We need a discriminator field to the root model to simplify the Validation errors
+    # Unfortunately cue export cannot add discriminator in OAS
+    # Even if we add it manually in OAS, `datamodel-code-generator` has bugs when discriminator field is enum type in member models.
+    # It will change the members to be incorrect like this
+    # class Service(BaseModel):
+    #     type: Literal["Service"] = Field("Service")  # notice the capital casing
+    # This is why we add it manually here
+
+    __root__: Union[
+        models.Service,
+        models.AsyncService,
+        models.Job,
+        models.Notebook,
+        models.Codeserver,
+        models.SSHServer,
+        models.RStudio,
+        models.Helm,
+        models.Volume,
+        models.ApplicationSet,
+        models.Workflow,
+        models.SparkJob,
+    ] = Field(..., description="", discriminator="type")
+
     def deploy(
         self, workspace_fqn: str, wait: bool = True, force: bool = False
     ) -> Deployment:
@@ -99,19 +138,3 @@ class Application(models.Application, DeployablePatchedModelBase):
                 wait=wait,
                 force=force,
             )
-
-
-class Workflow(models.Workflow, DeployablePatchedModelBase):
-    type: Literal["workflow"] = "workflow"
-    source: Union[models.RemoteSource, models.LocalSource] = Field(
-        default_factory=lambda: LocalSource(local_build=False)
-    )
-
-    def deploy(
-        self, workspace_fqn: str, wait: bool = True, force: bool = False
-    ) -> Deployment:
-        from truefoundry.deploy.v2.lib.deploy_workflow import deploy_workflow
-
-        return deploy_workflow(
-            workflow=self, workspace_fqn=workspace_fqn, wait=wait, force=force
-        )

truefoundry/workflow/__init__.py

@@ -9,6 +9,7 @@ except ImportError:
 
 from flytekit import conditional
 from flytekit.types.directory import FlyteDirectory
+from flytekit.types.error.error import FlyteError
 from flytekit.types.file import FlyteFile
 
 from truefoundry.common.constants import ENV_VARS
@@ -39,6 +40,7 @@ __all__ = [
     "PythonTaskConfig",
     "ExecutionConfig",
     "FlyteFile",
+    "FlyteError",
 ]
 
 

{truefoundry-0.9.0.dist-info → truefoundry-0.9.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truefoundry
-Version: 0.9.0
+Version: 0.9.1
 Summary: TrueFoundry CLI
 Author-email: TrueFoundry Team <abhishek@truefoundry.com>
 Requires-Python: <3.14,>=3.8.1

{truefoundry-0.9.0.dist-info → truefoundry-0.9.1.dist-info}/RECORD

@@ -67,7 +67,7 @@ truefoundry/deploy/cli/commands/apply_command.py,sha256=Y2e_C8HVpo8CssVod-3JRz-8
 truefoundry/deploy/cli/commands/ask_command.py,sha256=9TDZVKEKhBvNx3NVz51bbkbbeeEMI3Xc85THnOGHM1g,5970
 truefoundry/deploy/cli/commands/build_command.py,sha256=zJBywMatbpUlXx5O2aqpEVmPeBIJ9RNnG9abSc8C8CE,1234
 truefoundry/deploy/cli/commands/delete_command.py,sha256=i_lr_MocTEPKF2VwLe8B7oZWsgXK06EX_43_xdM5DIs,3875
-truefoundry/deploy/cli/commands/deploy_command.py,sha256=8aTBvzPaT9xg6KPmpcpqJlmdj4yXzWUfAy6slcoPN74,4123
+truefoundry/deploy/cli/commands/deploy_command.py,sha256=qpozW6RQa8EcLhrKtNrCoorBZU7p75XrmMUPX1c3EVM,4271
 truefoundry/deploy/cli/commands/deploy_init_command.py,sha256=g-jBfrEmhZ0TDWsyqPDn4K6q33EqJSGmBTt1eMYig-w,600
 truefoundry/deploy/cli/commands/get_command.py,sha256=bR8tAjQQhimzaTQ57L6BPJwcxQ_SGWCF5CqHDpxgG90,837
 truefoundry/deploy/cli/commands/k8s_exec_credential_command.py,sha256=EknpdufMAEnjSGMG7a-Jj7tkoiS5zmbJRREafb14Alw,2160
@@ -75,7 +75,7 @@ truefoundry/deploy/cli/commands/kubeconfig_command.py,sha256=v6LmDyf2YQPxPrueGlz
 truefoundry/deploy/cli/commands/login_command.py,sha256=kbEs4leyMYK2kz7L9ql2PXVgLVmCYo-LWtnntVVYLFY,1065
 truefoundry/deploy/cli/commands/logout_command.py,sha256=u3kfrEp0ETbrz40KjD4GCC3XEZ5YRAlrca_Df4U_mk0,536
 truefoundry/deploy/cli/commands/logs_command.py,sha256=osl2z5VaIceB9sYa6GtwsuyAPZKcw-A0oVEt3g1f62Q,4140
-truefoundry/deploy/cli/commands/patch_application_command.py,sha256=YdTlkWGI2gCMoVAgPJtybjuvRFqZq9tp6TqbCpwgXUY,2443
+truefoundry/deploy/cli/commands/patch_application_command.py,sha256=aRTHu2OmxQd7j9iE0RavsFCkCILp0rGh4DJO51Oij5I,2591
 truefoundry/deploy/cli/commands/patch_command.py,sha256=wA95khMO9uVz8SaJlgYMUwaX7HagtchjyxXXATq83Bk,1665
 truefoundry/deploy/cli/commands/terminate_comand.py,sha256=UKhOdbAej8ubX3q44vpLrOotAcvH4vHpRZJQrRf_AfM,1077
 truefoundry/deploy/cli/commands/trigger_command.py,sha256=_qSl-AShepZpbGUGTfLfJGd74VJJ_wd3eXYt2DfxIFo,4716
@@ -109,7 +109,7 @@ truefoundry/deploy/v2/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3h
 truefoundry/deploy/v2/lib/__init__.py,sha256=WEiVMZXOVljzEE3tpGJil14liIn_PCDoACJ6b3tZ6sI,188
 truefoundry/deploy/v2/lib/deploy.py,sha256=kAnh6RO4ci7AVjlIoN1Sr5FmcOU7nbkwNvbrS802spY,12625
 truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=G5BzMIbap8pgDX1eY-TITruUxQdkKhYtBmRwLL6lDeY,14342
-truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=xbHFD3pURflvCm8EODPvjfvRrv67mlSrjPUknY8SMB8,4060
+truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=xy7AGYmVpyk12RMrXxOHHkpNKpttPFi6AiLiKdpID20,4940
 truefoundry/deploy/v2/lib/models.py,sha256=ogc1UYs1Z2nBdGSKCrde9sk8d0GxFKMkem99uqO5CmM,1148
 truefoundry/deploy/v2/lib/patched_models.py,sha256=8ib9Y7b4-DoEml2zCv3V7QIqh4tLJUjzPj1AWomwvag,14775
 truefoundry/deploy/v2/lib/source.py,sha256=d6-8_6Zn5koBglqrBrY6ZLG_7yyPuLdyEmK4iZTw6xY,9405
@@ -365,7 +365,7 @@ truefoundry/ml/log_types/image/constants.py,sha256=wLtGEOA4T5fZHSlOXPuNDLX3lpbCt
 truefoundry/ml/log_types/image/image.py,sha256=sa0tBHdyluC8bELXY16E0HgFrUDnDBxHrteix4BFXcs,12479
 truefoundry/ml/log_types/image/image_normalizer.py,sha256=vrzfuSpVGgIxw_Q2sbFe7kQ_JpAndX0bMwC7wtfi41g,3104
 truefoundry/ml/log_types/image/types.py,sha256=inFQlyAyDvZtfliFpENirNCm1XO9beyZ8DNn97DoDKs,1568
-truefoundry/workflow/__init__.py,sha256=MNxnOh5fzAmDaK-cJy9qwtA-zH2CdOEP2h0q9lEiLgY,1588
+truefoundry/workflow/__init__.py,sha256=8wjsorcOGzCAWGqLRbAUf8eyezxpnB4NvXHX_rdO7ks,1656
 truefoundry/workflow/container_task.py,sha256=8arieePsX4__OnG337hOtCiNgJwtKJJCsZcmFmCBJtk,402
 truefoundry/workflow/map_task.py,sha256=f9vcAPRQy0Ttw6bvdZBKUVJMSm4eGQrbE1GHWhepHIU,1864
 truefoundry/workflow/python_task.py,sha256=SRXRLC4vdBqGjhkwuaY39LEWN6iPCpJAuW17URRdWTY,1128
@@ -375,7 +375,7 @@ truefoundry/workflow/remote_filesystem/__init__.py,sha256=LQ95ViEjJ7Ts4JcCGOxMPs
 truefoundry/workflow/remote_filesystem/logger.py,sha256=em2l7D6sw7xTLDP0kQSLpgfRRCLpN14Qw85TN7ujQcE,1022
 truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py,sha256=xcT0wQmQlgzcj0nP3tJopyFSVWT1uv3nhiTIuwfXYeg,12342
 truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py,sha256=nSGPZu0Gyd_jz0KsEE-7w_BmnTD8CVF1S8cUJoxaCbc,13305
-truefoundry-0.9.0.dist-info/METADATA,sha256=nfNsBa_uNPq_uhJD7zbHksMPNf5G7jVOnnp5hLP-fhw,2465
-truefoundry-0.9.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truefoundry-0.9.0.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
-truefoundry-0.9.0.dist-info/RECORD,,
+truefoundry-0.9.1.dist-info/METADATA,sha256=yLhzQUokgIWKvgpsTXBbzB6c0535kRCbWJdHnJMHmHo,2465
+truefoundry-0.9.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truefoundry-0.9.1.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
+truefoundry-0.9.1.dist-info/RECORD,,