flyte 2.0.0b22__py3-none-any.whl → 2.0.0b30__py3-none-any.whl

flyte/models.py

@@ -14,7 +14,7 @@ from flyte._interface import extract_return_annotation, literal_to_enum
 from flyte._logging import logger
 
 if TYPE_CHECKING:
-    from flyteidl.core import literals_pb2
+    from flyteidl2.core import literals_pb2
 
     from flyte._internal.imagebuild.image_builder import ImageCache
     from flyte.report import Report
@@ -77,6 +77,37 @@ class ActionID:
         return self.new_sub_action(new_name)
 
 
+@rich.repr.auto
+@dataclass
+class PathRewrite:
+    """
+    Configuration for rewriting paths during input loading.
+    """
+
+    # If set, rewrites any path starting with this prefix to the new prefix.
+    old_prefix: str
+    new_prefix: str
+
+    def __post_init__(self):
+        if not self.old_prefix or not self.new_prefix:
+            raise ValueError("Both old_prefix and new_prefix must be non-empty strings.")
+        if self.old_prefix == self.new_prefix:
+            raise ValueError("old_prefix and new_prefix must be different.")
+
+    @classmethod
+    def from_str(cls, pattern: str) -> PathRewrite:
+        """
+        Create a PathRewrite from a string pattern of the form `old_prefix->new_prefix`.
+        """
+        parts = pattern.split("->")
+        if len(parts) != 2:
+            raise ValueError(f"Invalid path rewrite pattern: {pattern}. Expected format 'old_prefix->new_prefix'.")
+        return cls(old_prefix=parts[0], new_prefix=parts[1])
+
+    def __repr__(self) -> str:
+        return f"{self.old_prefix}->{self.new_prefix}"
+
+
 @rich.repr.auto
 @dataclass(frozen=True, kw_only=True)
 class RawDataPath:
@@ -86,6 +117,7 @@ class RawDataPath:
     """
 
     path: str
+    path_rewrite: Optional[PathRewrite] = None
 
     @classmethod
     def from_local_folder(cls, local_folder: str | pathlib.Path | None = None) -> RawDataPath:
@@ -112,7 +144,7 @@ class RawDataPath:
 
     def get_random_remote_path(self, file_name: Optional[str] = None) -> str:
         """
-        Returns a random path for uploading a file/directory to.
+        Returns a random path for uploading a file/directory to. This file/folder will not be created, it's just a path.
 
         :param file_name: If given, will be joined after a randomly generated portion.
         :return:
@@ -128,13 +160,14 @@ class RawDataPath:
 
         protocol = get_protocol(file_prefix)
         if "file" in protocol:
-            local_path = pathlib.Path(file_prefix) / random_string
+            parent_folder = pathlib.Path(file_prefix)
+            parent_folder.mkdir(exist_ok=True, parents=True)
             if file_name:
-                # Only if file name is given do we create the parent, because it may be needed as a folder otherwise
-                local_path = local_path / file_name
-                if not local_path.exists():
-                    local_path.parent.mkdir(exist_ok=True, parents=True)
-                    local_path.touch()
+                random_folder = parent_folder / random_string
+                random_folder.mkdir()
+                local_path = random_folder / file_name
+            else:
+                local_path = parent_folder / random_string
             return str(local_path.absolute())
 
         fs = fsspec.filesystem(protocol)
@@ -162,6 +195,8 @@ class TaskContext:
     :param action: The action ID of the current execution. This is always set, within a run.
     :param version: The version of the executed task. This is set when the task is executed by an action and will be
       set on all sub-actions.
