sqlsaber 0.24.0__tar.gz → 0.25.0__tar.gz

{sqlsaber-0.24.0 → sqlsaber-0.25.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlsaber
-Version: 0.24.0
+Version: 0.25.0
 Summary: SQLsaber - Open-source agentic SQL assistant
 License-File: LICENSE
 Requires-Python: >=3.12
@@ -8,10 +8,10 @@ Requires-Dist: aiomysql>=0.2.0
 Requires-Dist: aiosqlite>=0.21.0
 Requires-Dist: asyncpg>=0.30.0
 Requires-Dist: cyclopts>=3.22.1
+Requires-Dist: duckdb>=0.9.2
 Requires-Dist: fastmcp>=2.9.0
 Requires-Dist: httpx>=0.28.1
 Requires-Dist: keyring>=25.6.0
-Requires-Dist: pandas>=2.0.0
 Requires-Dist: platformdirs>=4.0.0
 Requires-Dist: prompt-toolkit>3.0.51
 Requires-Dist: pydantic-ai
@@ -58,7 +58,7 @@ Ask your questions in natural language and `sqlsaber` will gather the right cont
 - 🧠 Memory management
 - 💬 Interactive REPL mode
 - 🧵 Conversation threads (store, display, and resume conversations)
-- 🗄️ Support for PostgreSQL, SQLite, and MySQL
+- 🗄️ Support for PostgreSQL, MySQL, SQLite, and DuckDB
 - 🔌 MCP (Model Context Protocol) server support
 - 🎨 Beautiful formatted output
 
@@ -170,6 +170,7 @@ saber -d mydb "count all orders"
 
 # You can also pass a connection string
 saber -d "postgresql://user:password@localhost:5432/mydb" "count all orders"
+saber -d "duckdb:///path/to/data.duckdb" "top customers"
 ```
 
 ## Examples

{sqlsaber-0.24.0 → sqlsaber-0.25.0}/README.md

@@ -37,7 +37,7 @@ Ask your questions in natural language and `sqlsaber` will gather the right cont
 - 🧠 Memory management
 - 💬 Interactive REPL mode
 - 🧵 Conversation threads (store, display, and resume conversations)
-- 🗄️ Support for PostgreSQL, SQLite, and MySQL
+- 🗄️ Support for PostgreSQL, MySQL, SQLite, and DuckDB
 - 🔌 MCP (Model Context Protocol) server support
 - 🎨 Beautiful formatted output
 
@@ -149,6 +149,7 @@ saber -d mydb "count all orders"
 
 # You can also pass a connection string
 saber -d "postgresql://user:password@localhost:5432/mydb" "count all orders"
+saber -d "duckdb:///path/to/data.duckdb" "top customers"
 ```
 
 ## Examples

{sqlsaber-0.24.0 → sqlsaber-0.25.0}/docs/src/content/docs/changelog.md

@@ -7,6 +7,14 @@ All notable changes to SQLsaber will be documented here.
 
 ### Unreleased
 
+### v0.25.0 - 2025-09-26
+
+#### Added
+
+- DuckDB support for efficient CSV and data file analysis
+  - Added `duckdb` provider to database options
+  - Enhanced CSV processing capabilities using DuckDB's optimized engine
+
 ### v0.24.0 - 2025-09-24
 
 #### Added

{sqlsaber-0.24.0 → sqlsaber-0.25.0}/docs/src/content/docs/guides/database-setup.mdx

@@ -5,13 +5,14 @@ description: Configure database connections in SQLsaber
 
 import { Aside } from '@astrojs/starlight/components';
 
-SQLsaber supports PostgreSQL, MySQL, and SQLite databases as well as CSV files. This guide covers all the ways to configure database connections.
+SQLsaber supports PostgreSQL, MySQL, SQLite, DuckDB, and CSV data sources. This guide covers all the ways to configure database connections.
 
 ### Supported Database Types
 
 - **PostgreSQL** - Full support with SSL
 - **MySQL** - Full support with SSL
 - **SQLite** - Local database files
+- **DuckDB** - Local DuckDB files or in-memory analytics
 - **CSV** - Local CSV files
 
 ### Adding Database
@@ -24,12 +25,14 @@ This will prompt you for all necessary connection details.
 
 ### Connection Strings
 
-SQLsaber supports direct connection strings for PostgreSQL and MySQL.
+SQLsaber supports direct connection strings for PostgreSQL and MySQL, plus file URIs for SQLite and DuckDB.
 
 ```bash
 saber -d "postgresql://user:password@localhost:5432/database"
 
 saber -d "mysql://user:password@localhost:3306/database"
+
+saber -d "duckdb:///path/to/data.duckdb"
 ```
 
 ### Files
@@ -44,6 +47,16 @@ saber -d "./data/sales.db" "Show me total sales by month"
 saber -d "~/Documents/mydata.db" "Count all records"
 ```
 
+#### DuckDB
+
+```bash
+# Query a DuckDB file
+saber -d "./warehouse/data.duckdb" "Show the latest partition"
+
+# Use an absolute path
+saber -d "duckdb:///Users/me/analytics.duckdb" "Describe orders"
+```
+
 #### CSV
 
 SQLsaber can also analyze CSV files directly.

{sqlsaber-0.24.0 → sqlsaber-0.25.0}/docs/src/content/docs/guides/getting-started.mdx

@@ -10,7 +10,7 @@ Welcome to SQLsaber! This guide will walk you through setting up SQLsaber and ru
 Before you begin, make sure you have:
 
 1. [Installed SQLsaber](/installation)
-2. Access to a database (PostgreSQL, MySQL, or SQLite)
+2. Access to a database (PostgreSQL, MySQL, SQLite, or DuckDB)
 3. An API key from a supported AI provider (Anthropic, OpenAI, Google, etc.)
 
 ### Quick Setup
@@ -54,6 +54,7 @@ This interactive command will ask you for:
   - PostgreSQL
   - MySQL
   - SQLite
+  - DuckDB
 - Connection details (host, port, database name, username)
 - SSL configuration (if needed)
 

{sqlsaber-0.24.0 → sqlsaber-0.25.0}/docs/src/content/docs/guides/queries.mdx

@@ -109,12 +109,15 @@ saber -d prod-db "What's our revenue this month?"
 
 #### Using Files Directly
 
-Work with SQLite or CSV files directly:
+Work with SQLite, DuckDB, or CSV files directly:
 
 ```bash
 # SQLite file
 saber -d "./data/sales.db" "Top selling products"
 
+# DuckDB file
+saber -d "./data/warehouse.duckdb" "Latest partitions"
+
 # CSV file
 saber -d "./customers.csv" "How many customers per state?"
 ```

{sqlsaber-0.24.0 → sqlsaber-0.25.0}/docs/src/content/docs/index.mdx

@@ -57,7 +57,7 @@ import {
   </p>
   <li class="flex items-center gap-3">
     <Icon name="seti:db" />
-    <span>PostgreSQL, MySQL, SQLite, CSV support</span>
+    <span>PostgreSQL, MySQL, SQLite, DuckDB, CSV support</span>
   </li>
   <li class="flex items-center gap-3">
     <Icon name="forward-slash" />

{sqlsaber-0.24.0 → sqlsaber-0.25.0}/docs/src/content/docs/reference/commands.md

@@ -10,6 +10,7 @@ This is a comprehensive reference for all SQLsaber commands and their options.
 The main SQLsaber command for running queries.
 
 **Usage:**
+
 ```bash
 # Interactive mode (default)
 saber
@@ -25,10 +26,12 @@ saber -d "postgresql://user:pass@host:5432/db" "User statistics for 2024"
 ```
 
 **Parameters:**
+
 - `QUERY-TEXT` - SQL query in natural language (optional, starts interactive mode if not provided)
-- `-d, --database` - Database connection name, file path (CSV/SQLite), or connection string
+- `-d, --database` - Database connection name, file path (CSV/SQLite/DuckDB), or connection string (postgresql://, mysql://, duckdb://)
 
 **Global Options:**
+
 - `--help, -h` - Display help message
 - `--version` - Show version information
 
@@ -43,6 +46,7 @@ Manage authentication configuration for AI providers.
 Configure authentication for SQLsaber (API keys and OAuth).
 
 **Usage:**
+
 ```bash
 saber auth setup
 ```
@@ -52,11 +56,13 @@ saber auth setup
 Check current authentication configuration.
 
 **Usage:**
+
 ```bash
 saber auth status
 ```
 
 **Output shows:**
+
 - Configured providers
 - Authentication methods (API key vs OAuth)
 
@@ -65,6 +71,7 @@ saber auth status
 Remove stored credentials for a provider.
 
 **Usage:**
+
 ```bash
 saber auth reset
 ```
@@ -80,15 +87,18 @@ Manage database connections.
 Add a new database connection.
 
 **Usage:**
+
 ```bash
 saber db add my-database [OPTIONS]
 ```
 
 **Parameters:**
+
 - `NAME` - Name for the database connection (required)
 
 **Options:**
-- `-t, --type` - Database type: `postgresql`, `mysql`, `sqlite` (default: postgresql)
+
+- `-t, --type` - Database type: `postgresql`, `mysql`, `sqlite`, `duckdb` (default: postgresql)
 - `-h, --host` - Database host
 - `-p, --port` - Database port
 - `--database, --db` - Database name
@@ -101,7 +111,8 @@ saber db add my-database [OPTIONS]
 
 **SSL Modes:**
 
-*PostgreSQL:*
+_PostgreSQL:_
+
 - `disable` - No SSL
 - `allow` - Try SSL, fallback to non-SSL
 - `prefer` - Try SSL first (default)
@@ -109,24 +120,26 @@ saber db add my-database [OPTIONS]
 - `verify-ca` - Require SSL and verify certificate
 - `verify-full` - Require SSL, verify certificate and hostname
 
-*MySQL:*
+_MySQL:_
+
 - `DISABLED` - No SSL
 - `PREFERRED` - Try SSL first (default)
 - `REQUIRED` - Require SSL
 - `VERIFY_CA` - Require SSL and verify certificate
 - `VERIFY_IDENTITY` - Require SSL, verify certificate and hostname
 
-
 #### `saber db list`
 
 List all configured database connections.
 
 **Usage:**
+
 ```bash
 saber db list
 ```
 
 **Output shows:**
+
 - Database names
 - Connection details (host, port, database)
 - Default database indicator
@@ -136,6 +149,7 @@ saber db list
 Set a database as the default connection.
 
 **Usage:**
+
 ```bash
 saber db set-default my-database
 ```
@@ -145,11 +159,13 @@ saber db set-default my-database
 Test a database connection.
 
 **Usage:**
+
 ```bash
 saber db test my-database
 ```
 
 **Output:**
+
 - Connection success/failure
 - Error details if connection fails
 
@@ -158,6 +174,7 @@ saber db test my-database
 Remove a database connection.
 
 **Usage:**
+
 ```bash
 saber db remove my-database
 ```
@@ -175,17 +192,21 @@ Manage database-specific memories and context.
 Add a new memory entry.
 
 **Usage:**
+
 ```bash
 saber memory add "Memory content here" [OPTIONS]
 ```
 
 **Parameters:**
+
 - `CONTENT` - Memory content to add (required)
 
 **Options:**
+
 - `-d, --database` - Database connection name (uses default if not specified)
 
 **Examples:**
+
 ```bash
 # Add memory to default database
 saber memory add "Active customers are those who made a purchase in the last 90 days"
@@ -205,14 +226,17 @@ saber memory add "Always format dates as YYYY-MM-DD for reports"
 List all memory entries for a database.
 
 **Usage:**
+
 ```bash
 saber memory list [OPTIONS]
 ```
 
 **Options:**
+
 - `-d, --database` - Database connection name (uses default if not specified)
 
 **Output shows:**
+
 - Memory ID
 - Memory content
 - Creation timestamp
@@ -222,11 +246,13 @@ saber memory list [OPTIONS]
 Remove a specific memory entry.
 
 **Usage:**
+
 ```bash
 saber memory remove a1b2c3d4
 ```
 
 **Parameters:**
+
 - `ID` - Memory ID from `saber memory list` output
 
 #### `saber memory clear`
@@ -234,11 +260,13 @@ saber memory remove a1b2c3d4
 Remove all memory entries for a database.
 
 **Usage:**
+
 ```bash
 saber memory clear [OPTIONS]
 ```
 
 **Options:**
+
 - `-d, --database` - Database connection name (uses default if not specified)
 
 **Confirmation required** - Will prompt before deletion.
@@ -254,6 +282,7 @@ Manage LLM models from different providers.
 List all available models for configured providers.
 
 **Usage:**
+
 ```bash
 saber models list
 ```
@@ -263,6 +292,7 @@ saber models list
 Set the default model.
 
 **Usage:**
+
 ```bash
 saber models set
 ```
@@ -272,6 +302,7 @@ saber models set
 Reset to the default model (Claude Sonnet 4).
 
 **Usage:**
+
 ```bash
 saber models reset
 ```
@@ -287,28 +318,32 @@ Manage conversation threads.
 List conversation threads.
 
 **Usage:**
+
 ```bash
 saber threads list [OPTIONS]
 ```
 
 **Options:**
+
 - `-d, --database` - Filter by database name
 - `-n, --limit` - Maximum threads to return (default: 50)
 
-
 #### `saber threads show`
 
 Show complete thread transcript.
 
 **Usage:**
+
 ```bash
 saber threads show a1b2c3d4
 ```
 
 **Parameters:**
+
 - `THREAD_ID` - Thread ID from `saber threads list`
 
 **Output shows:**
+
 - Thread metadata (database, model, timestamps)
 - Complete conversation history
 - SQL queries and results
@@ -319,17 +354,21 @@ saber threads show a1b2c3d4
 Resume an existing conversation thread.
 
 **Usage:**
+
 ```bash
 saber threads resume a1b2c3d4 [OPTIONS]
 ```
 
 **Parameters:**
+
 - `THREAD_ID` - Thread ID to resume
 
 **Options:**
+
 - `-d, --database` - Use different database than original thread
 
 **Features:**
+
 - Loads full conversation context
 - Uses same model as original thread
 - Connects to original database
@@ -340,21 +379,23 @@ saber threads resume a1b2c3d4 [OPTIONS]
 Clean up old conversation threads.
 
 **Usage:**
+
 ```bash
 saber threads prune
 ```
 
 ---
 
-
 ### Interactive Mode
 
 When in interactive mode (`saber` with no arguments), you have access to a few additional features:
 
 #### Slash Commands
+
 - `/clear` - Clear conversation history
 - `/exit` - Exit SQLsaber
 - `/quit` - Exit SQLsaber (alias for `/exit`)
 
 #### Autocomplete
+
 - **Table names** - Type `@table_name[TAB]` for completions

{sqlsaber-0.24.0 → sqlsaber-0.25.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sqlsaber"
-version = "0.24.0"
+version = "0.25.0"
 description = "SQLsaber - Open-source agentic SQL assistant"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -14,7 +14,7 @@ dependencies = [
     "httpx>=0.28.1",
     "aiomysql>=0.2.0",
     "aiosqlite>=0.21.0",
-    "pandas>=2.0.0",
+    "duckdb>=0.9.2",
     "fastmcp>=2.9.0",
     "cyclopts>=3.22.1",
     "prompt-toolkit>3.0.51",

{sqlsaber-0.24.0 → sqlsaber-0.25.0}/src/sqlsaber/agents/base.py

@@ -8,6 +8,7 @@ from typing import Any, AsyncIterator
 from sqlsaber.database.connection import (
     BaseDatabaseConnection,
     CSVConnection,
+    DuckDBConnection,
     MySQLConnection,
     PostgreSQLConnection,
     SQLiteConnection,
@@ -51,7 +52,9 @@ class BaseSQLAgent(ABC):
         elif isinstance(self.db, SQLiteConnection):
             return "SQLite"
         elif isinstance(self.db, CSVConnection):
-            return "SQLite"  # we convert csv to in-memory sqlite
+            return "DuckDB"
+        elif isinstance(self.db, DuckDBConnection):
+            return "DuckDB"
         else:
             return "database"  # Fallback
 

{sqlsaber-0.24.0 → sqlsaber-0.25.0}/src/sqlsaber/agents/pydantic_ai_agent.py

@@ -17,6 +17,7 @@ from sqlsaber.config.settings import Config
 from sqlsaber.database.connection import (
     BaseDatabaseConnection,
     CSVConnection,
+    DuckDBConnection,
     MySQLConnection,
     PostgreSQLConnection,
     SQLiteConnection,
@@ -169,7 +170,9 @@ def _get_database_type_name(db: BaseDatabaseConnection) -> str:
         return "MySQL"
     elif isinstance(db, SQLiteConnection):
         return "SQLite"
+    elif isinstance(db, DuckDBConnection):
+        return "DuckDB"
     elif isinstance(db, CSVConnection):
-        return "SQLite"
+        return "DuckDB"
     else:
         return "database"

{sqlsaber-0.24.0 → sqlsaber-0.25.0}/src/sqlsaber/cli/commands.py

@@ -46,7 +46,7 @@ def meta_handler(
         str | None,
         cyclopts.Parameter(
             ["--database", "-d"],
-            help="Database connection name, file path (CSV/SQLite), or connection string (postgresql://, mysql://) (uses default if not specified)",
+            help="Database connection name, file path (CSV/SQLite/DuckDB), or connection string (postgresql://, mysql://, duckdb://) (uses default if not specified)",
         ),
     ] = None,
 ):
@@ -59,8 +59,10 @@ def meta_handler(
         saber -d mydb "show me users"          # Run a query with specific database
         saber -d data.csv "show me users"      # Run a query with ad-hoc CSV file
         saber -d data.db "show me users"       # Run a query with ad-hoc SQLite file
+        saber -d data.duckdb "show me users"   # Run a query with ad-hoc DuckDB file
         saber -d "postgresql://user:pass@host:5432/db" "show users"  # PostgreSQL connection string
         saber -d "mysql://user:pass@host:3306/db" "show users"       # MySQL connection string
+        saber -d "duckdb:///data.duckdb" "show users"                 # DuckDB connection string
         echo "show me all users" | saber       # Read query from stdin
         cat query.txt | saber                  # Read query from file via stdin
     """
@@ -80,7 +82,7 @@ def query(
         str | None,
         cyclopts.Parameter(
             ["--database", "-d"],
-            help="Database connection name, file path (CSV/SQLite), or connection string (postgresql://, mysql://) (uses default if not specified)",
+            help="Database connection name, file path (CSV/SQLite/DuckDB), or connection string (postgresql://, mysql://, duckdb://) (uses default if not specified)",
         ),
     ] = None,
 ):
@@ -97,8 +99,10 @@ def query(
         saber "show me all users"         # Run a single query
         saber -d data.csv "show users"    # Run a query with ad-hoc CSV file
         saber -d data.db "show users"     # Run a query with ad-hoc SQLite file
+        saber -d data.duckdb "show users" # Run a query with ad-hoc DuckDB file
         saber -d "postgresql://user:pass@host:5432/db" "show users"  # PostgreSQL connection string
         saber -d "mysql://user:pass@host:3306/db" "show users"       # MySQL connection string
+        saber -d "duckdb:///data.duckdb" "show users"                 # DuckDB connection string
         echo "show me all users" | saber  # Read query from stdin
     """
 
@@ -111,6 +115,7 @@ def query(
         from sqlsaber.database.connection import (
             CSVConnection,
             DatabaseConnection,
+            DuckDBConnection,
             MySQLConnection,
             PostgreSQLConnection,
             SQLiteConnection,
@@ -149,15 +154,18 @@ def query(
                 # Single query mode with streaming
                 streaming_handler = StreamingQueryHandler(console)
                 # Compute DB type for the greeting line
-                db_type = (
-                    "PostgreSQL"
-                    if isinstance(db_conn, PostgreSQLConnection)
-                    else "MySQL"
-                    if isinstance(db_conn, MySQLConnection)
-                    else "SQLite"
-                    if isinstance(db_conn, (SQLiteConnection, CSVConnection))
-                    else "database"
-                )
+                if isinstance(db_conn, PostgreSQLConnection):
+                    db_type = "PostgreSQL"
+                elif isinstance(db_conn, MySQLConnection):
+                    db_type = "MySQL"
+                elif isinstance(db_conn, DuckDBConnection):
+                    db_type = "DuckDB"
+                elif isinstance(db_conn, SQLiteConnection):
+                    db_type = "SQLite"
+                elif isinstance(db_conn, CSVConnection):
+                    db_type = "DuckDB"
+                else:
+                    db_type = "database"
                 console.print(
                     f"[bold blue]Connected to:[/bold blue] {db_name} ({db_type})\n"
                 )

{sqlsaber-0.24.0 → sqlsaber-0.25.0}/src/sqlsaber/cli/database.py

@@ -31,7 +31,7 @@ def add(
         str,
         cyclopts.Parameter(
             ["--type", "-t"],
-            help="Database type (postgresql, mysql, sqlite)",
+            help="Database type (postgresql, mysql, sqlite, duckdb)",
         ),
     ] = "postgresql",
     host: Annotated[
@@ -87,17 +87,17 @@ def add(
         if not type or type == "postgresql":
             type = questionary.select(
                 "Database type:",
-                choices=["postgresql", "mysql", "sqlite"],
+                choices=["postgresql", "mysql", "sqlite", "duckdb"],
                 default="postgresql",
             ).ask()
 
-        if type == "sqlite":
-            # SQLite only needs database path
+        if type in {"sqlite", "duckdb"}:
+            # SQLite/DuckDB only need database file path
             database = database or questionary.path("Database file path:").ask()
             database = str(Path(database).expanduser().resolve())
             host = "localhost"
             port = 0
-            username = "sqlite"
+            username = type
             password = ""
         else:
             # PostgreSQL/MySQL need connection details
@@ -182,6 +182,17 @@ def add(
             port = 0
             username = "sqlite"
             password = ""
+        elif type == "duckdb":
+            if not database:
+                console.print(
+                    "[bold red]Error:[/bold red] Database file path is required for DuckDB"
+                )
+                sys.exit(1)
+            database = str(Path(database).expanduser().resolve())
+            host = "localhost"
+            port = 0
+            username = "duckdb"
+            password = ""
         else:
             if not all([host, database, username]):
                 console.print(
@@ -264,7 +275,7 @@ def list():
             if db.ssl_ca or db.ssl_cert:
                 ssl_status += " (certs)"
         else:
-            ssl_status = "disabled" if db.type != "sqlite" else "N/A"
+            ssl_status = "disabled" if db.type not in {"sqlite", "duckdb"} else "N/A"
 
         table.add_row(
             db.name,

{sqlsaber-0.24.0 → sqlsaber-0.25.0}/src/sqlsaber/cli/interactive.py

@@ -23,6 +23,7 @@ from sqlsaber.cli.display import DisplayManager
 from sqlsaber.cli.streaming import StreamingQueryHandler
 from sqlsaber.database.connection import (
     CSVConnection,
+    DuckDBConnection,
     MySQLConnection,
     PostgreSQLConnection,
     SQLiteConnection,
@@ -85,8 +86,12 @@ class InteractiveSession:
             if isinstance(self.db_conn, PostgreSQLConnection)
             else "MySQL"
             if isinstance(self.db_conn, MySQLConnection)
+            else "DuckDB"
+            if isinstance(self.db_conn, DuckDBConnection)
+            else "DuckDB"
+            if isinstance(self.db_conn, CSVConnection)
             else "SQLite"
-            if isinstance(self.db_conn, (SQLiteConnection, CSVConnection))
+            if isinstance(self.db_conn, SQLiteConnection)
             else "database"
         )