truefoundry 0.3.0rc9__py3-none-any.whl → 0.3.0rc10__py3-none-any.whl

truefoundry/deploy/__init__.py

@@ -16,6 +16,7 @@ from truefoundry.deploy.lib.dao.application import (
     list_job_runs,
     terminate_job_run,
     trigger_job,
+    trigger_workflow,
 )
 from truefoundry.deploy.lib.dao.version import (
     get_version as get_application_version,

truefoundry/deploy/cli/cli.py

@@ -13,6 +13,7 @@ from truefoundry.deploy.cli.commands import (
     get_logout_command,
     get_patch_application_command,
     get_patch_command,
+    get_trigger_command,
 )
 from truefoundry.deploy.cli.config import CliConfig
 from truefoundry.deploy.cli.const import GROUP_CLS
@@ -36,6 +37,7 @@ def create_truefoundry_cli():
     cli.add_command(get_deploy_command())
     cli.add_command(get_patch_application_command())
     cli.add_command(get_delete_command())
+    cli.add_command(get_trigger_command())
 
     if not (sys.platform.startswith("win32") or sys.platform.startswith("cygwin")):
         cli.add_command(get_patch_command())

truefoundry/deploy/cli/commands/trigger_command.py

@@ -82,6 +82,64 @@ def trigger_job(application_fqn: str, command: Optional[Sequence[str]], params):
     )
 
 
+@click.command(
+    name="workflow",
+    cls=COMMAND_CLS,
+    context_settings={"ignore_unknown_options": True, "allow_extra_args": True},
+)
+@click.option(
+    "--application-fqn",
+    "--application_fqn",
+    type=click.STRING,
+    required=True,
+    help="FQN of the workflow application",
+)
+@click.argument(
+    "inputs",
+    type=click.STRING,
+    nargs=-1,
+    required=False,
+)
+@handle_exception_wrapper
+def trigger_workflow(application_fqn: str, inputs):
+    """
+    Trigger a Workflow on TrueFoundry
+
+        [b]tfy trigger workflow --application-fqn "my-cluster:my-workspace:my-workflow"[/]
+
+    \n
+    Additionally, you can pass inputs (if defined in the workflow)\n\n
+
+    Passing inputs:
+
+        [b]tfy trigger workflow --application-fqn "my-cluster:my-workspace:my-workflow" -- --input1_name input1_value --input2_name input2_value ...[/]
+    """
+    if inputs:
+        inputs_dict = {}
+        if len(inputs) % 2 != 0:
+            raise ClickException(
+                f"Found odd number of argument pairs: {inputs}. "
+                "Perhaps you forgot to pass a value for one of the inputs? "
+                "inputs for workflow should be passed in the "
+                "format `--input1_name input1_value --input2_name input2_value ...`"
+            )
+        for i in range(0, len(inputs), 2):
+            key = inputs[i]
+            value = inputs[i + 1]
+            if not key.startswith("--"):
+                raise ClickException(
+                    f"Got ambiguous argument {key!r} in inputs: {inputs}. "
+                    f"input names should be prefixed with '--' i.e. "
+                    "inputs for workflow should be passed in the "
+                    "format `--input1_name input1_value --input2_name input2_value ...`"
+                )
+            key = key.lstrip("-")
+            inputs_dict[key] = value
+
+    application.trigger_workflow(application_fqn=application_fqn, inputs=inputs)
+
+
 def get_trigger_command():
     trigger_command.add_command(trigger_job)
+    trigger_command.add_command(trigger_workflow)
     return trigger_command

truefoundry/deploy/lib/clients/servicefoundry_client.py

@@ -616,6 +616,12 @@ class ServiceFoundryServiceClient:
         response = request_handling(res)
         return TriggerJobResult.parse_obj(response)
 