+    :param custom_context: Context metadata for the action. If an action receives context, it'll automatically pass it
+      to any actions it spawns. Context will not be used for cache key computation.
     """
 
     action: ActionID
@@ -178,6 +213,7 @@ class TaskContext:
     data: Dict[str, Any] = field(default_factory=dict)
     mode: Literal["local", "remote", "hybrid"] = "remote"
     interactive_mode: bool = False
+    custom_context: Dict[str, str] = field(default_factory=dict)
 
     def replace(self, **kwargs) -> TaskContext:
         if "data" in kwargs:
@@ -319,10 +355,18 @@ class NativeInterface:
         """
         Extract the native interface from the given function. This is used to create a native interface for the task.
         """
+        # Get function parameters, defaults, varargs info (POSITIONAL_ONLY, VAR_POSITIONAL, KEYWORD_ONLY, etc.).
         sig = inspect.signature(func)
 
         # Extract parameter details (name, type, default value)
         param_info = {}
+        try:
+            # Get fully evaluated, real Python types for type checking.
+            hints = typing.get_type_hints(func, include_extras=True)
+        except Exception as e:
+            logger.warning(f"Could not get type hints for function {func.__name__}: {e}")
+            raise
+
         for name, param in sig.parameters.items():
             if param.kind in (inspect.Parameter.VAR_POSITIONAL, inspect.Parameter.VAR_KEYWORD):
                 raise ValueError(f"Function {func.__name__} cannot have variable positional or keyword arguments.")
@@ -330,13 +374,14 @@ class NativeInterface:
                 logger.warning(
                     f"Function {func.__name__} has parameter {name} without type annotation. Data will be pickled."
                 )
-            if typing.get_origin(param.annotation) is Literal:
-                param_info[name] = (literal_to_enum(param.annotation), param.default)
+            arg_type = hints.get(name, param.annotation)
+            if typing.get_origin(arg_type) is Literal:
+                param_info[name] = (literal_to_enum(arg_type), param.default)
             else:
-                param_info[name] = (param.annotation, param.default)
+                param_info[name] = (arg_type, param.default)
 
         # Get return type
-        outputs = extract_return_annotation(sig.return_annotation)
+        outputs = extract_return_annotation(hints.get("return", sig.return_annotation))
         return cls(inputs=param_info, outputs=outputs)
 
     def convert_to_kwargs(self, *args, **kwargs) -> Dict[str, Any]:

flyte/remote/__init__.py

@@ -7,12 +7,15 @@ __all__ = [
     "ActionDetails",
     "ActionInputs",
     "ActionOutputs",
+    "Phase",
     "Project",
     "Run",
     "RunDetails",
     "Secret",
     "SecretTypes",
     "Task",
+    "Trigger",
+    "User",
     "create_channel",
     "upload_dir",
     "upload_file",
@@ -22,6 +25,8 @@ from ._action import Action, ActionDetails, ActionInputs, ActionOutputs
 from ._client.auth import create_channel
 from ._data import upload_dir, upload_file
 from ._project import Project
-from ._run import Run, RunDetails
+from ._run import Phase, Run, RunDetails
 from ._secret import Secret, SecretTypes
 from ._task import Task
+from ._trigger import Trigger
+from ._user import User

flyte/remote/_action.py

@@ -20,14 +20,15 @@ from typing import (
 import grpc
 import rich.pretty
 import rich.repr
+from flyteidl2.common import identifier_pb2, list_pb2
+from flyteidl2.task import common_pb2
+from flyteidl2.workflow import run_definition_pb2, run_service_pb2
+from flyteidl2.workflow.run_service_pb2 import WatchActionDetailsResponse
 from rich.console import Console
 from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn
 
 from flyte import types
-from flyte._initialize import ensure_client, get_client, get_common_config
-from flyte._protos.common import identifier_pb2, list_pb2
-from flyte._protos.workflow import run_definition_pb2, run_service_pb2
-from flyte._protos.workflow.run_service_pb2 import WatchActionDetailsResponse
+from flyte._initialize import ensure_client, get_client, get_init_config
 from flyte.remote._common import ToJSONMixin
 from flyte.remote._logs import Logs
 from flyte.syncify import syncify
@@ -67,14 +68,14 @@ def _action_rich_repr(action: run_definition_pb2.Action) -> rich.repr.Result:
     """
     Rich representation of the action.
     """
-    yield "run", action.id.run.name
+    yield "name", action.id.run.name
     if action.metadata.HasField("task"):
-        yield "task", action.metadata.task.id.name
+        yield "task name", action.metadata.task.id.name
         yield "type", action.metadata.task.task_type
     elif action.metadata.HasField("trace"):
         yield "trace", action.metadata.trace.name
         yield "type", "trace"
-    yield "name", action.id.name
+    yield "action name", action.id.name
     yield from _action_time_phase(action)
     yield "group", action.metadata.group
     yield "parent", action.metadata.parent
@@ -98,9 +99,10 @@ def _action_details_rich_repr(
     """
     yield "name", action.id.run.name
     yield from _action_time_phase(action)
