intuned-runtime 1.2.1__py3-none-any.whl → 1.2.2__py3-none-any.whl

intuned_cli/controller/save.py

@@ -0,0 +1,260 @@
+import io
+import json
+import re
+import uuid
+from typing import Any
+
+import httpx
+import pathspec
+import toml
+from anyio import Path
+from dotenv.main import DotEnv
+from pydantic import BaseModel
+from pydantic import ValidationError
+
+from intuned_cli.types import DirectoryNode
+from intuned_cli.types import FileNode
+from intuned_cli.types import FileNodeContent
+from intuned_cli.types import FileSystemTree
+from intuned_cli.utils.api_helpers import load_intuned_json
+from intuned_cli.utils.backend import get_base_url
+from intuned_cli.utils.console import console
+from intuned_cli.utils.error import CLIError
+from intuned_cli.utils.exclusions import exclusions
+from runtime.constants import api_key_header_name
+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):
+        class _Poetry(BaseModel):
+            dependencies: dict[str, Any]
+
+        poetry: _Poetry
+
+    tool: _Tool
+
+
+async def validate_intuned_project():
+    cwd = await Path().resolve()
+
+    pyproject_toml_path = cwd / "pyproject.toml"
+
+    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"
+    if not await api_folder.exists() or not await api_folder.is_dir():
+        raise CLIError("api directory does not exist in the current directory.")
+
+    if intuned_json.auth_sessions.enabled:
+        auth_sessions_folder = cwd / "auth-sessions"
+        if not await auth_sessions_folder.exists() or not await auth_sessions_folder.is_dir():
+            raise CLIError("auth-sessions directory does not exist in the api directory.")
+
+    return intuned_json
+
+
+def validate_project_name(project_name: str):
+    if len(project_name) > 50:
+        raise CLIError("Project name must be 50 characters or less.")
+
+    project_name_regex = r"^[a-z0-9]+(?:[-_][a-z0-9]+)*$"
+    if not re.match(project_name_regex, project_name):
+        raise CLIError("Project name can only contain lowercase letters, numbers, hyphens, and underscores in between.")
+
+    try:
+        import uuid
+
+        uuid.UUID(project_name)
+        raise CLIError("Project name cannot be a UUID.")
+    except ValueError:
+        # Not a valid UUID, continue
+        pass
+
+
+async def get_file_tree_from_project(path: Path, *, exclude: list[str] | None = None):
+    # Create pathspec object for gitignore-style pattern matching
+    spec = None
+    if exclude:
+        spec = pathspec.PathSpec.from_lines("gitwildmatch", exclude)
+
+    async def traverse(current_path: Path, tree: FileSystemTree):
+        async for entry in current_path.iterdir():
+            relative_path_name = entry.relative_to(path).as_posix()
+            basename = entry.name
+
+            # Check if this path should be excluded
+            if spec and spec.match_file(relative_path_name):
+                continue
+
+            if await entry.is_dir():
+                subtree = FileSystemTree(root={})
+                tree.root[basename] = DirectoryNode(directory=subtree)
+                # For directories, check if the directory itself is excluded
+                # If not excluded, traverse into it
+                await traverse(entry, subtree)
+            elif await entry.is_file():
+                tree.root[basename] = FileNode(file=FileNodeContent(contents=await entry.read_text()))
+
+    results = FileSystemTree(root={})
+    await traverse(path, results)
+    return results
+
+
+def mapFileTreeToIdeFileTree(file_tree: FileSystemTree):
+    """
+    Maps the file tree to IDE parameters format by processing parameters directory
+    and converting it to ____testParameters structure.
+    """
+
+    if not file_tree:
+        return
+
+    parameters_node = file_tree.root.get("parameters")
+    if parameters_node is None:
+        return
+
+    if not isinstance(parameters_node, DirectoryNode):
+        return
+
+    api_parameters_map: dict[str, list[dict[str, Any]]] = {}
+    cli_parameters = list(parameters_node.directory.root.keys())
+    test_parameters = DirectoryNode(directory=FileSystemTree(root={}))
+
+    for parameter_key in cli_parameters:
+        # If parameter of type directory, discard it and continue
+        parameter = parameters_node.directory.root[parameter_key]
+
+        if isinstance(parameter, DirectoryNode):
+            continue
+
+        if not parameter.file.contents.strip():
+            continue
+
+        try:
+            parameter_payload = json.loads(parameter.file.contents)
+        except json.JSONDecodeError:
+            continue
+
+        if "__api-name" not in parameter_payload:
+            continue
+
+        api = parameter_payload["__api-name"]
+        # Create parameter value by excluding __api-name
+        parameter_value = {k: v for k, v in parameter_payload.items() if k != "__api-name"}
+
+        test_parameter: dict[str, Any] = {
+            "name": parameter_key.replace(".json", ""),
+            "lastUsed": False,
+            "id": str(uuid.uuid4()),
+            "value": json.dumps(parameter_value),
+        }
+
+        if api not in api_parameters_map:
+            api_parameters_map[api] = []
+        api_parameters_map[api].append(test_parameter)
+
+    for api, parameters in api_parameters_map.items():
+        # By default, last one used is the last one in the map
+        if len(parameters) > 0:
+            parameters[-1]["lastUsed"] = True
+
+        test_parameters.directory.root[f"{api}.json"] = FileNode(
+            file=FileNodeContent(contents=json.dumps(parameters, indent=2))
+        )
+
+    del file_tree.root["parameters"]
+    file_tree.root["____testParameters"] = test_parameters
+
+
+class SaveProjectResponse(BaseModel):
+    id: str
+
+
+async def save_project(
+    *,
+    project_name: str,
+    workspace_id: str,
+    api_key: str,
+):
+    base_url = get_base_url()
+    url = f"{base_url}/api/v1/workspace/{workspace_id}/projects/{project_name}"
+    print(f"calling {url}")
+    headers = {
+        api_key_header_name: api_key,
+        "Content-Type": "application/json",
+    }
+    cwd = await Path().resolve()
+    file_tree = await get_file_tree_from_project(cwd, exclude=exclusions)
+    mapFileTreeToIdeFileTree(file_tree)
+
+    payload: dict[str, Any] = {
+        "codeTree": file_tree.model_dump(mode="json"),
+        "platformType": "CLI",
+        "language": "python",
+    }
+
+    async with httpx.AsyncClient() as client:
+        response = await client.put(url, headers=headers, json=payload)
+        if response.status_code < 200 or response.status_code >= 300:
+            if response.status_code == 401:
+                raise CLIError("Invalid API key. Please check your API key and try again.")
+
+            raise CLIError(
+                f"[red bold]Invalid response from server:[/red bold]\n [bright_red]{response.status_code} {response.text}[/bright_red][red bold]\nProject save failed.[/red bold]"
+            )
+
+    console.print("[green]Project saved successfully.[/green]")
+    try:
+        response = SaveProjectResponse.model_validate(response.json())
+    except ValidationError:
+        console.print(f"[yellow]Could not parse response:[/yellow]\n {response.text}")
+        return
+
+    dotenv_path = cwd / ".env"
+    if not await dotenv_path.exists():
+        content_to_write = f"""{workspace_env_var_key}={workspace_id}
+{project_env_var_key}={response.id}
+{api_key_env_var_key}={api_key}
+"""
+        await dotenv_path.write_text(content_to_write)
+        console.print("[green]Created .env with project credentials.[/green]")
+        return
+
+    dotenv_content = await dotenv_path.read_text()
+    dotenv = DotEnv(
+        dotenv_path=None,
+        stream=io.StringIO(dotenv_content),
+    ).dict()
+    content_to_append = ""
+    if dotenv.get(project_env_var_key) is None or dotenv.get(project_env_var_key) != response.id:
+        content_to_append += f"{project_env_var_key}={response.id}"
+    if dotenv.get(workspace_env_var_key) is None:
+        content_to_append += f"\n{workspace_env_var_key}={workspace_id}"
+    if dotenv.get(api_key_env_var_key) is None:
+        content_to_append += f"\n{api_key_env_var_key}={api_key}"
+
+    await dotenv_path.write_text(f"{dotenv_content}\n{content_to_append}\n")
+    console.print("[green]Updated .env with project credentials.[/green]")

