release v0.1.0

Author: viragtripathi

.github/workflows/python-ci.yml

@@ -0,0 +1,33 @@
+name: Python CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v3
+
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.11'
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -e .[dev]
+
+      - name: Lint with ruff (non-blocking)
+        run: ruff check . || true
+
+      - name: Format check with black
+        run: black --check crdb_mcp_server
+
+      - name: Run tests
+        run: pytest tests/

.gitignore

@@ -0,0 +1,29 @@
+*.sw?
+.#*
+*#
+*~
+.project
+.settings
+build
+target
+*.sublime-*
+/scratch
+*.iml
+*.log
+*.gfs
+*.idea
+logs/**
+versions.properties
+**/.DS_Store
+/.idea/
+*.egg-info
+/logs/
+build/
+dist/
+*.egg-info/
+__pycache__/
+venv/
+*.py[oc]
+wheels/
+.venv
+.coverage

CHANGELOG.md

@@ -0,0 +1,22 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+---
+
+## [v0.1.0] - 2025-06-11
+
+### Added
+- Initial public release of `cockroachdb-mcp-server`
+- FastAPI server implementing MCP spec
+- REST endpoints for context CRUD (`/contexts`)
+- SQLAlchemy integration with CockroachDB
+- Automatic dialect fix for `cockroachdb://` and `postgresql://` prefixes
+- CLI entrypoint via `cockroachdb-mcp-server serve`
+- Schema bootstrap via `--init-schema` or `MCP_AUTO_INIT_SCHEMA`
+- ASCII banner (`--banner`) and version info (`--version`)
+- Structured logging (`--log-level`)
+
+### Known Limitations
+- No `/run`, `/deploy`, or `/evaluate` endpoints yet
+- No auth or rate limiting (coming in future versions)

Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+# Install uv globally and install deps into system Python
+COPY pyproject.toml .
+RUN pip install --no-cache-dir uv && uv pip install --system .
+
+# Copy app
+COPY ./app ./app
+COPY ./schema ./schema
+
+COPY entrypoint.sh /entrypoint.sh
+RUN chmod +x /entrypoint.sh
+
+ENTRYPOINT ["/entrypoint.sh"]
+#CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8080"]

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Virag Tripathi
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

Makefile

