Add enum values to table_summary and truncate long column values

Author: txsmith

TODO.md

@@ -9,9 +9,3 @@
     - changing DB port
   - search
   - table_summary
-
-
-## schema include/exclude list in config
-
-## Add enum values to table_summary
-## Truncate sampled columns in table_summary

config_example.yaml

@@ -74,5 +74,4 @@ databases:
 settings:
   max_query_timeout: 30
   max_rows_per_query: 500
-  sample_size: 10
   enable_write_operations: false

pyproject.toml

@@ -1,9 +1,9 @@
 [project]
 name = "meeseeql"
-version = "0.1.0"
+version = "1.0.0"
 description = "A FastMCP server for exploring SQL databases"
 
-requires-python = ">=3.10"
+requires-python = ">=3.11"
 
 authors = [
     { name = "Thomas Smith", email = "thomas@apolloagriculture.com" },

src/meeseeql/main.py

@@ -103,10 +103,18 @@ async def table_summary(
     database: str,
     table_name: str,
     db_schema: str | None = None,
-    limit: int = 250,
+    limit: int = 100,
     page: int = 1,
 ) -> ToolResult:
-    """Get table structure including columns and foreign keys with pagination"""
+    """Get table structure including columns and foreign keys with pagination
+
+    Args:
+        database: Database name to query
+        table_name: Name of the table to summarize
+        db_schema: Optional schema name (uses default schema if not provided)
+        limit: Maximum items per page (default: 100)
+        page: Page number (default: 1)
+    """
     result = await tools.table_summary(
         get_or_init_db_manager(), database, table_name, db_schema, limit, page
     )

src/meeseeql/tools/sql/mssql/enum_values.sql

@@ -0,0 +1,4 @@
+SELECT
+    NULL as column_name,
+    NULL as enum_values
+LIMIT 0

src/meeseeql/tools/sql/mysql/enum_values.sql

@@ -0,0 +1,13 @@
+SELECT
+    c.column_name,
+    CASE
+        WHEN c.data_type = 'enum' THEN
+            REPLACE(REPLACE(SUBSTRING(c.column_type, 6, CHAR_LENGTH(c.column_type) - 6), '''', ''), ')', '')
+        ELSE NULL
+    END as enum_values
+FROM information_schema.columns c
+WHERE c.table_schema NOT IN ('information_schema', 'mysql', 'performance_schema', 'sys')
+AND LOWER(c.table_name) = LOWER('{{table_name}}')
+AND LOWER(c.table_schema) = LOWER('{{schema_name}}')
+AND c.data_type = 'enum'
+ORDER BY c.ordinal_position

src/meeseeql/tools/sql/postgresql/columns.sql

@@ -1,6 +1,9 @@
 SELECT
     c.column_name,
-    c.data_type,
+    CASE 
+        WHEN c.data_type = 'USER-DEFINED' THEN c.udt_name
+        ELSE c.data_type
+    END as data_type,
     c.is_nullable,
     c.column_default
 FROM information_schema.columns c

src/meeseeql/tools/sql/postgresql/enum_values.sql

@@ -0,0 +1,12 @@
+SELECT
+    c.column_name,
+    string_agg(e.enumlabel, ', ' ORDER BY e.enumsortorder) as enum_values
+FROM information_schema.columns c
+JOIN pg_type t ON t.typname = c.udt_name
+JOIN pg_enum e ON e.enumtypid = t.oid
+WHERE c.table_schema NOT IN ('information_schema', 'pg_catalog')
+AND LOWER(c.table_name) = LOWER('{{table_name}}')
+AND LOWER(c.table_schema) = LOWER('{{schema_name}}')
+AND t.typtype = 'e'
+GROUP BY c.column_name, c.ordinal_position
+ORDER BY c.ordinal_position

src/meeseeql/tools/sql/snowflake/enum_values.sql

@@ -0,0 +1,4 @@
+SELECT
+    NULL as column_name,
+    NULL as enum_values
+LIMIT 0

src/meeseeql/tools/sql/sqlite/enum_values.sql

@@ -0,0 +1,4 @@
+SELECT
+    NULL as column_name,
+    NULL as enum_values
+LIMIT 0