relationalai 0.13.1__py3-none-any.whl → 0.13.2__py3-none-any.whl

relationalai/tools/cli_helpers.py

@@ -1,6 +1,7 @@
 #pyright: reportPrivateImportUsage=false
 from __future__ import annotations
 import io
+import json
 import os
 import re
 import sys
@@ -9,10 +10,12 @@ import rich
 import click
 import functools
 import pytz
+from typing import NoReturn
 
 from relationalai.util.constants import TOP_LEVEL_PROFILE_NAME
+from relationalai.errors import RAIException
 from rich.table import Table
-from typing import Callable, Dict, Any
+from typing import Callable, Dict, Any, List, cast
 from ..clients import config
 from click.core import Context
 from rich.console import Console
@@ -23,9 +26,10 @@ from ..clients.config import ConfigFile
 from datetime import datetime, timedelta
 from click.formatting import HelpFormatter
 from ..clients.client import ResourcesBase
-from relationalai.tools.constants import GlobalProfile
+from relationalai.tools.constants import GlobalProfile, SHOW_FULL_TRACES
 from relationalai.tools.cli_controls import divider
 from relationalai.util.format import humanized_bytes, humanized_duration
+from InquirerPy.base.control import Choice
 
 #--------------------------------------------------
 # Helpers
@@ -152,6 +156,291 @@ def validate_engine_name(name:str) -> tuple[bool, str|None]:
         return False, ENGINE_NAME_ERROR
     return True, None
 