@@ -0,0 +1,43 @@
+.PHONY: help prepare install format lint test build publish clean
+
+help:
+	@echo "📦 Makefile for cockroachdb-mcp-server"
+	@echo ""
+	@echo "Available targets:"
+	@echo "  prepare     Set up virtualenv and install build tools"
+	@echo "  install     Install dependencies in editable mode"
+	@echo "  format      Format code using black"
+	@echo "  lint        Run static checks with ruff"
+	@echo "  test        Run tests with pytest"
+	@echo "  build       Build wheel and sdist packages"
+	@echo "  publish     Upload to PyPI (requires credentials)"
+	@echo "  clean       Remove build and metadata artifacts"
+
+prepare:
+	@echo "🔧 Preparing virtual environment..."
+	@python3 -m venv .venv
+	@. .venv/bin/activate && \
+		python -m ensurepip --upgrade && \
+		python -m pip install --upgrade pip build
+
+install:
+	pip install -e .[dev]
+
+format:
+	black cockroachdb_mcp_server
+	black app
+
+lint:
+	ruff cockroachdb_mcp_server
+
+test:
+	pytest tests/
+
+build:
+	. .venv/bin/activate && python -m build
+
+publish:
+	twine upload dist/*
+
+clean:
+	rm -rf dist/ build/ *.egg-info .pytest_cache .mypy_cache .venv

README.md

@@ -0,0 +1,141 @@
+# cockroachdb-mcp-server
+
+![PyPI](https://img.shields.io/pypi/v/cockroachdb-mcp-server)
+![Python](https://img.shields.io/pypi/pyversions/cockroachdb-mcp-server)
+![License](https://img.shields.io/github/license/viragtripathi/cockroachdb-mcp-server)
+![CI](https://github.com/viragtripathi/cockroachdb-mcp-server/actions/workflows/python-ci.yml/badge.svg)
+
+A Model Context Protocol (MCP) server implemented in Python using FastAPI and CockroachDB.
+
+---
+
+## 🧠 What This Is
+
+`cockroachdb-mcp-server` is a production-grade, spec-aligned MCP server that:
+
+- Implements the [Model Context Protocol](https://modelcontextprotocol.io/introduction)
+- Uses **CockroachDB** as a resilient, SQL-compatible backend
+- Exposes full **CRUD APIs** for managing model contexts
+- Stores context definitions as **JSONB**, allowing arbitrary input/output schema
+- Works seamlessly with the [`cockroachdb-mcp-client`](https://github.com/viragtripathi/cockroachdb-mcp-client) CLI
+
+---
+
+## ✅ Feature Highlights
+
+- ✅ REST API for MCP context management (`/contexts`)
+- ✅ Schema bootstrapping via CLI flag or env var
+- ✅ CRDB URL auto-detection and dialect fix
+- ✅ ASCII banner and version info
+- ✅ Structured logging and configurable log level
+- ✅ Ready for `/run`, `/deploy`, `/evaluate` extensions
+
+---
+
+## 🚀 Quickstart
+
+### 📦 Install from PyPI
+
+```bash
+pip install cockroachdb-mcp-server
+````
+
+### 🏃 Run with schema init
+
+```bash
+cockroachdb-mcp-server serve --init-schema --log-level INFO
+```
+
+Or:
+
+```bash
+export MCP_AUTO_INIT_SCHEMA=true
+cockroachdb-mcp-server serve
+```
+
+> Server runs at `http://localhost:8081` by default
+
+---
+
+## 🔧 CLI Usage
+
+```bash
+cockroachdb-mcp-server serve --init-schema
+cockroachdb-mcp-server serve --port 8081 --host 127.0.0.1 --reload
+cockroachdb-mcp-server --version
+cockroachdb-mcp-server --banner
+```
+
+---
+
+## 🔐 Configuring the Database
+
+### ✅ Set the `CRDB_URL` environment variable
+
+```bash
+export CRDB_URL="postgresql://root@localhost:26257/defaultdb?sslmode=disable"
+```
+
+> Automatically rewritten to `cockroachdb://...` under the hood for compatibility.
+
+Alternatively, set it directly:
+
+```bash
+export CRDB_URL="cockroachdb://root@localhost:26257/defaultdb?sslmode=disable"
+```
+
+✅ Both formats are supported.
+
+---
+
+## 🧪 API Endpoints
+
+| Method | Path             | Description       |
+|--------|------------------|-------------------|
+| POST   | `/contexts`      | Create a context  |
+| GET    | `/contexts`      | List all contexts |
+| GET    | `/contexts/{id}` | Get context by ID |
+| PUT    | `/contexts/{id}` | Update context    |
+| DELETE | `/contexts/{id}` | Delete context    |
+
+---
+
+## 🧱 Schema Auto-Bootstrap
+
+Run this manually:
+
+```bash
+cockroachdb-mcp-server serve --init-schema
+```
+
+Or automatically with:
+
+```bash
+export MCP_AUTO_INIT_SCHEMA=true
+```
+
+The schema created is:
+
+```sql
+CREATE TABLE IF NOT EXISTS mcp_contexts (
+  id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+  context_name STRING NOT NULL,
+  context_version STRING NOT NULL,
+  body JSONB NOT NULL,
+  created_at TIMESTAMP DEFAULT now()
+);
+```
+
+---
+
+## 🔗 Related Projects
+
+* [cockroachdb-mcp-client](https://github.com/viragtripathi/cockroachdb-mcp-client): CLI tool to manage MCP contexts, simulate LLM runs, export, and batch simulate across providers.
+
+---
+
+## 🙌 Contributions
+
+This project is designed for internal and community use.
+
+PRs welcome to extend functionality (auth, deployment support, `/evaluate`, telemetry, etc.).

app/__init__.py

@@ -0,0 +1,34 @@
+from sqlalchemy import create_engine
+from cockroachdb_mcp_server.config import resolve_crdb_url
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+def get_sqlalchemy_engine(opts=None):
+    if opts is None:
+        dsn = resolve_crdb_url()
+        return create_engine(dsn)
+
+    base = f"cockroachdb://root@{opts.get('host', 'localhost')}:{opts.get('port', 26257)}/{opts['db']}"
+    if opts.get("certs_dir"):
+        base += (
+            f"?sslmode=verify-full"
+            f"&sslrootcert={opts['certs_dir']}/ca.crt"
+            f"&sslcert={opts['certs_dir']}/client.root.crt"
+            f"&sslkey={opts['certs_dir']}/client.root.key"
+        )
+    else:
+        base += "?sslmode=disable"
+
+    logger.debug("Constructed CRDB DSN from opts: %s", base)
+    return create_engine(base)
+
+
+""""
+def get_psycopg_connection():
+    url = resolve_crdb_url()
+    pg_url = url.replace("cockroachdb://", "postgresql://")
+    print(f"[debug] Using psycopg2 connection: {pg_url}")
+    return psycopg2.connect(pg_url)
+"""

app/db/db_connection.py

@@ -0,0 +1,13 @@
+from fastapi import FastAPI
+from app.routes import health, contexts
+from cockroachdb_mcp_server import __version__
+
+app = FastAPI(title="CockroachDB MCP Server", version={__version__})
+
+app.include_router(health.router)
+app.include_router(contexts.router, prefix="/contexts")
+
+
+@app.get("/")
+def root():
+    return {"message": "MCP Server is running"}