truefoundry 0.3.0rc9__tar.gz → 0.3.0rc10__tar.gz

{truefoundry-0.3.0rc9 → truefoundry-0.3.0rc10}/PKG-INFO

@@ -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 → truefoundry-0.3.0rc10}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truefoundry"
-version = "0.3.0rc9"
+version = "0.3.0rc10"
 description = "Truefoundry CLI"
 authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
 readme = "README.md"
@@ -17,7 +17,7 @@ GitPython = ">=3.1.43,<3.2.0"
 importlib-metadata = ">=6.0.1,<8.0.0"
 importlib-resources = "^5.2.0"
 Mako = "^1.1.6"
-mlfoundry = { version = "0.11.2", optional = true }
+mlfoundry = { version = "0.11.3", optional = true }
 openai = ">=1.16.2,<2.0.0"
 packaging = ">=20.0,<25.0"
 pydantic = ">=1.10.0,<3"

{truefoundry-0.3.0rc9 → truefoundry-0.3.0rc10}/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-0.3.0rc9 → truefoundry-0.3.0rc10}/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-0.3.0rc9 → truefoundry-0.3.0rc10}/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-0.3.0rc9 → truefoundry-0.3.0rc10}/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-0.3.0rc9 → truefoundry-0.3.0rc10}/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}")