-    yield "task", action.resolved_task_spec.task_template.id.name
-    yield "task_type", action.resolved_task_spec.task_template.type
-    yield "task_version", action.resolved_task_spec.task_template.id.version
+    if action.HasField("task"):
+        yield "task", action.task.task_template.id.name
+        yield "task_type", action.task.task_template.type
+        yield "task_version", action.task.task_template.id.version
     yield "attempts", action.attempts
     yield "error", (f"{action.error_info.kind}: {action.error_info.message}" if action.HasField("error_info") else "NA")
     yield "phase", run_definition_pb2.Phase.Name(action.status.phase)
@@ -152,7 +154,7 @@ class Action(ToJSONMixin):
             key=sort_by[0],
             direction=(list_pb2.Sort.ASCENDING if sort_by[1] == "asc" else list_pb2.Sort.DESCENDING),
         )
-        cfg = get_common_config()
+        cfg = get_init_config()
         while True:
             req = list_pb2.ListRequest(
                 limit=100,
@@ -193,7 +195,7 @@ class Action(ToJSONMixin):
         :param name: The name of the action.
         """
         ensure_client()
-        cfg = get_common_config()
+        cfg = get_init_config()
         details: ActionDetails = await ActionDetails.get_details.aio(
             identifier_pb2.ActionIdentifier(
                 run=identifier_pb2.RunIdentifier(
@@ -458,7 +460,7 @@ class ActionDetails(ToJSONMixin):
         ensure_client()
         if not uri:
             assert name is not None and run_name is not None, "Either uri or name and run_name must be provided"
-        cfg = get_common_config()
+        cfg = get_init_config()
         return await cls.get_details.aio(
             identifier_pb2.ActionIdentifier(
                 run=identifier_pb2.RunIdentifier(
@@ -706,7 +708,7 @@ class ActionInputs(UserDict, ToJSONMixin):
     remote Union API.
     """
 
-    pb2: run_definition_pb2.Inputs
+    pb2: common_pb2.Inputs
     data: Dict[str, Any]
 
     def __repr__(self):
@@ -723,14 +725,14 @@ class ActionOutputs(tuple, ToJSONMixin):
     remote Union API.
     """
 
-    def __new__(cls, pb2: run_definition_pb2.Outputs, data: Tuple[Any, ...]):
+    def __new__(cls, pb2: common_pb2.Outputs, data: Tuple[Any, ...]):
         # Create the tuple part
         obj = super().__new__(cls, data)
         # Store extra data (you can't do this here directly since it's immutable)
         obj.pb2 = pb2
         return obj
 
-    def __init__(self, pb2: run_definition_pb2.Outputs, data: Tuple[Any, ...]):
+    def __init__(self, pb2: common_pb2.Outputs, data: Tuple[Any, ...]):
         # Normally you'd set instance attributes here,
         # but we've already set `pb2` in `__new__`
         self.pb2 = pb2

flyte/remote/_client/_protocols.py

@@ -1,13 +1,14 @@
 from typing import AsyncIterator, Protocol
 
 from flyteidl.admin import project_attributes_pb2, project_pb2, version_pb2
-from flyteidl.service import dataproxy_pb2
+from flyteidl.service import dataproxy_pb2, identity_pb2
+from flyteidl2.secret import payload_pb2
+from flyteidl2.task import task_service_pb2
+from flyteidl2.trigger import trigger_service_pb2
+from flyteidl2.workflow import run_logs_service_pb2, run_service_pb2
 from grpc.aio import UnaryStreamCall
 from grpc.aio._typing import RequestType
 
-from flyte._protos.secret import payload_pb2
-from flyte._protos.workflow import run_logs_service_pb2, run_service_pb2, task_service_pb2
-
 
 class MetadataServiceProtocol(Protocol):
     async def GetVersion(self, request: version_pb2.GetVersionRequest) -> version_pb2.GetVersionResponse: ...
@@ -131,3 +132,37 @@ class SecretService(Protocol):
     async def ListSecrets(self, request: payload_pb2.ListSecretsRequest) -> payload_pb2.ListSecretsResponse: ...
 
     async def DeleteSecret(self, request: payload_pb2.DeleteSecretRequest) -> payload_pb2.DeleteSecretResponse: ...
+
+
+class IdentityService(Protocol):
+    async def UserInfo(self, request: identity_pb2.UserInfoRequest) -> identity_pb2.UserInfoResponse: ...
+
+
+class TriggerService(Protocol):
+    async def DeployTrigger(
+        self, request: trigger_service_pb2.DeployTriggerRequest
+    ) -> trigger_service_pb2.DeployTriggerResponse: ...
+
+    async def GetTriggerDetails(
+        self, request: trigger_service_pb2.GetTriggerDetailsRequest
+    ) -> trigger_service_pb2.GetTriggerDetailsResponse: ...
+
+    async def GetTriggerRevisionDetails(
+        self, request: trigger_service_pb2.GetTriggerRevisionDetailsRequest
+    ) -> trigger_service_pb2.GetTriggerRevisionDetailsResponse: ...
+
+    async def ListTriggers(
+        self, request: trigger_service_pb2.ListTriggersRequest
+    ) -> trigger_service_pb2.ListTriggersResponse: ...
+
+    async def GetTriggerRevisionHistory(
+        self, request: trigger_service_pb2.GetTriggerRevisionHistoryRequest
+    ) -> trigger_service_pb2.GetTriggerRevisionHistoryResponse: ...
+
+    async def UpdateTriggers(
+        self, request: trigger_service_pb2.UpdateTriggersRequest
+    ) -> trigger_service_pb2.UpdateTriggersResponse: ...
+
+    async def DeleteTriggers(
+        self, request: trigger_service_pb2.DeleteTriggersRequest
+    ) -> trigger_service_pb2.DeleteTriggersResponse: ...

flyte/remote/_client/auth/_channel.py

@@ -7,6 +7,7 @@ import httpx
 from grpc.experimental.aio import init_grpc_aio
 
 from flyte._logging import logger
+from flyte._utils.org_discovery import hostname_from_url
 
 from ._authenticators.base import get_async_session
 from ._authenticators.factory import (
@@ -30,16 +31,19 @@ def bootstrap_ssl_from_server(endpoint: str) -> grpc.ChannelCredentials:
     :param endpoint: The endpoint URL to retrieve the SSL certificate from, may include port number
     :return: gRPC channel credentials created from the retrieved certificate
     """
+    hostname = hostname_from_url(endpoint)
+
     # Get port from endpoint or use 443
-    endpoint_parts = endpoint.rsplit(":", 1)
+    endpoint_parts = hostname.rsplit(":", 1)
     if len(endpoint_parts) == 2 and endpoint_parts[1].isdigit():
         server_address = (endpoint_parts[0], int(endpoint_parts[1]))
     else:
-        logger.warning(f"Unrecognized port in endpoint [{endpoint}], defaulting to 443.")
-        server_address = (endpoint, 443)
+        logger.warning(f"Unrecognized port in endpoint [{hostname}], defaulting to 443.")
+        server_address = (hostname, 443)
 
-    # Run the blocking SSL certificate retrieval in a thread pool
-    cert = ssl.get_server_certificate(server_address)
+    # Run the blocking SSL certificate retrieval with a timeout
+    logger.debug(f"Retrieving SSL certificate from {server_address}")
+    cert = ssl.get_server_certificate(server_address, timeout=10)
     return grpc.ssl_channel_credentials(str.encode(cert))
 
 
@@ -112,7 +116,7 @@ async def create_channel(
     if api_key:
         from flyte.remote._client.auth._auth_utils import decode_api_key
 
-        endpoint, client_id, client_secret, org = decode_api_key(api_key)
+        endpoint, client_id, client_secret, _org = decode_api_key(api_key)
         kwargs["auth_type"] = "ClientSecret"
         kwargs["client_id"] = client_id
         kwargs["client_secret"] = client_secret

flyte/remote/_client/controlplane.py

@@ -15,19 +15,22 @@ if "GRPC_VERBOSITY" not in os.environ:
 #### Has to be before grpc
 
 import grpc
-from flyteidl.service import admin_pb2_grpc, dataproxy_pb2_grpc
-
-from flyte._protos.secret import secret_pb2_grpc
-from flyte._protos.workflow import run_logs_service_pb2_grpc, run_service_pb2_grpc, task_service_pb2_grpc
+from flyteidl.service import admin_pb2_grpc, dataproxy_pb2_grpc, identity_pb2_grpc
+from flyteidl2.secret import secret_pb2_grpc
+from flyteidl2.task import task_service_pb2_grpc
+from flyteidl2.trigger import trigger_service_pb2_grpc
+from flyteidl2.workflow import run_logs_service_pb2_grpc, run_service_pb2_grpc
 
 from ._protocols import (
     DataProxyService,
+    IdentityService,
     MetadataServiceProtocol,
     ProjectDomainService,
     RunLogsService,
     RunService,
     SecretService,
     TaskService,
+    TriggerService,
 )
 from .auth import create_channel
 
@@ -38,7 +41,6 @@ class ClientSet:
         channel: grpc.aio.Channel,
         endpoint: str,
         insecure: bool = False,
-        data_proxy_channel: grpc.aio.Channel | None = None,
         **kwargs,
     ):
         self.endpoint = endpoint
@@ -50,6 +52,8 @@ class ClientSet:
         self._dataproxy = dataproxy_pb2_grpc.DataProxyServiceStub(channel=channel)
         self._log_service = run_logs_service_pb2_grpc.RunLogsServiceStub(channel=channel)
         self._secrets_service = secret_pb2_grpc.SecretServiceStub(channel=channel)
+        self._identity_service = identity_pb2_grpc.IdentityServiceStub(channel=channel)
+        self._trigger_service = trigger_service_pb2_grpc.TriggerServiceStub(channel=channel)
 
     @classmethod
     async def for_endpoint(cls, endpoint: str, *, insecure: bool = False, **kwargs) -> ClientSet:
@@ -105,5 +109,13 @@ class ClientSet:
     def secrets_service(self) -> SecretService:
         return self._secrets_service
 
+    @property
+    def identity_service(self) -> IdentityService:
+        return self._identity_service
+
+    @property
+    def trigger_service(self) -> TriggerService:
+        return self._trigger_service
+
     async def close(self, grace: float | None = None):
         return await self._channel.close(grace=grace)

flyte/remote/_console.py

@@ -9,8 +9,9 @@ def _get_http_domain(endpoint: str, insecure: bool) -> str:
     else:
         domain = parsed.netloc or parsed.path
     # TODO: make console url configurable
-    if domain.split(":")[0] == "localhost":
-        domain = "localhost:8080"
+    domain_split = domain.split(":")
+    if domain_split[0] == "localhost":
+        domain = domain if len(domain_split) > 1 else f"{domain}:8080"
     return f"{scheme}://{domain}"
 
 

flyte/remote/_data.py

@@ -15,7 +15,7 @@ import httpx
 from flyteidl.service import dataproxy_pb2
 from google.protobuf import duration_pb2
 
-from flyte._initialize import CommonInit, ensure_client, get_client, get_common_config
+from flyte._initialize import CommonInit, ensure_client, get_client, get_init_config, require_project_and_domain
 from flyte.errors import InitializationError, RuntimeSystemError
 from flyte.syncify import syncify
 
@@ -54,6 +54,7 @@ def hash_file(file_path: typing.Union[os.PathLike, str]) -> Tuple[bytes, str, in
     return h.digest(), h.hexdigest(), size
 
 
+@require_project_and_domain
 async def _upload_single_file(
     cfg: CommonInit, fp: Path, verify: bool = True, basedir: str | None = None
 ) -> Tuple[str, str]:
@@ -124,7 +125,7 @@ async def upload_file(fp: Path, verify: bool = True) -> Tuple[str, str]:
     """
     # This is a placeholder implementation. Replace with actual upload logic.
     ensure_client()
-    cfg = get_common_config()
+    cfg = get_init_config()
     if not fp.is_file():
         raise ValueError(f"{fp} is not a single file, upload arg must be a single file.")
     return await _upload_single_file(cfg, fp, verify=verify)
@@ -140,7 +141,7 @@ async def upload_dir(dir_path: Path, verify: bool = True) -> str:
     """
     # This is a placeholder implementation. Replace with actual upload logic.
     ensure_client()
-    cfg = get_common_config()
+    cfg = get_init_config()
     if not dir_path.is_dir():
         raise ValueError(f"{dir_path} is not a directory, upload arg must be a directory.")
 

flyte/remote/_logs.py

@@ -4,6 +4,9 @@ from dataclasses import dataclass
 from typing import AsyncGenerator, AsyncIterator
 
 import grpc
+from flyteidl2.common import identifier_pb2
+from flyteidl2.logs.dataplane import payload_pb2
+from flyteidl2.workflow import run_logs_service_pb2
 from rich.console import Console
 from rich.live import Live
 from rich.panel import Panel
@@ -11,9 +14,6 @@ from rich.text import Text
 
 from flyte._initialize import ensure_client, get_client
 from flyte._logging import logger
-from flyte._protos.common import identifier_pb2
-from flyte._protos.logs.dataplane import payload_pb2
-from flyte._protos.workflow import run_logs_service_pb2
 from flyte._tools import ipython_check, ipywidgets_check
 from flyte.errors import LogsNotYetAvailableError
 from flyte.syncify import syncify

flyte/remote/_run.py

@@ -5,10 +5,11 @@ from typing import AsyncGenerator, AsyncIterator, Literal, Tuple
 
 import grpc
 import rich.repr
+from flyteidl2.common import identifier_pb2, list_pb2
+from flyteidl2.workflow import run_definition_pb2, run_service_pb2
 
-from flyte._initialize import ensure_client, get_client, get_common_config
-from flyte._protos.common import identifier_pb2, list_pb2
-from flyte._protos.workflow import run_definition_pb2, run_service_pb2
+from flyte._initialize import ensure_client, get_client, get_init_config
+from flyte._logging import logger
 from flyte.syncify import syncify
 
 from . import Action, ActionDetails, ActionInputs, ActionOutputs
@@ -16,6 +17,11 @@ from ._action import _action_details_rich_repr, _action_rich_repr
 from ._common import ToJSONMixin
 from ._console import get_run_url
 
+# @kumare3 is sadpanda, because we have to create a mirror of phase types here, because protobuf phases are ghastly
+Phase = Literal[
+    "queued", "waiting_for_resources", "initializing", "running", "succeeded", "failed", "aborted", "timed_out"
+]
+
 
 @dataclass
 class Run(ToJSONMixin):
@@ -40,14 +46,16 @@ class Run(ToJSONMixin):
     @classmethod
     async def listall(
         cls,
-        filters: str | None = None,
+        in_phase: Tuple[Phase] | None = None,
+        created_by_subject: str | None = None,
         sort_by: Tuple[str, Literal["asc", "desc"]] | None = None,
         limit: int = 100,
     ) -> AsyncIterator[Run]:
         """
         Get all runs for the current project and domain.
 
-        :param filters: The filters to apply to the project list.
+        :param in_phase: Filter runs by one or more phases.
+        :param created_by_subject: Filter runs by the subject that created them. (this is not username, but the subject)
         :param sort_by: The sorting criteria for the project list, in the format (field, order).
         :param limit: The maximum number of runs to return.
         :return: An iterator of runs.
@@ -59,13 +67,44 @@ class Run(ToJSONMixin):
             key=sort_by[0],
             direction=(list_pb2.Sort.ASCENDING if sort_by[1] == "asc" else list_pb2.Sort.DESCENDING),
         )
-        cfg = get_common_config()
+        filters = []
+        if in_phase:
+            phases = [str(run_definition_pb2.Phase.Value(f"PHASE_{p.upper()}")) for p in in_phase]
+            logger.debug(f"Fetching run phases: {phases}")
+            if len(phases) > 1:
+                filters.append(
+                    list_pb2.Filter(
+                        function=list_pb2.Filter.Function.VALUE_IN,
+                        field="phase",
+                        values=phases,
+                    ),
+                )
+            else:
+                filters.append(
+                    list_pb2.Filter(
+                        function=list_pb2.Filter.Function.EQUAL,
+                        field="phase",
+                        values=phases[0],
+                    ),
+                )
+        if created_by_subject:
+            logger.debug(f"Fetching runs created by: {created_by_subject}")
+            filters.append(
+                list_pb2.Filter(
+                    function=list_pb2.Filter.Function.EQUAL,
+                    field="created_by",
+                    values=[created_by_subject],
+                ),
+            )
+
+        cfg = get_init_config()
         i = 0
         while True:
             req = list_pb2.ListRequest(
                 limit=min(100, limit),
                 token=token,
                 sort_by=sort_pb2,
+                filters=filters,
             )
             resp = await get_client().run_service.ListRuns(
                 run_service_pb2.ListRunsRequest(
@@ -225,6 +264,7 @@ class Run(ToJSONMixin):
         """
         Rich representation of the Run object.
         """
+        yield "url", f"[blue bold][link={self.url}]link[/link][/blue bold]"
         yield from _action_rich_repr(self.pb2.action)
 
     def __repr__(self) -> str:
@@ -276,7 +316,7 @@ class RunDetails(ToJSONMixin):
         :param name: The name of the run.
         """
         ensure_client()
-        cfg = get_common_config()
+        cfg = get_init_config()
         return await RunDetails.get_details.aio(
             run_id=identifier_pb2.RunIdentifier(
                 org=cfg.org,