mcp-server-sqlseed 0.1.0__tar.gz

mcp_server_sqlseed-0.1.0/.gitignore

@@ -0,0 +1,69 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# Virtual Environment
+.venv/
+env/
+ENV/
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+*~
+
+# Testing
+.pytest_cache/
+.coverage
+htmlcov/
+
+# Type checking
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Linting
+.ruff_cache/
+
+# Project specific
+*.db
+*.sqlite
+*.sqlite3
+snapshots/
+
+# AI cache
+.sqlseed_cache/
+
+# Archived temp files
+_archived_temp/
+
+# macOS
+.DS_Store
+
+# Trae IDE
+.trae/
+
+# Build artifacts
+dist/
+*.whl
+*.tar.gz

mcp_server_sqlseed-0.1.0/PKG-INFO

@@ -0,0 +1,20 @@
+Metadata-Version: 2.4
+Name: mcp-server-sqlseed
+Version: 0.1.0
+Summary: MCP server for sqlseed - AI-powered SQLite test data generation
+Project-URL: Homepage, https://github.com/sunbos/sqlseed
+Project-URL: Repository, https://github.com/sunbos/sqlseed/tree/main/plugins/mcp-server-sqlseed
+Author-email: SunBo <1443584939@qq.com>
+License-Expression: AGPL-3.0-or-later
+Classifier: Development Status :: 3 - Alpha
+Classifier: License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: >=3.10
+Requires-Dist: mcp>=1.0
+Requires-Dist: sqlseed
+Provides-Extra: ai
+Requires-Dist: sqlseed-ai; extra == 'ai'

mcp_server_sqlseed-0.1.0/pyproject.toml

@@ -0,0 +1,39 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "mcp-server-sqlseed"
+version = "0.1.0"
+requires-python = ">=3.10"
+description = "MCP server for sqlseed - AI-powered SQLite test data generation"
+license = "AGPL-3.0-or-later"
+authors = [
+    {name = "SunBo", email = "1443584939@qq.com"},
+]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+]
+dependencies = [
+    "sqlseed",
+    "mcp>=1.0",
+]
+
+[project.urls]
+Homepage = "https://github.com/sunbos/sqlseed"
+Repository = "https://github.com/sunbos/sqlseed/tree/main/plugins/mcp-server-sqlseed"
+
+[project.optional-dependencies]
+ai = ["sqlseed-ai"]
+
+[project.scripts]
+mcp-server-sqlseed = "mcp_server_sqlseed:main"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/mcp_server_sqlseed"]

mcp_server_sqlseed-0.1.0/src/mcp_server_sqlseed/__init__.py

@@ -0,0 +1,7 @@
+from __future__ import annotations
+
+from mcp_server_sqlseed.server import mcp
+
+
+def main() -> None:
+    mcp.run()

mcp_server_sqlseed-0.1.0/src/mcp_server_sqlseed/__main__.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+from mcp_server_sqlseed import main
+
+if __name__ == "__main__":
+    main()

mcp_server_sqlseed-0.1.0/src/mcp_server_sqlseed/config.py

@@ -0,0 +1,9 @@
+from __future__ import annotations
+
+from pydantic import BaseModel
+
+
+class MCPServerConfig(BaseModel):
+    db_path: str | None = None
+    host: str = "localhost"
+    port: int = 8000

mcp_server_sqlseed-0.1.0/src/mcp_server_sqlseed/server.py

