shaped-mcp 0.1.0__tar.gz

shaped_mcp-0.1.0/.gitignore

@@ -0,0 +1,233 @@
+.claude/
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib64/
+log_tensorboard/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+.hypothesis/
+.pytest_cache/
+**/log_tensorboard/
+**/tmp
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+**/.DS_Store
+.serverless
+
+docker_mount
+
+**/cdk.out
+.aws-sam
+.build
+
+# Wheel image builder
+.net
+
+.vscode/
+
+hydra_out/
+
+# dependencies
+/node_modules
+**/node_modules
+/.pnp
+.pnp.js
+
+# testing
+/coverage
+
+# next.js
+**/.next
+/out/
+
+# production
+/build
+
+# misc
+*.pem
+
+# debug
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+.pnpm-debug.log*
+
+# local env files
+.env*.local
+.env
+
+# vercel
+.vercel
+
+# typescript
+*.tsbuildinfo
+
+.contentlayer
+
+# terraform
+**/.terraform/*
+terraform.tfstate*
+cast-ai.key
+.kubeconfig
+
+# runhouse
+.rh
+
+.pymon
+
+# Rust
+target
+Cargo.lock
+
+# Handled by .gitignores in subdirectories
+!fibertide-infrastructure-modules/**
+!secrets/**
+!vpn/**
+
+# Cursor
+.cursor
+!.cursor/mcp.json.example
+.cursor/mcp.json
+PR_SUMMARY.md
+
+# SDK
+src/sdk/python/.openapi-generator/
+src/sdk/nodejs/.openapi-generator/
+src/sdk/nodejs/dist/
+src/sdk/nodejs/package-lock.json
+# Shaped-docs (Docusaurus site)
+sites/shaped-docs/node_modules/
+sites/shaped-docs/build/
+sites/shaped-docs/.docusaurus/
+sites/shaped-docs/.cache-loader/
+sites/shaped-docs/.env
+
+# Console test results
+src/pkg/images/console/test-results/
+src/pkg/images/console/playwright-report/
+src/pkg/images/console/coverage/
+terraform.tfvars
+
+# Claude
+.claude/
+
+.pnpm-store/*

shaped_mcp-0.1.0/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: shaped-mcp
+Version: 0.1.0
+Summary: MCP server for Shaped - retrieval for AI agents, search engines, and more
+Requires-Python: >=3.10
+Requires-Dist: httpx>=0.27.0
+Requires-Dist: mcp>=1.0.0
+Requires-Dist: shaped>=2.0.31
+Provides-Extra: dev
+Requires-Dist: pytest-asyncio>=0.21.0; extra == 'dev'
+Requires-Dist: pytest>=7.0.0; extra == 'dev'

shaped_mcp-0.1.0/README.md

@@ -0,0 +1,19 @@
+# Shaped MCP Server
+
+MCP (Model Context Protocol) server that exposes Shaped's API to AI agents. Enables Cursor, Claude Desktop, and other MCP clients to create and manage tables, views, engines, and run queries.
+
+## Quick Start
+
+1. Install: `pip install -e .` (from this directory) or `pip install shaped-mcp` (PyPI).
+2. Set `SHAPED_API_KEY` in your environment.
+3. Configure your AI tool (see [Usage Guide](../../docs/mcp-usage.md)).
+4. Run tests: `pytest mcp/tests/` (with `pip install -e ".[dev]"`).
+
+## Documentation
+
+- **[Build Plan](PLAN.md)** – Implementation plan, tools, and phases.
+- **[Usage Guide](../../docs/mcp-usage.md)** – How to configure Cursor, Claude, and other MCP clients.
+
+## Status
+
+Phase 1–3 implemented: STDIO, HTTP (Streamable), Bearer auth, prompts, Helm chart, PyPI publish, deploy workflow. See `PLAN.md` for full roadmap. For local Cursor testing, copy `.cursor/mcp.json.example` to `.cursor/mcp.json` and set `SHAPED_API_KEY`. To list in the [MCP Registry](https://registry.modelcontextprotocol.io/), use `registry-entry.md`.

shaped_mcp-0.1.0/pyproject.toml

@@ -0,0 +1,34 @@
+[project]
+name = "shaped-mcp"
+version = "0.1.0"
+description = "MCP server for Shaped - retrieval for AI agents, search engines, and more"
+requires-python = ">=3.10"
+dependencies = [
+    "mcp>=1.0.0",
+    "shaped>=2.0.31",
+    "httpx>=0.27.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=7.0.0",
+    "pytest-asyncio>=0.21.0",
+]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+pythonpath = ["src"]
+asyncio_mode = "auto"
+markers = [
+    "integration: tests that call the real Shaped API (require SHAPED_API_KEY)",
+]
+
+[project.scripts]
+shaped-mcp = "shaped_mcp.__main__:main"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/shaped_mcp"]

shaped_mcp-0.1.0/registry-entry.md

@@ -0,0 +1,44 @@
+# Shaped MCP Server – Registry Entry
+
+Use this information when submitting to the [MCP Registry](https://registry.modelcontextprotocol.io/).
+
+## Metadata
+
+- **Name:** Shaped
+- **Description:** MCP server for Shaped - create and manage tables, views, engines, and run retrieval queries. Enables AI agents to use Shaped's hybrid retrieval, trained scoring, and fine-tuning for recommendations and search.
+- **Repository:** https://github.com/shaped-ai/magnus (monorepo; MCP package in `mcp/`)
+- **PyPI:** https://pypi.org/project/shaped-mcp/
+- **Documentation:** See `docs/mcp-usage.md` in the repo
+
+## Installation
+
+```bash
+pip install shaped-mcp
+```
+
+## Configuration (STDIO)
+
+```json
+{
+  "mcpServers": {
+    "shaped": {
+      "command": "shaped-mcp",
+      "env": {
+        "SHAPED_API_KEY": "${SHAPED_API_KEY}"
+      }
+    }
+  }
+}
+```
+
+## Hosted Endpoint (HTTP)
+
+- **URL:** https://mcp.shaped.ai/mcp
+- **Auth:** `Authorization: Bearer <SHAPED_API_KEY>`
+
+## Tools
+
+Query: shaped_query, shaped_search, shaped_similar_items, shaped_feed, shaped_similar_users, shaped_rerank
+Tables: shaped_list_tables, shaped_get_table, shaped_create_table, shaped_delete_table, shaped_insert_rows
+Views: shaped_list_views, shaped_get_view, shaped_create_view, shaped_delete_view
+Engines: shaped_list_engines, shaped_get_engine, shaped_create_engine, shaped_update_engine, shaped_delete_engine, shaped_get_engine_status

shaped_mcp-0.1.0/src/shaped_mcp/__init__.py

@@ -0,0 +1,3 @@
+"""Shaped MCP server - exposes Shaped API to MCP-compatible agents."""
+
+__version__ = "0.1.0"

shaped_mcp-0.1.0/src/shaped_mcp/__main__.py

@@ -0,0 +1,35 @@
+"""Entry point for shaped-mcp CLI."""
+
+import argparse
+import os
+
+
+def main() -> None:
+    """Entry point for the shaped-mcp CLI."""
+    parser = argparse.ArgumentParser(description="Shaped MCP server")
+    parser.add_argument(
+        "--transport",
+        choices=["stdio", "streamable-http"],
+        default="stdio",
+        help="Transport: stdio (default) or streamable-http",
+    )
+    parser.add_argument(
+        "--host", default="0.0.0.0", help="Host for HTTP (default: 0.0.0.0)"
+    )
+    parser.add_argument(
+        "--port", type=int, default=8080, help="Port for HTTP (default: 8080)"
+    )
+    args = parser.parse_args()
+
+    # Set env before import so FastMCP picks up host/port.
+    if args.transport != "stdio":
+        os.environ["FASTMCP_HOST"] = args.host
+        os.environ["FASTMCP_PORT"] = str(args.port)
+
+    from shaped_mcp.server import run  # noqa: E402
+
+    run(transport=args.transport, host=args.host, port=args.port)
+
+
+if __name__ == "__main__":
+    main()