sqlsaber 0.29.0__py3-none-any.whl → 0.29.1__py3-none-any.whl

sqlsaber/application/auth_setup.py

@@ -104,7 +104,7 @@ async def setup_auth(
     """
     # Check if auth is already configured
     if auth_manager.has_auth_configured():
-        console.print("[green]✓ Authentication already configured![/green]")
+        console.print("[success]✓ Authentication already configured![/success]")
         return True, None
 
     # Select provider
@@ -137,7 +137,7 @@ async def setup_auth(
                 )
                 return True, provider
             else:
-                console.print("[red]✗ Anthropic OAuth setup failed.[/red]")
+                console.print("[error]✗ Anthropic OAuth setup failed.[/error]")
                 return False, None
 
     # API key flow
@@ -160,5 +160,5 @@ async def setup_auth(
         )
         return True, provider
     else:
-        console.print("[yellow]No API key provided.[/yellow]")
+        console.print("[warning]No API key provided.[/warning]")
         return False, None

sqlsaber/application/db_setup.py

@@ -88,7 +88,7 @@ async def collect_db_input(
         try:
             port = int(port_str)
         except ValueError:
-            console.print("[red]Invalid port number. Using default.[/red]")
+            console.print("[error]Invalid port number. Using default.[/error]")
             port = default_port
 
         database = await prompter.text("Database name:")
@@ -205,7 +205,7 @@ async def test_connection(config: DatabaseConfig, password: str | None) -> bool:
         await db_conn.close()
         return True
     except Exception as e:
-        console.print(f"[bold red]Connection failed:[/bold red] {e}", style="red")
+        console.print(f"[bold error]Connection failed:[/bold error] {e}", style="error")
         return False
 
 

sqlsaber/application/model_selection.py

@@ -34,7 +34,7 @@ async def choose_model(
         Selected model ID (provider:model_id) or None if cancelled
     """
     if not models:
-        console.print("[yellow]No models available[/yellow]")
+        console.print("[warning]No models available[/warning]")
         return None
 
     # Filter by provider if restricted
@@ -42,7 +42,7 @@ async def choose_model(
         models = [m for m in models if m.get("provider") == restrict_provider]
         if not models:
             console.print(
-                f"[yellow]No models available for {restrict_provider}[/yellow]"
+                f"[warning]No models available for {restrict_provider}[/warning]"
             )
             return None
 

sqlsaber/cli/auth.py

@@ -49,7 +49,7 @@ def setup():
     success, _ = asyncio.run(run_setup())
 
     if not success:
-        console.print("\n[yellow]No authentication configured.[/yellow]")
+        console.print("\n[warning]No authentication configured.[/warning]")
 
     console.print(
         "\nYou can change this anytime by running [cyan]saber auth setup[/cyan] again."
@@ -64,8 +64,10 @@ def status():
     console.print("\n[bold blue]Authentication Status[/bold blue]")
 
     if auth_method is None:
-        console.print("[yellow]No authentication method configured[/yellow]")
-        console.print("Run [cyan]saber auth setup[/cyan] to configure authentication.")
+        console.print("[warning]No authentication method configured[/warning]")
+        console.print(
+            "Run [primary]saber auth setup[/primary] to configure authentication."
+        )
         return
 
     # Show configured method summary
@@ -90,7 +92,7 @@ def status():
         elif from_keyring:
             console.print(f"> {provider}: [green]configured[/green]")
         else:
-            console.print(f"> {provider}: [yellow]not configured[/yellow]")
+            console.print(f"> {provider}: [warning]not configured[/warning]")
 
 
 @auth_app.command
@@ -105,7 +107,7 @@ def reset():
     ).ask()
 
     if provider is None:
-        console.print("[yellow]Reset cancelled.[/yellow]")
+        console.print("[warning]Reset cancelled.[/warning]")
         return
 
     api_key_manager = APIKeyManager()
@@ -121,7 +123,7 @@ def reset():
 
     if not api_key_present and not oauth_present:
         console.print(
-            f"[yellow]No stored credentials found for {provider}. Nothing to reset.[/yellow]"
+            f"[warning]No stored credentials found for {provider}. Nothing to reset.[/warning]"
         )
         return
 
@@ -153,7 +155,7 @@ def reset():
             # Already absent; treat as success
             pass
         except Exception as e:
-            console.print(f"Warning: Could not remove API key: {e}", style="yellow")
+            console.print(f"Warning: Could not remove API key: {e}", style="warning")
 
     # Optionally clear global auth method if removing Anthropic OAuth configuration
     if provider == "anthropic" and oauth_present:
@@ -169,7 +171,7 @@ def reset():
                 config_manager._save_config(config)
                 console.print("Global auth method unset.", style="green")
 
-    console.print("\n[bold green]✓ Reset complete.[/bold green]")
+    console.print("\n[success]✓ Reset complete.[/success]")
     console.print(
         "Environment variables are not modified by this command.", style="dim"
     )

sqlsaber/cli/commands.py

@@ -164,7 +164,7 @@ def query(
                 streaming_handler = StreamingQueryHandler(console)
                 db_type = sqlsaber_agent.db_type
                 console.print(
-                    f"[bold blue]Connected to:[/bold blue] {db_name} ({db_type})\n"
+                    f"[primary]Connected to:[/primary] {db_name} ({db_type})\n"
                 )
                 run = await streaming_handler.execute_streaming_query(
                     actual_query, sqlsaber_agent
@@ -200,13 +200,13 @@ def query(
         finally:
             # Clean up
             await db_conn.close()
-            console.print("\n[green]Goodbye![/green]")
+            console.print("\n[success]Goodbye![/success]")
 
     # Run the async function with proper error handling
     try:
         asyncio.run(run_session())
     except CLIError as e:
-        console.print(f"[bold red]Error:[/bold red] {e}")
+        console.print(f"[error]Error:[/error] {e}")
         sys.exit(e.exit_code)
 
 

sqlsaber/cli/database.py

@@ -95,7 +95,7 @@ def add(
         db_input = asyncio.run(collect_input())
 
         if db_input is None:
-            console.print("[yellow]Operation cancelled[/yellow]")
+            console.print("[warning]Operation cancelled[/warning]")
             return
 
         # Extract values from db_input
@@ -114,7 +114,7 @@ def add(
         if type == "sqlite":
             if not database:
                 console.print(
-                    "[bold red]Error:[/bold red] Database file path is required for SQLite"
+                    "[bold error]Error:[/bold error] Database file path is required for SQLite"
                 )
                 sys.exit(1)
             host = "localhost"
@@ -124,7 +124,7 @@ def add(
         elif type == "duckdb":
             if not database:
                 console.print(
-                    "[bold red]Error:[/bold red] Database file path is required for DuckDB"
+                    "[bold error]Error:[/bold error] Database file path is required for DuckDB"
                 )
                 sys.exit(1)
             database = str(Path(database).expanduser().resolve())
@@ -135,7 +135,7 @@ def add(
         else:
             if not all([host, database, username]):
                 console.print(
-                    "[bold red]Error:[/bold red] Host, database, and username are required"
+                    "[bold error]Error:[/bold error] Host, database, and username are required"
                 )
                 sys.exit(1)
 
@@ -179,7 +179,7 @@ def add(
             console.print(f"[blue]Set '{name}' as default database[/blue]")
 
     except Exception as e:
-        console.print(f"[bold red]Error adding database:[/bold red] {e}")
+        console.print(f"[bold error]Error adding database:[/bold error] {e}")
         sys.exit(1)
 
 
@@ -190,19 +190,19 @@ def list():
     default_name = config_manager.get_default_name()
 
     if not databases:
-        console.print("[yellow]No database connections configured[/yellow]")
+        console.print("[warning]No database connections configured[/warning]")
         console.print("Use 'sqlsaber db add <name>' to add a database connection")
         return
 
     table = Table(title="Database Connections")
     table.add_column("Name", style="cyan")
-    table.add_column("Type", style="magenta")
-    table.add_column("Host", style="green")
-    table.add_column("Port", style="yellow")
-    table.add_column("Database", style="blue")
-    table.add_column("Username", style="white")
-    table.add_column("SSL", style="bright_green")
-    table.add_column("Default", style="bold red")
+    table.add_column("Type", style="accent")
+    table.add_column("Host", style="success")
+    table.add_column("Port", style="warning")
+    table.add_column("Database", style="info")
+    table.add_column("Username", style="info")
+    table.add_column("SSL", style="success")
+    table.add_column("Default", style="error")
 
     for db in databases:
         is_default = "✓" if db.name == default_name else ""
@@ -239,7 +239,7 @@ def remove(
     """Remove a database connection."""
     if not config_manager.get_database(name):
         console.print(
-            f"[bold red]Error:[/bold red] Database connection '{name}' not found"
+            f"[bold error]Error:[/bold error] Database connection '{name}' not found"
         )
         sys.exit(1)
 
@@ -252,7 +252,7 @@ def remove(
             )
         else:
             console.print(
-                f"[bold red]Error:[/bold red] Failed to remove database connection '{name}'"
+                f"[bold error]Error:[/bold error] Failed to remove database connection '{name}'"
             )
             sys.exit(1)
     else:
@@ -269,14 +269,16 @@ def set_default(
     """Set the default database connection."""
     if not config_manager.get_database(name):
         console.print(
-            f"[bold red]Error:[/bold red] Database connection '{name}' not found"
+            f"[bold error]Error:[/bold error] Database connection '{name}' not found"
         )
         sys.exit(1)
 
     if config_manager.set_default_database(name):
         console.print(f"[green]Successfully set '{name}' as default database[/green]")
     else:
-        console.print(f"[bold red]Error:[/bold red] Failed to set '{name}' as default")
+        console.print(
+            f"[bold error]Error:[/bold error] Failed to set '{name}' as default"
+        )
         sys.exit(1)
 
 
@@ -299,14 +301,14 @@ def test(
             db_config = config_manager.get_database(name)
             if not db_config:
                 console.print(
-                    f"[bold red]Error:[/bold red] Database connection '{name}' not found"
+                    f"[bold error]Error:[/bold error] Database connection '{name}' not found"
                 )
                 sys.exit(1)
         else:
             db_config = config_manager.get_default_database()
             if not db_config:
                 console.print(
-                    "[bold red]Error:[/bold red] No default database configured"
+                    "[bold error]Error:[/bold error] No default database configured"
                 )
                 console.print(
                     "Use 'sqlsaber db add <name>' to add a database connection"
@@ -328,7 +330,7 @@ def test(
             )
 
         except Exception as e:
-            console.print(f"[bold red]✗ Connection failed:[/bold red] {e}")
+            console.print(f"[bold error]✗ Connection failed:[/bold error] {e}")
             sys.exit(1)
 
     asyncio.run(test_connection())

sqlsaber/cli/interactive.py

@@ -84,14 +84,14 @@ class InteractiveSession:
 
     def _banner(self) -> str:
         """Get the ASCII banner."""
-        return """
+        return """[primary]
 ███████  ██████  ██      ███████  █████  ██████  ███████ ██████
 ██      ██    ██ ██      ██      ██   ██ ██   ██ ██      ██   ██
 ███████ ██    ██ ██      ███████ ███████ ██████  █████   ██████
      ██ ██ ▄▄ ██ ██           ██ ██   ██ ██   ██ ██      ██   ██
 ███████  ██████  ███████ ███████ ██   ██ ██████  ███████ ██   ██
             ▀▀
-    """
+    [/primary]"""
 
     def _instructions(self) -> str:
         """Get the instruction text."""
@@ -123,12 +123,18 @@ class InteractiveSession:
     def show_welcome_message(self):
         """Display welcome message for interactive mode."""
         if self.first_message:
-            self.console.print(Panel.fit(self._banner()))
-            self.console.print(Markdown(self._instructions()))
+            self.console.print(Panel.fit(self._banner(), border_style="primary"))
+            self.console.print(
+                Markdown(
+                    self._instructions(),
+                    code_theme=self.tm.pygments_style_name,
+                    inline_code_theme=self.tm.pygments_style_name,
+                )
+            )
 
         db_name = self.database_name or "Unknown"
         self.console.print(
-            f"[heading]\n\nConnected to:[/heading] {db_name} ({self._db_type_name()})\n"
+            f"[heading]\nConnected to {db_name} ({self._db_type_name()})[/heading]\n"
         )
 
         if self._thread_id:

sqlsaber/cli/memory.py

@@ -29,7 +29,7 @@ def _get_database_name(database: str | None = None) -> str:
         db_config = config_manager.get_database(database)
         if not db_config:
             console.print(
-                f"[bold red]Error:[/bold red] Database connection '{database}' not found."
+                f"[bold error]Error:[/bold error] Database connection '{database}' not found."
             )
             sys.exit(1)
         return database
@@ -37,7 +37,7 @@ def _get_database_name(database: str | None = None) -> str:
         db_config = config_manager.get_default_database()
         if not db_config:
             console.print(
-                "[bold red]Error:[/bold red] No database connections configured."
+                "[bold error]Error:[/bold error] No database connections configured."
             )
             console.print("Use 'sqlsaber db add <name>' to add a database connection.")
             sys.exit(1)
@@ -64,7 +64,7 @@ def add(
         console.print(f"[dim]Memory ID:[/dim] {memory.id}")
         console.print(f"[dim]Content:[/dim] {memory.content}")
     except Exception as e:
-        console.print(f"[bold red]Error adding memory:[/bold red] {e}")
+        console.print(f"[bold error]Error adding memory:[/bold error] {e}")
         sys.exit(1)
 
 
@@ -85,7 +85,7 @@ def list(
 
     if not memories:
         console.print(
-            f"[yellow]No memories found for database '{database_name}'[/yellow]"
+            f"[warning]No memories found for database '{database_name}'[/warning]"
         )
         console.print("Use 'sqlsaber memory add \"<content>\"' to add memories")
         return
@@ -125,7 +125,7 @@ def show(
 
     if not memory:
         console.print(
-            f"[bold red]Error:[/bold red] Memory with ID '{memory_id}' not found for database '{database_name}'"
+            f"[bold error]Error:[/bold error] Memory with ID '{memory_id}' not found for database '{database_name}'"
         )
         sys.exit(1)
 
@@ -154,12 +154,12 @@ def remove(
     memory = memory_manager.get_memory_by_id(database_name, memory_id)
     if not memory:
         console.print(
-            f"[bold red]Error:[/bold red] Memory with ID '{memory_id}' not found for database '{database_name}'"
+            f"[bold error]Error:[/bold error] Memory with ID '{memory_id}' not found for database '{database_name}'"
         )
         sys.exit(1)
 
     # Show memory content before removal
-    console.print("[yellow]Removing memory:[/yellow]")
+    console.print("[warning]Removing memory:[/warning]")
     console.print(f"[dim]Content:[/dim] {memory.content}")
 
     if memory_manager.remove_memory(database_name, memory_id):
@@ -168,7 +168,7 @@ def remove(
         )
     else:
         console.print(
-            f"[bold red]Error:[/bold red] Failed to remove memory '{memory_id}'"
+            f"[bold error]Error:[/bold error] Failed to remove memory '{memory_id}'"
         )
         sys.exit(1)
 
@@ -198,14 +198,14 @@ def clear(
 
     if memories_count == 0:
         console.print(
-            f"[yellow]No memories to clear for database '{database_name}'[/yellow]"
+            f"[warning]No memories to clear for database '{database_name}'[/warning]"
         )
         return
 
     if not force:
         # Show confirmation
         console.print(
-            f"[yellow]About to clear {memories_count} memories for database '{database_name}'[/yellow]"
+            f"[warning]About to clear {memories_count} memories for database '{database_name}'[/warning]"
         )
 
         if not questionary.confirm("Are you sure you want to proceed?").ask():

sqlsaber/cli/models.py

@@ -106,7 +106,7 @@ class ModelManager:
                 results.sort(key=lambda x: (x["provider"], x["name"]))
                 return results
         except Exception as e:
-            console.print(f"[red]Error fetching models: {e}[/red]")
+            console.print(f"[error]Error fetching models: {e}[/error]")
             return []
 
     def get_current_model(self) -> str:
@@ -121,7 +121,7 @@ class ModelManager:
             config.set_model(model_id)
             return True
         except Exception as e:
-            console.print(f"[red]Error setting model: {e}[/red]")
+            console.print(f"[error]Error setting model: {e}[/error]")
             return False
 
     def reset_model(self) -> bool:
@@ -142,17 +142,17 @@ def list():
 
         if not models:
             console.print(
-                "[yellow]No models available or failed to fetch models[/yellow]"
+                "[warning]No models available or failed to fetch models[/warning]"
             )
             return
 
         table = Table(title="Available Models")
         table.add_column("Provider", style="magenta")
         table.add_column("ID", style="cyan")
-        table.add_column("Name", style="green")
-        table.add_column("Description", style="white")
-        table.add_column("Context", style="yellow", justify="right")
-        table.add_column("Current", style="bold red", justify="center")
+        table.add_column("Name", style="success")
+        table.add_column("Description", style="info")
+        table.add_column("Context", style="warning", justify="right")
+        table.add_column("Current", style="bold accent", justify="center")
 
         current_model = model_manager.get_current_model()
 
@@ -196,7 +196,7 @@ def set():
         models = await fetch_models(model_manager)
 
         if not models:
-            console.print("[red]Failed to fetch models. Cannot set model.[/red]")
+            console.print("[error]Failed to fetch models. Cannot set model.[/error]")
             sys.exit(1)
 
         prompter = AsyncPrompter()
@@ -208,10 +208,10 @@ def set():
             if model_manager.set_model(selected_model):
                 console.print(f"[green]✓ Model set to: {selected_model}[/green]")
             else:
-                console.print("[red]✗ Failed to set model[/red]")
+                console.print("[error]✗ Failed to set model[/error]")
                 sys.exit(1)
         else:
-            console.print("[yellow]Operation cancelled[/yellow]")
+            console.print("[warning]Operation cancelled[/warning]")
 
     asyncio.run(interactive_set())
 
@@ -236,10 +236,10 @@ def reset():
                     f"[green]✓ Model reset to default: {ModelManager.DEFAULT_MODEL}[/green]"
                 )
             else:
-                console.print("[red]✗ Failed to reset model[/red]")
+                console.print("[error]✗ Failed to reset model[/error]")
                 sys.exit(1)
         else:
-            console.print("[yellow]Operation cancelled[/yellow]")
+            console.print("[warning]Operation cancelled[/warning]")
 
     asyncio.run(interactive_reset())
 

sqlsaber/cli/onboarding.py

@@ -32,16 +32,16 @@ def needs_onboarding(database_arg: str | None = None) -> bool:
 
 def welcome_screen() -> None:
     """Display welcome screen to new users."""
-    banner = """
+    banner = """[primary]
 ███████  ██████  ██      ███████  █████  ██████  ███████ ██████
 ██      ██    ██ ██      ██      ██   ██ ██   ██ ██      ██   ██
 ███████ ██    ██ ██      ███████ ███████ ██████  █████   ██████
      ██ ██ ▄▄ ██ ██           ██ ██   ██ ██   ██ ██      ██   ██
 ███████  ██████  ███████ ███████ ██   ██ ██████  ███████ ██   ██
            ▀▀
-    """
+    [/primary]"""
 
-    console.print(Panel.fit(banner, style="bold blue"))
+    console.print(Panel.fit(banner, style="primary"))
     console.print()
 
     welcome_message = """
@@ -52,7 +52,9 @@ SQLsaber is an agentic SQL assistant that lets you query your database using nat
 Let's get you set up in just a few steps.
     """
 
-    console.print(Panel(welcome_message.strip(), border_style="blue", padding=(1, 2)))
+    console.print(
+        Panel.fit(welcome_message.strip(), border_style="primary", padding=(1, 2))
+    )
     console.print()
 
 
@@ -69,9 +71,7 @@ async def setup_database_guided() -> str | None:
     )
     from sqlsaber.application.prompts import AsyncPrompter
 
-    console.print("━" * 80, style="dim")
-    console.print("[bold cyan]Step 1 of 2: Database Connection[/bold cyan]")
-    console.print("━" * 80, style="dim")
+    console.print("[heading]Step 1 of 2: Database Connection[/heading]")
     console.print()
 
     try:
@@ -92,7 +92,7 @@ async def setup_database_guided() -> str | None:
         db_manager = DatabaseConfigManager()
         if db_manager.get_database(name):
             console.print(
-                f"[yellow]Database connection '{name}' already exists.[/yellow]"
+                f"[warning]Database connection '{name}' already exists.[/warning]"
             )
             return name
 
@@ -108,7 +108,7 @@ async def setup_database_guided() -> str | None:
         db_config = build_config(db_input)
 
         # Test the connection
-        console.print(f"[dim]Testing connection to '{name}'...[/dim]")
+        console.print(f"[muted]Testing connection to '{name}'...[/muted]")
         connection_success = await test_connection(db_config, db_input.password)
 
         if not connection_success:
@@ -119,25 +119,25 @@ async def setup_database_guided() -> str | None:
                 return await setup_database_guided()
             else:
                 console.print(
-                    "[yellow]You can add a database later using 'saber db add'[/yellow]"
+                    "[warning]You can add a database later using 'saber db add'[/warning]"
                 )
                 return None
 
         # Save the configuration
         try:
             save_database(db_manager, db_config, db_input.password)
-            console.print(f"[green]✓ Connection to '{name}' successful![/green]")
+            console.print(f"[success]✓ Connection to '{name}' successful![/success]")
             console.print()
             return name
         except Exception as e:
-            console.print(f"[bold red]Error saving database:[/bold red] {e}")
+            console.print(f"[error]Error saving database:[/error] {e}")
             return None
 
     except KeyboardInterrupt:
-        console.print("\n[yellow]Setup cancelled.[/yellow]")
+        console.print("\n[warning]Setup cancelled.[/warning]")
         return None
     except Exception as e:
-        console.print(f"[bold red]Unexpected error:[/bold red] {e}")
+        console.print(f"[error]Unexpected error:[/error] {e}")
         return None
 
 
@@ -151,14 +151,14 @@ async def select_model_for_provider(provider: str) -> str | None:
 
     try:
         console.print()
-        console.print(f"[dim]Fetching available {provider.title()} models...[/dim]")
+        console.print(f"[muted]Fetching available {provider.title()} models...[/muted]")
 
         model_manager = ModelManager()
         models = await fetch_models(model_manager, providers=[provider])
 
         if not models:
             console.print(
-                f"[yellow]Could not fetch models for {provider}. Using default.[/yellow]"
+                f"[warning]Could not fetch models for {provider}. Using default.[/warning]"
             )
             # Use provider-specific default or fallback to Anthropic
             default_model_id = ModelManager.RECOMMENDED_MODELS.get(
@@ -178,10 +178,10 @@ async def select_model_for_provider(provider: str) -> str | None:
         return selected_model
 
     except KeyboardInterrupt:
-        console.print("\n[yellow]Model selection cancelled.[/yellow]")
+        console.print("\n[warning]Model selection cancelled.[/warning]")
         return None
     except Exception as e:
-        console.print(f"[yellow]Error selecting model: {e}. Using default.[/yellow]")
+        console.print(f"[warning]Error selecting model: {e}. Using default.[/warning]")
         # Fallback to provider default
         if provider in ModelManager.RECOMMENDED_MODELS:
             return f"{provider}:{ModelManager.RECOMMENDED_MODELS[provider]}"
@@ -196,9 +196,7 @@ async def setup_auth_guided() -> tuple[bool, str | None]:
     from sqlsaber.application.auth_setup import setup_auth
     from sqlsaber.application.prompts import AsyncPrompter
 
-    console.print("━" * 80, style="dim")
-    console.print("[bold cyan]Step 2 of 2: Authentication[/bold cyan]")
-    console.print("━" * 80, style="dim")
+    console.print("[primary]Step 2 of 2: Authentication[/primary]")
     console.print()
 
     try:
@@ -218,7 +216,7 @@ async def setup_auth_guided() -> tuple[bool, str | None]:
 
         if not success:
             console.print(
-                "[yellow]You can set it up later using 'saber auth setup'[/yellow]"
+                "[warning]You can set it up later using 'saber auth setup'[/warning]"
             )
             console.print()
             return False, None
@@ -233,16 +231,16 @@ async def setup_auth_guided() -> tuple[bool, str | None]:
         if selected_model:
             model_manager = ModelManager()
             model_manager.set_model(selected_model)
-            console.print(f"[green]✓ Model set to: {selected_model}[/green]")
+            console.print(f"[success]✓ Model set to: {selected_model}[/success]")
         console.print()
         return True, selected_model
 
     except KeyboardInterrupt:
-        console.print("\n[yellow]Setup cancelled.[/yellow]")
+        console.print("\n[warning]Setup cancelled.[/warning]")
         console.print()
         return False, None
     except Exception as e:
-        console.print(f"[bold red]Unexpected error:[/bold red] {e}")
+        console.print(f"[error]Unexpected error:[/error] {e}")
         console.print()
         return False, None
 
@@ -251,35 +249,34 @@ def success_screen(
     database_name: str | None, auth_configured: bool, model_name: str | None = None
 ) -> None:
     """Display success screen after onboarding."""
-    console.print("━" * 80, style="dim")
-    console.print("[bold green]You're all set! 🚀[/bold green]")
-    console.print("━" * 80, style="dim")
+
+    console.print("[success]You're all set! 🚀[/success]")
     console.print()
 
     if database_name and auth_configured:
         console.print(
-            f"[green]✓ Database '{database_name}' connected and ready to use[/green]"
+            f"[success]✓ Database '{database_name}' connected and ready to use[/success]"
         )
-        console.print("[green]✓ Authentication configured[/green]")
+        console.print("[success]✓ Authentication configured[/success]")
         if model_name:
-            console.print(f"[green]✓ Model: {model_name}[/green]")
+            console.print(f"[success]✓ Model: {model_name}[/success]")
     elif database_name:
         console.print(
-            f"[green]✓ Database '{database_name}' connected and ready to use[/green]"
+            f"[success]✓ Database '{database_name}' connected and ready to use[/success]"
         )
         console.print(
-            "[yellow]⚠ AI authentication not configured - you'll be prompted when needed[/yellow]"
+            "[warning]⚠ AI authentication not configured - you'll be prompted when needed[/warning]"
         )
     elif auth_configured:
-        console.print("[green]✓ AI authentication configured[/green]")
+        console.print("[success]✓ AI authentication configured[/success]")
         if model_name:
-            console.print(f"[green]✓ Model: {model_name}[/green]")
+            console.print(f"[success]✓ Model: {model_name}[/success]")
         console.print(
-            "[yellow]⚠ No database configured - you'll need to provide one via -d flag[/yellow]"
+            "[warning]⚠ No database configured - you'll need to provide one via -d flag[/warning]"
         )
 
     console.print()
-    console.print("[dim]Starting interactive session...[/dim]")
+    console.print("[muted]Starting interactive session...[/muted]")
     console.print()
 
 
@@ -298,9 +295,9 @@ async def run_onboarding() -> bool:
 
         # If user cancelled database setup, exit
         if database_name is None:
-            console.print("[yellow]Database setup is required to continue.[/yellow]")
+            console.print("[warning]Database setup is required to continue.[/warning]")
             console.print(
-                "[dim]You can also provide a connection string using: saber -d <connection-string>[/dim]"
+                "[muted]You can also provide a connection string using: saber -d <connection-string>[/muted]"
             )
             return False
 
@@ -313,13 +310,13 @@ async def run_onboarding() -> bool:
         return True
 
     except KeyboardInterrupt:
-        console.print("\n[yellow]Onboarding cancelled.[/yellow]")
+        console.print("\n[warning]Onboarding cancelled.[/warning]")
         console.print(
-            "[dim]You can run setup commands manually:[/dim]\n"
-            "[dim]  - saber db add <name>  # Add database connection[/dim]\n"
-            "[dim]  - saber auth setup     # Configure authentication[/dim]"
+            "[muted]You can run setup commands manually:[/muted]\n"
+            "[muted]  - saber db add <name>  # Add database connection[/muted]\n"
+            "[muted]  - saber auth setup     # Configure authentication[/muted]"
         )
         sys.exit(0)
     except Exception as e:
-        console.print(f"[bold red]Onboarding failed:[/bold red] {e}")
+        console.print(f"[error]Onboarding failed:[/error] {e}")
         return False

sqlsaber/cli/threads.py

@@ -170,7 +170,7 @@ def _render_transcript(
                                     Panel.fit(
                                         content_str,
                                         title=f"Tool result: {name}",
-                                        border_style="yellow",
+                                        border_style="warning",
                                     )
                                 )
                     except Exception:
@@ -183,7 +183,7 @@ def _render_transcript(
                                 Panel.fit(
                                     content_str,
                                     title=f"Tool result: {name}",
-                                    border_style="yellow",
+                                    border_style="warning",
                                 )
                             )
                 else:
@@ -194,7 +194,7 @@ def _render_transcript(
                             Panel.fit(
                                 content_str,
                                 title=f"Tool result: {name}",
-                                border_style="yellow",
+                                border_style="warning",
                             )
                         )
         # Thinking parts omitted

sqlsaber/config/api_keys.py

@@ -36,7 +36,7 @@ class APIKeyManager:
                 return api_key
         except Exception as e:
             # Keyring access failed, continue to prompt
-            console.print(f"Keyring access failed: {e}", style="dim yellow")
+            console.print(f"Keyring access failed: {e}", style="muted warning")
 
         # 3. Prompt user for API key
         return self._prompt_and_store_key(provider, env_var_name, service_name)
@@ -72,7 +72,7 @@ class APIKeyManager:
             if not api_key.strip():
                 console.print(
                     "No API key provided. Some functionality may not work.",
-                    style="yellow",
+                    style="warning",
                 )
                 return None
 
@@ -83,16 +83,16 @@ class APIKeyManager:
             except Exception as e:
                 console.print(
                     f"Warning: Could not store API key in your operating system's credentials store: {e}",
-                    style="yellow",
+                    style="warning",
                 )
                 console.print(
-                    "You may need to enter it again next time", style="yellow"
+                    "You may need to enter it again next time", style="warning"
                 )
 
             return api_key.strip()
 
         except KeyboardInterrupt:
-            console.print("\nOperation cancelled", style="yellow")
+            console.print("\nOperation cancelled", style="warning")
             return None
         except Exception as e:
             console.print(f"Error prompting for API key: {e}", style="red")

sqlsaber/config/oauth_flow.py

@@ -131,7 +131,7 @@ class AnthropicOAuthFlow:
         if not questionary.confirm(
             "Continue with browser-based authentication?", default=True
         ).ask():
-            console.print("[yellow]Authentication cancelled.[/yellow]")
+            console.print("[warning]Authentication cancelled.[/warning]")
             return False
 
         try:
@@ -168,7 +168,7 @@ class AnthropicOAuthFlow:
             ).ask()
 
             if not auth_code:
-                console.print("[yellow]Authentication cancelled.[/yellow]")
+                console.print("[warning]Authentication cancelled.[/warning]")
                 return False
 
             # Step 2: Exchange code for tokens
@@ -198,23 +198,23 @@ class AnthropicOAuthFlow:
                 )
 
                 if self.token_manager.store_oauth_token("anthropic", oauth_token):
-                    console.print(
-                        "\n[bold green]✓ Authentication successful![/bold green]"
-                    )
+                    console.print("\n[success]✓ Authentication successful![/success]")
                     console.print(
                         "Your Claude Pro/Max subscription is now configured for SQLSaber."
                     )
                     return True
                 else:
-                    console.print("[red]✗ Failed to store authentication tokens.[/red]")
+                    console.print(
+                        "[error]✗ Failed to store authentication tokens.[/error]"
+                    )
                     return False
 
         except KeyboardInterrupt:
-            console.print("\n[yellow]Authentication cancelled by user.[/yellow]")
+            console.print("\n[warning]Authentication cancelled by user.[/warning]")
             return False
         except Exception as e:
             logger.error(f"OAuth authentication failed: {e}")
-            console.print(f"[red]✗ Authentication failed: {str(e)}[/red]")
+            console.print(f"[error]✗ Authentication failed: {str(e)}[/error]")
             return False
 
     def refresh_token_if_needed(self) -> OAuthToken | None:
@@ -255,13 +255,14 @@ class AnthropicOAuthFlow:
                 console.print("OAuth token refreshed successfully", style="green")
                 return refreshed_token
             else:
-                console.print("Failed to store refreshed token", style="yellow")
+                console.print("Failed to store refreshed token", style="warning")
                 return current_token
 
         except Exception as e:
             logger.warning(f"Token refresh failed: {e}")
             console.print(
-                "Token refresh failed. You may need to re-authenticate.", style="yellow"
+                "Token refresh failed. You may need to re-authenticate.",
+                style="warning",
             )
             return current_token
 

sqlsaber/config/oauth_tokens.py

@@ -97,14 +97,14 @@ class OAuthTokenManager:
             if token.is_expired():
                 console.print(
                     f"OAuth token for {provider} has expired and needs refresh",
-                    style="dim yellow",
+                    style="muted warning",
                 )
                 return token  # Return anyway for refresh attempt
 
             if token.expires_soon():
                 console.print(
                     f"OAuth token for {provider} expires soon, consider refreshing",
-                    style="dim yellow",
+                    style="muted warning",
                 )
 
             return token
@@ -126,7 +126,7 @@ class OAuthTokenManager:
             logger.error(f"Failed to store OAuth token for {provider}: {e}")
             console.print(
                 f"Warning: Could not store OAuth token in keyring: {e}",
-                style="yellow",
+                style="warning",
             )
             return False
 
@@ -137,7 +137,7 @@ class OAuthTokenManager:
         existing_token = self.get_oauth_token(provider)
         if not existing_token:
             console.print(
-                f"No existing OAuth token found for {provider}", style="yellow"
+                f"No existing OAuth token found for {provider}", style="warning"
             )
             return False
 
@@ -161,7 +161,9 @@ class OAuthTokenManager:
             return True
         except Exception as e:
             logger.error(f"Failed to remove OAuth token for {provider}: {e}")
-            console.print(f"Warning: Could not remove OAuth token: {e}", style="yellow")
+            console.print(
+                f"Warning: Could not remove OAuth token: {e}", style="warning"
+            )
             return False
 
     def has_oauth_token(self, provider: str) -> bool:

sqlsaber/database/schema.py

@@ -158,7 +158,7 @@ class SchemaManager:
             table["schema"] = table["table_schema"]
             table["type"] = table["table_type"]  # Map table_type to type for display
 
-        return {"tables": tables_list}
+        return {"tables": tables_list, "total_tables": len(tables_list)}
 
     async def close(self):
         """Close database connection."""

sqlsaber/theme/manager.py

@@ -18,14 +18,6 @@ from rich.theme import Theme
 DEFAULT_THEME_NAME = "nord"
 
 DEFAULT_ROLE_PALETTE = {
-    # base roles
-    "primary": "cyan",
-    "accent": "magenta",
-    "success": "green",
-    "warning": "yellow",
-    "error": "red",
-    "info": "cyan",
-    "muted": "dim",
     # components
     "table.header": "bold $primary",
     "panel.border.user": "$info",

{sqlsaber-0.29.0.dist-info → sqlsaber-0.29.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlsaber
-Version: 0.29.0
+Version: 0.29.1
 Summary: SQLsaber - Open-source agentic SQL assistant
 License-File: LICENSE
 Requires-Python: >=3.12

{sqlsaber-0.29.0.dist-info → sqlsaber-0.29.1.dist-info}/RECORD

@@ -5,29 +5,29 @@ sqlsaber/agents/base.py,sha256=40-MKEoz5rGrqVIylV1U2DaAUSPFcC75ohRin4E3-kk,2668
 sqlsaber/agents/mcp.py,sha256=Pn8tdDRUEVLYQyEi5nHRp9MKNePwHVVoeNI-uqWcr0Y,757
 sqlsaber/agents/pydantic_ai_agent.py,sha256=wBxKz0pjOkL-HI-TXV6B67bczZNgu7k26Rr3w5usR3o,10064
 sqlsaber/application/__init__.py,sha256=KY_-d5nEdQyAwNOsK5r-f7Tb69c63XbuEkHPeLpJal8,84
-sqlsaber/application/auth_setup.py,sha256=cbS2VTwL7mwm24UFRz84PhC13XMzC1F7JkN-ze7tApY,5104
-sqlsaber/application/db_setup.py,sha256=qtMxCd_KO7GsD4W_iRBpDRvLriiyvOXPvZdcvm6KVDM,6849
-sqlsaber/application/model_selection.py,sha256=xZI-nvUgYZikaTK38SCmEWvWSfRsDpFu2VthbVdI95g,3187
+sqlsaber/application/auth_setup.py,sha256=D94dyU9bOVfnNHLnnFJb5PaeWsKPTL21CiS_DLcY93A,5114
+sqlsaber/application/db_setup.py,sha256=ZSgR9rJJVHttIjsbYQS9GEIyzkM09k5RLrVGdegrfYc,6859
+sqlsaber/application/model_selection.py,sha256=fSC06MZNKinHDR-csMFVYYJFyK8MydKf6pStof74Jp0,3191
 sqlsaber/application/prompts.py,sha256=4rMGcWpYJbNWPMzqVWseUMx0nwvXOkWS6GaTAJ5mhfc,3473
 sqlsaber/cli/__init__.py,sha256=qVSLVJLLJYzoC6aj6y9MFrzZvAwc4_OgxU9DlkQnZ4M,86
-sqlsaber/cli/auth.py,sha256=ysDBXEFR8Jz7wYbIP6X7yWA2ivd8SDnUp_jUg_qYNWk,6088
-sqlsaber/cli/commands.py,sha256=LiHLMAZF9fNgQXShtF67dzDvp-w4LlSbmeZIa47ntC0,8634
+sqlsaber/cli/auth.py,sha256=elUpw8gypHGlxbHx4a4_z4wFznx2vr6V1h8lqpeC6OQ,6121
+sqlsaber/cli/commands.py,sha256=Pii_SlVKpNEtt57_QPQzwC1u-x6tA8kuG8yd43undWE,8628
 sqlsaber/cli/completers.py,sha256=g-hLDq5fiBx7gg8Bte1Lq8GU-ZxCYVs4dcPsmHPIcK4,6574
-sqlsaber/cli/database.py,sha256=hh8PdWnhaD0fO2jwvSSQyxsjwk-JyvmcY7f5tuHfnAQ,10663
+sqlsaber/cli/database.py,sha256=Tqy8H5MnjsrmOSPcbA5Qy-u-IOYJCIXRJVhk0veLNDk,10726
 sqlsaber/cli/display.py,sha256=WB5JCumhXadziDEX1EZHG3vN1Chol5FNAaTXHieqFK0,17892
-sqlsaber/cli/interactive.py,sha256=u1gvdMZJkFWelZQ2urOS6-EH1uShKF4u_edfr_BzDNk,13479
-sqlsaber/cli/memory.py,sha256=gKP-JJ0w1ya1YTM_Lk7Gw-7wL9ptyj6cZtg-uoW8K7A,7818
-sqlsaber/cli/models.py,sha256=NozZbnisSjbPKo7PW7CltJMIkGcPqTDpDQEY-C_eLhk,8504
-sqlsaber/cli/onboarding.py,sha256=l6FFWn8J1OVQUxr-xIAzKaFhAz8rFh6IEWwIyPWqR6U,11438
+sqlsaber/cli/interactive.py,sha256=PcY6mszImo_3PsqjjWmx_cOfj44OmKvD9ENOvGA-wjU,13715
+sqlsaber/cli/memory.py,sha256=IKq09DUbqpvvtATsyDlpm7rDlGqWEhdUX9wgnR-oiq4,7850
+sqlsaber/cli/models.py,sha256=iMTw3DCKIQbWfSfikwy-nQ5tiHWjoN4cbKA-G6RFlj4,8535
+sqlsaber/cli/onboarding.py,sha256=iBGT-W-OJFRvQoEpuHYyO1c9Mym5c97eIefRvxGHtTg,11265
 sqlsaber/cli/streaming.py,sha256=1XoZGPPMoTmBQVgp_Bqk483MR93j9oXxSV6Tx_-TpOg,6923
 sqlsaber/cli/theme.py,sha256=hP0kmsMLCtqaT7b5wB1dk1hW1hV94oP4BHdz8S6887A,4243
-sqlsaber/cli/threads.py,sha256=5EV4ckRzKqhWeTKpTfQSNCBuqs3onsJURKT09g4E4XM,13634
+sqlsaber/cli/threads.py,sha256=o9q9Hst1Wt7cxSyrpAtwG6pkUct6csgiAmN_0P_WO3k,13637
 sqlsaber/config/__init__.py,sha256=olwC45k8Nc61yK0WmPUk7XHdbsZH9HuUAbwnmKe3IgA,100
-sqlsaber/config/api_keys.py,sha256=bjogRmIuxNNGusyKXKi0ZpJWeS5Fyn53zrAD8hsoYx4,3671
+sqlsaber/config/api_keys.py,sha256=9RyhD5Bntq8NMFRPiZZo8YEHACK9MPyFGp8dsmQZ1iI,3678
 sqlsaber/config/auth.py,sha256=b5qB2h1doXyO9Bn8z0CcL8LAR2jF431gGXBGKLgTmtQ,2756
 sqlsaber/config/database.py,sha256=Yec6_0wdzq-ADblMNnbgvouYCimYOY_DWHT9oweaISc,11449
-sqlsaber/config/oauth_flow.py,sha256=VNbq4TZPk0hVJIcOh7JUO5qSxJnNzqDj0vwjCn-59Ok,10316
-sqlsaber/config/oauth_tokens.py,sha256=SC-lXVcKCV7uiWtBiU2mxvx1z7ryW8tOSKHBApPsXtE,5931
+sqlsaber/config/oauth_flow.py,sha256=P81lHhtICdhiQu8lNwyqn2m45FGEqCEzLgUQTLG5UW0,10343
+sqlsaber/config/oauth_tokens.py,sha256=V4U8GAQHjTfgUcTzwjRVaIE7DeN0tF9OsSjiasHw7Uc,5970
 sqlsaber/config/providers.py,sha256=JFjeJv1K5Q93zWSlWq3hAvgch1TlgoF0qFa0KJROkKY,2957
 sqlsaber/config/settings.py,sha256=iB4CnGQ4hw8gxkaa9CVLB_JEy6Y9h9FQTAams5OCVyI,6421
 sqlsaber/database/__init__.py,sha256=Gi9N_NOkD459WRWXDg3hSuGoBs3xWbMDRBvsTVmnGAg,2025
@@ -37,7 +37,7 @@ sqlsaber/database/duckdb.py,sha256=8HNKdx208aFK_YtwGjLz6LTne0xEmNevD-f9dRWlrFg,1
 sqlsaber/database/mysql.py,sha256=wMzDQqq4GFbfEdqXtv_sCb4Qbr9GSWqYAvOLeo5UryY,14472
 sqlsaber/database/postgresql.py,sha256=fuf2Wl29NKXvD3mqsR08PDleNQ1PG-fNvWSxT6HDh2M,13223
 sqlsaber/database/resolver.py,sha256=wSCcn__aCqwIfpt_LCjtW2Zgb8RpG5PlmwwZHli1q_U,3628
-sqlsaber/database/schema.py,sha256=68PrNcA-5eR9PZB3i-TUQw5_E7QatwiDU2wv9GgXgM4,6928
+sqlsaber/database/schema.py,sha256=CuV0ewoVaERe1gj_fJFJFWAP8aEPgepmn6X6B7bgkfQ,6962
 sqlsaber/database/sqlite.py,sha256=iReEIiSpkhhS1VzITd79ZWqSL3fHMyfe3DRCDpM0DvE,9421
 sqlsaber/mcp/__init__.py,sha256=COdWq7wauPBp5Ew8tfZItFzbcLDSEkHBJSMhxzy8C9c,112
 sqlsaber/mcp/mcp.py,sha256=tpNPHpkaCre1Xjp7c4DHXbTKeuYpDQ8qhmJZvAyr7Vk,3939
@@ -45,7 +45,7 @@ sqlsaber/memory/__init__.py,sha256=GiWkU6f6YYVV0EvvXDmFWe_CxarmDCql05t70MkTEWs,6
 sqlsaber/memory/manager.py,sha256=p3fybMVfH-E4ApT1ZRZUnQIWSk9dkfUPCyfkmA0HALs,2739
 sqlsaber/memory/storage.py,sha256=ne8szLlGj5NELheqLnI7zu21V8YS4rtpYGGC7tOmi-s,5745
 sqlsaber/theme/__init__.py,sha256=qCICX1Cg4B6yCbZ1UrerxglWxcqldRFVSRrSs73na_8,188
-sqlsaber/theme/manager.py,sha256=7QXoqPvl-aKfvqTRduWYwnHsySu66Gg8a3-QQkVM5Ss,6551
+sqlsaber/theme/manager.py,sha256=xOi1ZLNr31ZtE8-WzxdtvywCB1rbJ7z2DTzcqHsCavY,6377
 sqlsaber/threads/__init__.py,sha256=Hh3dIG1tuC8fXprREUpslCIgPYz8_6o7aRLx4yNeO48,139
 sqlsaber/threads/storage.py,sha256=rsUdxT4CR52D7xtGir9UlsFnBMk11jZeflzDrk2q4ME,11183
 sqlsaber/tools/__init__.py,sha256=x3YdmX_7P0Qq_HtZHAgfIVKTLxYqKk6oc4tGsujQWsc,586
@@ -55,8 +55,8 @@ sqlsaber/tools/instructions.py,sha256=X-x8maVkkyi16b6Tl0hcAFgjiYceZaSwyWTfmrvx8U
 sqlsaber/tools/registry.py,sha256=HWOQMsNIdL4XZS6TeNUyrL-5KoSDH6PHsWd3X66o-18,3211
 sqlsaber/tools/sql_guard.py,sha256=dTDwcZP-N4xPGzcr7MQtKUxKrlDzlc1irr9aH5a4wvk,6182
 sqlsaber/tools/sql_tools.py,sha256=ujmAcfLkNaBrb5LWEgWcINQEQSX0LRPX3VK5Dag1Sj4,9178
-sqlsaber-0.29.0.dist-info/METADATA,sha256=5Ocm_GD3MIdPLji_VpknTjiQ0ndDd2waD14dr6_IOjg,7174
-sqlsaber-0.29.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sqlsaber-0.29.0.dist-info/entry_points.txt,sha256=qEbOB7OffXPFgyJc7qEIJlMEX5RN9xdzLmWZa91zCQQ,162
-sqlsaber-0.29.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sqlsaber-0.29.0.dist-info/RECORD,,
+sqlsaber-0.29.1.dist-info/METADATA,sha256=kx_nQDGhcDUJp4DadT7Pi89QsUuf8S5W6IVqwaJfuXM,7174
+sqlsaber-0.29.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sqlsaber-0.29.1.dist-info/entry_points.txt,sha256=qEbOB7OffXPFgyJc7qEIJlMEX5RN9xdzLmWZa91zCQQ,162
+sqlsaber-0.29.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sqlsaber-0.29.1.dist-info/RECORD,,