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

truefoundry/deploy/v2/lib/deploy.py

@@ -262,6 +262,7 @@ def deploy_component(
     workspace_fqn: Optional[str] = None,
     wait: bool = True,
     force: bool = False,
+    trigger_on_deploy: bool = False,
 ) -> Deployment:
     workspace_fqn = _resolve_workspace_fqn(
         component=component, workspace_fqn=workspace_fqn
@@ -284,6 +285,7 @@ def deploy_component(
         workspace_id=workspace_id,
         application=updated_component,
         force=force,
+        trigger_on_deploy=trigger_on_deploy,
     )
     logger.info(
         "🚀 Deployment started for application '%s'. Deployment FQN is '%s'.",

truefoundry/deploy/v2/lib/deployable_patched_models.py

@@ -1,10 +1,49 @@
-from typing import Literal, Union
+import warnings
+from typing import Any, Literal, Union
 
+from truefoundry.common.warnings import TrueFoundryDeprecationWarning
 from truefoundry.deploy._autogen 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
+from truefoundry.pydantic_v1 import BaseModel, Field, conint, root_validator, validator
+
+_TRIGGER_ON_DEPLOY_DEPRECATION_MESSAGE = """
+Setting `trigger_on_deploy` in manifest has been deprecated and the field will be removed in future releases.
+
+Please remove it from the spec and instead use
+
+`trigger_on_deploy` argument on `.deploy`
+
+E.g.
+
+```
+job = Job(...)  # remove `trigger_on_deploy` from initialization
+job.deploy(..., trigger_on_deploy={arg_value})
+```
+
+OR
+
+`{flag}` option on `tfy deploy`
+
+E.g.
+
+```
+tfy deploy -f truefoundry.yaml {flag}
+```
+"""
+
+
+def _warn_if_trigger_on_deploy_used(_klass, v: Any) -> Any:
+    if v is not None:
+        # v is the value of trigger_on_deploy, which is also the arg_value for the message
+        flag = "--trigger-on-deploy" if v else "--no-trigger-on-deploy"
+        warnings.warn(
+            _TRIGGER_ON_DEPLOY_DEPRECATION_MESSAGE.format(arg_value=v, flag=flag),
+            TrueFoundryDeprecationWarning,
+            stacklevel=2,
+        )
+    return v
 
 
 class DeployablePatchedModelBase(BaseModel):
@@ -12,13 +51,18 @@ class DeployablePatchedModelBase(BaseModel):
         extra = "forbid"
 
     def deploy(
-        self, workspace_fqn: str, wait: bool = True, force: bool = False
+        self,
+        workspace_fqn: str,
+        wait: bool = True,
+        force: bool = False,
+        trigger_on_deploy: bool = False,
     ) -> Deployment:
         return deploy_component(
             component=self,
             workspace_fqn=workspace_fqn,
             wait=wait,
             force=force,
+            trigger_on_deploy=trigger_on_deploy,
         )
 
 
@@ -36,6 +80,10 @@ class Job(models.Job, DeployablePatchedModelBase):
     type: Literal["job"] = "job"
     resources: models.Resources = Field(default_factory=models.Resources)
 
+    @validator("trigger_on_deploy")
+    def _warn_if_trigger_on_deploy_used(cls, v: Any) -> Any:
+        return _warn_if_trigger_on_deploy_used(cls, v)
+
 
 class SparkJob(models.SparkJob, DeployablePatchedModelBase):
     type: Literal["spark-job"] = "spark-job"
@@ -100,10 +148,9 @@ class Application(models.Application, DeployablePatchedModelBase):
     # 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
+    # >>> 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,
@@ -119,8 +166,24 @@ class Application(models.Application, DeployablePatchedModelBase):
         models.SparkJob,
     ] = Field(..., description="", discriminator="type")
 
+    @root_validator(pre=True)
+    def _validate_spec(cls, values: Any) -> Any:
+        if isinstance(values, dict) and "__root__" in values:
+            root = values["__root__"]
+            if (
+                isinstance(root, dict)
+                and root.get("type") == "job"
+                and root.get("trigger_on_deploy") is not None
+            ):
+                _warn_if_trigger_on_deploy_used(cls, root.get("trigger_on_deploy"))
+        return values
+
     def deploy(
-        self, workspace_fqn: str, wait: bool = True, force: bool = False
+        self,
+        workspace_fqn: str,
+        wait: bool = True,
+        force: bool = False,
+        trigger_on_deploy: bool = False,
     ) -> Deployment:
         if isinstance(self.__root__, models.Workflow):
             from truefoundry.deploy.v2.lib.deploy_workflow import deploy_workflow
@@ -137,4 +200,5 @@ class Application(models.Application, DeployablePatchedModelBase):
                 workspace_fqn=workspace_fqn,
                 wait=wait,
                 force=force,
+                trigger_on_deploy=trigger_on_deploy,
             )

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truefoundry
-Version: 0.9.1
+Version: 0.9.2
 Summary: TrueFoundry CLI
 Author-email: TrueFoundry Team <abhishek@truefoundry.com>
 Requires-Python: <3.14,>=3.8.1
