flyte 2.0.0b26__py3-none-any.whl → 2.0.0b28__py3-none-any.whl

flyte/_code_bundle/_utils.py

@@ -193,10 +193,10 @@ def list_all_files(source_path: pathlib.Path, deref_symlinks, ignore_group: Opti
 def _file_is_in_directory(file: str, directory: str) -> bool:
     """Return True if file is in directory and in its children."""
     try:
-        return os.path.commonpath([file, directory]) == directory
-    except ValueError as e:
-        # ValueError is raised by windows if the paths are not from the same drive
-        logger.debug(f"{file} and {directory} are not in the same drive: {e!s}")
+        return pathlib.Path(file).resolve().is_relative_to(pathlib.Path(directory).resolve())
+    except OSError as e:
+        # OSError can be raised if paths cannot be resolved (permissions, broken symlinks, etc.)
+        logger.debug(f"Failed to resolve paths for {file} and {directory}: {e!s}")
         return False
 
 

flyte/_code_bundle/bundle.py

@@ -104,7 +104,7 @@ async def build_pkl_bundle(
                 import shutil
 
                 # Copy the bundle to the given path
-                shutil.copy(dest, copy_bundle_to)
+                shutil.copy(dest, copy_bundle_to, follow_symlinks=True)
                 local_path = copy_bundle_to / dest.name
                 return CodeBundle(pkl=str(local_path), computed_version=str_digest)
             return CodeBundle(pkl=str(dest), computed_version=str_digest)

flyte/_internal/runtime/trigger_serde.py

@@ -12,7 +12,7 @@ from flyte import Cron, FixedRate, Trigger, TriggerTime
 def _to_schedule(m: Union[Cron, FixedRate], kickoff_arg_name: str | None = None) -> common_pb2.Schedule:
     if isinstance(m, Cron):
         return common_pb2.Schedule(
-            cron_expression=m.expression,
+            cron_expression=m.timezone_expression,
             kickoff_time_input_arg=kickoff_arg_name,
         )
     elif isinstance(m, FixedRate):

flyte/_trigger.py

@@ -2,10 +2,611 @@ from __future__ import annotations
 
 from dataclasses import dataclass
 from datetime import datetime
-from typing import Any, Dict, Mapping, Union
+from typing import Any, Dict, Literal, Mapping, Union
 
 import rich.repr
 
+Timezone = Literal[
+    "Etc/GMT-5",
+    "Europe/London",
+    "America/Martinique",
+    "Asia/Sakhalin",
+    "Europe/Podgorica",
+    "America/Grand_Turk",
+    "America/Dawson_Creek",
+    "Africa/Asmera",
+    "Canada/Newfoundland",
+    "CST6CDT",
+    "ROC",
+    "Asia/Taipei",
+    "America/Danmarkshavn",
+    "Asia/Yakutsk",
+    "America/Catamarca",
+    "Asia/Gaza",
+    "America/Chihuahua",
+    "Africa/Lusaka",
+    "Atlantic/Azores",
+    "Canada/Atlantic",
+    "Africa/Abidjan",
+    "America/Port-au-Prince",
+    "Pacific/Noumea",
+    "Asia/Barnaul",
+    "Europe/Bucharest",
+    "Asia/Samarkand",
+    "Africa/Bangui",
+    "America/Yakutat",
+    "Africa/Porto-Novo",
+    "Etc/GMT+12",
+    "America/Fortaleza",
+    "Australia/Brisbane",
+    "America/Goose_Bay",
+    "America/Nassau",
+    "Arctic/Longyearbyen",
+    "Asia/Kolkata",
+    "Indian/Mayotte",
+    "Asia/Tel_Aviv",
+    "America/Cambridge_Bay",
+    "Africa/Nouakchott",
+    "Australia/North",
+    "Europe/Tirane",
+    "America/Ensenada",
+    "Asia/Rangoon",
+    "Pacific/Kanton",
+    "Africa/Tunis",
+    "Europe/Kyiv",
+    "America/Halifax",
+    "Europe/Guernsey",
+    "America/Cancun",
+    "Canada/Saskatchewan",
+    "Europe/Helsinki",
+    "Pacific/Norfolk",
+    "Chile/Continental",
+    "Eire",
+    "Africa/Mogadishu",
+    "Pacific/Midway",
+    "Etc/GMT-1",
+    "Etc/UTC",
+    "America/Argentina/San_Luis",
+    "Europe/Gibraltar",
+    "GMT+0",
+    "HST",
+    "America/Kentucky/Louisville",
+    "US/Alaska",
+    "Africa/Addis_Ababa",
+    "America/Costa_Rica",
+    "Pacific/Rarotonga",
+    "America/Matamoros",
+    "Europe/Vienna",
+    "America/Vancouver",
+    "Mexico/BajaNorte",
+    "America/Merida",
+    "Pacific/Efate",
+    "America/La_Paz",
+    "Pacific/Marquesas",
+    "America/Manaus",
+    "Antarctica/South_Pole",
+    "Pacific/Easter",
+    "Europe/Mariehamn",
+    "Atlantic/Madeira",
+    "Pacific/Bougainville",
+    "Antarctica/Syowa",
+    "America/Montevideo",
+    "Africa/Khartoum",
+    "America/St_Thomas",
+    "Africa/Mbabane",
+    "America/Campo_Grande",
+    "Australia/Queensland",
+    "Asia/Damascus",
+    "Etc/Universal",
+    "Asia/Amman",
+    "Europe/Samara",
+    "Australia/Tasmania",
+    "Africa/Douala",
+    "Indian/Antananarivo",
+    "Canada/Eastern",
+    "America/Argentina/Rio_Gallegos",
+    "Pacific/Samoa",
+    "Australia/Canberra",
+    "Australia/Sydney",
+    "Atlantic/Faeroe",
+    "Asia/Ust-Nera",
+    "Pacific/Palau",
+    "Africa/Kinshasa",
+    "Asia/Makassar",
+    "Asia/Hebron",
+    "ROK",
+    "America/St_Vincent",
+    "America/Argentina/Buenos_Aires",
+    "Africa/Asmara",
+    "Indian/Mahe",
+    "America/Dawson",
+    "Europe/Lisbon",
+    "Pacific/Pago_Pago",
+    "Brazil/West",
+    "America/Santarem",
+    "America/Virgin",
+    "America/Indianapolis",
+    "Asia/Bangkok",
+    "America/Indiana/Marengo",
+    "Atlantic/St_Helena",
+    "Europe/Moscow",
+    "Europe/Istanbul",
+    "Asia/Famagusta",
+    "Asia/Chongqing",
+    "America/Cuiaba",
+    "America/Detroit",
+    "America/Swift_Current",
+    "Greenwich",
+    "Antarctica/Davis",
+    "Africa/Conakry",
+    "America/Asuncion",
+    "Asia/Hovd",
+    "Africa/Freetown",
+    "America/Bahia_Banderas",
+    "Etc/GMT+6",
+    "Indian/Chagos",
+    "Pacific/Kiritimati",
+    "America/Toronto",
+    "EET",
+    "Indian/Cocos",
+    "America/Caracas",
+    "America/Indiana/Knox",
+    "America/Indiana/Winamac",
+    "Asia/Katmandu",
+    "America/Dominica",
+    "Asia/Hong_Kong",
+    "GMT0",
+    "Atlantic/Faroe",
+    "US/Michigan",
+    "UCT",
+    "Etc/GMT-9",
+    "America/Cordoba",
+    "Etc/GMT",
+    "Chile/EasterIsland",
+    "America/Resolute",
+    "America/Juneau",
+    "Europe/Chisinau",
+    "Africa/Djibouti",
+    "Antarctica/Vostok",
+    "Europe/Ulyanovsk",
+    "US/Hawaii",
+    "Africa/Juba",
+    "America/Chicago",
+    "America/Boa_Vista",
+    "Antarctica/DumontDUrville",
+    "Brazil/East",
+    "Mexico/BajaSur",
+    "Africa/Lubumbashi",
+    "America/Anguilla",
+    "Etc/GMT-13",
+    "Canada/Central",
+    "Europe/Busingen",
+    "America/Ciudad_Juarez",
+    "America/Edmonton",
+    "Atlantic/South_Georgia",
+    "America/Anchorage",
+    "America/Rosario",
+    "America/Araguaina",
+    "Asia/Shanghai",
+    "America/Tijuana",
+    "America/Cayenne",
+    "America/Regina",
+    "Australia/NSW",
+    "America/Santa_Isabel",
+    "Indian/Comoro",
+    "Europe/San_Marino",
+    "WET",
+    "Poland",
+    "US/Indiana-Starke",
+    "Asia/Saigon",
+    "Africa/Ceuta",
+    "Pacific/Niue",
+    "Australia/Darwin",
+    "Asia/Yekaterinburg",
+    "Pacific/Chuuk",
+    "Asia/Kathmandu",
+    "Asia/Almaty",
+    "America/Miquelon",
+    "Asia/Choibalsan",
+    "Australia/Melbourne",
+    "America/Managua",
+    "Portugal",
+    "Iceland",
+    "Africa/Malabo",
+    "America/North_Dakota/Center",
+    "Asia/Macau",
+    "EST5EDT",
+    "America/Louisville",
+    "America/Fort_Nelson",
+    "Europe/Amsterdam",
+    "America/Rio_Branco",
+    "Etc/GMT0",
+    "America/Thule",
+    "Pacific/Fakaofo",
+    "Africa/Bujumbura",
+    "Asia/Baku",
+    "Etc/GMT+8",
+    "Etc/GMT+10",
+    "America/Argentina/Tucuman",
+    "Africa/Lagos",
+    "Europe/Paris",
+    "Indian/Christmas",
+    "Asia/Qatar",
+    "Australia/ACT",
+    "Asia/Pyongyang",
+    "America/North_Dakota/Beulah",
+    "Factory",
+    "Europe/Copenhagen",
+    "Pacific/Fiji",
+    "America/Lower_Princes",
+    "Antarctica/Macquarie",
+    "America/Punta_Arenas",
+    "Antarctica/Rothera",
+    "America/Montserrat",
+    "Etc/GMT+0",
+    "NZ",
+    "America/Argentina/La_Rioja",
+    "America/Argentina/Catamarca",
+    "Antarctica/Troll",
+    "Europe/Vatican",
+    "Cuba",
+    "Africa/Windhoek",
+    "America/Havana",
+    "Asia/Atyrau",
+    "Australia/Eucla",
+    "America/Guatemala",
+    "US/Mountain",
+    "Europe/Saratov",
+    "Asia/Jakarta",
+    "US/Aleutian",
+    "Asia/Thimphu",
+    "Pacific/Enderbury",
+    "Africa/Luanda",
+    "Europe/Kirov",
+    "Pacific/Tongatapu",
+    "Africa/Sao_Tome",
+    "Africa/El_Aaiun",
+    "Iran",
+    "America/Godthab",
+    "MST7MDT",
+    "Europe/Sofia",
+    "America/Thunder_Bay",
+    "Australia/South",
+    "America/Tegucigalpa",
+    "Africa/Monrovia",
+    "Egypt",
+    "America/Scoresbysund",
+    "Singapore",
+    "Etc/GMT-7",
+    "Africa/Lome",
+    "America/Metlakatla",
+    "Asia/Singapore",
+    "Pacific/Chatham",
+    "America/Paramaribo",
+    "Asia/Ulaanbaatar",
+    "Antarctica/Mawson",
+    "Australia/Yancowinna",
+    "GMT-0",
+    "America/Belize",
+    "Asia/Magadan",
+    "America/Fort_Wayne",
+    "Pacific/Nauru",
+    "Europe/Warsaw",
+    "Asia/Muscat",
+    "Europe/Sarajevo",
+    "Etc/GMT-2",
+    "Africa/Gaborone",
+    "Africa/Bamako",
+    "Asia/Qyzylorda",
+    "GB",
+    "Etc/GMT+2",
+    "America/Lima",
+    "Asia/Omsk",
+    "Kwajalein",
+    "Asia/Tokyo",
+    "America/Inuvik",
+    "Asia/Tashkent",
+    "Jamaica",
+    "America/Argentina/ComodRivadavia",
+    "Africa/Algiers",
+    "Africa/Harare",
+    "America/Argentina/San_Juan",
+    "Pacific/Guam",
+    "Europe/Astrakhan",
+    "Africa/Nairobi",
+    "US/Arizona",
+    "EST",
+    "Australia/Hobart",
+    "America/Kentucky/Monticello",
+    "Asia/Urumqi",
+    "Etc/GMT-14",
+    "Pacific/Johnston",
+    "Etc/Zulu",
+    "Asia/Ashgabat",
+    "Atlantic/Jan_Mayen",
+    "America/Aruba",
+    "America/Argentina/Jujuy",
+    "Etc/Greenwich",
+    "America/St_Lucia",
+    "Australia/Currie",
+    "Asia/Jerusalem",
+    "America/Atka",
+    "America/St_Kitts",
+    "America/El_Salvador",
+    "Europe/Riga",
+    "US/Central",
+    "Etc/GMT+3",
+    "America/Montreal",
+    "Australia/Lord_Howe",
+    "W-SU",
+    "America/Noronha",
+    "Canada/Mountain",
+    "America/Indiana/Vincennes",
+    "Europe/Simferopol",
+    "Pacific/Gambier",
+    "Africa/Tripoli",
+    "Asia/Novosibirsk",
+    "Navajo",
+    "Asia/Harbin",
+    "America/Rankin_Inlet",
+    "Asia/Kuching",
+    "America/Argentina/Salta",
+    "Europe/Bratislava",
+    "America/Glace_Bay",
+    "America/Argentina/Mendoza",
+    "Asia/Tomsk",
+    "America/Nipigon",
+    "Asia/Pontianak",
+    "Australia/Perth",
+    "Indian/Reunion",
+    "Europe/Uzhgorod",
+    "Europe/Athens",
+    "Brazil/DeNoronha",
+    "Zulu",
+    "Asia/Qostanay",
+    "Europe/Malta",
+    "Indian/Maldives",
+    "Asia/Jayapura",
+    "America/Denver",
+    "Atlantic/Reykjavik",
+    "Australia/West",
+    "America/Phoenix",
+    "Europe/Volgograd",
+    "Asia/Kamchatka",
+    "America/Kralendijk",
+    "America/Creston",
+    "Africa/Dakar",
+    "Europe/Andorra",
+    "Europe/Madrid",
+    "Japan",
+    "Pacific/Kosrae",
+    "GMT",
+    "America/Maceio",
+    "America/Porto_Acre",
+    "Asia/Ho_Chi_Minh",
+    "Asia/Kashgar",
+    "US/Samoa",
+    "Africa/Banjul",
+    "Asia/Anadyr",
+    "Etc/GMT-6",
+    "Pacific/Truk",
+    "America/Winnipeg",
+    "Africa/Ndjamena",
+    "Africa/Bissau",
+    "Asia/Baghdad",
+    "Israel",
+    "America/Guadeloupe",
+    "America/Buenos_Aires",
+    "America/Adak",
+    "Asia/Vladivostok",
+    "Pacific/Tarawa",
+    "Antarctica/Casey",
+    "Antarctica/Palmer",
+    "Asia/Irkutsk",
+    "Asia/Colombo",
+    "America/Port_of_Spain",
+    "America/North_Dakota/New_Salem",
+    "Europe/Dublin",
+    "Pacific/Ponape",
+    "America/Boise",
+    "Pacific/Yap",
+    "America/Whitehorse",
+    "PRC",
+    "Australia/Adelaide",
+    "America/Indiana/Vevay",
+    "Europe/Berlin",
+    "America/Recife",
+    "Europe/Oslo",
+    "Turkey",
+    "Europe/Luxembourg",
+    "Europe/Zagreb",
+    "America/Grenada",
+    "Africa/Blantyre",
+    "Asia/Tbilisi",
+    "America/Coyhaique",
+    "Pacific/Apia",
+    "Africa/Niamey",
+    "America/Guyana",
+    "Asia/Yerevan",
+    "Pacific/Honolulu",
+    "America/Hermosillo",
+    "Asia/Macao",
+    "Europe/Belfast",
+    "America/Indiana/Tell_City",
+    "Asia/Dushanbe",
+    "Asia/Novokuznetsk",
+    "Africa/Maseru",
+    "Pacific/Funafuti",
+    "Antarctica/McMurdo",
+    "America/Menominee",
+    "NZ-CHAT",
+    "MET",
+    "Asia/Dhaka",
+    "America/Jujuy",
+    "Europe/Vaduz",
+    "Europe/Budapest",
+    "Asia/Kuwait",
+    "Africa/Maputo",
+    "Asia/Aqtau",
+    "Europe/Belgrade",
+    "Africa/Ouagadougou",
+    "America/Puerto_Rico",
+    "Europe/Vilnius",
+    "Asia/Chita",
+    "America/Yellowknife",
+    "America/Ojinaga",
+    "America/Shiprock",
+    "America/Bahia",
+    "America/Tortola",
+    "America/Antigua",
+    "Etc/GMT+11",
+    "Atlantic/Bermuda",
+    "Asia/Khandyga",
+    "US/Pacific",
+    "Asia/Nicosia",
+    "Etc/GMT-3",
+    "Asia/Kabul",
+    "America/St_Johns",
+    "Etc/GMT+5",
+    "Asia/Dubai",
+    "Pacific/Galapagos",
+    "Etc/GMT-0",
+    "America/Indiana/Petersburg",
+    "America/Blanc-Sablon",
+    "Etc/GMT-10",
+    "Pacific/Tahiti",
+    "America/Argentina/Cordoba",
+    "Europe/Tiraspol",
+    "America/Pangnirtung",
+    "Africa/Casablanca",
+    "Brazil/Acre",
+    "Pacific/Pitcairn",
+    "Europe/Ljubljana",
+    "Africa/Cairo",
+    "America/Nuuk",
+    "Asia/Chungking",
+    "Africa/Dar_es_Salaam",
+    "Asia/Bahrain",
+    "Etc/GMT-12",
+    "Asia/Krasnoyarsk",
+    "US/East-Indiana",
+    "Europe/Minsk",
+    "Asia/Dili",
+    "Etc/GMT+7",
+    "Asia/Seoul",
+    "Asia/Yangon",
+    "Europe/Zurich",
+    "America/Knox_IN",
+    "Atlantic/Cape_Verde",
+    "Asia/Ashkhabad",
+    "Pacific/Port_Moresby",
+    "Europe/Prague",
+    "Africa/Kampala",
+    "America/Belem",
+    "Asia/Vientiane",
+    "America/Moncton",
+    "Europe/Monaco",
+    "America/Mazatlan",
+    "Africa/Brazzaville",
+    "America/Marigot",
+    "Asia/Dacca",
+    "Etc/GMT-11",
+    "Atlantic/Stanley",
+    "Asia/Kuala_Lumpur",
+    "Hongkong",
+    "Asia/Manila",
+    "America/Santiago",
+    "Indian/Mauritius",
+    "Europe/Brussels",
+    "Europe/Isle_of_Man",
+    "Australia/Broken_Hill",
+    "Asia/Brunei",
+    "Europe/Zaporozhye",
+    "America/New_York",
+    "Asia/Tehran",
+    "America/Panama",
+    "Africa/Libreville",
+    "America/Santo_Domingo",
+    "Pacific/Wake",
+    "Pacific/Auckland",
+    "America/Argentina/Ushuaia",
+    "Asia/Aden",
+    "Africa/Timbuktu",
+    "Asia/Bishkek",
+    "Etc/GMT+1",
+    "Pacific/Pohnpei",
+    "America/Sao_Paulo",
+    "US/Eastern",
+    "Europe/Nicosia",
+    "Pacific/Kwajalein",
+    "America/Iqaluit",
+    "America/Cayman",
+    "America/Monterrey",
+    "America/Guayaquil",
+    "America/Nome",
+    "America/Barbados",
+    "Pacific/Majuro",
+    "Etc/GMT-8",
+    "Asia/Phnom_Penh",
+    "America/Rainy_River",
+    "America/Indiana/Indianapolis",
+    "America/Atikokan",
+    "America/St_Barthelemy",
+    "PST8PDT",
+    "Universal",
+    "Indian/Kerguelen",
+    "Etc/UCT",
+    "Australia/LHI",
+    "Europe/Stockholm",
+    "Asia/Ujung_Pandang",
+    "America/Los_Angeles",
+    "Asia/Riyadh",
+    "America/Curacao",
+    "Africa/Johannesburg",
+    "Etc/GMT+4",
+    "Canada/Yukon",
+    "GB-Eire",
+    "Asia/Karachi",
+    "America/Mendoza",
+    "Australia/Victoria",
+    "America/Jamaica",
+    "Australia/Lindeman",
+    "Asia/Srednekolymsk",
+    "America/Bogota",
+    "Asia/Beirut",
+    "Asia/Calcutta",
+    "MST",
+    "Europe/Jersey",
+    "Etc/GMT+9",
+    "Asia/Ulan_Bator",
+    "Europe/Rome",
+    "Pacific/Wallis",
+    "Etc/GMT-4",
+    "Libya",
+    "UTC",
+    "Asia/Thimbu",
+    "Canada/Pacific",
+    "Africa/Kigali",
+    "America/Eirunepe",
+    "Europe/Kaliningrad",
+    "Atlantic/Canary",
+    "America/Mexico_City",
+    "Europe/Kiev",
+    "CET",
+    "Europe/Skopje",
+    "Pacific/Saipan",
+    "America/Sitka",
+    "Africa/Accra",
+    "Asia/Aqtobe",
+    "Asia/Oral",
+    "Pacific/Guadalcanal",
+    "Asia/Istanbul",
+    "America/Porto_Velho",
+    "America/Coral_Harbour",
+    "Mexico/General",
+    "Europe/Tallinn",
+]
+
 
 class _trigger_time:
     """
@@ -33,9 +634,14 @@ class Cron:
     """
 
     expression: str
+    timezone: Timezone = "UTC"
+
+    @property
+    def timezone_expression(self) -> str:
+        return f"CRON_TZ={self.timezone} {self.expression}"
 
     def __str__(self):
-        return f"Cron Trigger: {self.expression}"
+        return f"Cron Trigger: {self.timezone_expression}"
 
 
 @rich.repr.auto
@@ -90,8 +696,9 @@ class Trigger:
        values for inputs to these defaults.
     :param env_vars: (Dict[str, str]) Optional environment variables for the trigger, default is None. If provided, will
         replace the environment variables set in the config of the task.
-    :param interruptible: (bool) Whether the trigger is interruptible, default is None. If provided,
-     it overrides whatever is set in the config of the task.
+    :param interruptible: (bool) Whether the trigger run is interruptible,
+      default is None (maintains the configured behavior). If provided, it overrides whatever is set in the config
+      of the task.
     :param overwrite_cache: (bool) Whether to overwrite the cache, default is False.
     :param queue: (str) Optional queue to run the trigger in, default is None.
     :param labels: (Mapping[str, str]) Optional labels to attach to the trigger, default is None.
@@ -142,7 +749,7 @@ class Trigger:
             auto_activate (bool): Whether the trigger should be automatically activated.
             inputs (Dict[str, Any] | None): Optional inputs for the trigger.
             env_vars (Dict[str, str] | None): Optional environment variables.
-            interruptible (bool | None): Whether the trigger is interruptible.
+            interruptible (bool | None): Whether the triggered run is interruptible.
             overwrite_cache (bool): Whether to overwrite the cache.
             queue (str | None): Optional queue to run the trigger in.
             labels (Mapping[str, str] | None): Optional labels to attach to the trigger.
@@ -380,3 +987,14 @@ class Trigger:
             labels=labels,
             annotations=annotations,
         )
+
+
+if __name__ == "__main__":
+    from typing import get_args
+
+    vals = get_args(Timezone)
+    with open("/tmp/timezones.txt", "w") as f:
+        for v in vals:
+            c = Cron(expression="0 0 * * *", timezone=v)
+            f.write(f"{c.timezone_expression}\n")
+    print(f"Wrote {len(vals)} timezones to /tmp/timezones.txt")

flyte/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '2.0.0b26'
-__version_tuple__ = version_tuple = (2, 0, 0, 'b26')
+__version__ = version = '2.0.0b28'
+__version_tuple__ = version_tuple = (2, 0, 0, 'b28')
 
-__commit_id__ = commit_id = 'g713072212'
+__commit_id__ = commit_id = 'gb62a542c8'

flyte/cli/_plugins.py

@@ -0,0 +1,208 @@
+"""CLI Plugin System for Flyte.
+
+This module provides a plugin system that allows external packages to:
+1. Register new top-level CLI commands (e.g., flyte my-command)
+2. Register new subcommands in existing groups (e.g., flyte get my-object)
+3. Modify behavior of existing commands via hooks
+
+Plugins are discovered via Python entry points.
+
+Entry Point Groups:
+- flyte.plugins.cli.commands: Register new commands
+  - Entry point name "foo" -> flyte foo (top-level command)
+  - Entry point name "get.bar" -> flyte get bar (adds subcommand to get group)
+  - Note: At most one dot is supported. For nested groups, register the entire
+    group hierarchy as a top-level command (without dots).
+
+- flyte.plugins.cli.hooks: Modify existing commands
+  - Entry point name "run" -> modifies flyte run
+  - Entry point name "get.project" -> modifies flyte get project
+  - Note: At most one dot is supported.
+
+Example Plugin Package:
+    # In your-plugin/pyproject.toml
+    [project.entry-points."flyte.plugins.cli.commands"]
+    my-command = "your_plugin.cli:my_command"
+    get.my-object = "your_plugin.cli:get_my_object"
+
+    [project.entry-points."flyte.plugins.cli.hooks"]
+    run = "your_plugin.hooks:modify_run"
+
+    # In your-plugin/your_plugin/cli.py
+    import rich_click as click
+
+    @click.command()
+    def my_command():
+        '''My custom top-level command.'''
+        click.echo("Hello from plugin!")
+
+    @click.command()
+    def get_my_object():
+        '''Get my custom object.'''
+        click.echo("Getting my object...")
+
+    # In your-plugin/your_plugin/hooks.py
+    def modify_run(command):
+        '''Add behavior to flyte run command.'''
+        # Wrap invoke() instead of callback to ensure Click's full machinery runs
+        original_invoke = command.invoke
+
+        def wrapper(ctx):
+            # Do something before
+            click.echo("Plugin: Starting task...")
+
+            result = original_invoke(ctx)
+
+            # Do something after
+            click.echo("Plugin: Task completed!")
+            return result
+
+        command.invoke = wrapper
+        return command
+"""
+
+from importlib.metadata import entry_points
+from typing import Callable
+
+import rich_click as click
+
+from flyte._logging import logger
+
+# Type alias for command hooks
+CommandHook = Callable[[click.Command], click.Command]
+
+
+def discover_and_register_plugins(root_group: click.Group):
+    """
+    Discover all CLI plugins from installed packages and register them.
+
+    This function:
+    1. Discovers command plugins and adds them to the CLI
+    2. Discovers hook plugins and applies them to existing commands
+
+    Args:
+        root_group: The root Click command group (main CLI group)
+    """
+    _load_command_plugins(root_group)
+    _load_hook_plugins(root_group)
+
+
+def _load_command_plugins(root_group: click.Group):
+    """Load and register command plugins."""
+    for ep in entry_points(group="flyte.plugins.cli.commands"):
+        try:
+            command = ep.load()
+            if not isinstance(command, click.Command):
+                logger.warning(f"Plugin {ep.name} did not return a click.Command, got {type(command).__name__}")
+                continue
+
+            # Check if this is a subcommand (contains dot notation)
+            if "." in ep.name:
+                group_name, command_name = ep.name.split(".", 1)
+
+                # Validate: only support one level of nesting (group.command)
+                if "." in command_name:
+                    logger.error(
+                        f"Plugin {ep.name} uses multiple dots, which is not supported. "
+                        f"Use at most one dot (e.g., 'group.command'). "
+                        f"For nested groups, register the entire group hierarchy as a top-level command."
+                    )
+                    continue
+
+                _add_subcommand_to_group(root_group, group_name, command_name, command)
+            else:
+                # Top-level command
+                root_group.add_command(command, name=ep.name)
+                logger.info(f"Registered plugin command: flyte {ep.name}")
+
+        except Exception as e:
+            logger.error(f"Failed to load plugin command {ep.name}: {e}")
+
+
+def _load_hook_plugins(root_group: click.Group):
+    """Load and apply hook plugins to existing commands."""
+    for ep in entry_points(group="flyte.plugins.cli.hooks"):
+        try:
+            hook = ep.load()
+            if not callable(hook):
+                logger.warning(f"Plugin hook {ep.name} is not callable")
+                continue
+
+            # Check if this is a subcommand hook (contains dot notation)
+            if "." in ep.name:
+                group_name, command_name = ep.name.split(".", 1)
+
+                # Validate: only support one level of nesting (group.command)
+                if "." in command_name:
+                    logger.error(
+                        f"Plugin hook {ep.name} uses multiple dots, which is not supported. "
+                        f"Use at most one dot (e.g., 'group.command')."
+                    )
+                    continue
+
+                _apply_hook_to_subcommand(root_group, group_name, command_name, hook)
+            else:
+                # Top-level command hook
+                _apply_hook_to_command(root_group, ep.name, hook)
+
+        except Exception as e:
+            logger.error(f"Failed to apply hook {ep.name}: {e}")
+
+
+def _add_subcommand_to_group(root_group: click.Group, group_name: str, command_name: str, command: click.Command):
+    """Add a subcommand to an existing command group."""
+    if group_name not in root_group.commands:
+        logger.warning(f"Cannot add plugin subcommand '{command_name}' - group '{group_name}' does not exist")
+        return
+
+    group = root_group.commands[group_name]
+    if not isinstance(group, click.Group):
+        logger.warning(f"Cannot add plugin subcommand '{command_name}' - '{group_name}' is not a command group")
+        return
+
+    group.add_command(command, name=command_name)
+    # lower to debug later
+    logger.info(f"Registered plugin subcommand: flyte {group_name} {command_name}")
+
+
+def _apply_hook_to_command(root_group: click.Group, command_name: str, hook: CommandHook):
+    """Apply a hook to a top-level command."""
+    if command_name not in root_group.commands:
+        logger.warning(f"Cannot apply hook - command '{command_name}' does not exist")
+        return
+
+    original_command = root_group.commands[command_name]
+    try:
+        modified_command = hook(original_command)
+        root_group.commands[command_name] = modified_command
+        # lower to debug later
+        logger.info(f"Applied hook to command: flyte {command_name}")
+    except Exception as e:
+        logger.error(f"Hook failed for command {command_name}: {e}")
+        root_group.commands[command_name] = original_command
+
+
+def _apply_hook_to_subcommand(root_group: click.Group, group_name: str, command_name: str, hook: CommandHook):
+    """Apply a hook to a subcommand within a group."""
+    if group_name not in root_group.commands:
+        logger.warning(f"Cannot apply hook - group '{group_name}' does not exist")
+        return
+
+    group = root_group.commands[group_name]
+    if not isinstance(group, click.Group):
+        logger.warning(f"Cannot apply hook - '{group_name}' is not a command group")
+        return
+
+    if command_name not in group.commands:
+        logger.warning(f"Cannot apply hook - subcommand '{command_name}' does not exist in group '{group_name}'")
+        return
+
+    original_command = group.commands[command_name]
+    original_command.callback
+    try:
+        modified_command = hook(original_command)
+        group.commands[command_name] = modified_command
+        logger.info(f"Applied hook to subcommand: flyte {group_name} {command_name}")
+    except Exception as e:
+        logger.error(f"Hook failed for subcommand {group_name}.{command_name}: {e}")
+        group.commands[command_name] = original_command

flyte/cli/_run.py

@@ -155,7 +155,8 @@ class RunArguments:
 
     @classmethod
     def from_dict(cls, d: Dict[str, Any]) -> RunArguments:
-        return cls(**d)
+        modified = {k: v for k, v in d.items() if k in {f.name for f in fields(cls)}}
+        return cls(**modified)
 
     @classmethod
     def options(cls) -> List[click.Option]:

flyte/cli/main.py

@@ -12,6 +12,7 @@ from ._delete import delete
 from ._deploy import deploy
 from ._gen import gen
 from ._get import get
+from ._plugins import discover_and_register_plugins
 from ._run import run
 from ._serve import serve
 from ._update import update
@@ -203,3 +204,6 @@ main.add_command(build)
 main.add_command(whoami)  # type: ignore
 main.add_command(update)  # type: ignore
 main.add_command(serve)  # type: ignore
+
+# Discover and register CLI plugins from installed packages
+discover_and_register_plugins(main)

{flyte-2.0.0b26.dist-info → flyte-2.0.0b28.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flyte
-Version: 2.0.0b26
+Version: 2.0.0b28
 Summary: Add your description here
 Author-email: Ketan Umare <kumare3@users.noreply.github.com>
 Requires-Python: >=3.10

{flyte-2.0.0b26.dist-info → flyte-2.0.0b28.dist-info}/RECORD

@@ -28,8 +28,8 @@ flyte/_task_plugins.py,sha256=4KcdSERpQ0yR9q0CO3H2VC84egaj7r1fqHg8YO2xsFM,1093
 flyte/_timeout.py,sha256=zx5sFcbYmjJAJbZWSGzzX-BpC9HC7Jfs35T7vVhKwkk,1571
 flyte/_tools.py,sha256=lB3OiJSAhxzSMCYjLUF6nZjlFsmNpaRXtr3_Fefcxbg,747
 flyte/_trace.py,sha256=-BIprs2MbupWl3vsC_Pn33SV3fSVku1rUIsnwfmrIy0,5204
-flyte/_trigger.py,sha256=RprrBjpv1Zcim2WBrudZlvU_sh8EXaleT2edAZGbdkY,15110
-flyte/_version.py,sha256=_oC0KBsEUQxmnKUJK6kSReRRGGXJDqNMiK5M0eaBYG8,722
+flyte/_trigger.py,sha256=fsDQv-_j7sw0UTdTjMNdtjRSXuSty8ihijYPi0VTf00,28973
+flyte/_version.py,sha256=CNHpRTV1VYNFWs_n4VMyt7o9G2upJyh5TAKj7gVAB1g,722
 flyte/errors.py,sha256=L_05C2BAfDSWqmU6hHnJyzkBQDPcLeIiW6oGZl3sQLw,6965
 flyte/extend.py,sha256=fP6VD_OvBZims0LG3ODxzfugUQp6L04ATX3kZk27PMg,562
 flyte/models.py,sha256=dqpMq_BGi3_VR1zKK1pw8WckNmDT_4rLWCKu_uhY3Nw,18249
@@ -45,8 +45,8 @@ flyte/_cache/policy_function_body.py,sha256=_AcyN6XKRXq16yV5lWuRJYCIVUlmyPvvWuYR
 flyte/_code_bundle/__init__.py,sha256=G7DJTQ0UN_ETvdh55pYcWsTrZJKXEcyQl9iQQNQOBXQ,328
 flyte/_code_bundle/_ignore.py,sha256=INTPvv8ironCBIl_sJ_VaXnMd7gJcx0hL-OvrZNXdRo,4127
 flyte/_code_bundle/_packaging.py,sha256=H-_boKm4Wlri2D1zR-VzjvDxM-_R-_hKUBWVPUWYFKU,7094
-flyte/_code_bundle/_utils.py,sha256=SZd6t3ofqygrnpyxW9rPWOViXUrkNfICKKv6NmkAVhk,12209
-flyte/_code_bundle/bundle.py,sha256=Cr6BGdBlQOwJ31IPeNTnRrGpKEUAyfIq_9BHRgzKxCo,8965
+flyte/_code_bundle/_utils.py,sha256=bXGNKGMTt7QMucuJtWpHxra3tOLsKfb6qWNVV4OaRfA,12250
+flyte/_code_bundle/bundle.py,sha256=0BkBsd-suB9I70Fo8VuDol8k8P_ecSFb3AbRceSeOqI,8987
 flyte/_debug/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flyte/_debug/constants.py,sha256=tI4410tMsCGdgsrCCdk28RAWu6lyTQs7yRvzrRoR1HY,1516
 flyte/_debug/utils.py,sha256=Nc6n1Y_OdLMa4VtvigP6U738D4Fpbuog94g37tPwu6k,596
@@ -80,7 +80,7 @@ flyte/_internal/runtime/reuse.py,sha256=Z7TU-H4hdWUfD2rMl8iU1pljKLEEvB2YKZPEdQon
 flyte/_internal/runtime/rusty.py,sha256=cbEq2ISfYE93CJclDPGOgZOUKKjvPhdfZHuhmFHWzcI,7648
 flyte/_internal/runtime/task_serde.py,sha256=IiCK53YeO4ZZ53n5unQl1K9pz90WeaFUxQ8BTeh9ZHI,14748
 flyte/_internal/runtime/taskrunner.py,sha256=XXKWaCd1kGaaCXTKxEOVkPVq3hX62-qjV3eY4Ru_fkY,7894
-flyte/_internal/runtime/trigger_serde.py,sha256=pTAFFKySkE0MzlNIJnMmKHUhfxR0tsCn-bNd1bLx-YE,5000
+flyte/_internal/runtime/trigger_serde.py,sha256=ionYTtv9Etx1DDQPtJPpuc7uwF772F1GF_dwYEFy1Gk,5009
 flyte/_internal/runtime/types_serde.py,sha256=gVUKagPBoQ3Yukce_0DSbiwtyXJGqbwa-0p2zTNDSPc,1814
 flyte/_keyring/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flyte/_keyring/file.py,sha256=P_61Jf3SnT5SEHX0JiROntTaaWEtFOgLb4P1in22HVA,2958
@@ -106,11 +106,12 @@ flyte/cli/_gen.py,sha256=7K2eYQLGVr26I2OC3Xe_bzAn4ANYA5mPlBW5m1476PM,6079
 flyte/cli/_get.py,sha256=g53ThkT7qWgWPaxyr-0-WBOKs7vcOtlAjO4INBuSuFc,12136
 flyte/cli/_option.py,sha256=oC1Gs0u0UrOC1SsrFo-iCuAkqQvI1wJWCdjYXA9rW4Q,1445
 flyte/cli/_params.py,sha256=ZCyd714-CKXAzIL3tb3zfheTYpV0aqZFT3YEL3voS4w,20051
-flyte/cli/_run.py,sha256=ALZRbr1NzZH0RrrN7Up2Bv3RP2cA8hxGmZ446ngKp-I,20553
+flyte/cli/_plugins.py,sha256=azyORquPhBHMzGbkC38_QjAO_JvpEQvPQ-dc30auZQM,7969
+flyte/cli/_run.py,sha256=EYF3qLJJjYgyQ_pLdWJvvHOfwsk-p57NnytahxqOnH4,20646
 flyte/cli/_serve.py,sha256=ZozcCrNa8syOUn_973YGI1wVGi_ok6PE32goWfz_ftg,4518
 flyte/cli/_update.py,sha256=oEG5hEe1xML6e8Q8k74ejdPjrMZd8b1trjqKNcFLLV0,1115
 flyte/cli/_user.py,sha256=x32vEXpJkV6ONoCCcq7DbCIltHcALcSvm3PalNI57kI,339
-flyte/cli/main.py,sha256=HHFYhD_pX_TtyEus08AMHJ05hIOAAYKQfT3fUnoAlnQ,5659
+flyte/cli/main.py,sha256=_nRj8AmN36VwNn6zWvty1ii9IhpcIWJWugWUG3fYNpM,5808
 flyte/config/__init__.py,sha256=MiwEYK5Iv7MRR22z61nzbsbvZ9Q6MdmAU_g9If1Pmb8,144
 flyte/config/_config.py,sha256=IPfKboFar4CELhdJf583R10gG0PIhzUl1YAAP0ccObc,11109
 flyte/config/_internal.py,sha256=cmRtr6JeIkSfGgw9gloDkTuGpFNdwoUbnBJgbmAhPUE,3055
@@ -181,11 +182,11 @@ flyte/types/_renderer.py,sha256=ygcCo5l60lHufyQISFddZfWwLlQ8kJAKxUT_XnR_6dY,4818
 flyte/types/_string_literals.py,sha256=Tg14opnmkB7MvfNyUACbcPAeph6vg88aW5yC_G9kPIA,4167
 flyte/types/_type_engine.py,sha256=997leiWW55t8rWbYLnif9hGrgVP2tzpSMfOoF_MSPvg,96036
 flyte/types/_utils.py,sha256=-4M__ImscXMqsySaKDrCBJA-xX-c1-Vbc0_zPVoGRyc,2627
-flyte-2.0.0b26.data/scripts/debug.py,sha256=hnX2tlv9QbqckoT5CJ3c3apJj3tGDpsrdV7ZAsE7j34,911
-flyte-2.0.0b26.data/scripts/runtime.py,sha256=_FymfqibynxcmaGu5HT8XITk_5i2VsUbfpxKUbqenVc,8061
-flyte-2.0.0b26.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-flyte-2.0.0b26.dist-info/METADATA,sha256=x1NKBu7yC-mTulUIiTXtCHAoPwlM0d5Y70SgOdFLFw8,10345
-flyte-2.0.0b26.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-flyte-2.0.0b26.dist-info/entry_points.txt,sha256=rb43Gfxw40iPH5B6EUs6Ter0ekLkGXsj7R890_MOTyk,136
-flyte-2.0.0b26.dist-info/top_level.txt,sha256=7dkyFbikvA12LEZEqawx8oDG1CMod6hTliPj7iWzgYo,6
-flyte-2.0.0b26.dist-info/RECORD,,
+flyte-2.0.0b28.data/scripts/debug.py,sha256=hnX2tlv9QbqckoT5CJ3c3apJj3tGDpsrdV7ZAsE7j34,911
+flyte-2.0.0b28.data/scripts/runtime.py,sha256=_FymfqibynxcmaGu5HT8XITk_5i2VsUbfpxKUbqenVc,8061
+flyte-2.0.0b28.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+flyte-2.0.0b28.dist-info/METADATA,sha256=VYe9eEW2pfQxSPMYC9zBCmvfcGeiJrpnwymOT5-t92Q,10345
+flyte-2.0.0b28.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+flyte-2.0.0b28.dist-info/entry_points.txt,sha256=rb43Gfxw40iPH5B6EUs6Ter0ekLkGXsj7R890_MOTyk,136
+flyte-2.0.0b28.dist-info/top_level.txt,sha256=7dkyFbikvA12LEZEqawx8oDG1CMod6hTliPj7iWzgYo,6
+flyte-2.0.0b28.dist-info/RECORD,,