intuned-runtime 1.3.2__py3-none-any.whl → 1.3.4__py3-none-any.whl

intuned_cli/commands/__init__.py

@@ -17,7 +17,7 @@ from .run_authsession_command import run__authsession as run__authsession  # typ
 from .run_authsession_create_command import authsession__create as authsession__create  # type: ignore
 from .run_authsession_create_command import run__authsession__create as run__authsession__create  # type: ignore
 from .run_authsession_update_command import authsession__update as authsession__update  # type: ignore
-from .run_authsession_update_command import run_authsession__update as run_authsession__update  # type: ignore
+from .run_authsession_update_command import run__authsession__update as run__authsession__update  # type: ignore
 from .run_authsession_validate_command import authsession__validate as authsession__validate  # type: ignore
 from .run_authsession_validate_command import run__authsession__validate as run__authsession__validate  # type: ignore
 from .run_command import run as run  # type: ignore

intuned_cli/commands/attempt_api_command.py

@@ -1,4 +1,3 @@
-import arguably
 import pytimeparse  # type: ignore
 
 from intuned_cli.controller.api import execute_attempt_api_cli
@@ -7,7 +6,6 @@ from intuned_cli.utils.auth_session_helpers import load_parameters
 from intuned_cli.utils.wrapper import cli_command
 
 
