Feature/notebooks

changelog/0.4.4.md

@@ -0,0 +1,20 @@
+# [0.4.4] - 2025-07-29
+
+## Added
+
+- `create_notebook_file` tool to generate notebooks and store them in a temp file
+- `upload_notebook_file` tool to upload notebooks to the user's space
+
+## Fixed
+
+- Renamed `list_of_regions`tool to `list_regions`
+- Removed internal tools
+  - `list_notebook_samples`
+  - `prepare_database_migration`
+  - `complete_database_migration`
+  - `get_notebook_path`
+  - `get_job_details`
+  - `list_job_executions`
+  - `create_scheduled_job`
+  - `organization_billing_usage`
+  - `check_if_file_exists`

pyproject.toml

@@ -11,6 +11,7 @@ dependencies = [
  "segment-analytics-python>=2.3.4",
  "singlestoredb>=1.14.2",
  "starlette>=0.46.2",
+ "jsonschema>=4.0.0",
 ]
 
 [project.scripts]

src/api/tools/__init__.py

@@ -1,14 +1,14 @@
-from .tools import (
-    tools,
-    terminate_virtual_workspace,
-    get_user_id,
-    list_of_regions,
-    workspaces_info,
-    run_sql,
-    choose_organization,
-    organization_info,
-    list_virtual_workspaces,
-)
+# Import from organized structure
+from .tools import tools
+
+# Import individual tools for backward compatibility
+from .user import get_user_id
+from .regions import list_regions as list_of_regions
+from .workspaces import workspaces_info
+from .database import run_sql
+from .organization import choose_organization, organization_info
+from .starter_workspaces import list_virtual_workspaces, terminate_virtual_workspace
+
 from .registery import register_tools
 
 __all__ = [

src/api/tools/database/__init__.py

@@ -0,0 +1,5 @@
+"""Database tools for SingleStore MCP server."""
+
+from .database import run_sql
+
+__all__ = ["run_sql"]