relationalai 1.0.0a3__py3-none-any.whl → 1.0.0a4__py3-none-any.whl

relationalai/tools/cli/docs.py

@@ -0,0 +1,394 @@
+"""
+Documentation generation and serving utilities.
+"""
+
+from __future__ import annotations
+
+import shutil
+import subprocess
+import webbrowser
+from pathlib import Path
+from typing import Optional
+
+import click
+from rich.console import Console
+
+console = Console()
+
+
+def find_docs_dir() -> Path:
+    """Find the documentation source directory."""
+    # Try multiple possible locations
+    possible_paths = [
+        Path("docs"),  # Root level
+        Path(__file__).parent.parent.parent.parent / "docs",  # From this file
+        Path.cwd() / "docs",  # Current working directory
+    ]
+
+    for path in possible_paths:
+        if path.exists() and (path / "package.json").exists():
+            return path.resolve()
+
+    raise click.ClickException(
+        "Documentation source directory not found. "
+        "Expected 'docs/' directory with package.json"
+    )
+
+
+def has_nodejs() -> bool:
+    """Check if Node.js is installed."""
+    try:
+        subprocess.run(
+            ["node", "--version"],
+            capture_output=True,
+            text=True,
+            check=True,
+        )
+        return True
+    except (subprocess.CalledProcessError, FileNotFoundError):
+        return False
+
+
+def has_npm() -> bool:
+    """Check if npm is installed."""
+    try:
+        subprocess.run(
+            ["npm", "--version"],
+            capture_output=True,
+            text=True,
+            check=True,
+        )
+        return True
+    except (subprocess.CalledProcessError, FileNotFoundError):
+        return False
+
+
+def print_generation_success(output_path: Path) -> None:
+    """
+    Print success message after documentation generation.
+
+    Args:
+        output_path: Path to the generated documentation output directory
+    """
+    console.print("[green]✓ Documentation generated")
+    console.print()
+    console.print("[cyan]The documentation site is ready to host![/cyan]")
+    console.print(f"[dim]Output location: {output_path}[/dim]")
+    console.print()
+    console.print("[yellow]You can serve it locally with:[/yellow]")
+    console.print("  [bold]rai docs serve[/bold]")
+    console.print()
+    console.print("[yellow]Or host the dist folder directly:[/yellow]")
+    console.print(
+        "[dim]The dist folder contains a complete static website. You can copy it to any "
+        "web server (Nginx, Apache, Flask, etc.), upload it to a CDN, or deploy it to "
+        "static hosting services like GitHub Pages, Netlify, or Vercel. All files use "
+        "relative paths, so it works regardless of where it's hosted.[/dim]"
+    )
+
+
+def generate_docs(
+    output_dir: Optional[str] = None,
+    docs_dir: Optional[Path] = None,
+) -> Path:
+    """
+    Generate static documentation site.
+
+    Args:
+        output_dir: Optional output directory (defaults to docs/dist)
+        docs_dir: Optional path to docs directory
+
+    Returns:
+        Path to the generated output directory
+    """
+    if docs_dir is None:
+        docs_dir = find_docs_dir()
+
+    # Check prerequisites
+    if not has_nodejs():
+        raise click.ClickException(
+            "Node.js is not installed. "
+            "Please install Node.js from https://nodejs.org/"
+        )
+
+    if not has_npm():
+        raise click.ClickException(
+            "npm is not installed. "
+            "npm should come with Node.js installation."
+        )
+
+    # Check if package.json exists
+    package_json = docs_dir / "package.json"
+    if not package_json.exists():
+        raise click.ClickException(
+            f"package.json not found in {docs_dir}. "
+            "This doesn't appear to be a valid SolidJS project."
+        )
+
+    # Install dependencies if node_modules doesn't exist
+    node_modules = docs_dir / "node_modules"
+    if not node_modules.exists():
+        console.print("[blue]Installing dependencies...[/blue]")
+        try:
+            subprocess.run(
+                ["npm", "install"],
+                cwd=docs_dir,
+                check=True,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+            )
+            console.print("[green]✓ Dependencies installed[/green]")
+        except subprocess.CalledProcessError as e:
+            error_msg = e.stderr.decode() if e.stderr else "Unknown error"
+            raise click.ClickException(
+                f"Failed to install dependencies: {error_msg}"
+            )
+
+    # Build the documentation
+    console.print("[blue]Building documentation site...[/blue]")
+    try:
+        subprocess.run(
+            ["npm", "run", "build"],
+            cwd=docs_dir,
+            check=True,
+            capture_output=True,
+            text=True,
+        )
+        console.print("[green]✓ Build completed successfully[/green]")
+    except subprocess.CalledProcessError as e:
+        error_msg = e.stderr if e.stderr else e.stdout if e.stdout else "Unknown error"
+        console.print(f"[red]Build error output:[/red]\n{error_msg}")
+        raise click.ClickException(f"Build failed: {error_msg}")
+
+    # Determine output directory
+    build_output = docs_dir / "dist"
+    if not build_output.exists():
+        raise click.ClickException(
+            "Build output directory not found. "
+            "Build may have failed."
+        )
+
+    # Copy to custom output directory if specified
+    if output_dir:
+        output_path = Path(output_dir).resolve()
+        if output_path.exists():
+            console.print(f"[yellow]Output directory '{output_path}' exists. Cleaning...[/yellow]")
+            shutil.rmtree(output_path)
+        shutil.copytree(build_output, output_path)
+        print_generation_success(output_path)
+        return output_path
+
+    print_generation_success(build_output)
+    return build_output
+
+
+def serve_docs(
+    build_dir: str,
+    host: str = "127.0.0.1",
+    port: int = 8000,
+    open_browser: bool = True,
+) -> None:
+    """
+    Serve generated documentation site.
+
+    Args:
+        build_dir: Directory containing built documentation
+        host: Host to bind to
+        port: Port to serve on
+        open_browser: Whether to open browser automatically
+    """
+    build_path = Path(build_dir).resolve()
+
+    if not build_path.exists():
+        raise click.ClickException(
+            "Build directory does not exist. "
+            "Run 'rai docs generate' first."
+        )
+
+    index_html = build_path / "index.html"
+    if not index_html.exists():
+        raise click.ClickException(
+            "index.html not found. "
+            "This doesn't appear to be a valid build directory. "
+            "Run 'rai docs generate' first."
+        )
+
+    # Import FastAPI here to avoid requiring it if not using server
+    try:
+        from fastapi import FastAPI
+        from fastapi.staticfiles import StaticFiles
+        import uvicorn
+    except ImportError:
+        raise click.ClickException(
+            "FastAPI and uvicorn are required for serving documentation. "
+            "Install with: pip install fastapi uvicorn"
+        )
+
+    app = FastAPI(title="RAI Documentation")
+    app.mount("/", StaticFiles(directory=str(build_path), html=True), name="static")
+
+    url = f"http://{host}:{port}"
+    console.print(f"[green]✓ Documentation server running at {url}[/green]")
+    console.print("[yellow]Press Ctrl+C to stop the server[/yellow]")
+
+    if open_browser:
+        try:
+            webbrowser.open(url)
+        except Exception:
+            console.print(f"[yellow]Could not open browser automatically. Visit {url}[/yellow]")
+
+    try:
+        uvicorn.run(app, host=host, port=port, log_level="warning")
+    except KeyboardInterrupt:
+        console.print("\n[yellow]Server stopped[/yellow]")
+    except OSError as e:
+        if "Address already in use" in str(e):
+            raise click.ClickException(
+                f"Port {port} is already in use. "
+                "Try a different port with --port option."
+            )
+        raise
+
+
+def dev_docs(
+    docs_dir: Optional[Path] = None,
+    host: str = "127.0.0.1",
+    port: int = 5173,
+    open_browser: bool = True,
+) -> None:
+    """
+    Start development server with hot module replacement.
+
+    This runs Vite's dev server, which provides:
+    - Hot module replacement (HMR) - changes reflect instantly
+    - Fast refresh - no need to rebuild on every change
+    - Source maps for debugging
+
+    Args:
+        docs_dir: Optional path to docs directory
+        host: Host to bind to
+        port: Port to serve on (default: 5173, Vite's default)
+        open_browser: Whether to open browser automatically
+    """
+    if docs_dir is None:
+        docs_dir = find_docs_dir()
+
+    # Check prerequisites
+    if not has_nodejs():
+        raise click.ClickException(
+            "Node.js is not installed. "
+            "Please install Node.js from https://nodejs.org/"
+        )
+
+    if not has_npm():
+        raise click.ClickException(
+            "npm is not installed. "
+            "npm should come with Node.js installation."
+        )
+
+    # Check if package.json exists
+    package_json = docs_dir / "package.json"
+    if not package_json.exists():
+        raise click.ClickException(
+            f"package.json not found in {docs_dir}. "
+            "This doesn't appear to be a valid SolidJS project."
+        )
+
+    # Install dependencies if node_modules doesn't exist
+    node_modules = docs_dir / "node_modules"
+    if not node_modules.exists():
+        console.print("[blue]Installing dependencies...[/blue]")
+        try:
+            subprocess.run(
+                ["npm", "install"],
+                cwd=docs_dir,
+                check=True,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+            )
+            console.print("[green]✓ Dependencies installed[/green]")
+        except subprocess.CalledProcessError as e:
+            error_msg = e.stderr.decode() if e.stderr else "Unknown error"
+            raise click.ClickException(
+                f"Failed to install dependencies: {error_msg}"
+            )
+
+    # Start dev server
+    url = f"http://{host}:{port}"
+    console.print("[green]✓ Starting development server...[/green]")
+    console.print(f"[blue]Server will be available at {url}[/blue]")
+    console.print("[yellow]Press Ctrl+C to stop the server[/yellow]")
+    console.print(
+        "[dim]Note: Changes to files will automatically reload in the browser[/dim]"
+    )
+
+    if open_browser:
+        # Open browser after a short delay to let server start
+        import threading
+        import time
+
+        def open_browser_delayed():
+            time.sleep(1.5)  # Give server time to start
+            try:
+                webbrowser.open(url)
+            except Exception:
+                console.print(f"[yellow]Could not open browser automatically. Visit {url}[/yellow]")
+
+        threading.Thread(target=open_browser_delayed, daemon=True).start()
+
+    try:
+        # Run npm run dev with custom host/port
+        # Vite accepts --host and --port flags directly
+        subprocess.run(
+            ["npm", "run", "dev", "--", "--host", host, "--port", str(port)],
+            cwd=docs_dir,
+            check=True,
+        )
+    except KeyboardInterrupt:
+        console.print("\n[yellow]Development server stopped[/yellow]")
+    except subprocess.CalledProcessError as e:
+        error_msg = e.stderr.decode() if e.stderr else e.stdout.decode() if e.stdout else "Unknown error"
+        raise click.ClickException(f"Dev server failed: {error_msg}")
+    except OSError as e:
+        if "Address already in use" in str(e):
+            raise click.ClickException(
+                f"Port {port} is already in use. "
+                "Try a different port with --port option."
+            )
+        raise
+
+
+def cleanup_docs(docs_dir: Optional[Path] = None) -> None:
+    """
+    Clean up generated files and dependencies.
+
+    Silently removes:
+    - node_modules/ directory
+    - dist/ directory (build output)
+
+    Args:
+        docs_dir: Optional path to docs directory
+    """
+    if docs_dir is None:
+        docs_dir = find_docs_dir()
+
+    removed_items = []
+
+    # Remove node_modules
+    node_modules = docs_dir / "node_modules"
+    if node_modules.exists() and node_modules.is_dir():
+        try:
+            shutil.rmtree(node_modules)
+            removed_items.append("node_modules")
+        except Exception as e:
+            raise click.ClickException(f"Failed to remove node_modules: {e}")
+
+    # Remove dist
+    dist = docs_dir / "dist"
+    if dist.exists() and dist.is_dir():
+        try:
+            shutil.rmtree(dist)
+            removed_items.append("dist")
+        except Exception as e:
+            raise click.ClickException(f"Failed to remove dist: {e}")
+

