intuned-runtime 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

intuned_cli/utils/import_function.py

@@ -0,0 +1,15 @@
+from anyio import Path
+
+from runtime.run.run_api import import_function_from_api_dir
+from runtime.run.run_api import ImportFunction
+
+
+async def get_cli_import_function() -> ImportFunction:
+    """
+    Import a function from the API directory for CLI usage.
+    """
+    cwd = await Path().resolve()
+
+    return lambda file_path, name=None: import_function_from_api_dir(
+        file_path=file_path, automation_function_name=name, base_dir=str(cwd)
+    )

intuned_cli/utils/timeout.py

@@ -0,0 +1,25 @@
+import asyncio
+from contextlib import asynccontextmanager
+
+from runtime.context.context import IntunedContext
+from runtime.errors.run_api_errors import AutomationError
+
+
+@asynccontextmanager
+async def extendable_timeout(timeout: float):
+    try:
+        async with asyncio.timeout(timeout) as tm:
+            existing_extend_timeout = IntunedContext.current().extend_timeout
+
+            async def extend_timeout():
+                tm.reschedule(asyncio.timeout(timeout).when())
+                if existing_extend_timeout:
+                    await existing_extend_timeout()
+
+            IntunedContext.current().extend_timeout = extend_timeout
+            try:
+                yield
+            finally:
+                IntunedContext.current().extend_timeout = existing_extend_timeout
+    except asyncio.TimeoutError as e:
+        raise AutomationError(Exception("Timed out")) from e

{cli → intuned_internal_cli}/__init__.py

@@ -18,7 +18,7 @@ def run():
         load_dotenv(dotenv, override=True)
     try:
         with IntunedContext():
-            arguably.run()
+            arguably.run(name="intuned-internal")
     except ValueError as e:
         print(bold(red(str(e))))
         sys.exit(1)

{cli → intuned_internal_cli}/commands/__init__.py