+#--------------------------------------------------
+# Engine types & selection helpers (Snowflake)
+#--------------------------------------------------
+
+def format_state_with_color(state: str) -> str:
+    """Format engine state with colors for display."""
+    if not state:
+        return ""
+    state_upper = state.upper()
+    if state_upper == "READY":
+        return f"[green]{state_upper}[/green]"
+    if state_upper == "SUSPENDED":
+        return f"[yellow]{state_upper}[/yellow]"
+    if state_upper in ("PENDING", "SYNCING", "PROCESSING"):
+        return f"[bold yellow]{state_upper}[/bold yellow]"
+    if state_upper in ("ABORTED", "QUARANTINED", "GONE"):
+        return f"[red]{state_upper}[/red]"
+    return state_upper
+
+def _get_engine_type_api():
+    # Local import to avoid importing snowflake modules for non-snowflake usage
+    from relationalai.clients.resources.snowflake import EngineType
+    return EngineType
+
+def _get_internal_engine_sizes() -> list[str]:
+    # Local import to avoid snowflake imports for non-snowflake usage
+    from relationalai.clients.resources.snowflake import INTERNAL_ENGINE_SIZES
+    return list(INTERNAL_ENGINE_SIZES)
+
+def get_engine_type_choices(cfg: config.Config, exclude_types: List[str] | None = None) -> List[Choice]:
+    """Get sorted list of engine type choices for interactive selection."""
+    EngineType = _get_engine_type_api()
+    if exclude_types is None:
+        exclude_types = []
+    exclude_types_upper = [et.upper() for et in exclude_types]
+    engine_types_list = [EngineType.LOGIC, EngineType.ML, EngineType.SOLVER]
+    engine_types_list = [et for et in engine_types_list if et.upper() not in exclude_types_upper]
+    engine_types_list.sort(key=lambda et: EngineType.get_label(et))
+    return [
+        Choice(value=et, name=f"{EngineType.get_label(et)}: {EngineType.get_description(et)}")
+        for et in engine_types_list
+    ]
+
+def select_engine_type_interactive(cfg: config.Config) -> str:
+    """Show interactive engine type selection and return the selected type."""
+    from . import cli_controls as controls
+    rich.print("")
+    engine_type_choices = get_engine_type_choices(cfg)
+    return controls.select("Engine type:", cast("list[str | Choice]", engine_type_choices), None, newline=True)
+
+def select_engine_interactive(
+    provider: ResourcesBase,
+    prompt: str = "Select an engine:",
+    engine_name: str | None = None,
+    engines: List[Dict[str, Any]] | None = None,
+) -> tuple[str, str | None] | None:
+    """Interactive engine picker returning (name, type)."""
+    from . import cli_controls as controls
+
+    engine_map: Dict[str, tuple[str, str | None]] = {}
+
+    def get_engines():
+        engine_list = engines if engines is not None else provider.list_engines()
+        engine_map.clear()
+        items: List[str] = []
+        EngineType = _get_engine_type_api()
+        for engine in engine_list:
+            eng_name = engine.get("name", "")
+            if engine_name and eng_name.upper() != engine_name.upper():
+                continue
+            eng_type = engine.get("type", "")
+            eng_size = engine.get("size", "")
+            if eng_type:
+                label = f"{EngineType.get_label(eng_type)} ({eng_type})" if EngineType.is_valid(eng_type) else f"{eng_type} ({eng_type})"
+                display = f"{eng_name}, {label}, {eng_size}"
+            else:
+                display = f"{eng_name}, {eng_size}" if eng_size else eng_name
+            engine_map[display] = (eng_name, eng_type or None)
+            items.append(display)
+        return items
+
+    # Auto-select when engine_name uniquely identifies an engine
+    if engine_name:
+        engine_list = engines if engines is not None else provider.list_engines()
+        matches = [e for e in engine_list if e.get("name", "").upper() == engine_name.upper()]
+        if len(matches) == 1:
+            e = matches[0]
+            return (e.get("name", ""), e.get("type"))
+        if len(matches) == 0:
+            return None
+
+    selected = controls.fuzzy_with_refetch(prompt, "engines", get_engines)
+    if not selected or isinstance(selected, Exception):
+        return None
+    return engine_map.get(selected)
+
+def select_engine_with_state_filter(
+    provider: ResourcesBase,
+    engine_name: str | None,
+    engine_type: str | None,
+    state_filter: str,
+    prompt_no_name: str,
+    prompt_with_name: str,
+    error_no_engines: str,
+    error_no_matching: str,
+) -> tuple[str, str | None] | None:
+    """Select an engine with optional state filtering + optional name filtering."""
+    EngineType = _get_engine_type_api()
+
+    if not engine_name:
+        filtered = provider.list_engines(state_filter)
+        if not filtered:
+            exit_with_error(error_no_engines)
+        return select_engine_interactive(provider, prompt_no_name, engines=filtered)
+
+    # If type provided and valid, return directly
+    if engine_type and EngineType.is_valid(engine_type):
+        return (engine_name, engine_type)
+
+    # Filter by name + state; selection handles (name,type)
+    filtered = provider.list_engines(state_filter, name=engine_name)
+    if not filtered:
+        exit_with_error(error_no_matching)
+    return select_engine_interactive(provider, prompt_with_name, engine_name=engine_name, engines=filtered)
+
+def ensure_engine_type_for_snowflake(
+    provider: ResourcesBase,
+    engine_name: str,
+    engine_type: str | None,
+    error_message: str,
+) -> str:
+    """Ensure engine_type is provided and valid; default to LOGIC if omitted."""
+    EngineType = _get_engine_type_api()
+    # If --type was omitted, default to LOGIC for backwards compatibility
+    if engine_type is None:
+        return EngineType.LOGIC
+    assert isinstance(engine_type, str)
+    if engine_type == "" or not EngineType.is_valid(engine_type):
+        cfg = get_config()
+        if engine_type == "":
+            rich.print(f"[yellow]Empty engine type provided for engine '{engine_name}'.")
+        else:
+            rich.print(f"[yellow]Invalid engine type '{engine_type}' for engine '{engine_name}'.")
+        return select_engine_type_interactive(cfg)
+    return engine_type
+
+def build_engine_operation_messages(
+    provider: ResourcesBase,
+    engine_name: str,
+    engine_type: str | None,
+    action: str,
+    action_past: str,
+) -> tuple[str, str]:
+    EngineType = _get_engine_type_api()
+    if engine_type:
+        label = EngineType.get_label(engine_type) if EngineType.is_valid(engine_type) else engine_type
+        return (f"{action} {label} engine '{engine_name}'", f"{label} Engine '{engine_name}' {action_past.lower()}")
+    return (f"{action} '{engine_name}' engine", f"Engine '{engine_name}' {action_past.lower()}")
+
+def prompt_and_validate_engine_name(name: str | None) -> str:
+    """Prompt for engine name if missing; validate using ENGINE_NAME_REGEX."""
+    from . import cli_controls as controls
+    if not name:
+        name = controls.prompt(
+            "Engine name:",
+            name,
+            validator=ENGINE_NAME_REGEX.match,
+            invalid_message=ENGINE_NAME_ERROR,
+            newline=True,
+        )
+    assert isinstance(name, str)
+    return name
+
+def validate_auto_suspend_mins(auto_suspend_mins: int | str | None) -> int | None:
+    if auto_suspend_mins is None:
+        return None
+    if isinstance(auto_suspend_mins, int):
+        return auto_suspend_mins
+    error_msg = f"[yellow]Error: auto_suspend_mins must be an integer instead of {type(auto_suspend_mins)}"
+    try:
+        return int(auto_suspend_mins)
+    except ValueError:
+        exit_with_error(error_msg)
+    return None
+
+def get_engine_type_for_creation(provider: ResourcesBase, cfg: config.Config, engine_type: str | None) -> str | None:
+    """Get engine type for engine creation; defaults to LOGIC when omitted."""
+    EngineType = _get_engine_type_api()
+    if engine_type is None:
+        return EngineType.LOGIC
+    if engine_type == "" or not EngineType.is_valid(engine_type):
+        if engine_type == "":
+            rich.print("[yellow]Empty engine type provided.")
+        else:
+            valid_types_display = ", ".join(EngineType.get_all_types())
+            rich.print(f"[yellow]Invalid engine type '{engine_type}'. Valid types: {valid_types_display}")
+        return select_engine_type_interactive(cfg)
+    return engine_type
+
+def get_and_validate_engine_size(
+    provider: ResourcesBase,
+    cfg: config.Config,
+    size: str | None,
+    engine_type: str | None = None,
+) -> str:
+    from . import cli_controls as controls
+    EngineType = _get_engine_type_api()
+    internal_sizes = set(_get_internal_engine_sizes())
+
+    cloud_provider = provider.get_cloud_provider()
+    valid_sizes = provider.get_engine_sizes(cloud_provider)
+
+    # Engine-type-aware filtering:
+    # Internal sizes (XS/S/M/L) are only valid for LOGIC engines (and only on some accounts).
+    # For ML/SOLVER, hide them to avoid presenting invalid options.
+    if engine_type and EngineType.is_valid(engine_type) and engine_type != EngineType.LOGIC:
+        valid_sizes = [s for s in valid_sizes if s not in internal_sizes]
+
+    # Ask if missing and not in config
+    if not size and not cfg.get("engine_size", None):
+        rich.print("")
+        # This refers to the cloud backing your Snowflake account (AWS/Azure), not the engine "platform".
+        size = controls.fuzzy(f"Engine size (Snowflake cloud: {cloud_provider.upper()}):", choices=valid_sizes)
+    elif size is None and cfg.get("engine_size", None):
+        size = cfg.get("engine_size", None)
+    if not isinstance(size, str) or size not in valid_sizes:
+        exit_with_error(f"\nInvalid engine size [yellow]{size}[/yellow] provided. Please check your config.\n\nValid sizes: [green]{valid_sizes}[/green]")
+    return size
+
+def create_engine_with_spinner(
+    provider: ResourcesBase,
+    engine_name: str,
+    engine_size: str,
+    engine_type: str | None,
+    auto_suspend_mins: int | None,
+) -> None:
+    """Create an engine with appropriate spinner messages and error handling."""
+    from .cli_controls import Spinner
+
+    EngineType = _get_engine_type_api()
+    # Build creation message with engine type when available
+    if engine_type:
+        creation_message = (
+            f"Creating {EngineType.get_label(engine_type)} engine '{engine_name}' with size {engine_size}... "
+            f"(this may take several minutes)"
+        )
+    else:
+        creation_message = (
+            f"Creating engine '{engine_name}' with size {engine_size}... "
+            f"(this may take several minutes)"
+        )
+
+    with Spinner(
+        creation_message,
+        f"Engine '{engine_name}' created!",
+        failed_message=None,  # We handle error display ourselves below
+    ):
+        try:
+            provider.create_engine(
+                engine_name,
+                type=engine_type,
+                size=engine_size,
+                auto_suspend_mins=auto_suspend_mins,
+            )
+        except Exception as e:
+            # Prefer richer error messages when available
+            error_msg = None
+
+            # EngineProvisioningFailed has a format_message() method that provides better error details
+            if hasattr(e, "format_message"):
+                try:
+                    error_msg = getattr(e, "format_message")()
+                except Exception:
+                    pass
+
+            # Try content/message fallbacks
+            if not error_msg and hasattr(e, "content"):
+                error_msg = getattr(e, "content", None)
+            if not error_msg and hasattr(e, "message"):
+                error_msg = str(getattr(e, "message", ""))
+            if not error_msg:
+                error_msg = str(e)
+
+            raise Exception(error_msg)
+
 #--------------------------------------------------
 # Tables
 #--------------------------------------------------