intuned_cli/utils/backend.py

@@ -1,5 +1,31 @@
 import os
 
+from intuned_cli.types import IntunedJson
+from intuned_cli.utils.error import CLIError
+
 
 def get_base_url():
     return os.environ.get("INTUNED_API_BASE_URL") or os.environ.get("INTUNED_API_DOMAIN") or "https://app.intuned.io"
+
+
+async def get_intuned_api_auth_credentials(
+    *, intuned_json: IntunedJson, workspace_id: str | None, api_key: str | None
+) -> tuple[str, str]:
+    """
+    Retrieves the Intuned API authentication credentials from environment variables.
+
+    Returns:
+        tuple: A tuple containing the workspace ID and API key.
+    """
+    workspace_id = workspace_id or intuned_json.workspace_id
+    api_key = api_key or os.environ.get("INTUNED_API_KEY")
+
+    if not workspace_id:
+        raise CLIError("Workspace ID is required. Please provide it via command line options or Intuned.json")
+
+    if not api_key:
+        raise CLIError(
+            "API key is required. Please provide it via command line options or INTUNED_API_KEY environment variable."
+        )
+
+    return workspace_id, api_key

intuned_cli/utils/error.py

@@ -2,6 +2,7 @@ import traceback
 
 from intuned_cli.utils.console import console
 from runtime.errors.run_api_errors import AutomationError