{relationalai-1.0.0a3.dist-info → relationalai-1.0.0a4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: relationalai
-Version: 1.0.0a3
+Version: 1.0.0a4
 Summary: RelationalAI Library and CLI
 Author-email: RelationalAI <support@relational.ai>
 Requires-Python: >=3.10
@@ -15,10 +15,9 @@ Requires-Dist: numpy<2
 Requires-Dist: pandas
 Requires-Dist: colorama
 Requires-Dist: inquirerpy
-Requires-Dist: click==8.2.1
+Requires-Dist: click
 Requires-Dist: gravis
 Requires-Dist: toml
-Requires-Dist: tomlkit
 Requires-Dist: requests
 Requires-Dist: pyarrow
 Requires-Dist: websockets
@@ -35,6 +34,9 @@ Requires-Dist: sqlglot
 Requires-Dist: pyyaml
 Requires-Dist: pydantic<2.12.0,>=2.0.0
 Requires-Dist: pydantic-settings<2.11.0,>=2.0.0
+Requires-Dist: tomlkit
+Requires-Dist: fastapi
+Requires-Dist: uvicorn[standard]
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"

{relationalai-1.0.0a3.dist-info → relationalai-1.0.0a4.dist-info}/RECORD

@@ -2,7 +2,7 @@ relationalai/__init__.py,sha256=hHhZc1h-IDj0pzPF67763g0a4Pdh7W-4GQpyqL60dec,152
 relationalai/config/__init__.py,sha256=QL7MxBN7xy50h4Sqg3jotuyg-6EisCEtJCHCWQrL1ME,1524
 relationalai/config/config.py,sha256=CUBkOYWu3IgCAa4oHVwEozfMbZgAX3hENo-Va2gsdIs,10477
 relationalai/config/config_fields.py,sha256=Gskc1Ugsi4xowepDlma1HNaYdQ_wS2RiMpUaG6aNdv4,4081
-relationalai/config/shims.py,sha256=Xoqvie0Rlc7Qr3FvipPdcDiGwHJmFk3m5TQ_vNRPZgA,14
+relationalai/config/shims.py,sha256=YM3rtPNpRSrQomRw5xNSekU5Da3dcYbxxzGryQt7RME,41
 relationalai/config/connections/__init__.py,sha256=5709qbuJxqg2WedSoejpdDrXpvwowG_xDS8adoexeOA,1171
 relationalai/config/connections/base.py,sha256=9DkPLosMc9yb6uFFks_FP_e-47K5wMD14tW_cJ05s_A,759
 relationalai/config/connections/duckdb.py,sha256=PhhJXARTJf2ObQa9CYMJlZMGohMNCg879cKHvUr1M5Q,891
@@ -13,22 +13,25 @@ relationalai/config/external/dbt_models.py,sha256=Px_C5nID_3te3o1wfsYJjrKkZcTPUY
 relationalai/config/external/snowflake_converter.py,sha256=Vnaqgazd-XwZybqbOU2YK3qYf0if-KwdTORaP0xTff4,1280
 relationalai/config/external/snowflake_models.py,sha256=6hCaVMdhESxbW-HF0-s51Z4WIc97bS1FrZRKZ4DXE6A,2681
 relationalai/config/external/utils.py,sha256=IJsZUCLI4Y2erETERQKcd-Alo4wNMTi1AHkdb92Z-n0,422
-relationalai/semantics/__init__.py,sha256=N3rTlD4oIyIkCqTNNWWUjQbJMyEbot6XNR_NvmtiyDo,4340
+relationalai/semantics/__init__.py,sha256=Td5zPCjwRbYV-UA05kXwnuL1BW-iRxiWEV6sJksrzTY,4640
 relationalai/semantics/backends/lqp/annotations.py,sha256=TUk_fuXyBWs7H4BHx-4jZGonSD8MdgPswt-D2k7QC-4,390
 relationalai/semantics/backends/sql/sql_compiler.py,sha256=p_1YjROlcmu0kwvI32qUWAmMQntp_A6H3_vgGDDLiQk,10628
 relationalai/semantics/frontend/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-relationalai/semantics/frontend/base.py,sha256=yUBeXvF0cRYzKk4sTSh7Ja5PUSwEszvyuQjT1E25Cjc,65783
-relationalai/semantics/frontend/core.py,sha256=mwPG9HxQWrh77O5n7j_M9lt96gTyYsoUE_BFz7qcfXw,7396
-relationalai/semantics/frontend/front_compiler.py,sha256=q84SDS1AQllKSnZya1k8gUG_SD12zIl9QbWIU9zFliU,57403
-relationalai/semantics/frontend/pprint.py,sha256=IMkKwkZs3ohOIjYUIjT10poa1L39F_PrLdSjV-D4wpk,14977
+relationalai/semantics/frontend/base.py,sha256=xxs0udBqK4ErwA9pPAtFXV-Ky4keEOJQ0k4gsOKrMUQ,66007
+relationalai/semantics/frontend/core.py,sha256=nTFDsF18WemplUmxOeCbUTHGhrCmxWIeRjC5amKkdDE,8399
+relationalai/semantics/frontend/front_compiler.py,sha256=RIa12ifKoWJ0HqwrBSiDniKF5K-AGEJ1L8VC4wOyuck,58615
+relationalai/semantics/frontend/pprint.py,sha256=mc2i4MlmPpsjXWx6bO_J7lMrxUeDfO2du4eaPZi6pFU,14982
 relationalai/semantics/metamodel/__init__.py,sha256=mISE8niHeEAD6QjX8pi_apFfMIVCZ7dGeCXFGtFPTvk,364
 relationalai/semantics/metamodel/builtins.py,sha256=rmeVdjw3kDSsx7TjpnFpy8W6OrHuUV1GQwazWhaWrG0,7009
 relationalai/semantics/metamodel/metamodel.py,sha256=TgIufxJslJZkf5_RleTU19i2h0DTW6CodSYqwyVcLEc,13899
 relationalai/semantics/metamodel/metamodel_analyzer.py,sha256=tusDRO2az1yiF951SDg-cHOaNQ1M7qw1HG1cxV3YuxU,16228
 relationalai/semantics/metamodel/metamodel_compiler.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 relationalai/semantics/metamodel/pprint.py,sha256=65t7tZTfBP2qOlKShYM8nwrbsiS3gKaCYkUJepu0a2o,15684
-relationalai/semantics/metamodel/rewriter.py,sha256=CEkYxsyFPF-nEHktJxwzOY6RQzo9Nr4Welhw2YDq9q8,8313
-relationalai/semantics/metamodel/typer.py,sha256=Q35spCuIRMGE2BslgWiyZvQWb3Z0VzCx1YF_Lzk1Sts,55268
+relationalai/semantics/metamodel/rewriter.py,sha256=14T2fzoBLMvYZNOy8VjRq_NS7rwU9EECbkQXJxfdvDw,8436
+relationalai/semantics/metamodel/typer.py,sha256=Acc1NiS9oCrqt9ZP_uI4GyT0sduOwmu1tS5U-3Irndo,57654
+relationalai/semantics/reasoners/__init__.py,sha256=T1ysWgmX85pULUqdxZroPexY45DvfzsV3J2xJQHr3nQ,377
+relationalai/semantics/reasoners/graph/__init__.py,sha256=1hFTc-a8tj4LJv7bYUSGqRCKAuSygbhqwFH25lebWDs,1356
+relationalai/semantics/reasoners/graph/core.py,sha256=prUtjt8nb87Xp9z38Y6bD_iqFaYWCZLwBGhWOmCs20I,399827
 relationalai/semantics/std/__init__.py,sha256=_cbDmeCT634GtdGJ0CqgfJw6pFaJR0ajR3B_bWCr_9A,2627
 relationalai/semantics/std/aggregates.py,sha256=uCwoBLmuRBBq9lTf5ACD3yosvtp5Wi8IQUCF_dKWS80,6259
 relationalai/semantics/std/common.py,sha256=raWuvX2u4juKPU0BKtqF5xvJSDjKHF1BWQVFioBmOf8,1658
@@ -44,14 +47,15 @@ relationalai/semantics/std/strings.py,sha256=w0k06gDfyMS6GODPjhG0gl2SkJ0aH5aYezS
 relationalai/shims/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 relationalai/shims/executor.py,sha256=LtpBx74mhbJ1bCUVvetDlk5iDAEi_x-wNZ6iq5WPW2s,7531
 relationalai/shims/helpers.py,sha256=1v6D-wDq3nCwDz4egdGUeleRC0fK6wfo9fIQIeQCfZA,4508
-relationalai/shims/hoister.py,sha256=YXXi9pWpKQ7Sd5DzRVl_78z4OYCoKzFfIMMp6OIN3EM,8797
-relationalai/shims/mm2v0.py,sha256=_fqEfd6b3sVwmkuozfggoopRdNeqbsrtJgj8b9Teij8,63957
+relationalai/shims/hoister.py,sha256=maebjwse3GvOFCHJWgBXBTwpiqeny8DAyoFW6HtUzI8,9423
+relationalai/shims/mm2v0.py,sha256=G1Gct9rtH6KF1z0yN6yZj0WFTACfeJhcbRCZTyc1eRU,64436
 relationalai/tools/debugger.py,sha256=B8QIE1-WFdfkiPoFFn1yztr9t6yf5PI32Y9Y9XzQOJk,10798
 relationalai/tools/typer_debugger.py,sha256=__oLODEBwgt6pYIn1jgIPlenFmiK7iqXfTY_mozHxXA,3182
 relationalai/tools/cli/__init__.py,sha256=jkIlp7wftUuSIlczwZD4_siT5cs6mzpsNGrQ_kMkLZ4,74
-relationalai/tools/cli/cli.py,sha256=aNXpFb-8GaBjv1J8qk32EeRXu4bDuBnlVsVkNasnfvw,2278
+relationalai/tools/cli/cli.py,sha256=N2quQWjcgq7qqiW1WhkDGOVZZL0nfNfUc59zsUj95pY,5958
 relationalai/tools/cli/config_template.py,sha256=l52KOI3ZBkhtM9V7EQSUOEddHVdMnjl21hPnnQSwdBE,949
 relationalai/tools/cli/dev.py,sha256=1GjfjTLi4BXbV5fXKlF5g1bc8-YsA9oKOBSntquBd8g,340
+relationalai/tools/cli/docs.py,sha256=GSUMHVqHMzrpYw1wOHTKfbkZW-dozRfzoKpgb2o4wvQ,12648
 relationalai/tools/cli/components/__init__.py,sha256=H91hUt3lohr6kAVGOHgTKpDgwRc2Ka4a-iw2domfm5o,143
 relationalai/tools/cli/components/progress_reader.py,sha256=mrAFI4p1T7tgDNd_S9QAF9QXuF5RlWGSg5fzQY1_MRY,56767
 relationalai/tools/cli/components/utils.py,sha256=n_st7K3B8bDLsi8rjYgilPtx7TJRcU5xZlauGUrjyv8,1700
@@ -72,7 +76,7 @@ v0/relationalai/debugging.py,sha256=AuzbwLT1s71tctuIbYNhfWY2vR_opB7ItbmlqFVHgrY,
 v0/relationalai/dependencies.py,sha256=tL113efcISkJUiDXYHmRdU_usdD7gmee-VRHA7N4EFA,16574
 v0/relationalai/docutils.py,sha256=1gVv9mk0ytdMB2W7_NvslJefmSQtTOg8LHTCDcGCjyE,1554
 v0/relationalai/dsl.py,sha256=9BdpqzDNcW4717GyxVAYIMZ7ctE_bvXS-5sNLIbXQZ0,66128
-v0/relationalai/errors.py,sha256=2rFE3tWDhhZ3GCzG1UNuCkns3xxsscUU6wx_3VvoVWI,95976
+v0/relationalai/errors.py,sha256=bwKNODY5aROWRRW-Df2zxHk5ACxWnR0zOhOdOHO9Yi0,96831
 v0/relationalai/metagen.py,sha256=o10PNvR_myr_61DC8g6lkB093bFo9qXGUkZKgKyfXiE,26821
 v0/relationalai/metamodel.py,sha256=P1hliwHd1nYxbXON4LZeaYZD6T6pZm97HgmFBFrWyCk,32886
 v0/relationalai/rel.py,sha256=ePmAXx4NxOdsPcHNHyGH3Jkp_cB3QzfKu5p_EQSHPh0,38293
@@ -90,6 +94,7 @@ v0/relationalai/clients/__init__.py,sha256=LQ_yHsutRMpoW2mOTmOPGF8mrbP0OiV5E68t8
 v0/relationalai/clients/client.py,sha256=gk_V9KS7_MM2dLL2OCO7EPLHD9dsRwR6R-30SW8lDwU,35759
 v0/relationalai/clients/config.py,sha256=PriDlD0_VZFV0fEQng6U-F5xjGwz5gC2F5_lM159AjM,24481
 v0/relationalai/clients/direct_access_client.py,sha256=p3hqjwdiXQQytRke1wIeNRM-HlZLAxH1U6-yPhsPOdQ,6454
+v0/relationalai/clients/exec_txn_poller.py,sha256=veEHerCeKIJe9X7Yjnxab-lXUI4lWBW8tmY-tgMwKwQ,3123
 v0/relationalai/clients/hash_util.py,sha256=NXVtUFgRokZz-45CwV-r5Z5KvmEXD2dh4ypNOrEWAwY,1501
 v0/relationalai/clients/local.py,sha256=vo5ikSWg38l3xQAh9yL--4sMAj_T5Tn7YEZiw7TCH08,23504
 v0/relationalai/clients/profile_polling.py,sha256=c1ixxAOKvy-dDfVqkKENwfcsvBB8jhLA1oT9RIFIrY8,2571
@@ -98,15 +103,15 @@ v0/relationalai/clients/types.py,sha256=VU2LRtlnFsBRYdokxUccrsUkEYLE9N62EUiWtA_q
 v0/relationalai/clients/util.py,sha256=Vrw_kr-Oqp_DKTkJ_rTbzpDm4_Z4h8w10T8GUyGLxbA,12323
 v0/relationalai/clients/resources/__init__.py,sha256=pymn8gB86Q3C2bVoFei0KAL8pX_U04uDY9TE4TKzTBs,260
 v0/relationalai/clients/resources/azure/azure.py,sha256=TDapfM5rLoHrPrXg5cUe827m3AO0gSqQjNid1VUlUFo,20631
-v0/relationalai/clients/resources/snowflake/__init__.py,sha256=9VR-hSIw4ZSEWisKcWhNEcRVBmBfueXNCTOOfLt-8rs,871
+v0/relationalai/clients/resources/snowflake/__init__.py,sha256=Ofyf1RZu9GLQdvsjpHDUHEQHHVODb9vKYI4hMOxczH4,923
 v0/relationalai/clients/resources/snowflake/cache_store.py,sha256=A-qd11wcwN3TkIqvlN0_iFUU3aEjJal3T2pqFBwkkzQ,3966
 v0/relationalai/clients/resources/snowflake/cli_resources.py,sha256=xTIcCzvgbkxuNAEvzZoRpj0n-js0hZCK30q7IZXztbI,3252
-v0/relationalai/clients/resources/snowflake/direct_access_resources.py,sha256=Xvh1e6TxUW2dTSS-9HadrfWVKrxNQ5GikqM4yjohJkM,29849
+v0/relationalai/clients/resources/snowflake/direct_access_resources.py,sha256=a_vfArYM8IAd7gAwx4VEis0KOOr2Vej7vMwY3csxGmQ,30134
 v0/relationalai/clients/resources/snowflake/engine_state_handlers.py,sha256=SQBu4GfbyABU6xrEV-koivC-ubsVrfCBTF0FEQgJM5g,12054
 v0/relationalai/clients/resources/snowflake/error_handlers.py,sha256=581G2xOihUoiPlucC_Z2FOzhKu_swdIc3uORd0yJQuA,8805
 v0/relationalai/clients/resources/snowflake/resources_factory.py,sha256=4LGd4IQ6z8hGeGlO1TIjSFJEeUNHutaB7j9q1a9rYfQ,3385
-v0/relationalai/clients/resources/snowflake/snowflake.py,sha256=UiSE2jbJVGbe2_UfLI9Gud9aA5SJysuVu3F0McOz95k,133046
-v0/relationalai/clients/resources/snowflake/use_index_poller.py,sha256=4lPMgaeuxUdWo-ds_78OJpc7pyogWhSVgGzuTzj13wE,48460
+v0/relationalai/clients/resources/snowflake/snowflake.py,sha256=buPPwLR9_gX5zo9sN90rGD1qIkn1qX4yOcaXmNlOP4U,134362
+v0/relationalai/clients/resources/snowflake/use_index_poller.py,sha256=AE7z2pL4QioPRDL7-O9J4KHSuxbYbRAvd886mZPpCQI,48871
 v0/relationalai/clients/resources/snowflake/use_index_resources.py,sha256=69PNWHI_uf-Aw_evfwC6j8HLVdjhp84vs8hLkjnhwbg,6462
 v0/relationalai/clients/resources/snowflake/util.py,sha256=BEnm1B1-nqqHdm41RNxblbb-zqXbtqEGGZmTdAYeN_M,13841
 v0/relationalai/early_access/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -372,7 +377,7 @@ v0/relationalai/semantics/lqp/__init__.py,sha256=XgcQZxK-zz_LqPDVtwREhsIvjTuUIt4
 v0/relationalai/semantics/lqp/builtins.py,sha256=Xst9-UxkCAwDqL-hFOby8B3GKh705oidFUjEozxn71Y,585
 v0/relationalai/semantics/lqp/compiler.py,sha256=xpGIQAqTMwLi3Vrbe79Yal7O_mfNxpBLioZjqeV2hzc,955
 v0/relationalai/semantics/lqp/constructors.py,sha256=x7G85vbJzLIXzUCMKVdJVsr4PrcUsAOzCCfvimHLMVU,2366
-v0/relationalai/semantics/lqp/executor.py,sha256=3tJVxFsUMdevC-HGYR2BjexC4nBoblUBSf87xlIZ_g0,21469
+v0/relationalai/semantics/lqp/executor.py,sha256=OkxegnEp4_U7uslhDS0cBi8OQOg499NPYEYomWpEqlI,21550
 v0/relationalai/semantics/lqp/intrinsics.py,sha256=8kgH4ndYKBCNnoNOn7iWOU0wnMTSiov0kYaKocQ45RY,880
 v0/relationalai/semantics/lqp/ir.py,sha256=6W9mUH0W7u5eIfF1S3o33uSOfQuM3UcqEkxrxpr1X_8,1867
 v0/relationalai/semantics/lqp/model2lqp.py,sha256=J_QofcYgq_Yb5dhOOP4lXvxsxFMEGHVnaNXoCMK_ak8,38639
@@ -387,7 +392,7 @@ v0/relationalai/semantics/lqp/rewrite/__init__.py,sha256=V9ERED9qdh4VvY9Ud_M8Zn8
 v0/relationalai/semantics/lqp/rewrite/annotate_constraints.py,sha256=zOm-OUzWBhk2iMc_eURII1-okIGJ-teDbEcJ30WDjUU,2325
 v0/relationalai/semantics/lqp/rewrite/cdc.py,sha256=cx_fpQrzkzkeAGHE_kV10nd1-0Jduws6yiIRq650f4c,10412
 v0/relationalai/semantics/lqp/rewrite/extract_common.py,sha256=johHGsS9SDDVbonMIE8Ck9rCuidllwdeJZDruW_5kQs,14705
-v0/relationalai/semantics/lqp/rewrite/extract_keys.py,sha256=lDO1e5pC_Z-nYMXt5zeLBxWQUjNgkhLMiaRLYkF7ZqU,21070
+v0/relationalai/semantics/lqp/rewrite/extract_keys.py,sha256=8C5te8XWU5JiLpc7WojIVKg0_wyMM-gdlxmJdHQSnQQ,22165
 v0/relationalai/semantics/lqp/rewrite/function_annotations.py,sha256=WAHpqRVBqpUIDL2MA1aBmyOTvcpH_yfdeWBS1vAEtzA,4675
 v0/relationalai/semantics/lqp/rewrite/functional_dependencies.py,sha256=LTwMvU19FMdgCG8nJwV2wyiW8Hu9oCj-hIKKSMOYZYE,12727
 v0/relationalai/semantics/lqp/rewrite/quantify_vars.py,sha256=3l_pDMc4r-izuE274MP_ffNgZ6g7Rk5ysg3CXqKKlBU,12016
@@ -420,7 +425,7 @@ v0/relationalai/semantics/reasoners/graph/core.py,sha256=Bth2oe5ulItBChZCoFTgP-o
 v0/relationalai/semantics/reasoners/optimization/__init__.py,sha256=HZlyy3Od7ArXI_HMhWrxrQPnyZGScgiVdmMul51lx7Q,2218
 v0/relationalai/semantics/reasoners/optimization/common.py,sha256=gmJrJAPNyaBqik2_QaxjCoFPZKqdroxmNXUeoDGwOtY,3130
 v0/relationalai/semantics/reasoners/optimization/solvers_dev.py,sha256=MqCvdDk1l_XSBlHd5-XpHVfcvD8INjCNbCttaomv6SY,24517
-v0/relationalai/semantics/reasoners/optimization/solvers_pb.py,sha256=1RobVkhUDP0OCRBS7f6MMM8wXQD56y5WKWZEO0fFB6k,48015
+v0/relationalai/semantics/reasoners/optimization/solvers_pb.py,sha256=MbubEIDJbXpe1gJ3lT_yBhVSDzA_EVuDvxRU4bLOjUk,60159
 v0/relationalai/semantics/rel/__init__.py,sha256=pMlVTC_TbQ45mP1LpzwFBBgPxpKc0H3uJDvvDXEWzvs,55
 v0/relationalai/semantics/rel/builtins.py,sha256=0M9r5GQb9CnkkbdhO8Psw9TgiZk5CC3z5RMncUa6uuM,1571
 v0/relationalai/semantics/rel/compiler.py,sha256=t7gitonfRewTk8IQ53DTqPegytwax1jGu43VgQOqBWA,43062
@@ -491,8 +496,8 @@ v0/relationalai/util/span_tracker.py,sha256=7vyvEBdI9uCj1dxVZCQT_W5vPMxIzxewrlRR
 v0/relationalai/util/spans_file_handler.py,sha256=w34WVw19ubtrjPBkKmg0Hb-9MldICmowF1OW85KCJN0,3209
 v0/relationalai/util/timeout.py,sha256=2o6BVNFnFc-B2j-i1pEkZcQbMRto9ps2emci0XwiA4I,783
 v0/relationalai/util/tracing_handler.py,sha256=wJQN52PRw8R2XB1_qd-POgfGZStrj9OFL7wa5Xr6SEM,1732
-relationalai-1.0.0a3.dist-info/METADATA,sha256=4hBJy9pOGv9uDUiDfIbGTGOX0M_N94_EptAIzzR7csc,1368
-relationalai-1.0.0a3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-relationalai-1.0.0a3.dist-info/entry_points.txt,sha256=u_anMN5_VCOQNA5E2mTLT2LWc9r1i-F9yQmy3ZNAAck,91
-relationalai-1.0.0a3.dist-info/top_level.txt,sha256=Y9cfzWf-p2omqqmVy_98m287xf0OJbj6OB5RRdkGql4,16
-relationalai-1.0.0a3.dist-info/RECORD,,
+relationalai-1.0.0a4.dist-info/METADATA,sha256=IBtzvgqfMTby0BJHn885Ub5UgZ9O3f_flCGqY84qCWM,1417
+relationalai-1.0.0a4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+relationalai-1.0.0a4.dist-info/entry_points.txt,sha256=u_anMN5_VCOQNA5E2mTLT2LWc9r1i-F9yQmy3ZNAAck,91
+relationalai-1.0.0a4.dist-info/top_level.txt,sha256=Y9cfzWf-p2omqqmVy_98m287xf0OJbj6OB5RRdkGql4,16
+relationalai-1.0.0a4.dist-info/RECORD,,

v0/relationalai/clients/exec_txn_poller.py

@@ -0,0 +1,91 @@
+from __future__ import annotations
+
+import time
+from typing import Dict, Optional, TYPE_CHECKING
+
+from v0.relationalai import debugging
+from v0.relationalai.clients.util import poll_with_specified_overhead
+from v0.relationalai.tools.cli_controls import create_progress
+from v0.relationalai.util.format import format_duration
+
+if TYPE_CHECKING:
+    from v0.relationalai.clients.resources.snowflake import Resources
+
+# Polling behavior constants
+POLL_OVERHEAD_RATE = 0.1  # Overhead rate for exponential backoff
+
+# Text color constants
+GREEN_COLOR = '\033[92m'
+GRAY_COLOR = '\033[90m'
+ENDC = '\033[0m'
+
+
+class ExecTxnPoller:
+    """
+    Encapsulates the polling logic for exec_async transaction completion.
+    """
+
+    def __init__(
+        self,
+        resource: "Resources",
+        txn_id: str,
+        headers: Optional[Dict] = None,
+        txn_start_time: Optional[float] = None,
+    ):
+        self.res = resource
+        self.txn_id = txn_id
+        self.headers = headers or {}
+        self.txn_start_time = txn_start_time or time.time()
+
+    def poll(self) -> bool:
+        """
+        Poll for transaction completion with interactive progress display.
+
+        Returns:
+            True if transaction completed successfully, False otherwise
+        """
+
+        # Don't show duration summary - we handle our own completion message
+        with create_progress(
+            description="Evaluating Query...",
+            success_message="",  # We'll handle this ourselves
+            leading_newline=False,
+            trailing_newline=False,
+            show_duration_summary=False,
+        ) as progress:
+            def check_status() -> bool:
+                """Check if transaction is complete."""
+                elapsed = time.time() - self.txn_start_time
+                # Update the main status with elapsed time
+                progress.update_main_status(
+                    query_progress_message(self.txn_id, elapsed)
+                )
+                return self.res._check_exec_async_status(self.txn_id, headers=self.headers)
+
+            with debugging.span("wait", txn_id=self.txn_id):
+                poll_with_specified_overhead(check_status, overhead_rate=POLL_OVERHEAD_RATE)
+
+            # Calculate final duration
+            total_duration = time.time() - self.txn_start_time
+
+            # Update to success message with duration
+            progress.update_main_status(
+                query_complete_message(self.txn_id, total_duration)
+            )
+
+        return True
+
+def query_progress_message(id: str, duration: float) -> str:
+    return (
+        # Print with whitespace to align with the end of the transaction ID
+        f"Evaluating Query... {format_duration(duration):>18}\n" +
+        f"{GRAY_COLOR}Query: {id}{ENDC}"
+    )
+
+def query_complete_message(id: str, duration: float, status_header: bool = False) -> str:
+    return (
+        (f"{GREEN_COLOR}✅ " if status_header else "") +
+        # Print with whitespace to align with the end of the transaction ID
+        f"Query Complete: {format_duration(duration):>24}\n" +
+        f"{GRAY_COLOR}Query: {id}{ENDC}"
+    )

v0/relationalai/clients/resources/snowflake/__init__.py

@@ -2,7 +2,7 @@
 Snowflake resources module.
 """
 # Import order matters - Resources must be imported first since other classes depend on it
-from .snowflake import Resources, Provider, Graph, SnowflakeClient, APP_NAME, PYREL_ROOT_DB, ExecContext, INTERNAL_ENGINE_SIZES, ENGINE_SIZES_AWS, ENGINE_SIZES_AZURE, PrimaryKey
+from .snowflake import Resources, Provider, Graph, SnowflakeClient, APP_NAME, PYREL_ROOT_DB, ExecContext, INTERNAL_ENGINE_SIZES, ENGINE_SIZES_AWS, ENGINE_SIZES_AZURE, PrimaryKey, PRINT_TXN_PROGRESS_FLAG
 
 # These imports depend on Resources, so they come after
 from .cli_resources import CLIResources
@@ -14,7 +14,7 @@ __all__ = [
     'Resources', 'DirectAccessResources', 'Provider', 'Graph', 'SnowflakeClient',
     'APP_NAME', 'PYREL_ROOT_DB', 'CLIResources', 'UseIndexResources', 'ExecContext',
     'INTERNAL_ENGINE_SIZES', 'ENGINE_SIZES_AWS', 'ENGINE_SIZES_AZURE', 'PrimaryKey',
-    'create_resources_instance',
+    'PRINT_TXN_PROGRESS_FLAG', 'create_resources_instance',
 ]