-@arguably.command  # type: ignore
 @cli_command
 async def attempt__api(
     api_name: str,

intuned_cli/commands/attempt_authsession_check_command.py

@@ -1,4 +1,3 @@
-import arguably
 import pytimeparse  # type: ignore
 
 from intuned_cli.controller.authsession import execute_attempt_check_auth_session_cli
@@ -6,7 +5,6 @@ from intuned_cli.utils.auth_session_helpers import assert_auth_enabled
 from intuned_cli.utils.wrapper import cli_command
 
 
-@arguably.command  # type: ignore
 @cli_command
 async def attempt__authsession__check(
     id: str,

intuned_cli/commands/attempt_authsession_command.py

@@ -1,10 +1,7 @@
-import arguably
-
 from intuned_cli.utils.help import print_help_and_exit
 from intuned_cli.utils.wrapper import cli_command
 
 
-@arguably.command  # type: ignore
 @cli_command
 async def attempt__authsession():
     """Execute san Intuned authsession attempt"""

intuned_cli/commands/attempt_authsession_create_command.py

@@ -1,4 +1,3 @@
-import arguably
 import pytimeparse  # type: ignore
 
 from intuned_cli.controller.authsession import execute_attempt_create_auth_session_cli
@@ -7,7 +6,6 @@ from intuned_cli.utils.auth_session_helpers import load_parameters
 from intuned_cli.utils.wrapper import cli_command
 
 
-@arguably.command  # type: ignore
 @cli_command
 async def attempt__authsession__create(
     parameters: str,

intuned_cli/commands/attempt_command.py

@@ -1,10 +1,7 @@
-import arguably
-
 from intuned_cli.utils.help import print_help_and_exit
 from intuned_cli.utils.wrapper import cli_command
 
 
-@arguably.command  # type: ignore
 @cli_command
 async def attempt():
     """Executes an Intuned attempt."""

intuned_cli/commands/authsession_command.py

@@ -1,10 +1,7 @@
-import arguably
-
 from intuned_cli.utils.help import print_help_and_exit
 from intuned_cli.utils.wrapper import cli_command
 
 
-@arguably.command  # type: ignore
 @cli_command
 async def authsession():
     """Manage Auth Sessions"""

intuned_cli/commands/authsession_record_command.py

@@ -1,4 +1,3 @@
-import arguably
 import pytimeparse  # type: ignore
 
 from intuned_cli.controller.authsession import execute_record_auth_session_cli
@@ -7,7 +6,6 @@ from intuned_cli.utils.auth_session_helpers import get_auth_session_recorder_par
 from intuned_cli.utils.wrapper import cli_command
 
 
-@arguably.command  # type: ignore
 @cli_command
 async def authsession__record(
     *,

intuned_cli/commands/command.py

@@ -1,12 +1,9 @@
-import arguably
-
 from intuned_cli.utils.console import console
 from intuned_cli.utils.error import CLIExit
 from intuned_cli.utils.help import print_help_and_exit
 from intuned_cli.utils.wrapper import cli_command
 
 
-@arguably.command  # type: ignore
 @cli_command
 async def __root__(
     *,

intuned_cli/commands/deploy_command.py

@@ -1,5 +1,3 @@
-import arguably
-
 from intuned_cli.controller.deploy import deploy_project
 from intuned_cli.controller.save import validate_intuned_project
 from intuned_cli.controller.save import validate_project_name
@@ -8,7 +6,6 @@ from intuned_cli.utils.error import CLIError
 from intuned_cli.utils.wrapper import cli_command
 
 
-@arguably.command  # type: ignore
 @cli_command
 async def deploy(
     project_name: str | None = None,

intuned_cli/commands/init_command.py

@@ -1,12 +1,9 @@
 from typing import Any
 
-import arguably
-
 from intuned_cli.utils.error import CLIError
 from intuned_cli.utils.wrapper import cli_command
 
 
-@arguably.command  # type: ignore
 @cli_command
 async def init(
     *args: Any,

intuned_cli/commands/run_api_command.py

@@ -1,4 +1,3 @@
-import arguably
 import pytimeparse  # type: ignore
 
 from intuned_cli.controller.api import AuthSessionInput
@@ -8,7 +7,6 @@ from intuned_cli.utils.auth_session_helpers import load_parameters
 from intuned_cli.utils.wrapper import cli_command
 
 
-@arguably.command  # type: ignore
 @cli_command
 async def run__api(
     api_name: str,

intuned_cli/commands/run_authsession_command.py

@@ -1,10 +1,7 @@
-import arguably
-
 from intuned_cli.utils.help import print_help_and_exit
 from intuned_cli.utils.wrapper import cli_command
 
 
-@arguably.command  # type: ignore
 @cli_command
 async def run__authsession():
     """Executes an Intuned authsession run"""

intuned_cli/commands/run_authsession_create_command.py

@@ -1,7 +1,6 @@
 from functools import WRAPPER_ASSIGNMENTS
 from functools import wraps
 
-import arguably
 import pytimeparse  # type: ignore
 
 from intuned_cli.controller.authsession import execute_run_create_auth_session_cli
@@ -10,7 +9,6 @@ from intuned_cli.utils.auth_session_helpers import load_parameters
 from intuned_cli.utils.wrapper import cli_command
 
 
-@cli_command
 async def _run_auth_session_create_impl(
     parameters: str,
     /,
@@ -61,8 +59,7 @@ async def _run_auth_session_create_impl(
     )
 
 
-# @wraps will give the wrapped function all the properties of the original function (parameter types, docstring, name, etc...). We want to do that for everything except the name, to register it as two different commands
-@arguably.command  # type: ignore
+@cli_command
 @wraps(_run_auth_session_create_impl, (a for a in WRAPPER_ASSIGNMENTS if a != "__name__"))
 async def run__authsession__create(
     *args,  # type: ignore
@@ -71,7 +68,7 @@ async def run__authsession__create(
     return await _run_auth_session_create_impl(*args, **kwargs)  # type: ignore
 
 
-@arguably.command  # type: ignore
+@cli_command
 @wraps(_run_auth_session_create_impl, (a for a in WRAPPER_ASSIGNMENTS if a != "__name__"))
 async def authsession__create(
     *args,  # type: ignore

intuned_cli/commands/run_authsession_update_command.py

@@ -1,7 +1,6 @@
 from functools import WRAPPER_ASSIGNMENTS
 from functools import wraps
 
-import arguably
 import pytimeparse  # type: ignore
 
 from intuned_cli.controller.authsession import execute_run_update_auth_session_cli
@@ -10,7 +9,6 @@ from intuned_cli.utils.auth_session_helpers import load_parameters
 from intuned_cli.utils.wrapper import cli_command
 
 
-@cli_command
 async def _run_update_authsession_impl(
     id: str,
     /,
@@ -62,17 +60,16 @@ async def _run_update_authsession_impl(
     )
 
 
-# @wraps will give the wrapped function all the properties of the original function (parameter types, docstring, name, etc...). We want to do that for everything except the name, to register it as two different commands
-@arguably.command  # type: ignore
+@cli_command
 @wraps(_run_update_authsession_impl, (a for a in WRAPPER_ASSIGNMENTS if a != "__name__"))
-async def run_authsession__update(
+async def run__authsession__update(
     *args,  # type: ignore
     **kwargs,  # type: ignore
 ):
     return await _run_update_authsession_impl(*args, **kwargs)  # type: ignore
 
 
-@arguably.command  # type: ignore
+@cli_command
 @wraps(_run_update_authsession_impl, (a for a in WRAPPER_ASSIGNMENTS if a != "__name__"))
 async def authsession__update(
     *args,  # type: ignore

intuned_cli/commands/run_authsession_validate_command.py

@@ -1,7 +1,6 @@
 from functools import WRAPPER_ASSIGNMENTS
 from functools import wraps
 
-import arguably
 import pytimeparse  # type: ignore
 
 from intuned_cli.controller.authsession import execute_run_validate_auth_session_cli
@@ -9,7 +8,6 @@ from intuned_cli.utils.auth_session_helpers import assert_auth_enabled
 from intuned_cli.utils.wrapper import cli_command
 
 
-@cli_command
 async def _run_validate_authsession_impl(
     id: str,
     /,
@@ -59,8 +57,7 @@ async def _run_validate_authsession_impl(
     )
 
 
-# @wraps will give the wrapped function all the properties of the original function (parameter types, docstring, name, etc...). We want to do that for everything except the name, to register it as two different commands
-@arguably.command  # type: ignore
+@cli_command
 @wraps(_run_validate_authsession_impl, (a for a in WRAPPER_ASSIGNMENTS if a != "__name__"))
 async def run__authsession__validate(
     *args,  # type: ignore
@@ -69,7 +66,7 @@ async def run__authsession__validate(
     return await _run_validate_authsession_impl(*args, **kwargs)  # type: ignore
 
 
-@arguably.command  # type: ignore
+@cli_command
 @wraps(_run_validate_authsession_impl, (a for a in WRAPPER_ASSIGNMENTS if a != "__name__"))
 async def authsession__validate(
     *args,  # type: ignore

intuned_cli/commands/run_command.py

@@ -1,10 +1,7 @@
-import arguably
-
 from intuned_cli.utils.help import print_help_and_exit
 from intuned_cli.utils.wrapper import cli_command
 
 
-@arguably.command  # type: ignore
 @cli_command
 async def run():
     """Executes an Intuned run."""

intuned_cli/commands/save_command.py

@@ -1,5 +1,3 @@
-import arguably
-
 from intuned_cli.controller.save import save_project
 from intuned_cli.controller.save import validate_intuned_project
 from intuned_cli.controller.save import validate_project_name
@@ -8,7 +6,6 @@ from intuned_cli.utils.error import CLIError
 from intuned_cli.utils.wrapper import cli_command
 
 
-@arguably.command  # type: ignore
 @cli_command
 async def save(
     project_name: str | None = None,

intuned_cli/controller/deploy.py

@@ -11,8 +11,6 @@ from intuned_cli.utils.backend import get_base_url
 from intuned_cli.utils.console import console
 from intuned_cli.utils.error import CLIError
 
-supported_playwright_versions = ["1.46.0", "1.52.0"]
-
 project_deploy_timeout = 10 * 60
 project_deploy_check_period = 5
 

intuned_cli/controller/save.py

@@ -6,7 +6,6 @@ from typing import Any
 
 import httpx
 import pathspec
-import toml
 from anyio import Path
 from dotenv.main import DotEnv
 from pydantic import BaseModel
@@ -27,8 +26,6 @@ from runtime.env import api_key_env_var_key
 from runtime.env import project_env_var_key
 from runtime.env import workspace_env_var_key
 
-supported_playwright_versions = ["1.46.0", "1.52.0"]
-
 
 class IntunedPyprojectToml(BaseModel):
     class _Tool(BaseModel):
@@ -48,21 +45,6 @@ async def validate_intuned_project():
     if not await pyproject_toml_path.exists():
         raise CLIError("pyproject.toml file is missing in the current directory.")
 
-    content = await pyproject_toml_path.read_text()
-    json_content = toml.loads(content)
-    try:
-        pyproject_toml = IntunedPyprojectToml.model_validate(json_content)
-    except Exception as e:
-        raise CLIError(f"Failed to parse pyproject.toml: {e}") from e
-
-    playwright_version = pyproject_toml.tool.poetry.dependencies.get("playwright")
-
-    if playwright_version not in supported_playwright_versions:
-        raise CLIError(
-            f"Unsupported Playwright version '{playwright_version}'. "
-            f"Supported versions are: {', '.join(supported_playwright_versions)}."
-        )
-
     intuned_json = await load_intuned_json()
 
     api_folder = cwd / "api"

intuned_cli/utils/wrapper.py

@@ -4,6 +4,10 @@ from collections.abc import Callable
 from functools import wraps
 from typing import Any
 from typing import cast
+from typing import ParamSpec
+from typing import TypeVar
+
+import arguably
 
 from intuned_cli.utils.browser import close_cli_browser
 from intuned_cli.utils.browser import is_cli_browser_launched
@@ -12,11 +16,17 @@ from intuned_cli.utils.error import CLIError
 from intuned_cli.utils.error import CLIExit
 from intuned_cli.utils.error import log_automation_error
 from runtime.errors.run_api_errors import RunApiError
+from runtime.utils.anyio import run_sync
+
+P = ParamSpec("P")
+R = TypeVar("R")
 
 
-def cli_command[F: Callable[..., Awaitable[Any]]](fn: F) -> F:
+def cli_command(fn: Callable[P, Awaitable[R]]) -> Callable[P, R]:
+    @arguably.command  # type: ignore
     @wraps(fn)
-    async def wrapper(*args: Any, **kwargs: Any):
+    @run_sync
+    async def wrapper(*args: Any, **kwargs: Any) -> R:
         keep_open = kwargs.get("keep_browser_open", False)
         if keep_open:
             console.print(
@@ -47,7 +57,7 @@ def cli_command[F: Callable[..., Awaitable[Any]]](fn: F) -> F:
                 await _wait_for_user_input()
             await close_cli_browser()
 
-    return cast(F, wrapper)
+    return cast(Callable[P, R], wrapper)
 
 
 async def _wait_for_user_input():

intuned_internal_cli/commands/__init__.py

@@ -1,23 +1,15 @@
-from .ai_source import ai_source  # type: ignore
-from .ai_source import ai_source__deploy  # type: ignore
-from .browser import browser__save_state  # type: ignore
-from .init import init  # type: ignore
-from .project import project  # type: ignore
-from .project import project__run  # type: ignore
-from .project import project__type_check  # type: ignore
-from .project.auth_session import project__auth_session__check  # type: ignore
-from .project.auth_session import project__auth_session__create  # type: ignore
-from .project.auth_session import project__auth_session__load  # type: ignore
-from .publish_packages import publish_packages  # type: ignore
-from .root import __root__  # type: ignore
+from .browser import browser__save_state
+from .project import project
+from .project import project__run
+from .project import project__type_check
+from .project.auth_session import project__auth_session__check
+from .project.auth_session import project__auth_session__create
+from .project.auth_session import project__auth_session__load
+from .root import __root__
 
 __all__ = [
     "project__run",
-    "publish_packages",
-    "init",
     "project",
-    "ai_source__deploy",
-    "ai_source",
     "project__auth_session__load",
     "project__auth_session__create",
     "project__auth_session__check",

intuned_internal_cli/commands/browser/__init__.py

@@ -1,3 +1,3 @@
-from .save_state import browser__save_state  # type: ignore
+from .save_state import browser__save_state
 
 __all__ = ["browser__save_state"]

intuned_internal_cli/commands/browser/save_state.py

@@ -1,13 +1,12 @@
 import json
 import os
 
-import arguably
-
+from intuned_internal_cli.utils.wrapper import internal_cli_command
 from runtime.browser import launch_browser
 from runtime.browser.storage_state import get_storage_state
 
 
-@arguably.command  # type: ignore
+@internal_cli_command
 async def browser__save_state(
     *,
     cdp_address: str,

intuned_internal_cli/commands/project/__init__.py

@@ -1,11 +1,10 @@
-from .auth_session import project__auth_session__check  # type: ignore
-from .auth_session import project__auth_session__create  # type: ignore
-from .auth_session import project__auth_session__load  # type: ignore
-from .project import project  # type: ignore
-from .run import project__run  # type: ignore
-from .run_interface import project__run_interface  # type: ignore
-from .type_check import project__type_check  # type: ignore
-from .upgrade import project__upgrade  # type: ignore
+from .auth_session import project__auth_session__check
+from .auth_session import project__auth_session__create
+from .auth_session import project__auth_session__load
+from .project import project
+from .run import project__run
+from .run_interface import project__run_interface
+from .type_check import project__type_check
 
 __all__ = [
     "run",
@@ -15,6 +14,5 @@ __all__ = [
     "project__auth_session__load",
     "project__auth_session__create",
     "project__auth_session__check",
-    "project__upgrade",
     "project__run_interface",
 ]

intuned_internal_cli/commands/project/auth_session/__init__.py

@@ -1,5 +1,5 @@
-from .check import project__auth_session__check  # type: ignore
-from .create import project__auth_session__create  # type: ignore
-from .load import project__auth_session__load  # type: ignore
+from .check import project__auth_session__check
+from .create import project__auth_session__create
+from .load import project__auth_session__load
 
 __all__ = ["project__auth_session__check", "project__auth_session__create", "project__auth_session__load"]

intuned_internal_cli/commands/project/auth_session/check.py

@@ -2,7 +2,6 @@ import json
 import os
 from typing import Any
 
-import arguably
 import pydantic  # type: ignore
 from more_termcolor import bold  # type: ignore
 from more_termcolor import green  # type: ignore
@@ -12,6 +11,7 @@ from tenacity import retry_if_not_result
 from tenacity import RetryError
 from tenacity import stop_after_attempt
 
+from intuned_internal_cli.utils.wrapper import internal_cli_command
 from runtime.context.context import IntunedContext
 from runtime.errors.run_api_errors import RunApiError
 from runtime.run.intuned_settings import load_intuned_settings
@@ -27,7 +27,7 @@ from runtime.types.run_types import StorageState
 from runtime.types.run_types import TracingDisabled
 
 
-@arguably.command  # type: ignore
+@internal_cli_command
 async def project__auth_session__check(
     *,
     no_headless: bool = False,

intuned_internal_cli/commands/project/auth_session/create.py

@@ -2,8 +2,7 @@ import json
 import os
 from typing import Any
 
-import arguably
-
+from intuned_internal_cli.utils.wrapper import internal_cli_command
 from runtime.context.context import IntunedContext
 from runtime.errors.run_api_errors import RunApiError
 from runtime.run.intuned_settings import load_intuned_settings
@@ -16,7 +15,7 @@ from runtime.types.run_types import StandaloneRunOptions
 from runtime.types.run_types import TracingDisabled
 
 
-@arguably.command  # type: ignore
+@internal_cli_command
 async def project__auth_session__create(
     *,
     no_headless: bool = False,

intuned_internal_cli/commands/project/auth_session/load.py

@@ -1,15 +1,14 @@
 import json
 import os
 
-import arguably
-
+from intuned_internal_cli.utils.wrapper import internal_cli_command
 from runtime.browser import launch_browser
 from runtime.browser.storage_state import set_storage_state
 from runtime.run.intuned_settings import load_intuned_settings
 from runtime.types.run_types import StorageState
 
 
-@arguably.command  # type: ignore
+@internal_cli_command
 async def project__auth_session__load(
     *,
     cdp_address: str,

intuned_internal_cli/commands/project/project.py

@@ -1,7 +1,7 @@
-import arguably
+from intuned_internal_cli.utils.wrapper import internal_cli_command
 
 
-@arguably.command  # type: ignore
+@internal_cli_command
 def project():
     """
     Commands to run on automation projects.

intuned_internal_cli/commands/project/run.py

@@ -10,7 +10,6 @@ from importlib import import_module
 from typing import Any
 from typing import cast
 
-import arguably
 from dotenv import find_dotenv
 from dotenv import load_dotenv
 from more_termcolor import bold  # type: ignore
@@ -21,6 +20,7 @@ from more_termcolor import on_blue  # type: ignore
 from more_termcolor import underline  # type: ignore
 from more_termcolor import yellow  # type: ignore
 
+from intuned_internal_cli.utils.wrapper import internal_cli_command
 from runtime.run.intuned_settings import load_intuned_settings
 from runtime.types import Payload
 from runtime.types.run_types import Auth
@@ -38,7 +38,7 @@ class Mode(StrEnum):
     ide = "ide"
 
 
-@arguably.command  # type: ignore
+@internal_cli_command
 async def project__run(
     *,
     api_name: str = "default",