sqlsaber 0.17.0__py3-none-any.whl → 0.18.0__py3-none-any.whl

sqlsaber/cli/commands.py

@@ -7,24 +7,8 @@ from typing import Annotated
 import cyclopts
 from rich.console import Console
 
-from sqlsaber.agents import build_sqlsaber_agent
-from sqlsaber.cli.auth import create_auth_app
-from sqlsaber.cli.database import create_db_app
-from sqlsaber.cli.interactive import InteractiveSession
-from sqlsaber.cli.memory import create_memory_app
-from sqlsaber.cli.models import create_models_app
-from sqlsaber.cli.streaming import StreamingQueryHandler
-from sqlsaber.cli.threads import create_threads_app
+# Lazy imports - only import what's needed for CLI parsing
 from sqlsaber.config.database import DatabaseConfigManager
-from sqlsaber.database.connection import (
-    CSVConnection,
-    DatabaseConnection,
-    MySQLConnection,
-    PostgreSQLConnection,
-    SQLiteConnection,
-)
-from sqlsaber.database.resolver import DatabaseResolutionError, resolve_database
-from sqlsaber.threads import ThreadStorage
 
 
 class CLIError(Exception):
@@ -108,6 +92,21 @@ def query(
     """
 
     async def run_session():
+        # Import heavy dependencies only when actually running a query
+        # This is only done to speed up startup time
+        from sqlsaber.agents import build_sqlsaber_agent
+        from sqlsaber.cli.interactive import InteractiveSession
+        from sqlsaber.cli.streaming import StreamingQueryHandler
+        from sqlsaber.database.connection import (
+            CSVConnection,
+            DatabaseConnection,
+            MySQLConnection,
+            PostgreSQLConnection,
+            SQLiteConnection,
+        )
+        from sqlsaber.database.resolver import DatabaseResolutionError, resolve_database
+        from sqlsaber.threads import ThreadStorage
+
         # Check if query_text is None and stdin has data
         actual_query = query_text
         if query_text is None and not sys.stdin.isatty():
@@ -196,25 +195,45 @@ def query(
         sys.exit(e.exit_code)
 
 
-# Add authentication management commands
-auth_app = create_auth_app()
-app.command(auth_app, name="auth")
+# Use lazy imports for fast CLI startup time
+@app.command(name="auth")
+def auth(*args, **kwargs):
+    """Manage authentication configuration."""
+    from sqlsaber.cli.auth import create_auth_app
+
+    return create_auth_app()(*args, **kwargs)
+
+
+@app.command(name="db")
+def db(*args, **kwargs):
+    """Manage database connections."""
+    from sqlsaber.cli.database import create_db_app
+
+    return create_db_app()(*args, **kwargs)
+
+
+@app.command(name="memory")
+def memory(*args, **kwargs):
+    """Manage database-specific memories."""
+    from sqlsaber.cli.memory import create_memory_app
+
+    return create_memory_app()(*args, **kwargs)
+
+
+@app.command(name="models")
+def models(*args, **kwargs):
+    """Select and manage models."""
+    from sqlsaber.cli.models import create_models_app
 
-# Add database management commands after main callback is defined
-db_app = create_db_app()
-app.command(db_app, name="db")
+    return create_models_app()(*args, **kwargs)
 
-# Add memory management commands
-memory_app = create_memory_app()
-app.command(memory_app, name="memory")
 
-# Add model management commands
-models_app = create_models_app()
-app.command(models_app, name="models")
+@app.command(name="threads")
+def threads(*args, **kwargs):
+    """Manage SQLsaber threads."""
+    from sqlsaber.cli.threads import create_threads_app
 
-# Add threads management commands
-threads_app = create_threads_app()
-app.command(threads_app, name="threads")
+    return create_threads_app()(*args, **kwargs)
 
 
 def main():

sqlsaber/database/connection.py

@@ -9,7 +9,6 @@ from urllib.parse import parse_qs, urlparse
 import aiomysql
 import aiosqlite
 import asyncpg
-import pandas as pd
 
 
 class BaseDatabaseConnection(ABC):
@@ -324,6 +323,10 @@ class CSVConnection(BaseDatabaseConnection):
             return
 
         try:
+            # Import pandas only when needed for CSV operations
+            # This improves CLI load times
+            import pandas as pd
+
             # Read CSV file using pandas
             df = pd.read_csv(
                 self.csv_path,

{sqlsaber-0.17.0.dist-info → sqlsaber-0.18.0.dist-info}/METADATA

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

{sqlsaber-0.17.0.dist-info → sqlsaber-0.18.0.dist-info}/RECORD

@@ -6,7 +6,7 @@ sqlsaber/agents/mcp.py,sha256=GcJTx7YDYH6aaxIADEIxSgcWAdWakUx395JIzVnf17U,768
 sqlsaber/agents/pydantic_ai_agent.py,sha256=dGdsgyxCZvfK-v-MH8KimKOr-xb2aSfSWY8CMcOUCT8,6795
 sqlsaber/cli/__init__.py,sha256=qVSLVJLLJYzoC6aj6y9MFrzZvAwc4_OgxU9DlkQnZ4M,86
 sqlsaber/cli/auth.py,sha256=jTsRgbmlGPlASSuIKmdjjwfqtKvjfKd_cTYxX0-QqaQ,7400
-sqlsaber/cli/commands.py,sha256=ffEJq8WOfX7YBJzn5UCgeXBpU5lXnVBUMkbJEQ0R9WY,8169
+sqlsaber/cli/commands.py,sha256=CmCqDC6KiE8JD6Vkpsry4lBQInCiS8TBeKKx3gdxZcM,8689
 sqlsaber/cli/completers.py,sha256=HsUPjaZweLSeYCWkAcgMl8FylQ1xjWBWYTEL_9F6xfU,6430
 sqlsaber/cli/database.py,sha256=atwg3l8acQ3YTDuhq7vNrBN6tpOv0syz6V62KTF-Bh8,12910
 sqlsaber/cli/display.py,sha256=wa7BjTBwXwqLT145Q1AEL0C28pQJTrvDN10mnFMjqsg,8554
@@ -24,7 +24,7 @@ sqlsaber/config/oauth_tokens.py,sha256=C9z35hyx-PvSAYdC1LNf3rg9_wsEIY56hkEczelba
 sqlsaber/config/providers.py,sha256=JFjeJv1K5Q93zWSlWq3hAvgch1TlgoF0qFa0KJROkKY,2957
 sqlsaber/config/settings.py,sha256=vgb_RXaM-7DgbxYDmWNw1cSyMqwys4j3qNCvM4bljwI,5586
 sqlsaber/database/__init__.py,sha256=a_gtKRJnZVO8-fEZI7g3Z8YnGa6Nio-5Y50PgVp07ss,176
-sqlsaber/database/connection.py,sha256=sJtIIe0GVbo-1Py9-j66UxJoY1aKL9gqk68jkDL-Kvk,15123
+sqlsaber/database/connection.py,sha256=kwx18bnwr4kyTUfQT0OW-DXzJUNWIQJP54spJBqU_48,15243
 sqlsaber/database/resolver.py,sha256=RPXF5EoKzvQDDLmPGNHYd2uG_oNICH8qvUjBp6iXmNY,3348
 sqlsaber/database/schema.py,sha256=r12qoN3tdtAXdO22EKlauAe7QwOm8lL2vTMM59XEMMY,26594
 sqlsaber/mcp/__init__.py,sha256=COdWq7wauPBp5Ew8tfZItFzbcLDSEkHBJSMhxzy8C9c,112
@@ -40,8 +40,8 @@ sqlsaber/tools/enums.py,sha256=CH32mL-0k9ZA18911xLpNtsgpV6tB85TktMj6uqGz54,411
 sqlsaber/tools/instructions.py,sha256=X-x8maVkkyi16b6Tl0hcAFgjiYceZaSwyWTfmrvx8U8,9024
 sqlsaber/tools/registry.py,sha256=HWOQMsNIdL4XZS6TeNUyrL-5KoSDH6PHsWd3X66o-18,3211
 sqlsaber/tools/sql_tools.py,sha256=hM6tKqW5MDhFUt6MesoqhTUqIpq_5baIIDoN1MjDCXY,9647
-sqlsaber-0.17.0.dist-info/METADATA,sha256=epsxWGdmHWrFxLbbxWmOVsGqFMYmH1xUNAIGyGxpl_w,6141
-sqlsaber-0.17.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sqlsaber-0.17.0.dist-info/entry_points.txt,sha256=qEbOB7OffXPFgyJc7qEIJlMEX5RN9xdzLmWZa91zCQQ,162
-sqlsaber-0.17.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sqlsaber-0.17.0.dist-info/RECORD,,
+sqlsaber-0.18.0.dist-info/METADATA,sha256=HAXen2RBeZ4wIdxoLIMPKiNgOXJolWFqqXl5_cCOsJA,6141
+sqlsaber-0.18.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sqlsaber-0.18.0.dist-info/entry_points.txt,sha256=qEbOB7OffXPFgyJc7qEIJlMEX5RN9xdzLmWZa91zCQQ,162
+sqlsaber-0.18.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sqlsaber-0.18.0.dist-info/RECORD,,