@@ -190,7 +479,7 @@ def format_value(value) -> str:
 def format_row(key: str, value) -> dict:
     result = {}
     result[key] = value
-    if "status" or "state" in key.lower():
+    if "status" in key.lower() or "state" in key.lower():
         result["style"] = get_color_by_state(value)
     if key == "query_size" and isinstance(value, int):
         result[key] = humanized_bytes(value)
@@ -216,6 +505,28 @@ def show_dictionary_table(dict, format_fn:Callable|None=None):
 
 
 class RichGroup(click.Group):
+    def invoke(self, ctx: Context) -> Any:
+        """Invoke the CLI command, suppressing tracebacks for handled RAIExceptions.
+
+        Any `RAIException` is expected to already know how to render itself nicely via
+        `pprint()`. When such an exception bubbles up to the top-level Click runner,
+        Click will otherwise print a full Python traceback, which is noisy for users.
+        """
+        try:
+            return super().invoke(ctx)
+        except RAIException as exc:
+            # Respect config-based full-trace setting when available.
+            try:
+                show_full_traces = get_config().get("show_full_traces", SHOW_FULL_TRACES)
+            except Exception:
+                show_full_traces = SHOW_FULL_TRACES
+
+            if show_full_traces:
+                raise
+
+            exc.pprint()
+            raise click.exceptions.Exit(1) from None
+
     def format_help(self, ctx: Context, formatter: HelpFormatter) -> None:
         is_latest, current_ver, latest_ver = is_latest_cli_version()
 