+    def trigger_workflow(self, application_id: str, inputs: Dict[str, Any]):
+        url = f"{self._api_server_url}/{VERSION_PREFIX}/workflow/{application_id}/executions"
+        res = requests.post(url, json=inputs, headers=self._get_header())
+        response = request_handling(res)
+        return response
+
     @check_min_cli_version
     def get_docker_registry_creds(
         self, docker_registry_fqn: str, cluster_id: str

truefoundry/deploy/lib/dao/application.py

@@ -241,3 +241,23 @@ def terminate_job_run(
         job_run_name=job_run_name,
     )
     return response
+
+
+def trigger_workflow(application_fqn: str, inputs: Optional[Dict[str, Any]] = None):
+    inputs = inputs or {}
+    client = ServiceFoundryServiceClient()
+    _application_info = client.get_application_info_by_fqn(
+        application_fqn=application_fqn
+    )
+    application_info = client.get_application_info(
+        application_id=_application_info.applicationId
+    )
+    client.trigger_workflow(
+        application_id=application_info.id,
+        inputs=inputs,
+    )
+    logger.info(f"Started Execution for Workflow: {application_info.name}")
+    executions_page = (
+        f"{client.base_url.strip('/')}/deployments/{application_info.id}?tab=executions"
+    )
+    logger.info(f"You can check the executions at {executions_page}")

{truefoundry-0.3.0rc9.dist-info → truefoundry-0.3.0rc10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.3.0rc9
+Version: 0.3.0rc10
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com
@@ -24,7 +24,7 @@ Requires-Dist: flytekit (==1.12.2) ; extra == "workflow"
 Requires-Dist: gitignorefile (>=1.1.2,<1.2.0)
 Requires-Dist: importlib-metadata (>=6.0.1,<8.0.0)
 Requires-Dist: importlib-resources (>=5.2.0,<6.0.0)
-Requires-Dist: mlfoundry (==0.11.2) ; extra == "ml"
+Requires-Dist: mlfoundry (==0.11.3) ; extra == "ml"
 Requires-Dist: openai (>=1.16.2,<2.0.0)
 Requires-Dist: packaging (>=20.0,<25.0)
 Requires-Dist: pydantic (>=1.10.0,<3)

{truefoundry-0.3.0rc9.dist-info → truefoundry-0.3.0rc10.dist-info}/RECORD

@@ -25,7 +25,7 @@ truefoundry/autodeploy/utils/diff.py,sha256=Ef8Y-VffDKel_-q-GxRam6gqiv8qTLMcqVg6
 truefoundry/autodeploy/utils/pydantic_compat.py,sha256=hEAUy5kLjhPdzw7yGZ2iXGMXbbMVXVlGzIofmyHafXQ,412
 truefoundry/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/cli/__main__.py,sha256=Jap_IddZ9zNyMIyIkCw75xHQCN0WtV2dPZJ_pzdLsVc,916
-truefoundry/deploy/__init__.py,sha256=hyBoxhL3SG37vduJBdUYJuy1EHGeSLPI4S8MuKhyCFo,2259
+truefoundry/deploy/__init__.py,sha256=ugawKF2G02EmEXX35oZ2tec12d9oWN28Sf6mtGGIERY,2281
 truefoundry/deploy/auto_gen/models.py,sha256=inof7aH6Wk1z15YYtCNJo7sp3Tixv5yaJzB_4fiZux4,78412
 truefoundry/deploy/builder/__init__.py,sha256=a1qR6nicHGcxRaeNTxWRsmDs8zsmXc7j13-I8q0qqVk,4938
 truefoundry/deploy/builder/builders/__init__.py,sha256=tlFLXqyDaKLd4iZbo4Hcu_8gOmgtL6drnXpbmQ6x1P8,636
@@ -36,7 +36,7 @@ truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py,sha256=n7Mw
 truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py,sha256=vFmFeK38-t8booJEGREapEjrIL8xnyOQeRSJQq7d3ZQ,6183
 truefoundry/deploy/builder/docker_service.py,sha256=vQS15790njzlFJZ3JW6txYLBdT11ltxqqpf78ZFL_Ng,5208
 truefoundry/deploy/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/cli/cli.py,sha256=UM6nuyZJ1CfYmN8-A65GGDN3n3j0qmSZb283aMLIUOY,2649
+truefoundry/deploy/cli/cli.py,sha256=l1vRYoovhx40HYZdfsC9JyiYzW9Jx6__7QDzQpN315I,2717
 truefoundry/deploy/cli/commands/__init__.py,sha256=oOstv608ueOJTxR3qlQbZBIdo-FsFe2O0NYNOXi01-s,1276
 truefoundry/deploy/cli/commands/apply_command.py,sha256=Xzky2csXwLARTpQBcPj87imnDu8tIDRN0GnqY8ebEuw,1863
 truefoundry/deploy/cli/commands/build_command.py,sha256=QwKkZ3nVecPMzs-R57YP-_ih9rjTSnhwBWX5tqHVsvU,1255
@@ -53,7 +53,7 @@ truefoundry/deploy/cli/commands/patch_application_command.py,sha256=Zuswj36o-qni
 truefoundry/deploy/cli/commands/patch_command.py,sha256=OQCmxcn1Qd2iZfBUaqs48oUKjHVZ13SoMsWLhksPvks,1682
 truefoundry/deploy/cli/commands/redeploy_command.py,sha256=-wMQLeMcBdJsQBMgTv7rS6hCqH1EJsSKgPrAXU9HjxU,1032
 truefoundry/deploy/cli/commands/terminate_comand.py,sha256=RTNuykw5DBvid44u2nCka_Eu0XSK3Qrzwob9fybHLgQ,1097
-truefoundry/deploy/cli/commands/trigger_command.py,sha256=TdGjhBFLlfVV7Ba044YsbFRSzDekSVEYdqoyLweJQdY,2748
+truefoundry/deploy/cli/commands/trigger_command.py,sha256=Js6x7-TEuQQN1fTBoILTJiCt76TdH9qpAnkCnei6vYo,4730
 truefoundry/deploy/cli/config.py,sha256=tf8w4UfVzcC6eYkENvuuCPYt_V3sqVpO1bclORV9tAk,206
 truefoundry/deploy/cli/console.py,sha256=9-dMy4YPisCJQziRKTg8Qa0UJnOGl1soiUnJjsnLDvE,242
 truefoundry/deploy/cli/const.py,sha256=dVHPo1uAiDSSMXwXoT2mR5kNQjExT98QNVRz98Hz_Ts,510
@@ -83,12 +83,12 @@ truefoundry/deploy/lib/auth/credential_file_manager.py,sha256=DXeXWoVakfZI2Geu8F
 truefoundry/deploy/lib/auth/credential_provider.py,sha256=MwTN8170TXi7g9m2Fw4VRReZqk1DzBFG1bMVgd7jYZk,4375
 truefoundry/deploy/lib/auth/servicefoundry_session.py,sha256=2OahwRg-8l3QUwpC2iLA8lHavkeTxKXUP_0AN27HPS8,1859
 truefoundry/deploy/lib/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=fq7kvQa2OgTs_TobCEPC_Yy0A-g0w7enOu0jdSfTyVc,26083
+truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=NASbj67lrQKZwhu7mVu6ngE9A8C76FqqUpfFYQKjVAU,26394
 truefoundry/deploy/lib/clients/shell_client.py,sha256=tMrc0Ha1DmGtUCJrZD8eusOzfe8R_WIe6AAH7nxL0xA,461
 truefoundry/deploy/lib/clients/utils.py,sha256=rK7DrvA71kSTjy23-Bk4LTQjgViBWeHtcV_SlBLZw6M,1282
 truefoundry/deploy/lib/const.py,sha256=Yk_nXeZWzwKs-6hEXDjVDyjwEGh35T5TWaBxyJgP-Zw,1395
 truefoundry/deploy/lib/dao/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/dao/application.py,sha256=I-4YicbU5tazo9JLkW9FlVpB-nRamdOliJNxApMaHJE,8419
+truefoundry/deploy/lib/dao/application.py,sha256=uUTFSQkLUrFCtQQgS2Jm9BpyHyhMkN4GI1yx9oJo4_E,9161
 truefoundry/deploy/lib/dao/apply.py,sha256=sXnQY6RVzLVm1fX2BKuWHAoKlKISirrcByHEhY3x4zo,2570
 truefoundry/deploy/lib/dao/version.py,sha256=AtdW_4O1DPUKdfv2qy6iUJsZ_95vM6z0AqeEy3WDKs8,1130
 truefoundry/deploy/lib/dao/workspace.py,sha256=jm8UWytwVajVcrYyHSTCwWYDYl-RHuk0zAf9Caj4GzQ,2356
@@ -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.0rc9.dist-info/METADATA,sha256=MoFxdl5KpAr9mCShojPeE2O2dXsgyB82DSjYKPIR9b0,2693
-truefoundry-0.3.0rc9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-truefoundry-0.3.0rc9.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.3.0rc9.dist-info/RECORD,,
+truefoundry-0.3.0rc10.dist-info/METADATA,sha256=ycIiBcPg1h3rNv8TSK3OHc4PHWDH6DGVP9f0MC5acSo,2694
+truefoundry-0.3.0rc10.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+truefoundry-0.3.0rc10.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.3.0rc10.dist-info/RECORD,,