truefoundry 0.6.2__py3-none-any.whl → 0.6.3rc2__py3-none-any.whl

truefoundry/deploy/_autogen/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2025-03-10T17:05:18+00:00
+#   timestamp: 2025-03-20T05:09:43+00:00
 
 from __future__ import annotations
 
@@ -389,6 +389,10 @@ class JwtAuthConfig(BaseModel):
     claims: Optional[List[Claim]] = Field(
         None, description="List of key-value pairs of claims to verify in the JWT token"
     )
+    bypass_auth_paths: Optional[List[constr(regex=r"^/[^*]*")]] = Field(
+        None,
+        description="List of paths that will bypass auth.\nneeds to start with a forward slash(/) and should not contain wildcards(*)",
+    )
 
 
 class KafkaMetricConfig(BaseModel):
@@ -510,15 +514,36 @@ class NvidiaGPU(BaseModel):
 
 class Profile(str, Enum):
     """
-    Name of the MIG profile to use. One of [1g.5gb, 2g.10gb, 3g.20gb, 1g.10gb, 2g.20gb, 3g.40gb]
+    Name of the MIG profile to use. One of the following based on gpu type
+    Please refer to https://docs.nvidia.com/datacenter/tesla/mig-user-guide/#supported-mig-profiles for more details
+    A100 40 GB - [1g.5gb, 1g.10gb, 2g.10gb, 3g.20gb, 4g.20gb]
+    A100 80 GB / H100 80 GB - [1g.10gb, 1g.20gb, 2g.20gb, 3g.40gb, 4g.40gb]
+    H100 94 GB - [1g.12gb, 1g.24gb, 2g.24gb, 3g.47gb, 4g.47gb]
+    H100 96 GB - [1g.12gb, 1g.24gb, 2g.24gb, 3g.48gb, 4g.48gb]
+    H200 141 GB - [1g.18gb, 1g.35gb, 2g.35gb, 3g.71gb, 4g.71gb]
     """
 
     field_1g_5gb = "1g.5gb"
-    field_2g_10gb = "2g.10gb"
-    field_3g_20gb = "3g.20gb"
     field_1g_10gb = "1g.10gb"
+    field_1g_12gb = "1g.12gb"
+    field_1g_18gb = "1g.18gb"
+    field_1g_20gb = "1g.20gb"
+    field_1g_24gb = "1g.24gb"
+    field_1g_35gb = "1g.35gb"
+    field_2g_10gb = "2g.10gb"
     field_2g_20gb = "2g.20gb"
+    field_2g_24gb = "2g.24gb"
+    field_2g_35gb = "2g.35gb"
+    field_3g_20gb = "3g.20gb"
     field_3g_40gb = "3g.40gb"
+    field_3g_47gb = "3g.47gb"
+    field_3g_48gb = "3g.48gb"
+    field_3g_71gb = "3g.71gb"
+    field_4g_20gb = "4g.20gb"
+    field_4g_40gb = "4g.40gb"
+    field_4g_47gb = "4g.47gb"
+    field_4g_48gb = "4g.48gb"
+    field_4g_71gb = "4g.71gb"
 
 
 class NvidiaMIGGPU(BaseModel):
@@ -529,7 +554,7 @@ class NvidiaMIGGPU(BaseModel):
     )
     profile: Profile = Field(
         ...,
-        description="Name of the MIG profile to use. One of [1g.5gb, 2g.10gb, 3g.20gb, 1g.10gb, 2g.20gb, 3g.40gb]",
+        description="Name of the MIG profile to use. One of the following based on gpu type\nPlease refer to https://docs.nvidia.com/datacenter/tesla/mig-user-guide/#supported-mig-profiles for more details\nA100 40 GB - [1g.5gb, 1g.10gb, 2g.10gb, 3g.20gb, 4g.20gb]\nA100 80 GB / H100 80 GB - [1g.10gb, 1g.20gb, 2g.20gb, 3g.40gb, 4g.40gb]\nH100 94 GB - [1g.12gb, 1g.24gb, 2g.24gb, 3g.47gb, 4g.47gb]\nH100 96 GB - [1g.12gb, 1g.24gb, 2g.24gb, 3g.48gb, 4g.48gb]\nH200 141 GB - [1g.18gb, 1g.35gb, 2g.35gb, 3g.71gb, 4g.71gb]",
     )
 
 
@@ -918,6 +943,10 @@ class TrueFoundryArtifactSource(BaseModel):
 
 class TrueFoundryInteractiveLogin(BaseModel):
     type: Literal["truefoundry_oauth"] = Field(..., description="")