@@ -383,16 +694,109 @@ def show_engines(engines):
         table = Table(show_header=True, border_style="dim", header_style="bold", box=rich_box.SIMPLE_HEAD)
         table.add_column("#")
         table.add_column("Name")
+        # Show type column if present
+        table.add_column("Type")
         table.add_column("Size")
         table.add_column("State")
+        table.add_column("Created By")
+        table.add_column("Created On")
+        EngineType = _get_engine_type_api()
         for index, engine in enumerate(engines):
-            table.add_row(f"{index+1}", engine.get("name"), engine.get("size"), engine.get("state"))
+            engine_type = engine.get("type", "")
+            type_display = EngineType.get_label_with_value(engine_type) if engine_type and EngineType.is_valid(engine_type) else (engine_type or "")
+            created_on = format_value(engine.get("created_on"))
+            state_display = format_state_with_color(engine.get("state", ""))
+            table.add_row(
+                f"{index+1}",
+                engine.get("name"),
+                type_display,
+                engine.get("size"),
+                state_display,
+                engine.get("created_by", ""),
+                created_on,
+            )
         rich.print(table)
 
-def exit_with_error(message:str):
+
+def show_engine_details(engine: dict[str, Any]) -> None:
+    """Print a vertical table of engine details (one field per row)."""
+    from relationalai.clients.resources.snowflake import EngineType as _EngineType
+
+    table = Table(
+        show_header=True,
+        border_style="dim",
+        header_style="bold",
+        box=rich_box.SIMPLE_HEAD,
+    )
+    table.add_column("Field")
+    table.add_column("Value", overflow="fold")
+
+    engine_type_from_db = engine.get("type", "")
+    type_display = (
+        _EngineType.get_label_with_value(engine_type_from_db)
+        if engine_type_from_db and _EngineType.is_valid(engine_type_from_db)
+        else engine_type_from_db
+    )
+
+    rows: list[tuple[str, str]] = [
+        ("Name", str(engine.get("name", ""))),
+        ("Type", str(type_display)),
+        ("Size", str(engine.get("size", ""))),
+        ("State", str(format_state_with_color(engine.get("state", "")))),
+        ("Created By", str(engine.get("created_by", ""))),
+        ("Created On", str(format_value(engine.get("created_on")))),
+    ]
+
+    # Optional fields (may not exist on older backends).
+    for key, label, formatter in [
+        ("version", "Version", lambda v: "" if v is None else str(v)),
+        ("updated_on", "Updated On", format_value),
+        ("suspends_at", "Suspends At", format_value),
+    ]:
+        if key in engine:
+            try:
+                val = engine.get(key)
+                rows.append((label, str(formatter(val))))
+            except Exception:
+                rows.append((label, str(engine.get(key))))
+
+    # Auto-suspend minutes is represented differently across backends:
+    # - list_engines tends to return auto_suspend_mins
+    # - get_engine (Snowflake EngineServiceSQL) returns auto_suspend
+    if "auto_suspend_mins" in engine or "auto_suspend" in engine:
+        auto_suspend_val = engine.get("auto_suspend_mins", engine.get("auto_suspend"))
+        rows.append(("Auto Suspend (mins)", "" if auto_suspend_val is None else str(auto_suspend_val)))
+
+    settings = engine.get("settings")
+    if settings in (None, {}, ""):
+        settings_str = ""
+    elif isinstance(settings, dict):
+        settings_str = json.dumps(settings, indent=2, sort_keys=True)
+    else:
+        settings_str = str(settings)
+    rows.append(("Settings", settings_str))
+
+    for field, value in rows:
+        table.add_row(field, value)
+
+    rich.print(table)
+
+def exit_with_error(message: str) -> NoReturn:
     rich.print(message, file=sys.stderr)
     exit_with_divider(1)
 