@@ -9,6 +9,7 @@ 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
 
 __all__ = [
     "project__run",
@@ -22,4 +23,5 @@ __all__ = [
     "project__auth_session__check",
     "project__type_check",
     "browser__save_state",
+    "__root__",
 ]

{cli → intuned_internal_cli}/commands/ai_source/deploy.py

@@ -6,7 +6,7 @@ from more_termcolor import cyan  # type: ignore
 from more_termcolor import green  # type: ignore
 from more_termcolor import red  # type: ignore
 
-from cli.utils.code_tree import convert_project_to_code_tree
+from intuned_internal_cli.utils.code_tree import convert_project_to_code_tree
 
 from ...utils.ai_source_project import AiSourceInfo
 from ...utils.ai_source_project import deploy_ai_source

{cli → intuned_internal_cli}/commands/project/type_check.py

@@ -9,6 +9,7 @@ current_dir = os.path.dirname(os.path.abspath(__file__))
 PYRIGHT_CONFIG_PATH = os.path.join(current_dir, "..", "..", "pyright_type_check.json")
 PYRIGHT_CONFIG_PATH = os.path.abspath(PYRIGHT_CONFIG_PATH)
 
+
 @arguably.command  # type: ignore
 async def project__type_check():
     """
@@ -37,44 +38,50 @@ async def project__type_check():
     project_dir = os.getcwd()
     print("📦 Checking Types...")
 
-    try:  
-      pyright_issues = []    
-      pyright_result = subprocess.run(
-          ["pyright", "--outputjson", project_dir, "--project", PYRIGHT_CONFIG_PATH],
-          capture_output=True,
-          text=True,
-          check=False
-      )
+    try:
+        pyright_issues = []
+        pyright_result = subprocess.run(
+            ["pyright", "--outputjson", project_dir, "--project", PYRIGHT_CONFIG_PATH],
+            capture_output=True,
+            text=True,
+            check=False,
+        )
+
+        if pyright_result.stdout:
+            pyright_data = json.loads(pyright_result.stdout)
+            for diagnostic in pyright_data.get("generalDiagnostics", []):
+                severity = diagnostic.get("severity", "").lower()
+                severity_emoji = "ℹ️" if severity == "information" else "⚠️" if severity == "warning" else "🔴"
+
+                pyright_issues.append(
+                    {
+                        "path": diagnostic.get("file", ""),
+                        "line": diagnostic.get("range", {}).get("start", {}).get("line", 0) + 1,
+                        "severity": diagnostic.get("severity", ""),
+                        "message": diagnostic.get("message", ""),
+                        "rule": "type-check",
+                    }
+                )
 
-      if pyright_result.stdout:
-          pyright_data = json.loads(pyright_result.stdout)
-          for diagnostic in pyright_data.get("generalDiagnostics", []):
-              severity = diagnostic.get("severity", "").lower()
-              severity_emoji = "ℹ️" if severity == "information" else "⚠️" if severity == "warning" else "🔴"
-              
-              pyright_issues.append(
-                      {"path": diagnostic.get("file", ""), "line": diagnostic.get("range", {}).get("start", {}).get("line", 0) + 1, "severity": diagnostic.get("severity", ""), "message": diagnostic.get("message", ""), "rule": "type-check"}
-              )
-              
-              file_path = diagnostic.get("file", "")
-              if "api/" in file_path:
-                  file_path = file_path[file_path.index("api/"):]
-              line_num = diagnostic.get("range", {}).get("start", {}).get("line", 0) + 1
-              message = diagnostic.get("message", "")
-              print(f"{severity_emoji} {file_path}:{line_num} - {message}")
-              
-              if severity.lower() == "error":
-                  print("\n🔴 Type check failed")
-                  sys.exit(1)
+                file_path = diagnostic.get("file", "")
+                if "api/" in file_path:
+                    file_path = file_path[file_path.index("api/") :]
+                line_num = diagnostic.get("range", {}).get("start", {}).get("line", 0) + 1
+                message = diagnostic.get("message", "")
+                print(f"{severity_emoji} {file_path}:{line_num} - {message}")
 
-          if pyright_issues:
+                if severity.lower() == "error":
+                    print("\n🔴 Type check failed")
+                    sys.exit(1)
+
+            if pyright_issues:
                 has_warnings = any(issue["severity"].lower() == "warning" for issue in pyright_issues)
                 if has_warnings:
                     print("\n⚠️ Type check passed with warnings")
                     sys.exit(0)
-          
-          print("✨ Python type checking passed without errors.")
-          sys.exit(0)
+
+            print("✨ Python type checking passed without errors.")
+            sys.exit(0)
     except json.JSONDecodeError:
         print("🔴 Failed to parse pyright output as JSON")
         sys.exit(1)

intuned_internal_cli/commands/root.py

@@ -0,0 +1,15 @@
+import arguably
+
+
+@arguably.command  # type: ignore
+async def __root__():
+    """Internal Intuned CLI.
+
+    This command is intended for internal use by Intuned and is not intended for general users.
+    Breaking changes and experimental features may be present.
+
+    If you are not an Intuned developer, please use the main Intuned CLI instead.
+    """
+    if arguably.is_target():
+        print("-h/--help")
+        exit(1)

{intuned_runtime-1.0.0.dist-info → intuned_runtime-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: intuned-runtime
-Version: 1.0.0
+Version: 1.1.0
 Summary: Runtime commands for Intuned platform Python scrapers
 License: Elastic-2.0
 Keywords: runtime,intuned
@@ -24,7 +24,9 @@ Requires-Dist: pathspec (>=0.12.1,<0.13.0)
 Requires-Dist: pydantic (>=2.10.6,<3.0.0)
 Requires-Dist: pyright (>=1.1.387,<2.0.0)
 Requires-Dist: python-dotenv (==1.0.1)
+Requires-Dist: pytimeparse (>=1.1.8,<2.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
+Requires-Dist: rich (>=14.1.0,<15.0.0)
 Requires-Dist: ruff (>=0.7.2,<0.8.0)
 Requires-Dist: semver (>=3.0.4,<4.0.0)
 Requires-Dist: tenacity (>=8.5.0,<9.0.0)

intuned_runtime-1.1.0.dist-info/RECORD

@@ -0,0 +1,96 @@
+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/commands/attempt_authsession_command.py,sha256=hir9y1XyW9VYioOWT6C1-dH43f3JcHhIlfEEijMg2Lc,277
+intuned_cli/commands/attempt_authsession_create_command.py,sha256=ynAKlNoHuaVPgx4gv2AJquIt7ZVIe8Nh1IIHwNiDeNA,1708
+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/init_command.py,sha256=MV5WR5o4KH1J_Zai6K5S84NG41KZbsJ-I8NPGXurt2o,671
+intuned_cli/commands/run_api_command.py,sha256=Y6FkSFI9wybuoW204YqOAiQ794__hIw12Eaf9px_ovE,3013
+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_command.py,sha256=JN1yCewcyb4zFquMcv0wEZ_aRmhJZIBMheY8L9yBeDE,245
+intuned_cli/constants/__init__.py,sha256=AIfIQYQ55zSdGKL2p15wv4H9ANeRjYEkeUiObu6JFt8,37
+intuned_cli/constants/readme.py,sha256=3OUkZNmz9kss7hA3V2j5h49dv7ADy4LON5-iJkrJUA4,5240
+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_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/init.py,sha256=TfkPkq9rbYam2-fchwrpMIt9GJIkY-TQnWpwvPomRZE,3774
+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=NFMAqHFslxad9KYirUz3b0BsRz1aQqOGhZRUJdEiO8k,106
+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/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
+intuned_cli/utils/timeout.py,sha256=DkoeoU9XvKKKSQ06CpwqcNvxWqLPAOVuAMw6kSr4Tuo,886
+intuned_internal_cli/__init__.py,sha256=iz-mTdcPhNCknHZzdO5-9PHFSapbXiLCwg_cTaYfww4,1337
+intuned_internal_cli/commands/__init__.py,sha256=gZ-r8UIXvRakeCfvmOZCNWr8CJHBcYzRzbMjyKYPXV0,1003
+intuned_internal_cli/commands/ai_source/__init__.py,sha256=lg7owgcK8owNn2a4VBUP9RKxzFsLruhtnnQV0F_z6pc,149
+intuned_internal_cli/commands/ai_source/ai_source.py,sha256=2woQtCmhxKvLfEz832eUoCT9gMsuSvEE6rMnHSYXC7w,138
+intuned_internal_cli/commands/ai_source/deploy.py,sha256=NpomuP_2mo5SLKe7BZoY5H0Pp36aDtkYU_CyihqOvdI,2457
+intuned_internal_cli/commands/browser/__init__.py,sha256=AuVbvh7aSBTFKYvewXZyPoIvfBTe2uHiPcnaAkzapas,95
+intuned_internal_cli/commands/browser/save_state.py,sha256=eHKfvBfeFR_U9VQbsjOnIZjWepyDjNn9NL3naDYWP2s,841
+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/create.py,sha256=r-eYu3uLUo2mzF836CbVgu4oBzcIIDekzzFwwekxmg0,3374
+intuned_internal_cli/commands/project/auth_session/load.py,sha256=UUvg9Vyj15xiR44XlJzojLoFm5esv-o4E3qA3JqnBsE,1201
+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_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
+intuned_internal_cli/commands/publish_packages.py,sha256=sijkaG7_s0I1EWgLekGy1qm8Aqi_gYY8poXbMX0B6Yw,10505
+intuned_internal_cli/commands/root.py,sha256=o75rSb5WYa5JMD8HLN3eiNu4fM0mar3y15A63-Qm4s4,426
+intuned_internal_cli/logger.py,sha256=bZK3q-KUdGxk_qzDb6pn-n0LOhKJvi6a9p8oSwZtq3s,594
+intuned_internal_cli/utils/ai_source_project.py,sha256=xUCM6p3i1XN4bJbuQz8LCzeI4BwqAdSvCl_vwDAEi0k,831
+intuned_internal_cli/utils/code_tree.py,sha256=1wfxZoQ5kRCfqs2SEPAicbAIPTiD6P1LxSuwYu_eeaI,2790
+intuned_internal_cli/utils/run_apis.py,sha256=Zee4zkgt9R8XY1XCGzj2Nc4zJ3jlRz1xnO493wotuWw,4690
+intuned_internal_cli/utils/unix_socket.py,sha256=UISmkJMHrir5iBLUm6vxC3uzTGIFyOk_wa0C9LUw4Cc,1889
+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=zuaf4mwYHSm5RTedhMdU66jAMAzdPYPMmXJE_V1xoyk,2869
+runtime/backend_functions/get_auth_session_parameters.py,sha256=pOvB7XiWpphEuBpazdKALw9EWgBU1PeY3gkzBfVLpkc,869
+runtime/browser/__init__.py,sha256=CRBpMS319LBz2FLxDjjCvQdOUCgHLIrQFscr_Zi5Hm0,160
+runtime/browser/launch_chromium.py,sha256=OVaE8ezjxCYA1NUSDdd3FczrfNCpG_528Vmsv4S6QGU,7916
+runtime/browser/storage_state.py,sha256=fwLg8sP-H-vgt_6AJKNl03CpgyMVCQWWcN2cqswTQMs,3603
+runtime/context/__init__.py,sha256=hg8ejm4bJy4tNkwmZ9lKgYJx6bU7OgOdBS684Uv5XGg,73
+runtime/context/context.py,sha256=pl_0x77_d5CiAznz1qGSk6o9cW-msNvlCt-2eFoMKlA,1739
+runtime/env.py,sha256=h4BJI-XVSZKgtTxjkzj2HsyN3DlY5Ml9GZqeH4CKDE8,238
+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=LdmOEHoUk7wjWSk0HQYqslfJNNmxVgga_0bankzvX-s,3341
+runtime/errors/trace_errors.py,sha256=Lzfo0sH3zGaWz1kn5DHcAXQMn3aR2y2bnauj6xP1LYE,110
+runtime/helpers/__init__.py,sha256=jozYPKHgZJ7Na5U1Wjt83egzjPATMZ_OMInEI6swSbY,234
+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=UZvP502fk4Hk8xtI9jMlyJ7jHLJsWt2SZyZjhet7L1A,576
+runtime/run/pydantic_encoder.py,sha256=wJCljwwINSICvCJ0i2izp2RLkQ15nYglUQCyyjM40Jk,332
+runtime/run/run_api.py,sha256=iYekBi-mkBuBNvLIBXQ6RWvEDN7JhjSlX3i7a2td2P8,8952
+runtime/run/traces.py,sha256=fKzh11LqV47ujgq_9I2tdp-dgld566wffWaHwU_4gis,1123
+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_helpers/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+intuned_runtime-1.1.0.dist-info/LICENSE,sha256=9LIjQdgyU_ptzNIfItNCR7VmEHqYnrY1f1XwOreKFI0,3714
+intuned_runtime-1.1.0.dist-info/METADATA,sha256=SO-KZ2dpNlLDzqEQBZcOMC8_oie3r2wyw2D9ZSigXWQ,5217
+intuned_runtime-1.1.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+intuned_runtime-1.1.0.dist-info/entry_points.txt,sha256=ToMS2cqDeRmF1FGkflwoeD-Xz6jJV5p1zIbw9G7IxMg,85
+intuned_runtime-1.1.0.dist-info/RECORD,,

intuned_runtime-1.1.0.dist-info/entry_points.txt

@@ -0,0 +1,4 @@
+[console_scripts]
+intuned=intuned_cli:run
+intuned-internal=intuned_internal_cli:run
+

runtime/__init__.py

@@ -1,3 +1,4 @@
-from .browser import launch_chromium, dangerous_launch_chromium
+from .browser import dangerous_launch_chromium
+from .browser import launch_chromium
 
 __all__ = ["launch_chromium", "dangerous_launch_chromium"]

runtime/backend_functions/_call_backend_function.py

@@ -32,11 +32,6 @@ async def call_backend_function[T: BaseModel](
         # todo
         raise Exception("No run context found.")
 
-    auth_session_id = context.run_context.auth_session_id
-    if auth_session_id is None:
-        # todo
-        raise Exception("No auth session ID found.")
-
     async with AsyncClient() as client:
         if context.functions_token:
             client.headers["Authorization"] = f"Bearer {context.functions_token}"

runtime/browser/__init__.py

@@ -1,3 +1,4 @@
-from .launch_chromium import launch_chromium, dangerous_launch_chromium
+from .launch_chromium import dangerous_launch_chromium
+from .launch_chromium import launch_chromium
 
 __all__ = ["launch_chromium", "dangerous_launch_chromium"]

runtime/browser/launch_chromium.py

@@ -1,21 +1,21 @@
+import asyncio
 import json
+import logging
 import os
-import tempfile
 from contextlib import asynccontextmanager
-from os.path import join
 from typing import Any
 from typing import Optional
-from typing import Literal
-import logging
-import aiofiles
-from playwright.async_api import async_playwright
-from playwright.async_api import Browser
-from playwright.async_api import ProxySettings
+from typing import TYPE_CHECKING
+
+import anyio
+
+if TYPE_CHECKING:
+    from playwright.async_api import ProxySettings
 
 logger = logging.getLogger(__name__)
 
 
-def get_proxy_env() -> Optional[ProxySettings]:
+def get_proxy_env() -> Optional["ProxySettings"]:
     server = os.getenv("PROXY_SERVER")
     username = os.getenv("PROXY_USERNAME")
     password = os.getenv("PROXY_PASSWORD")
@@ -64,12 +64,12 @@ chromium_launch_args_to_ignore = [
 
 async def create_user_dir_with_preferences():
     # Create a temporary directory
-    playwright_temp_dir = tempfile.mkdtemp(prefix="pw-")
-    user_dir = join(playwright_temp_dir, "userdir")
-    default_dir = join(user_dir, "Default")
+    playwright_temp_dir = anyio.Path(await anyio.mkdtemp(prefix="pw-"))
+    user_dir = playwright_temp_dir / "userdir"
+    default_dir = user_dir / "Default"
 
     # Create the default directory recursively
-    os.makedirs(default_dir, exist_ok=True)
+    await default_dir.mkdir(parents=True, exist_ok=True)
 
     # Preferences data
     preferences = {
@@ -79,10 +79,10 @@ async def create_user_dir_with_preferences():
     }
 
     # Write preferences to file
-    async with aiofiles.open(join(default_dir, "Preferences"), mode="w") as f:
+    async with await (default_dir / "Preferences").open("w") as f:
         await f.write(json.dumps(preferences))
 
-    return os.path.abspath(user_dir)
+    return await user_dir.absolute(), await playwright_temp_dir.absolute()
 
 
 extra_args = [
@@ -95,7 +95,9 @@ extra_args = [
     "--disable-blink-features=AutomationControlled",
 ]
 
-default_user_agent = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36"
+default_user_agent = (
+    "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36"
+)
 
 
 @asynccontextmanager
@@ -105,13 +107,17 @@ async def launch_chromium(
     cdp_address: str | None = None,
     **kwargs: Any,
 ):
+    from playwright.async_api import async_playwright
+    from playwright.async_api import Browser
+
     async with async_playwright() as playwright:
         if cdp_address is not None:
             browser: Browser = await playwright.chromium.connect_over_cdp(cdp_address)
             context = browser.contexts[0]
-            dir = None
+            user_preferences_dir = None
+            dir_to_clean = None
         else:
-            dir = await create_user_dir_with_preferences()
+            user_preferences_dir, dir_to_clean = await create_user_dir_with_preferences()
             if kwargs.get("proxy") is None:
                 proxy_env = get_proxy_env()
             else:
@@ -126,7 +132,7 @@ async def launch_chromium(
                 extra_args.append("--headless=new")
 
             context = await playwright.chromium.launch_persistent_context(
-                dir,
+                os.fspath(user_preferences_dir),
                 headless=headless,
                 viewport=viewport,
                 proxy=proxy_env,
@@ -138,9 +144,19 @@ async def launch_chromium(
         context.set_default_timeout(timeout * 1000)
 
         async def remove_dir_after_close(*_: Any, **__: Any) -> None:
-            if not dir:
+            if not dir_to_clean:
+                return
+            if not await dir_to_clean.exists():
                 return
-            os.system(f"rm -rf {os.path.realpath(dir)}")
+
+            process = await asyncio.create_subprocess_exec(
+                "rm",
+                "-rf",
+                os.fspath(dir_to_clean),  # Using subprocess to remove the directory
+                stdout=asyncio.subprocess.DEVNULL,
+                stderr=asyncio.subprocess.DEVNULL,
+            )
+            await process.wait()
 
         context.once("close", remove_dir_after_close)
         yield context, context.pages[0]
@@ -149,31 +165,25 @@ async def launch_chromium(
 async def dangerous_launch_chromium(
     headless: bool = True,
     timeout: int = 10,
-    web_socket: str | None = None,
     cdp_url: str | None = None,
-    connection_method: Literal["ws", "cdp"] | None = None,
+    port: int | None = None,
     **kwargs: Any,
 ):
+    from playwright.async_api import async_playwright
+    from playwright.async_api import Browser
+
     playwright = await async_playwright().start()
-    if web_socket is not None and connection_method == "ws":
-        logging.info(f"Connecting to ws: {web_socket}")
-        browser: Browser = await playwright.chromium.connect(web_socket)
-        browser.on("disconnected", lambda: logging.info("Browser Session disconnected"))
-        await browser.new_context(
-            viewport={"width": 1280, "height": 800}, user_agent=default_user_agent
-        )
-        context = browser.contexts[0]
-        dir = None
-    elif cdp_url is not None and connection_method == "cdp":
+    if cdp_url is not None:
         logging.info(f"Connecting to cdp: {cdp_url}")
         browser: Browser = await playwright.chromium.connect_over_cdp(cdp_url)
-        browser.on("disconnected", lambda: logging.info("Browser Session disconnected"))
+        browser.on("disconnected", lambda _: logging.info("Browser Session disconnected"))
         context = browser.contexts[0]
-        dir = None
-    elif web_socket is None and cdp_url is None and connection_method is None:
+        user_preferences_dir = None
+        dir_to_clean = None
+    else:
         logging.info("Launching local browser")
-        dir = await create_user_dir_with_preferences()
-        logging.info(f"Using user data directory: {dir}")
+        user_preferences_dir, dir_to_clean = await create_user_dir_with_preferences()
+        logging.info(f"Using user data directory: {user_preferences_dir}")
         if kwargs.get("proxy") is None:
             proxy_env = get_proxy_env()
         else:
@@ -187,8 +197,11 @@ async def dangerous_launch_chromium(
             chromium_launch_args_to_ignore.append("--headless")
             extra_args.append("--headless=new")
 
+        if port:
+            extra_args.append(f"--remote-debugging-port={port}")
+
         context = await playwright.chromium.launch_persistent_context(
-            dir,
+            os.fspath(user_preferences_dir),
             headless=headless,
             viewport=viewport,
             proxy=proxy_env,
@@ -197,16 +210,22 @@ async def dangerous_launch_chromium(
             args=extra_args,
             **kwargs,
         )
-    else:
-        raise ValueError(
-            "You have to provide method if you are launching a remote browser with ws or cdp"
-        )
-    context.set_default_timeout(timeout * 1000)
+        context.set_default_timeout(timeout * 1000)
 
-    async def remove_dir_after_close(*_: Any, **__: Any) -> None:
-        if not dir:
-            return
-        os.system(f"rm -rf {os.path.realpath(dir)}")
+        async def remove_dir_after_close(*_: Any, **__: Any) -> None:
+            if not dir_to_clean:
+                return
+            if not await dir_to_clean.exists():
+                return
 
-    context.once("close", remove_dir_after_close)
+            process = await asyncio.create_subprocess_exec(
+                "rm",
+                "-rf",
+                os.fspath(dir_to_clean),  # Using subprocess to remove the directory
+                stdout=asyncio.subprocess.DEVNULL,
+                stderr=asyncio.subprocess.DEVNULL,
+            )
+            await process.wait()
+
+        context.once("close", remove_dir_after_close)
     return playwright, context

runtime/browser/storage_state.py

@@ -1,23 +1,22 @@
 from typing import Any
+from typing import TYPE_CHECKING
 
-from playwright.async_api import BrowserContext
-from playwright.async_api import Error as PlaywrightError
+if TYPE_CHECKING:
+    from playwright.async_api import BrowserContext
+
+import logging
 
 from ..types.run_types import Cookie
 from ..types.run_types import Origin
 from ..types.run_types import SessionStorageOrigin
 from ..types.run_types import StorageState
 
-import logging
-
 logger = logging.getLogger(__name__)
 
 
-async def set_storage_state(context: BrowserContext, state: StorageState):
+async def set_storage_state(context: "BrowserContext", state: StorageState):
     # Add cookies if they exist
-    await context.add_cookies(
-        [cookie.model_dump(by_alias=True) for cookie in state.cookies]
-    )  # type: ignore
+    await context.add_cookies([cookie.model_dump(by_alias=True) for cookie in state.cookies])  # type: ignore
 
     # Apply localStorage for each origin
     page = await context.new_page()
@@ -67,7 +66,9 @@ async def set_storage_state(context: BrowserContext, state: StorageState):
     await page.close()
 
 
-async def get_storage_state(context: BrowserContext) -> StorageState:
+async def get_storage_state(context: "BrowserContext") -> StorageState:
+    from playwright.async_api import Error as PlaywrightError
+
     storage_state = await context.storage_state()
     cookies = storage_state.get("cookies") or []
     origins = storage_state.get("origins") or []
@@ -92,9 +93,7 @@ async def get_storage_state(context: BrowserContext) -> StorageState:
             session_storage.append(SessionStorageOrigin(**session_data))
         except PlaywrightError as e:
             if "SecurityError" in e.message:
-                logger.warning(
-                    f"Could not get storage state for page due '{page.url}' to security error."
-                )
+                logger.warning(f"Could not get storage state for page due '{page.url}' to security error.")
                 continue
             raise e
 

runtime/errors/run_api_errors.py

@@ -66,22 +66,30 @@ class AutomationNotCoroutineError(RunApiError):
 
 
 class AutomationError(RunApiError):
-    def __init__(self, exception: BaseException):
+    _error: BaseException
+
+    def __init__(self, error: BaseException):
         # Get all public attributes of the exception
-        error_props = {key: str(value) for key, value in exception.__dict__.items() if not key.startswith("_")}
+        error_props = {key: str(value) for key, value in error.__dict__.items() if not key.startswith("_")}
 
         super().__init__(
-            str(exception),
+            str(error),
             "AutomationError",
         )
 
+        self._error = error
+
         self.details = {
             "error_props": error_props,
-            "error_type": exception.__class__.__name__,
-            "name": exception.__class__.__name__,
-            "message": str(exception),
+            "error_type": error.__class__.__name__,
+            "name": error.__class__.__name__,
+            "message": str(error),
         }
 
+    @property
+    def error(self) -> BaseException:
+        return self._error
+
 
 class AutomationTimeoutError(RunApiError):
     def __init__(self):
@@ -111,10 +119,6 @@ class InternalInvalidInputError(RunApiError):
     def __init__(self, message: str, details: Any | None = None):
         super().__init__(
             f"Internal error: {message}. Please report this issue to the Intuned team.",
-            RunApiResponse(
-                status_code=500,
-                response={"error": "Internal error", "message": f"Internal error: {message}"},
-            ),
             "InternalInvalidInputError",
         )
         self.details = details