+    bypass_auth_paths: Optional[List[constr(regex=r"^/[^*]*")]] = Field(
+        None,
+        description="List of paths that will bypass auth.\nneeds to start with a forward slash(/) and should not contain wildcards(*)",
+    )
 
 
 class VolumeBrowser(BaseModel):

truefoundry/deploy/v2/lib/deploy_workflow.py

@@ -132,6 +132,7 @@ def _validate_workflow_entities(  # noqa: C901
         `from truefoundry.workflow import PythonTaskConfig, ContainerTaskConfig`
          """
     tasks_without_truefoundry_worflow_package = []
+    task_names = set()
     for task in tasks:
         if _is_dynamic_task(task):
             raise ValueError("Dynamic workflows are not supported yet.")
@@ -141,6 +142,12 @@ def _validate_workflow_entities(  # noqa: C901
                     task.template.id.name
                 )
             )
+        task_name = task.template.id.name
+        if task_name in task_names:
+            raise ValueError(
+                f"Task name should be unique, task with name {task_name} is repeated in the workflow"
+            )
+        task_names.add(task_name)
         task_image_spec = task.template.custom["truefoundry"]["image"]
         if task_image_spec["type"] == "task-python-build":
             is_tfy_wf_present_in_task_python_build = (

{truefoundry-0.6.2.dist-info → truefoundry-0.6.3rc2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truefoundry
-Version: 0.6.2
+Version: 0.6.3rc2
 Summary: TrueFoundry CLI
 Author-email: TrueFoundry Team <abhishek@truefoundry.com>
 Requires-Python: <3.14,>=3.8.1
@@ -30,7 +30,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.0.6
+Requires-Dist: truefoundry-sdk==0.0.8
 Requires-Dist: typing-extensions>=4.0
 Requires-Dist: urllib3<3,>=1.26.18
 Requires-Dist: yq<4.0.0,>=3.1.0

{truefoundry-0.6.2.dist-info → truefoundry-0.6.3rc2.dist-info}/RECORD

@@ -50,7 +50,7 @@ truefoundry/common/utils.py,sha256=Ev4lATS0jT9BDNPmQEP3aDn8IIeh9pmahv5pcNIjuuw,6
 truefoundry/common/warnings.py,sha256=rs6BHwk7imQYedo07iwh3TWEOywAR3Lqhj0AY4khByg,504
 truefoundry/deploy/__init__.py,sha256=lso6tFZVNyB3KmjEZICq_Sd6R_Yip-67XAPXYAQCvGw,2597
 truefoundry/deploy/python_deploy_codegen.py,sha256=AainOFR20XvhNeztJkLPWGZ40lAT_nwc-ZmG77Kum4o,6525
-truefoundry/deploy/_autogen/models.py,sha256=oi0Q80dQsXCMVL3SJudogxaBVdIOvBQDEaOi791Dj-4,66926
+truefoundry/deploy/_autogen/models.py,sha256=3yTDqmWSO9F56ZsMky8wNtpB0othiSV0ewLT836P-BI,68662
 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
@@ -105,7 +105,7 @@ truefoundry/deploy/lib/model/entity.py,sha256=Up-DDOezkwM2tdqibfLdZO6jmT2pVq6SSh
 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=23Sc2OxtAMvDzKfNmVv5vuaB6GqP-e1WAZWIezMzm3c,12413
-truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=7NlL5eASwsDLcmsWhhh-YR-yBxS1CPwM-ceP87lNm_M,12860
+truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=TmrxnXJcdtv3SxQ34kGWYIogyHPQfq-qB3FkK3eqUhU,13141
 truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=xbHFD3pURflvCm8EODPvjfvRrv67mlSrjPUknY8SMB8,4060
 truefoundry/deploy/v2/lib/models.py,sha256=ogc1UYs1Z2nBdGSKCrde9sk8d0GxFKMkem99uqO5CmM,1148
 truefoundry/deploy/v2/lib/patched_models.py,sha256=Lowm-tyCGGv6-4iY9vfgmHrr4zPFy2u9wZEY3xQBv4E,16359
@@ -371,7 +371,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.6.2.dist-info/METADATA,sha256=dooURHmZMoy9716p3g3tFJkt_LwqIqMVPU1vJgvp1VQ,2349
-truefoundry-0.6.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truefoundry-0.6.2.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
-truefoundry-0.6.2.dist-info/RECORD,,
+truefoundry-0.6.3rc2.dist-info/METADATA,sha256=izHRNtbS_jpAioPdbTK_35BgXLqiJ9eIvFW_GA3-4tM,2352
+truefoundry-0.6.3rc2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truefoundry-0.6.3rc2.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
+truefoundry-0.6.3rc2.dist-info/RECORD,,