-def exit_with_divider(exit_code:int=0):
+def exit_with_handled_exception(context: str, exc: Exception) -> NoReturn:
+    """Exit with a nicely formatted message for handled RAIExceptions.
+
+    - If `exc` is a RAIException (i.e. produced by an error handler), print its rich
+        formatted content via `exc.pprint()`.
+    - Otherwise, print a raw one-line error including the context and exception string.
+    """
+    if isinstance(exc, RAIException):
+        exc.pprint()
+        sys.exit(1)
+    exit_with_error(f"\n\n[yellow]{context}: {exc}")
+
+def exit_with_divider(exit_code: int = 0) -> NoReturn:
     divider()
     sys.exit(exit_code)

{relationalai-0.13.1.dist-info → relationalai-0.13.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: relationalai
-Version: 0.13.1
+Version: 0.13.2
 Summary: RelationalAI Library and CLI
 Author-email: RelationalAI <support@relational.ai>
 License-File: LICENSE

{relationalai-0.13.1.dist-info → relationalai-0.13.2.dist-info}/RECORD

@@ -4,7 +4,7 @@ relationalai/debugging.py,sha256=wqGly2Yji4ErdV9F_S1Bny35SiGqI3C7S-hap7B8yUs,119
 relationalai/dependencies.py,sha256=tL113efcISkJUiDXYHmRdU_usdD7gmee-VRHA7N4EFA,16574
 relationalai/docutils.py,sha256=1gVv9mk0ytdMB2W7_NvslJefmSQtTOg8LHTCDcGCjyE,1554
 relationalai/dsl.py,sha256=UJr93X8kwnnyUY-kjPzp_jhsp2pYBUnDfu8mhNXPNII,66116
-relationalai/errors.py,sha256=PaWzKsfWwjca7J0huILkiR515_PwJXP1Hlq6owOjL0Y,96828
+relationalai/errors.py,sha256=LNMUXoe7e19c4azqx3WDYpQs-59IEwCH_i3iXxGptE8,96830
 relationalai/metagen.py,sha256=o10PNvR_myr_61DC8g6lkB093bFo9qXGUkZKgKyfXiE,26821
 relationalai/metamodel.py,sha256=P1hliwHd1nYxbXON4LZeaYZD6T6pZm97HgmFBFrWyCk,32886
 relationalai/rel.py,sha256=ePmAXx4NxOdsPcHNHyGH3Jkp_cB3QzfKu5p_EQSHPh0,38293
@@ -19,27 +19,28 @@ relationalai/auth/oauth_callback_server.py,sha256=vbcpz77n_WKMDZ4sac6IYyrpxScR0D
 relationalai/auth/token_handler.py,sha256=d5aueGEiK6BmzkxSab3reCdkhtjdphb0CeecyvvNaQU,19443
 relationalai/auth/util.py,sha256=oXOUwW5gaBhEV5v5A-q2ME1VnfjfRWswvRlXW4oBYpk,1116
 relationalai/clients/__init__.py,sha256=LQ_yHsutRMpoW2mOTmOPGF8mrbP0OiV5E68t8uVwDyQ,833
-relationalai/clients/client.py,sha256=gk_V9KS7_MM2dLL2OCO7EPLHD9dsRwR6R-30SW8lDwU,35759
+relationalai/clients/client.py,sha256=3yK1mCXYSECAS7vBa9uovTeRbfOvnBAJ9USWzhtr_bk,36715
 relationalai/clients/config.py,sha256=hERaKjc3l4kd-kf0l-NUOHrWunCn8gmFWpuE0j3ScJg,24457
 relationalai/clients/direct_access_client.py,sha256=VGjQ7wzduxCo04BkxSZjlPAgqK-aBc32zIXcMfAzzSU,6436
 relationalai/clients/exec_txn_poller.py,sha256=JbmrTvsWGwxM5fcZVjeJihQDqhMf-ySfHN3Jn0HGwG0,3108
 relationalai/clients/hash_util.py,sha256=pZVR1FX3q4G_19p_r6wpIR2tIM8_WUlfAR7AVZJjIYM,1495
-relationalai/clients/local.py,sha256=vo5ikSWg38l3xQAh9yL--4sMAj_T5Tn7YEZiw7TCH08,23504
+relationalai/clients/local.py,sha256=X_v4h8HNyZEYeyQuoDBfavLuedjCVKg5GvnEtOUn5WM,23952
 relationalai/clients/profile_polling.py,sha256=pUH7WKH4nYDD0SlQtg3wsWdj0K7qt6nZqUw8jTthCBs,2565
 relationalai/clients/result_helpers.py,sha256=wDSD02Ngx6W-YQqBIGKnpXD4Ju3pA1e9Nz6ORRI6SRI,17808
-relationalai/clients/types.py,sha256=eNo6akcMTbnBFbBbHd5IgVeY-zuAgtXlOs8Bo1SWmVU,2890
+relationalai/clients/types.py,sha256=xHkmrbqOfxxhBrTiQDUQqN0Lqnsv8pkAzstffYNdqqg,3038
 relationalai/clients/util.py,sha256=NJC8fnrWHR01NydwESPSetIHRWf7jQJURYpaWJjmDyE,12311
 relationalai/clients/resources/__init__.py,sha256=pymn8gB86Q3C2bVoFei0KAL8pX_U04uDY9TE4TKzTBs,260
-relationalai/clients/resources/azure/azure.py,sha256=TDapfM5rLoHrPrXg5cUe827m3AO0gSqQjNid1VUlUFo,20631
-relationalai/clients/resources/snowflake/__init__.py,sha256=Ofyf1RZu9GLQdvsjpHDUHEQHHVODb9vKYI4hMOxczH4,923
+relationalai/clients/resources/azure/azure.py,sha256=3RQucsot4M8sU4yqeCi5IIeC4BgVFMQm1q9uhWY54W0,21676
+relationalai/clients/resources/snowflake/__init__.py,sha256=kHjuNWr37viTd-n8tNMVo0FGhlKlrStb8_V9yJx_CNw,975
 relationalai/clients/resources/snowflake/cache_store.py,sha256=A-qd11wcwN3TkIqvlN0_iFUU3aEjJal3T2pqFBwkkzQ,3966
-relationalai/clients/resources/snowflake/cli_resources.py,sha256=xTIcCzvgbkxuNAEvzZoRpj0n-js0hZCK30q7IZXztbI,3252
-relationalai/clients/resources/snowflake/direct_access_resources.py,sha256=5LZIQCSiX_TT2tVHqbTBzJTBXZzHmQI7cxFOY6CDDBM,28871
-relationalai/clients/resources/snowflake/engine_state_handlers.py,sha256=SQBu4GfbyABU6xrEV-koivC-ubsVrfCBTF0FEQgJM5g,12054
-relationalai/clients/resources/snowflake/error_handlers.py,sha256=581G2xOihUoiPlucC_Z2FOzhKu_swdIc3uORd0yJQuA,8805
+relationalai/clients/resources/snowflake/cli_resources.py,sha256=DjvGyxYLwfCA57FgxpmyBfAyyHIHsbpa-UsoYziEowk,3689
+relationalai/clients/resources/snowflake/direct_access_resources.py,sha256=HF9rowNOBrjimtBGJlDSRsS0jGEHu9FdQhRxImdCM0M,30624
+relationalai/clients/resources/snowflake/engine_service.py,sha256=CYW0RoQjIlEr2w1FeUXwl2gj5o0HoNg9qLmiLo_7rQY,13492
+relationalai/clients/resources/snowflake/engine_state_handlers.py,sha256=gHdqotbNkfAsXQq9XR7aRcsW4QBk7sKAxdra9YS5bsM,12079
+relationalai/clients/resources/snowflake/error_handlers.py,sha256=R5f-3WhOBOFN373BjhkCNLBKVWBIMNPEm2VbAvyvELo,10733
 relationalai/clients/resources/snowflake/export_procedure.py.jinja,sha256=00iLO2qmvJoqAeJUWt3bAsFDDnof7Ab2spggzelbwK4,10566
 relationalai/clients/resources/snowflake/resources_factory.py,sha256=4LGd4IQ6z8hGeGlO1TIjSFJEeUNHutaB7j9q1a9rYfQ,3385
-relationalai/clients/resources/snowflake/snowflake.py,sha256=SuFwAdhaFrd3JZFEjVJMVVFuLrC_T3dqXS27ABrXZAw,136973
+relationalai/clients/resources/snowflake/snowflake.py,sha256=wlGd0AA2fWo2ew681l3Q-N7It0PmcAmiJ4SekzOLcf4,135366
 relationalai/clients/resources/snowflake/use_index_poller.py,sha256=xF_9XAymO82YZcZOQqz0r9oZZTqvbWW-WjDTZm4-tFM,48859
 relationalai/clients/resources/snowflake/use_index_resources.py,sha256=69PNWHI_uf-Aw_evfwC6j8HLVdjhp84vs8hLkjnhwbg,6462
 relationalai/clients/resources/snowflake/util.py,sha256=BEnm1B1-nqqHdm41RNxblbb-zqXbtqEGGZmTdAYeN_M,13841
@@ -360,7 +361,7 @@ relationalai/semantics/metamodel/rewrite/flatten.py,sha256=oyCYyJrq84YpNC6UpLDfS
 relationalai/semantics/metamodel/rewrite/format_outputs.py,sha256=n0IxC3RL3UMly6MWsq342EGfL2yGj3vOgVG_wg7kt-o,6225
 relationalai/semantics/metamodel/typer/__init__.py,sha256=E3ydmhWRdm-cAqWsNR24_Qd3NcwiHx8ElO2tzNysAXc,143
 relationalai/semantics/metamodel/typer/checker.py,sha256=frY0gilDO6skbDiYFiIpDUOWyt9s9jAJsRBs848DcG0,19184
-relationalai/semantics/metamodel/typer/typer.py,sha256=6yyLfGXPuh71TbAvK4I9AkMI4Ap5LlHhhMuDYWzj6Dw,62774
+relationalai/semantics/metamodel/typer/typer.py,sha256=cgb8ckAIQSIzkS2VLvj9tl2UdGALKZXrsn_io35ojNk,62906
 relationalai/semantics/reasoners/__init__.py,sha256=cLrGNKFX859EdPjk2n6MdYLvueaFtTEnfVYnBriQMfI,303
 relationalai/semantics/reasoners/experimental/__init__.py,sha256=ZWXb3Oun7m_G2c3ijKnqxbEsAzTMVa7ciBXjdi4dCNI,144
 relationalai/semantics/reasoners/graph/README.md,sha256=QgKEXTllp5PO-yK8oDfMx1PNTYF2uVoneMRKsWTY5GU,23953
@@ -417,9 +418,9 @@ relationalai/std/re.py,sha256=7B0dPaYyEdIlEgZfDzs7HJ_MTqccYTfIZieqst6MHbk,4457
 relationalai/std/strings.py,sha256=vHvex_W5GHhhsVws6Dfyl4w1EHdbDE3nPjT09DnpvSE,4260
 relationalai/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 relationalai/tools/cleanup_snapshots.py,sha256=s1q1ophi1RmiTZ9QQolQ6q1K9ZwseCaOuDtI1Rm99MQ,3434
-relationalai/tools/cli.py,sha256=z6FpwRe3pVro78QDlTkfq6D-Tpb7KpgyscGpVKhJGVM,76841
+relationalai/tools/cli.py,sha256=KM-_4KCNkW6loi3LjfLNRcApKTcusd_8gFS8166ZEAs,84147
 relationalai/tools/cli_controls.py,sha256=84K0sjfsVwqi9C8ZlpvPi6XZPhjyW-R06Pir9Ex_Xtg,70468
-relationalai/tools/cli_helpers.py,sha256=dUIx9Z1I1rTZ3VKtn4jlKGCY5TNk1rl1CY8g3-m99A0,14431
+relationalai/tools/cli_helpers.py,sha256=z7D4-9hX8DMm_WrG_5JYpsRhe9WzznsbKh2kN4OvGuw,30846
 relationalai/tools/constants.py,sha256=A8Qz0CPs2cRUs5xaZ-Y7BCyvqmI6RLvipzwLDYZ3_PU,4505
 relationalai/tools/debugger.py,sha256=iWO1tMJud3dXHIkrjdZ5Temp0MR2AwN5_QGDt-ddmcs,6934
 relationalai/tools/debugger_client.py,sha256=g9vBaeJLKVq6vW63W-KKSaPJ--g5Wuf4LcIuYRA_pk4,3729
@@ -444,7 +445,7 @@ relationalai/util/spans_file_handler.py,sha256=a0sDwDPBBvGsM6be2En3mId9sXpuJlXia
 relationalai/util/timeout.py,sha256=2o6BVNFnFc-B2j-i1pEkZcQbMRto9ps2emci0XwiA4I,783
 relationalai/util/tracing_handler.py,sha256=H919ETAxh7Z1tRz9x8m90qP51_264UunHAPw8Sr6x2g,1729
 relationalai_test_util/__init__.py,sha256=Io_9_IQXXnrUlaL7S1Ndv-4YHilNxy36LrL723MI7lw,118
-relationalai_test_util/fixtures.py,sha256=WhpPto6LtZTGWQhYV9Tmz9yJGoVAkYlGRrgM_7errm4,9455
+relationalai_test_util/fixtures.py,sha256=1fCU-45pUMuWstkOfQ1u_37lJnPPqC43ks2nYW-UA7A,9478
 relationalai_test_util/snapshot.py,sha256=FeH2qYBzLxr2-9qs0yElPIgWUjm_SrzawB3Jgn-aSuE,9291
 relationalai_test_util/traceback.py,sha256=lD0qaEmCyO-7xg9CNf6IzwS-Q-sTS8N9YIv8RroAE50,3298
 frontend/debugger/dist/.gitignore,sha256=JAo-DTfS6GthQGP1NH6wLU-ZymwlTea4KHH_jZVTKn0,14
@@ -452,8 +453,8 @@ frontend/debugger/dist/index.html,sha256=0wIQ1Pm7BclVV1wna6Mj8OmgU73B9rSEGPVX-Wo
 frontend/debugger/dist/assets/favicon-Dy0ZgA6N.png,sha256=tPXOEhOrM4tJyZVJQVBO_yFgNAlgooY38ZsjyrFstgg,620
 frontend/debugger/dist/assets/index-Cssla-O7.js,sha256=MxgIGfdKQyBWgufck1xYggQNhW5nj6BPjCF6Wleo-f0,298886
 frontend/debugger/dist/assets/index-DlHsYx1V.css,sha256=21pZtAjKCcHLFjbjfBQTF6y7QmOic-4FYaKNmwdNZVE,60141
-relationalai-0.13.1.dist-info/METADATA,sha256=zO4iuv-8tLuz3rByVdd1X3MSdh5MU6XZzqzXcAvdf4k,2561
-relationalai-0.13.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-relationalai-0.13.1.dist-info/entry_points.txt,sha256=fo_oLFJih3PUgYuHXsk7RnCjBm9cqRNR--ab6DgI6-0,88
-relationalai-0.13.1.dist-info/licenses/LICENSE,sha256=pPyTVXFYhirkEW9VsnHIgUjT0Vg8_xsE6olrF5SIgpc,11343
-relationalai-0.13.1.dist-info/RECORD,,
+relationalai-0.13.2.dist-info/METADATA,sha256=C_6NBEgmM09pft0zVQWwdSXVuUDJG6IGrY_C0WKuVCE,2561
+relationalai-0.13.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+relationalai-0.13.2.dist-info/entry_points.txt,sha256=fo_oLFJih3PUgYuHXsk7RnCjBm9cqRNR--ab6DgI6-0,88
+relationalai-0.13.2.dist-info/licenses/LICENSE,sha256=pPyTVXFYhirkEW9VsnHIgUjT0Vg8_xsE6olrF5SIgpc,11343
+relationalai-0.13.2.dist-info/RECORD,,

relationalai_test_util/fixtures.py

@@ -78,14 +78,14 @@ def create_engine(engine_name: str, size: str,  use_direct_access=False):
 
     provider = rai.Resources(config=config)
     print(f"Creating engine {engine_name}")
-    provider.create_engine(name=engine_name, size=size)
+    provider.create_engine(name=engine_name, type="LOGIC", size=size)
     print(f"Engine {engine_name} created")
 
 def delete_engine(engine_name: str, use_direct_access=False):
     print(f"Deleting engine {engine_name}")
     config = make_config(engine_name, use_direct_access=use_direct_access)
     provider = rai.Resources(config=config)
-    provider.delete_engine(engine_name)
+    provider.delete_engine(engine_name, "LOGIC")
     print(f"Engine {engine_name} deleted")
 
 def make_config(engine_name: str | None = None, fetch_profile: bool = True, use_package_manager = USE_PACKAGE_MANAGER, use_direct_access = False, show_full_traces = True, show_debug_logs = True, reuse_model = False) -> cfg.Config: