snowglobe-cli 0.1.8__tar.gz → 0.2.0__tar.gz

{snowglobe_cli-0.1.8 → snowglobe_cli-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowglobe-cli
-Version: 0.1.8
+Version: 0.2.0
 Summary: Explainable cost and access visibility for Snowflake
 Author-email: Jaryd Thornton <jaryd90@gmail.com>
 License-Expression: Apache-2.0

{snowglobe_cli-0.1.8 → snowglobe_cli-0.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "snowglobe-cli"
-version = "0.1.8"
+version = "0.2.0"
 description = "Explainable cost and access visibility for Snowflake"
 readme = "README.md"
 requires-python = ">=3.12.3"

{snowglobe_cli-0.1.8 → snowglobe_cli-0.2.0}/snowglobe/cli/optimizer.py

@@ -51,7 +51,7 @@ def query(
     if not no_ai:
         typer.echo("\nGenerating AI suggestions...")
         ai = optimizer_service.ai_suggestion(model=model)
-        typer.echo(ai)
+        cli.print_ai_suggestion(ai)
 
 
 @opt_app.command()

{snowglobe_cli-0.1.8 → snowglobe_cli-0.2.0}/snowglobe/cli/shell.py

@@ -1321,6 +1321,13 @@ def _cmd_optimize(ctx: SnowglobeContext, args: list):
     opt_exp = optimizer_service.expensive_operators()
     typer.echo(cli.format_expensive_operators(opt_exp))
 
+    typer.echo("\nGenerating AI suggestions (Cortex)…")
+    try:
+        ai = optimizer_service.ai_suggestion()
+        cli.print_ai_suggestion(ai)
+    except Exception as e:
+        typer.secho(f"  AI suggestion failed: {e}", fg=typer.colors.YELLOW)
+
 
 def _cmd_status(ctx: SnowglobeContext, args: list):
     """Show current shell working state."""

{snowglobe_cli-0.1.8 → snowglobe_cli-0.2.0}/snowglobe/core/cost_service.py

@@ -830,26 +830,36 @@ class CostService:
 
     # --- Budget status (Snowflake-native budgets) ---
 
-    def get_budget_status(self) -> tuple[pd.DataFrame, str | None]:
+    def get_budget_status(self, days: int = 30) -> tuple[pd.DataFrame, str | None]:
         """
         Surface Snowflake-native budget status.
-        Calls the account root budget spending history and lists custom budgets.
+        GET_SPENDING_HISTORY is a stored procedure, not a table function — must use CALL.
         Returns (df, error_message) — error_message is set if budgets are not activated.
         """
         conn = self.context.connect()
         try:
             with conn:
-                # Get spending history from account budget
-                rows = conn.query("""
-                    SELECT *
-                    FROM TABLE(SNOWFLAKE.LOCAL.ACCOUNT_ROOT_BUDGET!GET_SPENDING_HISTORY())
+                rows = conn.query(f"""
+                    CALL SNOWFLAKE.LOCAL.ACCOUNT_ROOT_BUDGET!GET_SPENDING_HISTORY(
+                        TIME_LOWER_BOUND => DATEADD('days', -{int(days)}, CURRENT_TIMESTAMP()),
+                        TIME_UPPER_BOUND => CURRENT_TIMESTAMP()
+                    )
                 """)
             df = pd.DataFrame(rows)
             return df, None
         except Exception as e:
             err_msg = str(e)
-            if "not activated" in err_msg.lower() or "does not exist" in err_msg.lower():
-                return pd.DataFrame(), "Budgets are not activated on this account. Use Snowsight or CALL SNOWFLAKE.LOCAL.ACCOUNT_ROOT_BUDGET!ACTIVATE() to enable."
+            not_activated = (
+                "ACCOUNT_ROOT_BUDGET_NOT_ACTIVATED" in err_msg
+                or "not activated" in err_msg.lower()
+                or "does not exist" in err_msg.lower()
+            )
+            if not_activated:
+                return pd.DataFrame(), (
+                    "Snowflake budgets are not activated on this account.\n\n"
+                    "To enable, run the following in a Snowsight worksheet as ACCOUNTADMIN:\n\n"
+                    "  CALL SNOWFLAKE.LOCAL.ACCOUNT_ROOT_BUDGET!ACTIVATE();"
+                )
             return pd.DataFrame(), f"Could not retrieve budget status: {err_msg}"
 
     # --- Replication costs ---
@@ -941,3 +951,23 @@ class CostService:
         except Exception:
             # View may not exist if no MVs are used
             return pd.DataFrame(), None
+
+    def get_day_detail(self, date: str) -> pd.DataFrame:
+        """Warehouse credit breakdown for a single calendar day."""
+        conn = self.context.connect()
+        with conn:
+            rows = conn.query(f"""
+                SELECT WAREHOUSE_NAME,
+                       ROUND(SUM(CREDITS_USED), 4) AS CREDITS
+                FROM SNOWFLAKE.ACCOUNT_USAGE.WAREHOUSE_METERING_HISTORY
+                WHERE START_TIME::DATE = '{date}'
+                  AND WAREHOUSE_NAME IS NOT NULL
+                GROUP BY 1
+                ORDER BY 2 DESC
+                LIMIT 30
+            """)
+        if not rows:
+            return pd.DataFrame(columns=["WAREHOUSE_NAME", "CREDITS"])
+        df = pd.DataFrame(rows)
+        df["CREDITS"] = df["CREDITS"].astype(float)
+        return df

{snowglobe_cli-0.1.8 → snowglobe_cli-0.2.0}/snowglobe/core/report_service.py

@@ -37,8 +37,15 @@ class ReportService:
         ai_df, _, _ = self.cost_service.get_ai_costs(days)
         ai_total = float(ai_df["TOTAL_CREDITS"].sum()) if not ai_df.empty else 0
 
-        # Storage
+        # AI costs by user — exclude zero-cost users
+        ai_users_df, _, _ = self.cost_service.get_ai_costs_by_user(days)
+        if not ai_users_df.empty:
+            ai_users_df = ai_users_df[ai_users_df["TOTAL_CREDITS"] > 0].reset_index(drop=True)
+
+        # Storage — exclude databases with no stored data
         storage_df, _ = self.cost_service.get_storage_usage(days)
+        if not storage_df.empty:
+            storage_df = storage_df[storage_df["TOTAL_TB"] > 0].reset_index(drop=True)
         storage_total_tb = float(storage_df["TOTAL_TB"].sum()) if not storage_df.empty else 0
         storage_total_cost = float(storage_df["EST_MONTHLY_COST"].sum()) if not storage_df.empty else 0
         storage_rate = self.cost_service.get_storage_rate()
@@ -56,6 +63,7 @@ class ReportService:
             "cost_total": cost_total,
             "ai_costs": ai_df.to_dict("records") if not ai_df.empty else [],
             "ai_total": ai_total,
+            "ai_users": ai_users_df.to_dict("records") if not ai_users_df.empty else [],
             "storage": storage_df.to_dict("records") if not storage_df.empty else [],
             "storage_total_tb": storage_total_tb,
             "storage_total_cost": storage_total_cost,

{snowglobe_cli-0.1.8 → snowglobe_cli-0.2.0}/snowglobe/output/cli.py

@@ -311,6 +311,57 @@ def format_query_insights(query_id, insights):
     return output
 
 
+def format_ai_suggestion(result) -> str:
+    """
+    Convert the Cortex AI result (dict or string) to readable Markdown.
+    Safe to pass to a Textual Markdown widget or print as plain text.
+    """
+    import json
+
+    if isinstance(result, str):
+        try:
+            result = json.loads(result)
+        except (json.JSONDecodeError, ValueError):
+            return result  # plain message — "No AI suggestions", error text, etc.
+
+    if not isinstance(result, dict):
+        return str(result)
+
+    lines = ["## AI Optimization Suggestions", ""]
+
+    summary = result.get("summary", "")
+    if summary:
+        lines += ["**Summary**", "", summary, ""]
+
+    for i, opt in enumerate(result.get("optimizations", []), 1):
+        problem = opt.get("problem", "")
+        solution = opt.get("solution", "")
+        explanation = opt.get("explanation", "")
+        sql = opt.get("sql", "").strip()
+
+        lines += ["---", "", f"### {i}. {problem}", ""]
+        if solution:
+            lines += [f"**Solution:** {solution}", ""]
+        if explanation:
+            lines += [f"**Explanation:** {explanation}", ""]
+        if sql:
+            lines += ["```sql", sql, "```", ""]
+
+    improvement = result.get("expected_improvement", "")
+    if improvement:
+        lines += ["---", "", f"**Expected Improvement:** {improvement}"]
+
+    return "\n".join(lines)
+
+
+def print_ai_suggestion(result) -> None:
+    """Render the AI suggestion as formatted Markdown in the terminal via Rich."""
+    from rich.console import Console
+    from rich.markdown import Markdown as RichMarkdown
+
+    Console().print(RichMarkdown(format_ai_suggestion(result)))
+
+
 def format_drift_text(drift: dict) -> str:
     """Format access drift detection results."""
     if drift.get("error"):

{snowglobe_cli-0.1.8 → snowglobe_cli-0.2.0}/snowglobe/templates/report.md.j2

@@ -30,6 +30,18 @@
 
 ---
 
+## AI Costs by User
+
+| User | Functions | Analyst | Agent | Code | Intelligence | Total |
+|---|---:|---:|---:|---:|---:|---:|
+{% for row in ai_users -%}
+| {{ row.USER_NAME }} | {{ "%.2f"|format(row.CORTEX_FUNCTIONS) }} | {{ "%.2f"|format(row.CORTEX_ANALYST) }} | {{ "%.2f"|format(row.CORTEX_AGENT) }} | {{ "%.2f"|format(row.CORTEX_CODE) }} | {{ "%.2f"|format(row.SNOWFLAKE_INTELLIGENCE) }} | {{ "%.2f"|format(row.TOTAL_CREDITS) }} |
+{% else -%}
+*No AI usage in this period.*
+{% endfor -%}
+
+---
+
 ## Storage
 
 | Database | Size (TB) | Est. Monthly Cost |

{snowglobe_cli-0.1.8 → snowglobe_cli-0.2.0}/snowglobe/tui/screens/cost.py

@@ -38,11 +38,12 @@ class CostScreen(Vertical):
 
     # Active view: "summary" | "trend" | "top_queries" | "warehouses" | "users"
     # | "ai" | "ai_users" | "services" | "storage" | "replication" | "mv" | "budget"
-    # | "drill_service" | "drill_warehouse" | "drill_user"
+    # | "drill_service" | "drill_warehouse" | "drill_user" | "drill_day"
     _current_view: str = "summary"
     _drill_service: str | None = None
     _drill_warehouse: str | None = None
     _drill_user: str | None = None
+    _drill_day: str | None = None
     # Becomes True after the user explicitly picks a view, so Select.Changed
     # events fired during the initial mount don't auto-trigger a Snowflake fetch.
     _user_initiated: bool = False
@@ -136,6 +137,9 @@ class CostScreen(Vertical):
         elif self._current_view == "users":
             self._drill_user = key
             self._fetch_user_drill(key)
+        elif self._current_view == "trend":
+            self._drill_day = key
+            self._fetch_day_drill(key)
         elif self._current_view == "top_queries":
             # Hand off to Tune via the App's content switcher.
             try:
@@ -179,6 +183,8 @@ class CostScreen(Vertical):
             self._fetch_warehouse_drill(self._drill_warehouse)
         elif self._current_view == "drill_user" and self._drill_user:
             self._fetch_user_drill(self._drill_user)
+        elif self._current_view == "drill_day" and self._drill_day:
+            self._fetch_day_drill(self._drill_day)
 
     def action_back_from_drill(self) -> None:
         if self._current_view == "drill_service":
@@ -190,6 +196,9 @@ class CostScreen(Vertical):
         elif self._current_view == "drill_user":
             self._drill_user = None
             self._fetch_users()
+        elif self._current_view == "drill_day":
+            self._drill_day = None
+            self._fetch_trend()
 
     # --- Days helper --------------------------------------------------
 
@@ -268,7 +277,7 @@ class CostScreen(Vertical):
         avg = float(df["CREDITS"].mean())
         suffix = f"  ·  cached {cache_age}m ago" if cache_age else ""
         self._set_status(
-            f"Daily trend — {days}d  ·  total {total:,.2f}  ·  avg/day {avg:,.2f}{suffix}"
+            f"Daily trend — {days}d  ·  total {total:,.2f}  ·  avg/day {avg:,.2f}{suffix}  ·  click a date to drill in"
         )
 
         table = self.query_one(DataTable)
@@ -536,6 +545,44 @@ class CostScreen(Vertical):
                 avg_str,
             )
 
+    # --- Drill: day detail -------------------------------------------
+
+    def _fetch_day_drill(self, date: str) -> None:
+        self._current_view = "drill_day"
+        self._set_status(f"Loading breakdown for {date}…  Esc to return")
+        self._day_drill_worker(date=date)
+
+    @work(thread=True, exclusive=True, group="cost")
+    def _day_drill_worker(self, *, date: str) -> None:
+        try:
+            df = self.app.get_cost_service().get_day_detail(date)
+        except Exception as e:
+            self.app.call_from_thread(self._fetch_failed, e)
+            return
+        self.app.call_from_thread(self._render_day_drill, df, date)
+
+    def _render_day_drill(self, df: pd.DataFrame, date: str) -> None:
+        table = self._reset_table()
+        if df is None or df.empty:
+            table.add_columns("(no warehouse activity on this date)")
+            self._set_status(f"{date}  ·  no data  ·  Esc to return")
+            return
+        total = float(df["CREDITS"].sum())
+        max_c = float(df["CREDITS"].max()) if total > 0 else 1.0
+        table.add_columns("WAREHOUSE", "CREDITS", "% OF DAY", "BAR")
+        for _, row in df.iterrows():
+            credits = float(row["CREDITS"])
+            pct = (credits / total * 100) if total > 0 else 0
+            table.add_row(
+                str(row["WAREHOUSE_NAME"]),
+                f"{credits:,.4f}",
+                f"{pct:.1f}%",
+                _bar(credits / max_c),
+            )
+        self._set_status(
+            f"Day detail — {date}  ·  {total:,.4f} credits  ·  Esc to return"
+        )
+
     # --- View 6: AI services -----------------------------------------
 
     def _fetch_ai(self, force: bool = False) -> None:

{snowglobe_cli-0.1.8 → snowglobe_cli-0.2.0}/snowglobe/tui/screens/home.py

@@ -1,13 +1,8 @@
 """Home — landing dashboard.
 
-Three KPI cards (Cache / Connection / This week) at the top, quick-action
-hotkeys + recent expensive queries DataTable below. KPI data comes from
-local SQLite (instant); recent queries fire one `get_top_queries(7, 5)`
-worker on mount.
-
-Hotkeys jump to common workflows:
-  a  Access check        c  Cost summary        r  Refresh cache
-  w  Who-access          s  Risk scan
+Three KPI cards (Cache / Connection / This week) at the top.
+Below: 30d spend trend (left) and recent expensive queries (right).
+Both panels fire Snowflake workers on mount; KPI data is local/instant.
 """
 from datetime import datetime, timezone
 
@@ -48,16 +43,10 @@ class HomeScreen(Vertical):
                 yield Static("loading…",   id="kpi-week", classes="kpi-body")
 
         with Horizontal(id="home-body"):
-            with Vertical(id="home-actions", classes="panel"):
-                yield Static("Quick actions", classes="panel-title")
-                yield Static(
-                    "[a]  Access check\n"
-                    "[w]  Who-access\n"
-                    "[c]  Cost\n"
-                    "\\[s]  Risk scan\n"
-                    "\\[r]  Refresh\n",
-                    classes="actions-list",
-                )
+            with Vertical(id="home-trend", classes="panel"):
+                yield Static("30d spend trend", classes="panel-title")
+                yield DataTable(id="home-trend-table",
+                                cursor_type="none", zebra_stripes=True)
             with Vertical(id="home-queries", classes="panel"):
                 yield Static("Recent expensive queries (7d)", classes="panel-title")
                 yield DataTable(id="home-queries-table",
@@ -69,7 +58,8 @@ class HomeScreen(Vertical):
         self._refresh_kpis()
         # Re-render KPIs whenever the app-level cache age ticks (every 30s).
         self.watch(self.app, "cache_age_minutes", lambda _v: self._refresh_kpis())
-        # Fire one query fetch when Home first mounts.
+        # Fire both Snowflake workers on mount (different groups → concurrent).
+        self._fetch_trend()
         self._fetch_queries()
 
     # --- Hotkey action -----------------------------------------------
@@ -172,6 +162,39 @@ class HomeScreen(Vertical):
         )
         self.query_one("#kpi-week", Static).update(text)
 
+    # --- 30d spend trend (one Snowflake call on mount) ---------------
+
+    @work(thread=True, exclusive=True, group="trend")
+    def _fetch_trend(self) -> None:
+        try:
+            df, _ = self.app.get_cost_service().get_daily_trend(days=30)
+        except Exception as e:
+            self.app.call_from_thread(self._trend_failed, e)
+            return
+        self.app.call_from_thread(self._render_trend, df)
+
+    def _trend_failed(self, err: Exception) -> None:
+        table = self.query_one("#home-trend-table", DataTable)
+        table.clear(columns=True)
+        table.add_columns(f"(could not load — {err})")
+
+    def _render_trend(self, df: pd.DataFrame) -> None:
+        table = self.query_one("#home-trend-table", DataTable)
+        table.clear(columns=True)
+        if df is None or df.empty:
+            table.add_columns("(no trend data)")
+            return
+        table.add_columns("DATE", "CREDITS", "TREND")
+        max_c = float(df["CREDITS"].max()) if not df.empty else 1.0
+        for _, row in df.iterrows():
+            credits = float(row["CREDITS"])
+            bar_width = max(1, int(credits / max_c * 20)) if max_c > 0 else 0
+            table.add_row(
+                str(row["DATE"]),
+                f"{credits:,.2f}",
+                "█" * bar_width,
+            )
+
     # --- Recent expensive queries (one Snowflake call on mount) -----
 
     @work(thread=True, exclusive=True, group="cost")

{snowglobe_cli-0.1.8 → snowglobe_cli-0.2.0}/snowglobe/tui/screens/reports.py

@@ -14,7 +14,7 @@ from typing import Any
 
 from textual import work
 from textual.app import ComposeResult
-from textual.containers import Horizontal, Vertical
+from textual.containers import Horizontal, Vertical, VerticalScroll
 from textual.suggester import SuggestFromList
 from textual.widgets import Button, Input, Markdown, Select, Static
 
@@ -50,11 +50,11 @@ class ReportsScreen(Vertical):
                 yield Select(_TYPES, value="full", id="rp-type",
                              classes="form-select", allow_blank=False)
 
-            with Horizontal(classes="form-row"):
+            with Horizontal(classes="form-row", id="rp-days-row"):
                 yield Static("Window:", classes="form-label")
                 yield Input(value="30", id="rp-days", classes="form-input")
 
-            with Horizontal(classes="form-row"):
+            with Horizontal(classes="form-row", id="rp-top-row"):
                 yield Static("Top queries:", classes="form-label")
                 yield Input(value="10", id="rp-top", classes="form-input")
 
@@ -76,15 +76,16 @@ class ReportsScreen(Vertical):
             "Preview will appear here after Generate. Save writes to the output path.",
             id="rp-status", classes="hint",
         )
-        yield Markdown(
-            "No report yet — fill the form and click Generate.",
-            id="rp-preview",
-        )
+        with VerticalScroll(id="rp-preview-scroll"):
+            yield Markdown(
+                "No report yet — fill the form and click Generate.",
+                id="rp-preview",
+            )
 
     # --- Lifecycle ----------------------------------------------------
 
     def on_mount(self) -> None:
-        # Username suggester from user_graph (only useful for user reports).
+        self._update_form_visibility("full")
         self.watch(self.app, "user_graph",
                    lambda _v: self._refresh_user_suggester())
 
@@ -103,9 +104,16 @@ class ReportsScreen(Vertical):
 
     def on_select_changed(self, event: Select.Changed) -> None:
         if event.select.id == "rp-type":
-            # Update the default output path to match the type
-            new_path = _default_path(str(event.value))
-            self.query_one("#rp-path", Input).value = new_path
+            rtype = str(event.value)
+            self.query_one("#rp-path", Input).value = _default_path(rtype)
+            self._update_form_visibility(rtype)
+
+    def _update_form_visibility(self, rtype: str) -> None:
+        is_user = rtype == "user"
+        is_full = rtype == "full"
+        self.query_one("#rp-days-row").display = not is_user
+        self.query_one("#rp-top-row").display = is_full
+        self.query_one("#rp-user-row").display = is_user
 
     def on_button_pressed(self, event: Button.Pressed) -> None:
         if event.button.id == "rp-generate":

{snowglobe_cli-0.1.8 → snowglobe_cli-0.2.0}/snowglobe/tui/screens/tune.py

@@ -17,7 +17,7 @@ from rich.syntax import Syntax
 
 from textual import work
 from textual.app import ComposeResult
-from textual.containers import Horizontal, Vertical
+from textual.containers import Horizontal, Vertical, VerticalScroll
 from textual.widgets import (
     Button, DataTable, Input, Markdown, Static, TabbedContent, TabPane, Tree,
 )
@@ -46,13 +46,15 @@ class TuneScreen(Vertical):
         with Horizontal(id="tu-body"):
             with Vertical(id="tu-sql-pane", classes="panel"):
                 yield Static("SQL", classes="panel-title")
-                yield Static("Run an analysis to see SQL here.",
-                             id="tu-sql", classes="sql-pane")
+                with VerticalScroll(id="tu-sql-scroll"):
+                    yield Static("Run an analysis to see SQL here.",
+                                 id="tu-sql", classes="sql-pane")
             with Vertical(id="tu-analysis-pane", classes="panel"):
                 yield Static("Analysis", classes="panel-title")
                 with TabbedContent(initial="tu-tab-heuristics"):
                     with TabPane("Heuristics", id="tu-tab-heuristics"):
-                        yield Static("Pending.", id="tu-heuristics")
+                        with VerticalScroll(id="tu-heuristics-scroll"):
+                            yield Static("Pending.", id="tu-heuristics")
                     with TabPane("Insights", id="tu-tab-insights"):
                         yield DataTable(id="tu-insights-table",
                                         cursor_type="row", zebra_stripes=True)
@@ -65,11 +67,12 @@ class TuneScreen(Vertical):
                         with Horizontal(classes="actions-row"):
                             yield Button("Generate AI suggestion (Cortex)",
                                          id="tu-ai-run", variant="warning")
-                        yield Markdown(
-                            "AI suggestions will appear here after you Analyse "
-                            "and then click Generate.",
-                            id="tu-ai",
-                        )
+                        with VerticalScroll(id="tu-ai-scroll"):
+                            yield Markdown(
+                                "AI suggestions will appear here after you Analyse "
+                                "and then click Generate.",
+                                id="tu-ai",
+                            )
 
     # --- Events -------------------------------------------------------
 
@@ -249,6 +252,7 @@ class TuneScreen(Vertical):
         self.query_one("#tu-ai-run", Button).disabled = False
         self.app.notify(f"AI generation failed: {err}", severity="error", timeout=8)
 
-    def _render_ai(self, text: str) -> None:
-        self.query_one("#tu-ai", Markdown).update(text or "(empty AI response)")
+    def _render_ai(self, text) -> None:
+        from snowglobe.output.cli import format_ai_suggestion
+        self.query_one("#tu-ai", Markdown).update(format_ai_suggestion(text) or "(empty AI response)")
         self.query_one("#tu-ai-run", Button).disabled = False

{snowglobe_cli-0.1.8 → snowglobe_cli-0.2.0}/snowglobe_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowglobe-cli
-Version: 0.1.8
+Version: 0.2.0
 Summary: Explainable cost and access visibility for Snowflake
 Author-email: Jaryd Thornton <jaryd90@gmail.com>
 License-Expression: Apache-2.0