@@ -14,7 +14,6 @@ Requires-Dist: gitpython<4.0.0,>=3.1.43
 Requires-Dist: importlib-metadata<9.0.0,>=4.11.3
 Requires-Dist: importlib-resources<7.0.0,>=5.2.0
 Requires-Dist: mako<2.0.0,>=1.1.6
-Requires-Dist: mcp==1.8.1; python_version >= '3.10'
 Requires-Dist: numpy<3.0.0,>=1.23.0
 Requires-Dist: openai<2.0.0,>=1.16.2
 Requires-Dist: packaging<26.0,>=20.0
@@ -35,6 +34,8 @@ 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
+Provides-Extra: ai
+Requires-Dist: mcp==1.9.1; (python_version >= '3.10') and extra == 'ai'
 Provides-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-0.9.1.dist-info → truefoundry-0.9.2.dist-info}/RECORD

@@ -3,6 +3,10 @@ truefoundry/_client.py,sha256=Y3qHi_Lg4Sx6GNvsjAHIoAfFr8PJnqgCrXmpNAI3ECg,1417
 truefoundry/logger.py,sha256=u-YCNjg5HBwE70uQcpjIG64Ghos-K2ulTWaxC03BSj4,714
 truefoundry/pydantic_v1.py,sha256=jSuhGtz0Mbk1qYu8jJ1AcnIDK4oxUsdhALc4spqstmM,345
 truefoundry/version.py,sha256=bqiT4Q-VWrTC6P4qfK43mez-Ppf-smWfrl6DcwV7mrw,137
+truefoundry/_ask/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+truefoundry/_ask/cli.py,sha256=zPaDvxhX2dITmPTtut2Iu6WAIaizrwR-U_dDZ6xv2io,5814
+truefoundry/_ask/client.py,sha256=4vWO04jWbSF0XD3q8DwXjvL4HW-WBg7nsQ0DydNwHmM,18479
+truefoundry/_ask/llm_utils.py,sha256=ayjz7JtVu142lrm8t0cVoxLxUpx76b71y8R62z_WurY,13537
 truefoundry/autodeploy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/autodeploy/cli.py,sha256=9ZxKu_MGIpraMzaW4ZyuQZhlKIQYE3biBrBV4S1h6Fo,14167
 truefoundry/autodeploy/constants.py,sha256=vTh2nA7cjqghqbW2rNh3FbtcIk2scdAWZuuQCmVBO80,1273
@@ -28,15 +32,15 @@ truefoundry/autodeploy/utils/client.py,sha256=PvbSkfgAjAogGjisinqmh4mP4svowxAC0I
 truefoundry/autodeploy/utils/diff.py,sha256=Ef8Y-VffDKel_-q-GxRam6gqiv8qTLMcqVg6iifXfcA,5358
 truefoundry/autodeploy/utils/pydantic_compat.py,sha256=hEAUy5kLjhPdzw7yGZ2iXGMXbbMVXVlGzIofmyHafXQ,412
 truefoundry/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/cli/__main__.py,sha256=k9wgXDoUOYdsK5Ply_GpDV-Ns4DNn0jRKNUQSCRnZ-0,3976
+truefoundry/cli/__main__.py,sha256=6qCyQ4iXJYwSqepQurw9rifYeiEmuu7WLJgEWFqvQJk,4004
 truefoundry/cli/config.py,sha256=f7z0_gmYZiNImB7Bxz0AnOlrxY2X4lFnX4jYW1I7NHQ,139
 truefoundry/cli/console.py,sha256=9-dMy4YPisCJQziRKTg8Qa0UJnOGl1soiUnJjsnLDvE,242
 truefoundry/cli/const.py,sha256=dVHPo1uAiDSSMXwXoT2mR5kNQjExT98QNVRz98Hz_Ts,510
-truefoundry/cli/display_util.py,sha256=s0_eWUUAK1dbmqW5h_qAG93roH81dh-g1nLjuQVFm6k,5130
-truefoundry/cli/util.py,sha256=7DmKXY5OPslPu2LO6vrUUfDtoHeo12sJTDUA0GOi8IM,3922
+truefoundry/cli/display_util.py,sha256=9vzN3mbQqU6OhS7qRUiMRana4PTHa4sDTA0Hn7OVjCI,3108
+truefoundry/cli/util.py,sha256=pezUfF2GC6ru7s8VeH2a7uvXTU0xN9ka7yLXkIgC3dY,4998
 truefoundry/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/common/auth_service_client.py,sha256=N3YxKlx63r6cPZqbgb2lqBOPI69ShB7D7RCIq4FSCjc,7949
-truefoundry/common/constants.py,sha256=eWcElAYIVb0jnHUAcsHvgnkdKf2E1nCg_Ybbi8ibxF0,4365
+truefoundry/common/constants.py,sha256=pAEuXtUNtg_qQIU40HBtS-L-KSunVAsF_UqzyIPnpEw,4487
 truefoundry/common/credential_file_manager.py,sha256=1yEk1Zm2xS4G0VDFwKSZ4w0VUrcPWQ1nJnoBaz9xyKA,4251
 truefoundry/common/credential_provider.py,sha256=_OhJ2XFlDaVsrUO-FyywxctcGGqDdC2pgcvwEKqQD0Q,4071
 truefoundry/common/entities.py,sha256=b4R6ss06-ygDS3C4Tqa_GOq5LFKDYbt7x4Mghnfz6yo,4007
@@ -50,7 +54,7 @@ truefoundry/common/utils.py,sha256=j3QP0uOsaGD_VmDDR68JTwoYE1okkAq6OqpVkzVf48Q,6
 truefoundry/common/warnings.py,sha256=rs6BHwk7imQYedo07iwh3TWEOywAR3Lqhj0AY4khByg,504
 truefoundry/deploy/__init__.py,sha256=6D22iiCgd5xlzBaG34q9Cx4rGgwf5qIAKQrOCgaCXYY,2746
 truefoundry/deploy/python_deploy_codegen.py,sha256=AainOFR20XvhNeztJkLPWGZ40lAT_nwc-ZmG77Kum4o,6525
-truefoundry/deploy/_autogen/models.py,sha256=gGH63evQTTnU0fEjtNgCsD0aqIxhdp316GL3Xb65NJk,71461
+truefoundry/deploy/_autogen/models.py,sha256=8zuCebKyr3Ake4OR4IRfagAo0VbV3WVOpvPmiB25RAg,71444
 truefoundry/deploy/builder/__init__.py,sha256=nGQiR3r16iumRy7xbVQ6q-k0EApmijspsfVpXDE-9po,4953
 truefoundry/deploy/builder/constants.py,sha256=amUkHoHvVKzGv0v_knfiioRuKiJM0V0xW0diERgWiI0,508
 truefoundry/deploy/builder/docker_service.py,sha256=sm7GWeIqyrKaZpxskdLejZlsxcZnM3BTDJr6orvPN4E,3948
@@ -62,16 +66,15 @@ truefoundry/deploy/builder/builders/tfy_notebook_buildpack/dockerfile_template.p
 truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py,sha256=_fjqHKn80qKi68SAMMALge7_A6e1sTsQWichw8uoGIw,2025
 truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py,sha256=f4l3fH21E2b8W3-JotMKc0AdPcCxV7LRPxxYJa7z_UQ,9134
 truefoundry/deploy/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/cli/commands/__init__.py,sha256=f7sXiQK9UuxDJmvBa-QCFNyumUpGGMhZbCxdwJzWXwQ,1116
+truefoundry/deploy/cli/commands/__init__.py,sha256=qv818jxqSAygJ3h-6Ul8t-5VOgR_UrSgsVtNCl3e5G0,1408
 truefoundry/deploy/cli/commands/apply_command.py,sha256=Y2e_C8HVpo8CssVod-3JRz-89qStC5JRaNzJ7O2mRlY,2039
-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=qpozW6RQa8EcLhrKtNrCoorBZU7p75XrmMUPX1c3EVM,4271
+truefoundry/deploy/cli/commands/deploy_command.py,sha256=fN6yVXdSGD8xWyAj6KcwayCjA_sV5aKCpxLuNCrUl8U,4681
 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
-truefoundry/deploy/cli/commands/kubeconfig_command.py,sha256=v6LmDyf2YQPxPrueGlzIq_leN6kKi1ks2zyWb4FmbXU,3150
+truefoundry/deploy/cli/commands/kubeconfig_command.py,sha256=WTYCv_itwUb6kwlsGkq3hwn5i-Gjd7cVrbBhoLZDnJI,3146
 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
@@ -79,7 +82,7 @@ truefoundry/deploy/cli/commands/patch_application_command.py,sha256=aRTHu2OmxQd7
 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
-truefoundry/deploy/cli/commands/utils.py,sha256=44XkNPxGK1wUEHfUTAn9pHZXosvUlby4PVy2cY3yBNc,4723
+truefoundry/deploy/cli/commands/utils.py,sha256=mIMYbHuAxnT0yz_0PU8LDC9sAZPU_xURZFMOrGoasuc,3694
 truefoundry/deploy/core/__init__.py,sha256=j61bMWj4BkWihdssKMSFhieo7afJDtpc7qO7zk1rDB4,140
 truefoundry/deploy/core/login.py,sha256=N2VrW3nlBzoyoYulkipxwQvCpjBhi3sfsmhxK1ktWhg,236
 truefoundry/deploy/core/logout.py,sha256=TpWLq4_DsxYS5GX2OJQGDhekNOfiOLb-vO5khQueHXw,80
@@ -95,8 +98,7 @@ truefoundry/deploy/lib/session.py,sha256=fLdgR6ZDp8-hFl5NTON4ngnWLsMzGxvKtfpDOOw
 truefoundry/deploy/lib/util.py,sha256=J7r8San2wKo48A7-BlH2-OKTlBO67zlPjLEhMsL8os0,1059
 truefoundry/deploy/lib/win32.py,sha256=1RcvPTdlOAJ48rt8rCbE2Ufha2ztRqBAE9dueNXArrY,5009
 truefoundry/deploy/lib/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/clients/ask_client.py,sha256=77106708EC16wsi2M1n1_5HgOVboEZoq9_obKsf24M0,13494
-truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=fmRlPYCimk1ZLbMgdzfJVCbcKRCVnFYL5T3j2uJA0Tc,27037
+truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=HLBXRQqzwrJfxFnXnxKInx4UWPA8rF8T_ouk-OX7HaE,27128
 truefoundry/deploy/lib/dao/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/dao/application.py,sha256=oMszpueXPUfTUuN_XdKwoRjQyqAgWHhZ-10cbprCVdM,9226
 truefoundry/deploy/lib/dao/apply.py,sha256=5IFERe5sLmZGlavaKTIxL4xPHAme4ZS2Ww0a2rKTyT0,3029
@@ -107,9 +109,9 @@ truefoundry/deploy/lib/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 truefoundry/deploy/lib/model/entity.py,sha256=Bp9sLB-M5INCpw5lPmFdygHWS1zvnLicnSiSCi2iqhQ,8591
 truefoundry/deploy/v2/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/v2/lib/__init__.py,sha256=WEiVMZXOVljzEE3tpGJil14liIn_PCDoACJ6b3tZ6sI,188
-truefoundry/deploy/v2/lib/deploy.py,sha256=kAnh6RO4ci7AVjlIoN1Sr5FmcOU7nbkwNvbrS802spY,12625
+truefoundry/deploy/v2/lib/deploy.py,sha256=HfSUdAS3gSpFAFtV0Mq9LscfpkaXqA2LHW4VXqk9Y0g,12707
 truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=G5BzMIbap8pgDX1eY-TITruUxQdkKhYtBmRwLL6lDeY,14342
-truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=xy7AGYmVpyk12RMrXxOHHkpNKpttPFi6AiLiKdpID20,4940
+truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=mUi-OjPf7bc8rzfrPLdFb79LKuDq7F36RxL4V-AXebs,6830
 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
@@ -375,7 +377,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.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,,
+truefoundry-0.9.2.dist-info/METADATA,sha256=2jpQDTSr8gNEmXZp1N5Dk8-hsHbfBiW4O8TuNY2Wjdg,2504
+truefoundry-0.9.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truefoundry-0.9.2.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
+truefoundry-0.9.2.dist-info/RECORD,,