@@ -0,0 +1,148 @@
+from __future__ import annotations
+
+import hashlib
+import json
+from typing import Any
+
+from mcp.server.fastmcp import FastMCP
+
+mcp = FastMCP("sqlseed")
+
+
+def _serialize_schema_context(ctx: dict[str, Any]) -> dict[str, Any]:
+    return {
+        "table_name": ctx["table_name"],
+        "columns": [
+            {
+                "name": c.name,
+                "type": c.type,
+                "nullable": c.nullable,
+                "default": c.default,
+                "is_primary_key": c.is_primary_key,
+                "is_autoincrement": c.is_autoincrement,
+            }
+            for c in ctx["columns"]
+        ],
+        "foreign_keys": [
+            {"column": fk.column, "ref_table": fk.ref_table, "ref_column": fk.ref_column}
+            for fk in ctx["foreign_keys"]
+        ],
+        "indexes": ctx["indexes"],
+        "sample_data": ctx["sample_data"],
+        "all_table_names": ctx["all_table_names"],
+    }
+
+
+def _compute_schema_hash(schema_ctx: dict[str, Any]) -> str:
+    hash_input = json.dumps(
+        {
+            "columns": [
+                {"name": c.name, "type": c.type, "nullable": c.nullable}
+                for c in schema_ctx["columns"]
+            ],
+            "foreign_keys": [
+                {"column": fk.column, "ref_table": fk.ref_table}
+                for fk in schema_ctx["foreign_keys"]
+            ],
+        },
+        sort_keys=True,
+    )
+    return hashlib.sha256(hash_input.encode()).hexdigest()[:16]
+
+
+@mcp.resource("sqlseed://schema/{db_path}/{table_name}")
+def get_schema_resource(db_path: str, table_name: str) -> str:
+    from sqlseed.core.orchestrator import DataOrchestrator
+
+    with DataOrchestrator(db_path) as orch:
+        ctx = orch.get_schema_context(table_name)
+        serializable_ctx = _serialize_schema_context(ctx)
+        return json.dumps(serializable_ctx, ensure_ascii=False, indent=2)
+
+
+@mcp.tool()
+def sqlseed_inspect_schema(db_path: str, table_name: str | None = None) -> dict[str, Any]:
+    """Inspect database schema. Returns column info, foreign keys, indexes,
+    sample data, and schema_hash for specified table or all tables."""
+    from sqlseed.core.orchestrator import DataOrchestrator
+
+    with DataOrchestrator(db_path) as orch:
+        tables = [table_name] if table_name else orch._db.get_table_names()
+        result: dict[str, Any] = {}
+        for tbl in tables:
+            ctx = orch.get_schema_context(tbl)
+            result[tbl] = _serialize_schema_context(ctx)
+            result[tbl]["schema_hash"] = _compute_schema_hash(ctx)
+        return result
+
+
+@mcp.tool()
+def sqlseed_generate_yaml(db_path: str, table_name: str, max_retries: int = 3) -> str:
+    """Generate YAML config for a table using AI analysis with self-correction.
+    Returns YAML string for human review. Requires sqlseed-ai plugin and API key."""
+    import yaml
+    from sqlseed_ai.analyzer import SchemaAnalyzer
+    from sqlseed_ai.refiner import AiConfigRefiner, AISuggestionFailedError
+
+    analyzer = SchemaAnalyzer()
+    refiner = AiConfigRefiner(analyzer, db_path)
+
+    try:
+        result = refiner.generate_and_refine(
+            table_name=table_name,
+            max_retries=max_retries,
+        )
+    except AISuggestionFailedError as e:
+        return f"# AI suggestion failed: {e}"
+    except Exception as e:
+        return f"# Error: {e}"
+
+    if result:
+        output = {"db_path": db_path, "provider": "mimesis", "locale": "zh_CN", "tables": [result]}
+        return yaml.dump(output, allow_unicode=True, sort_keys=False, default_flow_style=False)
+    return "# No AI suggestions available. Ensure sqlseed-ai plugin is installed and API key is configured."
+
+
+@mcp.tool()
+def sqlseed_execute_fill(
+    db_path: str,
+    table_name: str,
+    count: int = 1000,
+    yaml_config: str | None = None,
+) -> dict[str, Any]:
+    """Execute data generation for a table. Optionally provide YAML config string for column rules."""
+    from sqlseed.core.orchestrator import DataOrchestrator
+
+    with DataOrchestrator(db_path) as orch:
+        column_configs = None
+        clear_before = False
+        seed = None
+
+        if yaml_config:
+            import yaml as _yaml
+
+            from sqlseed.config.models import GeneratorConfig
+
+            data = _yaml.safe_load(yaml_config)
+            config = GeneratorConfig(**data)
+            for t in config.tables:
+                if t.name == table_name:
+                    column_configs = t.columns
+                    clear_before = t.clear_before
+                    seed = t.seed
+                    break
+
+        result = orch.fill_table(
+            table_name=table_name,
+            count=count,
+            column_configs=column_configs,
+            clear_before=clear_before,
+            seed=seed,
+        )
+
+        return {
+            "table_name": result.table_name,
+            "count": result.count,
+            "elapsed": result.elapsed,
+            "errors": result.errors,
+        }