+from runtime.errors.run_api_errors import RunApiError
 
 
 class CLIError(Exception):
@@ -20,8 +21,11 @@ class CLIError(Exception):
         self.auto_color = auto_color
 
 
-def log_automation_error(e: AutomationError):
+def log_automation_error(e: RunApiError):
     console.print("[bold red]An error occurred while running the API:[/bold red]")
 
-    stack_trace = traceback.format_exception(type(e.error), value=e.error, tb=e.error.__traceback__)
-    console.print(f"[red]{''.join(stack_trace)}[/red]")
+    if isinstance(e, AutomationError):
+        stack_trace = traceback.format_exception(type(e.error), value=e.error, tb=e.error.__traceback__)
+        console.print(f"[red]{''.join(stack_trace)}[/red]")
+    else:
+        console.print(f"[red]{e}[/red]")

intuned_internal_cli/commands/project/auth_session/check.py

@@ -94,7 +94,6 @@ async def project__auth_session__check(
                     if cdp_address is not None
                     else StandaloneRunOptions(headless=not no_headless),
                     auth=Auth(
-                        run_check=False,
                         session=StateSession(
                             state=auth_session,
                         ),

intuned_internal_cli/commands/project/run.py

@@ -258,7 +258,6 @@ async def _run_payloads_and_extend(
                 headless=headless,
                 cdp_address=cdp_address,
                 auth=Auth(
-                    run_check=True,
                     session=StateSession(state=session),
                 )
                 if session is not None
@@ -292,7 +291,6 @@ async def run_api_for_ide_mode(
             session=StateSession(
                 state=session,
             ),
-            run_check=False,
         )
         if session is not None
         else None,

intuned_runtime/__init__.py

@@ -1,5 +1,6 @@
 from runtime.helpers import extend_payload
 from runtime.helpers import extend_timeout
+from runtime.helpers.attempt_store import attempt_store
 from runtime.helpers.get_auth_session_parameters import get_auth_session_parameters
 
-__all__ = ["extend_payload", "extend_timeout", "get_auth_session_parameters"]
+__all__ = ["extend_payload", "extend_timeout", "get_auth_session_parameters", "attempt_store"]

{intuned_runtime-1.2.1.dist-info → intuned_runtime-1.2.2.dist-info}/METADATA

@@ -1,8 +1,9 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: intuned-runtime
-Version: 1.2.1
+Version: 1.2.2
 Summary: Runtime commands for Intuned platform Python scrapers
 License: Elastic-2.0
+License-File: LICENSE
 Keywords: runtime,intuned
 Author: Intuned Developers
 Author-email: engineering@intunedhq.com
@@ -15,6 +16,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Requires-Dist: aiofiles (>=24.1.0,<25.0.0)
 Requires-Dist: arguably (>=1.3.0,<2.0.0)
 Requires-Dist: browserforge[all]

{intuned_runtime-1.2.1.dist-info → intuned_runtime-1.2.2.dist-info}/RECORD

@@ -1,32 +1,34 @@
-intuned_cli/__init__.py,sha256=SNcUoVkqPTmIxjEsp05GZBKaAzgCt-rBkvkzndjpfPo,1124
-intuned_cli/commands/__init__.py,sha256=qndefhU1fGHYYu7f7wxcv03F27NPMIwsxUd3K-fmOD8,1236
-intuned_cli/commands/attempt_api_command.py,sha256=f4Oi8vjNzGLMgk_1XXO5KzOIAw9sq9N8A62d3QIZxso,1954
-intuned_cli/commands/attempt_authsession_check_command.py,sha256=RdKRltXIp51wnXamRB1Xr45TUBUs1AQCmAzW8tMQnxo,1334
+intuned_cli/__init__.py,sha256=COYx7VKi1Qdsusc58Tt3P1DpGUd6LFomjbf53GTCVKI,1466
+intuned_cli/commands/__init__.py,sha256=pN4XuY6EAAaiJRoRcpcyKw4Lei9ZfS7zmoaoNuwbz9I,1291
+intuned_cli/commands/attempt_api_command.py,sha256=3iN3L_vd--wMb96zgW-qNfyaqrGTUdcxKhYgifQu9XU,1949
+intuned_cli/commands/attempt_authsession_check_command.py,sha256=y7ODZLLSF5Pzm3LHOdRr7frGeuFn18NKHNtkg6_NH9Q,1329
 intuned_cli/commands/attempt_authsession_command.py,sha256=hir9y1XyW9VYioOWT6C1-dH43f3JcHhIlfEEijMg2Lc,277
-intuned_cli/commands/attempt_authsession_create_command.py,sha256=ynAKlNoHuaVPgx4gv2AJquIt7ZVIe8Nh1IIHwNiDeNA,1708
+intuned_cli/commands/attempt_authsession_create_command.py,sha256=tgnBqtboV2PRfGGr0kxTVo6yAOf2M2kD1DBlwNPW7pA,1703
 intuned_cli/commands/attempt_command.py,sha256=7NZC2dXA8GViCH3ZS2PMziR43hv2RmS6-jgW7l9bu40,253
 intuned_cli/commands/command.py,sha256=b0OlQIFhoCjCo5QIerfysccBKcU9iIsvqiU7oxshA2M,727
-intuned_cli/commands/deploy_command.py,sha256=0XG8lo5fGeorZMxMtEx12EaqLFMvmFYJ7Zu9sgJCmQ8,1540
+intuned_cli/commands/deploy_command.py,sha256=vZQy6AJUuyfdsM_7DIvMIBhmk7RUxBsUDLOB8uPNzVw,1549
 intuned_cli/commands/init_command.py,sha256=JIDr9vY_SRKCvVbXE_lCEeHQlmC48LMZz0cghMPAs90,492
-intuned_cli/commands/run_api_command.py,sha256=Y6FkSFI9wybuoW204YqOAiQ794__hIw12Eaf9px_ovE,3013
+intuned_cli/commands/run_api_command.py,sha256=Hwltx9eMEzksUoZUcpzTe2teGSD7tyu8f08uXSGBJOs,3008
 intuned_cli/commands/run_authsession_command.py,sha256=kM_TANy8M3yx8iBUsgSDO42byzccikLOd9KJfytfLmQ,269
-intuned_cli/commands/run_authsession_create_command.py,sha256=SiuIZ7LTMqT3gnd8DUt-VmP3cyuA0I_i27daI3IJ2tw,2089
-intuned_cli/commands/run_authsession_update_command.py,sha256=Op7epDgvEhjp17qs2dIXerTckHz7wlOsZsBDp8UJLjI,2145
-intuned_cli/commands/run_authsession_validate_command.py,sha256=5eET02paijq1f1rzsheINvAqVYj1JGUA_iGD5Py3LVM,2008
+intuned_cli/commands/run_authsession_create_command.py,sha256=kVxQR8j8yARGRKY2FyUhW_E3hRA3KmGzP2zG3gVLRBc,2084
+intuned_cli/commands/run_authsession_update_command.py,sha256=vK3rJcHkqL-mVfL5IGrLb_4tWRQ6xpufwXymBr0kTCw,2140
+intuned_cli/commands/run_authsession_validate_command.py,sha256=lri5CyV-1cHOgKa83BTHHLeMT6_xlpbkS5vH1pUjHNA,2003
 intuned_cli/commands/run_command.py,sha256=JN1yCewcyb4zFquMcv0wEZ_aRmhJZIBMheY8L9yBeDE,245
+intuned_cli/commands/save_command.py,sha256=iW8i42B_NBrhmuGveV0XXkhaKW0Xx63ofOgMfH1FxTM,1543
 intuned_cli/controller/__test__/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-intuned_cli/controller/__test__/test_api.py,sha256=8Hk3V52qKhWIqjirKFwN68Tv5er70gTxemjI35HyZiU,19295
+intuned_cli/controller/__test__/test_api.py,sha256=yuKxW6yWlAs1Pj6QBJ4DyryR8A8suG-A7xtBiUcMJ-E,19242
 intuned_cli/controller/__test__/test_authsession.py,sha256=y7O3dkDPHZTf2W7WRLEhysYYRati0hg93-J8BvGjRUM,35110
-intuned_cli/controller/api.py,sha256=1wHgatavA5zr6kYMf1BJ3lC5JR1IvlVBiab_ahOSVnY,7172
-intuned_cli/controller/authsession.py,sha256=z8C97T-beEORNemLFx2x_JsHr2gXRakwKGV3Rgivnu8,14309
-intuned_cli/controller/deploy.py,sha256=5cQgtpT2nDQ5XSUnIoeBR9xdyVoLWcY-2btuUxUD7Es,12343
+intuned_cli/controller/api.py,sha256=cIxfuKsMibn7XRbTQCELy1QVtWH-UMWO8KGyOCvKx2A,7135
+intuned_cli/controller/authsession.py,sha256=KGpRYXO9W6DFbuM1ofdH5aba0Zs4A_A3nhbNj-KaCE8,14272
+intuned_cli/controller/deploy.py,sha256=krbVm0-c1XDIsiOPIf0lesFmFuVF_VfGYNWCOQ70Mmo,5452
+intuned_cli/controller/save.py,sha256=_ujsh4C9cgFGW4GTkj43JDSpIq0TuBLhZRJsH_1wgX8,9025
 intuned_cli/types.py,sha256=Lsykui4mbq5T1_1L7Gg5OZpJvCOmaz9EiQcZVfUEoLc,766
 intuned_cli/utils/api_helpers.py,sha256=57gvXVYM_9hMGIkMqSEN_jzs3mYWlXzinPjYXzzqgg8,1122
 intuned_cli/utils/auth_session_helpers.py,sha256=acKAPUjOfdybstLur4Lk3huaLFP2Ipl4AjPSqQPQLzY,1899
-intuned_cli/utils/backend.py,sha256=9HmaIvyKRkybUL-lEiJYHbIclinEZEBNBFLUG0hNC0A,150
+intuned_cli/utils/backend.py,sha256=CtGoX6eWO6dUnND_FUurIgBXgp9Z4y89k8j3tYK9G-k,1030
 intuned_cli/utils/confirmation.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 intuned_cli/utils/console.py,sha256=jgin2xB-0kpANPMoiDBOnWhrc8Ey5zT4IHRxQa8f5CQ,93
-intuned_cli/utils/error.py,sha256=DmNlXBb7zfpHDRya00ouEcRgUEZuGaTqIy1SIjDWVfo,842
+intuned_cli/utils/error.py,sha256=KzqWG0e8PLerCY-AL534zygCLbIBD6uReo2WEuej-EA,990
 intuned_cli/utils/exclusions.py,sha256=Qe7NkWA3lsX73dOC9WprdD0KyYc_vuiwkrUCwD105gk,796
 intuned_cli/utils/get_auth_parameters.py,sha256=HmMSjBE8bPulkUdX319Ipr383Ko2Gtz3y8_WT9CK3Kw,798
 intuned_cli/utils/import_function.py,sha256=UmE2yw5std1ENMFdBU-TUBuQ01qsv7Qr5ElnAhqE6Yc,453
@@ -41,11 +43,11 @@ intuned_internal_cli/commands/browser/save_state.py,sha256=_gfspwLkk6j1zBzqY1Qr7
 intuned_internal_cli/commands/init.py,sha256=8rWBenWZfwNtLxOBqhEMbOATyQNEnmDUmrFJ1xBGyxI,4384
 intuned_internal_cli/commands/project/__init__.py,sha256=t97wvhSenerYRdbSeCKXqHASA6EWA3lc1hnRhF9baOE,734
 intuned_internal_cli/commands/project/auth_session/__init__.py,sha256=gt7mlaW6xmqAc_4-pfF_FiecsR51C6fqCaq_NFbcbwA,300
-intuned_internal_cli/commands/project/auth_session/check.py,sha256=AFILp7m34nAO_RD3IfRpuJm5Zh0wnCRtBXqIrerdbwo,4565
+intuned_internal_cli/commands/project/auth_session/check.py,sha256=UDG2iF6ULmYH6_WdUTVRTAZq3fBHuVSzhWB1tGkkGb0,4524
 intuned_internal_cli/commands/project/auth_session/create.py,sha256=r-eYu3uLUo2mzF836CbVgu4oBzcIIDekzzFwwekxmg0,3374
 intuned_internal_cli/commands/project/auth_session/load.py,sha256=Y0V6bFZQeHq_lTtR-rQvM9SSbExWhVVMFu-Uykl-9k4,1199
 intuned_internal_cli/commands/project/project.py,sha256=_MSh6Xor2Cbh-ItifwgOPq_BP8UDuKB7S6w796FULKQ,137
-intuned_internal_cli/commands/project/run.py,sha256=FDYYkU24aURYbljyYLFo8wLF-nvb86EVr9gMEjAfeE0,12274
+intuned_internal_cli/commands/project/run.py,sha256=7lqrFrTOaW0W5UF6WTVwwDauAoGbABNv6RVQ6_DEIfQ,12209
 intuned_internal_cli/commands/project/run_interface.py,sha256=4RyR8WZriIF7Va4z1wt-q6zZDQOI31n62Ho2dyimzUY,8717
 intuned_internal_cli/commands/project/type_check.py,sha256=QFnXL93Y-z250EQRln5sRqwkVqjGJ-eP_p1jGrXB7NA,3522
 intuned_internal_cli/commands/project/upgrade.py,sha256=XmkZLflM4O-mwvhwcswlZpazRotwi3xesLgE0Zz8fTI,3061
@@ -57,42 +59,47 @@ intuned_internal_cli/utils/code_tree.py,sha256=1wfxZoQ5kRCfqs2SEPAicbAIPTiD6P1Lx
 intuned_internal_cli/utils/run_apis.py,sha256=Zee4zkgt9R8XY1XCGzj2Nc4zJ3jlRz1xnO493wotuWw,4690
 intuned_internal_cli/utils/setup_ide_functions_token.py,sha256=72-hf5HOPO9hj_eo3MTSVEPIwtkaIma_NRepsw_FHQM,304
 intuned_internal_cli/utils/unix_socket.py,sha256=UISmkJMHrir5iBLUm6vxC3uzTGIFyOk_wa0C9LUw4Cc,1889
-intuned_runtime/__init__.py,sha256=XBrEiE9yNC8Lgn8NgIkqNXbI6e4ap237E83Zj_nlhCQ,249
+intuned_runtime/__init__.py,sha256=1BPzEc-qC2WAYiCWDOJChpgnFyO3ZNYRKHEZqdHUGwM,322
 runtime/__init__.py,sha256=87gDXuxUv_kGzQfuB1mh6DF-dDysJN8r684c7jGnHxc,144
 runtime/backend_functions/__init__.py,sha256=j2EaK4FK8bmdFtqc5FxtFwx1KhIn_7qKPChrrAhJI3s,119
-runtime/backend_functions/_call_backend_function.py,sha256=h6UUsQ2pLUTbEQNYM1TLdesc5yWWaUrPUaTbCtCIxlU,2769
+runtime/backend_functions/_call_backend_function.py,sha256=NpbQmEieuRan0fgbJQ0L_skU6MgjMmR99bWsQQYuFtI,3402
 runtime/backend_functions/get_auth_session_parameters.py,sha256=pOvB7XiWpphEuBpazdKALw9EWgBU1PeY3gkzBfVLpkc,869
 runtime/browser/__init__.py,sha256=EPWfa4ZmdR8GJqh2qcsx1ZvHmCYiUYrQ-zeHYVapH9s,285
-runtime/browser/helpers.py,sha256=b1Xp005adbl7ZeJrSEYgH2OPzrZEHxTPgGTsnS3OqS0,554
-runtime/browser/launch_browser.py,sha256=f71hLUDbIBEv5hhQVOQFTXleUm_X1kmEpp0kTx0SqZo,966
+runtime/browser/helpers.py,sha256=CwgiBToawPgwAY9nIGkGHW544N7Db_OgKmS-SHkN2pU,1255
+runtime/browser/launch_browser.py,sha256=ym97J4RffOGUwhi9iNjAR5Ek2f8pKiAtAcDQFSqMJw0,1899
 runtime/browser/launch_camoufox.py,sha256=TBOAwwipNGlbtMdFYnGkVM0ppLU44vWNkMGZA5uPZCE,1787
-runtime/browser/launch_chromium.py,sha256=QY9Mxw8FImT_W23cT2zjTAeLvApzVHA_YSu0b77nfeA,7813
+runtime/browser/launch_chromium.py,sha256=YvZig28k1e_txdNyheOAnh1P8r2o5sVQVcKIEXkjQRA,8038
 runtime/browser/storage_state.py,sha256=fwLg8sP-H-vgt_6AJKNl03CpgyMVCQWWcN2cqswTQMs,3603
+runtime/constants.py,sha256=YMYQgCWZdJXUpxz_IN2TvZO5rFye9k_Lk9CS8m-shLg,34
 runtime/context/__init__.py,sha256=hg8ejm4bJy4tNkwmZ9lKgYJx6bU7OgOdBS684Uv5XGg,73
-runtime/context/context.py,sha256=pl_0x77_d5CiAznz1qGSk6o9cW-msNvlCt-2eFoMKlA,1739
-runtime/env.py,sha256=OXxzLpM56AJVlX0gmG7Ph82xAfqZboW3kv2232lzXb4,306
+runtime/context/context.py,sha256=uZxt5cF7Q66Ng6F5J83LMCn9UG7swWOM3f-Wz4MYq7w,1805
+runtime/env.py,sha256=DYEU5a-XtPbImdmLEDqdWn40JGSOvG-V7mIXrt0sM9A,660
 runtime/errors/__init__.py,sha256=oqiBSvT_yFLQ3hG0AbCUA3WYFaxkTDVkDMSy59xvBCo,688
 runtime/errors/auth_session_errors.py,sha256=6b4XTI8UCDHDPX4jEA8_HyrNUp4VZ1TrEA8DRh6Z3rM,228
-runtime/errors/run_api_errors.py,sha256=Bjs_gsgdIUbVLN67E_s-804qB2lyzIljDEaAcymgwK0,3906
+runtime/errors/run_api_errors.py,sha256=D3zHsJnsjSOy0JUac_CDgHWOYEsOG2CUo7boVcteuCI,4070
 runtime/errors/trace_errors.py,sha256=Lzfo0sH3zGaWz1kn5DHcAXQMn3aR2y2bnauj6xP1LYE,110
-runtime/helpers/__init__.py,sha256=jozYPKHgZJ7Na5U1Wjt83egzjPATMZ_OMInEI6swSbY,234
+runtime/helpers/__init__.py,sha256=rkywrr_76AMaIsAAjsVmmfLh8BjB5_E6AeJY5sWFm6U,292
+runtime/helpers/attempt_store.py,sha256=YPwGHkdJNbC8rSiQ1vOReKNOQrjPT1wpO9ZDcQf530I,339
 runtime/helpers/extend_payload.py,sha256=towZF08WTpTTDBL4AV1bUU3XpKAQHEB66kGUfTICDe0,246
 runtime/helpers/extend_timeout.py,sha256=KjfSLEUrqoz7v00rhnPAKq2OmUzEzcv-eQ3M8c2U46s,348
 runtime/helpers/get_auth_session_parameters.py,sha256=7bopGhJ7vjKAn_UxnHSAah-k2rVOPbq0zi9FQOOCFds,472
 runtime/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 runtime/run/__init__.py,sha256=zxMYVb7hn147YTrhMLsrcX6-KTd71HLrYHstJOWeWXQ,52
 runtime/run/intuned_settings.py,sha256=vy2-ktEzUfUp5Z90dp3l7jPKHNjgB-8GSMDgAY-rYaU,1074
-runtime/run/playwright_constructs.py,sha256=EIfnRlAi1k1wRiTT2OpPFVmrWwEeeiryuORibaLD1Xw,573
+runtime/run/playwright_context.py,sha256=pSTWIg_XpnuHtV5noApabAlZGZJYqtX82YW77n1FGFU,4675
+runtime/run/playwright_tracing.py,sha256=oqr9b6xjqdYPCEOUvIVE1M7ord7LAvXAUQAz38Jwjw8,708
 runtime/run/pydantic_encoder.py,sha256=wJCljwwINSICvCJ0i2izp2RLkQ15nYglUQCyyjM40Jk,332
-runtime/run/run_api.py,sha256=JiO-UBpCBP7H4TZbKXmYmGZmYrkqidqEg7epaxj_3jA,10105
+runtime/run/run_api.py,sha256=AFhZerLTyHGznGCJl0fLbiTqhZ_WRmphaJK-RqrupTQ,8981
+runtime/run/setup_context_hook.py,sha256=KTX4mRmeUEmYS9zrmobR1V09GakOk6uz81Uo_xXTJZk,1156
 runtime/run/traces.py,sha256=fKzh11LqV47ujgq_9I2tdp-dgld566wffWaHwU_4gis,1123
+runtime/run/types.py,sha256=RR4RGiYVBIK6f2qXvzfLhQMZ8kmrziu265k5eqoIh98,346
 runtime/types/__init__.py,sha256=IJkDfqsau8F8R_j8TO6j-JwW4ElQr6aU6LNaWRehg5U,401
 runtime/types/payload.py,sha256=sty8HgDEn3nJbZrwEOMCXyuG7_ICGDwlBIIWSON5ABY,124
-runtime/types/run_types.py,sha256=-j-XKXfRkzlyoW-Doe0og2jbqMMQWjOTIUqRFEc8lHA,4582
-runtime_helpers/__init__.py,sha256=XBrEiE9yNC8Lgn8NgIkqNXbI6e4ap237E83Zj_nlhCQ,249
+runtime/types/run_types.py,sha256=GcYLkL2BHxOjT4O3KvBP6xjBKsmJbjltMt_5bCVnfCI,4554
+runtime_helpers/__init__.py,sha256=1BPzEc-qC2WAYiCWDOJChpgnFyO3ZNYRKHEZqdHUGwM,322
 runtime_helpers/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-intuned_runtime-1.2.1.dist-info/LICENSE,sha256=9LIjQdgyU_ptzNIfItNCR7VmEHqYnrY1f1XwOreKFI0,3714
-intuned_runtime-1.2.1.dist-info/METADATA,sha256=HF-_MS8G5PkAx2xWevN0giX-4Vrs9VpqWICGmdSAsYM,5299
-intuned_runtime-1.2.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-intuned_runtime-1.2.1.dist-info/entry_points.txt,sha256=ToMS2cqDeRmF1FGkflwoeD-Xz6jJV5p1zIbw9G7IxMg,85
-intuned_runtime-1.2.1.dist-info/RECORD,,
+intuned_runtime-1.2.2.dist-info/METADATA,sha256=IyR56IaaeamglTzQ0OOENJP78Kb9GNzSegBhcactZM0,5372
+intuned_runtime-1.2.2.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+intuned_runtime-1.2.2.dist-info/entry_points.txt,sha256=ToMS2cqDeRmF1FGkflwoeD-Xz6jJV5p1zIbw9G7IxMg,85
+intuned_runtime-1.2.2.dist-info/licenses/LICENSE,sha256=9LIjQdgyU_ptzNIfItNCR7VmEHqYnrY1f1XwOreKFI0,3714
+intuned_runtime-1.2.2.dist-info/RECORD,,

{intuned_runtime-1.2.1.dist-info → intuned_runtime-1.2.2.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.3
+Generator: poetry-core 2.2.1
 Root-Is-Purelib: true
 Tag: py3-none-any

runtime/backend_functions/_call_backend_function.py

@@ -5,8 +5,11 @@ from typing import Literal
 from httpx import AsyncClient
 from pydantic import BaseModel
 
+from runtime.constants import api_key_header_name
 from runtime.context.context import IntunedContext
+from runtime.env import get_api_key
 from runtime.env import get_functions_domain
+from runtime.env import get_is_running_in_cli
 from runtime.env import get_project_id
 from runtime.env import get_workspace_id
 
@@ -24,19 +27,27 @@ async def call_backend_function[T: BaseModel](
     functions_domain, workspace_id, project_id = get_functions_domain(), get_workspace_id(), get_project_id()
 
     if functions_domain is None or workspace_id is None or project_id is None:
-        # todo
+        if get_is_running_in_cli():
+            raise Exception(
+                "API credentials not set - make sure to save your project to Intuned to set up the correct API credentials."
+            )
+
         raise Exception("No workspace ID or project ID found.")
 
     context = IntunedContext.current()
 
     async with AsyncClient() as client:
+        api_key = get_api_key()
+        if api_key is not None:
+            client.headers[api_key_header_name] = api_key
+
         if context.functions_token:
             client.headers["Authorization"] = f"Bearer {context.functions_token}"
         if params:
             client.headers["Content-Type"] = "application/json"
         path = f"{functions_domain}/api/{workspace_id}/functions/{project_id}/{name}"
         body = params.model_dump() if params else None
-        print("Calling backend function", method, path, context.functions_token, json.dumps(body, indent=2))
+
         response = await client.request(
             method,
             path,
@@ -56,6 +67,11 @@ async def call_backend_function[T: BaseModel](
             )
         if 200 <= response.status_code < 300:
             return validation_model.model_validate(response_json)
+        if response.status_code == 401 and get_is_running_in_cli():
+            raise CallBackendException(
+                response.status_code,
+                "Unauthorized backend function call - make sure to save your project to Intuned to set up the correct API credentials",
+            )
         raise CallBackendException(
             response.status_code,
             f"Calling backend function errored with status {response.status_code}: {response_json}",

runtime/browser/helpers.py

@@ -1,7 +1,10 @@
+import asyncio
 import os
 from typing import Optional
 from typing import TYPE_CHECKING
 
+import tenacity as tx
+from httpx import AsyncClient
 from playwright.async_api import ProxySettings
 
 if TYPE_CHECKING:
@@ -19,3 +22,22 @@ def get_proxy_env() -> Optional["ProxySettings"]:
         "username": username,
         "password": password,
     }
+
+
+def get_local_cdp_address(port: int) -> str:
+    return f"http://localhost:{port}"
+
+
+async def wait_on_cdp_address(cdp_address: str) -> None:
+    cdp_address_without_protocol = (
+        cdp_address.replace("http://", "").replace("https://", "").replace("localhost", "127.0.0.1")
+    )
+
+    @tx.retry(stop=tx.stop_after_delay(5), wait=tx.wait_fixed(0.1))
+    async def _check_cdp() -> None:
+        async with AsyncClient(timeout=1) as client:
+            response = await client.get(f"http://{cdp_address_without_protocol}/json/version")
+            response.raise_for_status()
+
+    await asyncio.sleep(0.1)
+    await _check_cdp()