sqlsaber 0.28.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

@@ -11,6 +11,7 @@ from sqlsaber.cli.database import create_db_app
 from sqlsaber.cli.memory import create_memory_app
 from sqlsaber.cli.models import create_models_app
 from sqlsaber.cli.onboarding import needs_onboarding, run_onboarding
+from sqlsaber.cli.theme import create_theme_app
 from sqlsaber.cli.threads import create_threads_app
 
 # Lazy imports - only import what's needed for CLI parsing
@@ -35,6 +36,7 @@ app.command(create_auth_app(), name="auth")
 app.command(create_db_app(), name="db")
 app.command(create_memory_app(), name="memory")
 app.command(create_models_app(), name="models")
+app.command(create_theme_app(), name="theme")
 app.command(create_threads_app(), name="threads")
 
 console = create_console()
@@ -162,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
@@ -198,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())