From 602415819eb1eab8ac07b98b1367a70d46917483 Mon Sep 17 00:00:00 2001
From: Cody Fincher <cody@litestar.dev>
Date: Mon, 11 Aug 2025 23:25:15 +0000
Subject: [PATCH] feat: register CLI with litestar extension

---
 sqlspec/extensions/litestar/cli.py    | 2 +-
 sqlspec/extensions/litestar/plugin.py | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/sqlspec/extensions/litestar/cli.py b/sqlspec/extensions/litestar/cli.py
index 24f29919..f27f0673 100644
--- a/sqlspec/extensions/litestar/cli.py
+++ b/sqlspec/extensions/litestar/cli.py
@@ -39,7 +39,7 @@ def get_database_migration_plugin(app: "Litestar") -> "SQLSpec":
     raise ImproperConfigurationError(msg)
 
 
-@click.group(cls=LitestarGroup, name="database")
+@click.group(cls=LitestarGroup, name="db")
 def database_group(ctx: "click.Context") -> None:
     """Manage SQLSpec database components."""
     ctx.obj = {"app": ctx.obj, "configs": get_database_migration_plugin(ctx.obj.app).config}
diff --git a/sqlspec/extensions/litestar/plugin.py b/sqlspec/extensions/litestar/plugin.py
index 22e7ff84..d0ffac80 100644
--- a/sqlspec/extensions/litestar/plugin.py
+++ b/sqlspec/extensions/litestar/plugin.py
@@ -1,7 +1,7 @@
 from typing import TYPE_CHECKING, Any, Union
 
 from litestar.di import Provide
-from litestar.plugins import InitPluginProtocol
+from litestar.plugins import CLIPlugin, InitPluginProtocol
 
 from sqlspec.base import SQLSpec as SQLSpecBase
 from sqlspec.config import AsyncConfigT, DatabaseConfigProtocol, DriverT, SyncConfigT
@@ -17,7 +17,7 @@
 logger = get_logger("extensions.litestar")
 
 
-class SQLSpec(InitPluginProtocol, SQLSpecBase):
+class SQLSpec(InitPluginProtocol, CLIPlugin, SQLSpecBase):
     """Litestar plugin for SQLSpec database integration."""
 
     __slots__ = ("_config", "_plugin_configs")