Skip to main content

MCP server for the Qiskit Code Assistant

Project description

qiskit-code-assistant-mcp-server

MCP Registry

MCP server for Qiskit Code Assistant

Components

Tools

The server implements four tools:

  • accept_model_disclaimer_tool: Accept the disclaimer for a given model
  • get_completion_tool: Get completion for a given prompt
  • get_rag_completion_tool: Get RAG completion for answering conceptual or descriptive questions about Qiskit or Quantum
  • accept_completion_tool: Accept a given completion

Prerequisites

  • Python 3.10 or higher
  • uv package manager (recommended)
  • IBM Quantum account and API token
  • Access to Qiskit Code Assistant service

Installation

Install from PyPI

The easiest way to install is via pip:

pip install qiskit-code-assistant-mcp-server

Install from Source

This project uses uv for virtual environments and dependencies management. If you don't have uv installed, check out the instructions in https://docs.astral.sh/uv/getting-started/installation/

Setting up the Project with uv

  1. Initialize or sync the project:

    # This will create a virtual environment and install dependencies
    uv sync
    
  2. Configure environment variables:

    # Copy the example environment file
    cp .env.example .env
    
    # Edit .env and add your IBM Quantum API token
    # Get your token from: https://cloud.quantum.ibm.com/
    

Configuration

Environment Variables

The server can be configured using environment variables in your .env file:

  • QISKIT_IBM_TOKEN - Your IBM Quantum API token (required)
  • QCA_TOOL_API_BASE - Qiskit Code Assistant API base URL (default: https://qiskit-code-assistant.quantum.ibm.com)
  • QCA_TOOL_MODEL_NAME - Default model name (default: mistral-small-3.2-24b-qiskit)
  • QCA_REQUEST_TIMEOUT - Request timeout in seconds (default: 30.0)
  • QCA_MCP_DEBUG_LEVEL - Logging level: DEBUG, INFO, WARNING, ERROR, CRITICAL (default: INFO)

Model Selection

The server includes an automatic model availability guardrail that:

  • Checks available models from the Qiskit Code Assistant service at startup
  • Uses the configured QCA_TOOL_MODEL_NAME if available
  • Automatically falls back to the first available model if the default is unavailable
  • Logs warnings when using a fallback model
  • Gracefully handles API errors by using the configured default

This ensures the server can start and function even when the default model is temporarily unavailable.

Quick Start

Running the Server

uv run qiskit-code-assistant-mcp-server

The server will start and listen for MCP connections.

Synchronous Usage

For frameworks that don't support async operations, all async functions have a .sync attribute for synchronous execution:

from qiskit_code_assistant_mcp_server.qca import (
    get_completion,
    get_rag_completion,
    list_models
)

# Use .sync for synchronous execution
result = get_completion.sync("Write a quantum circuit for a Bell state")
print(result)

# Works in Jupyter notebooks (handles nested event loops automatically)
rag_result = get_rag_completion.sync("What is quantum entanglement?")
print(rag_result)

# List available models
models = list_models.sync()
print(models)

Available functions (all support .sync):

  • list_models() - List available models
  • get_model(model_id) - Get model info
  • get_completion(prompt) - Get code completion
  • get_rag_completion(prompt) - Get RAG-based completion
  • accept_completion(completion_id) - Accept a completion
  • get_service_status() - Get service status

LangChain Integration Example:

Note: To run LangChain examples you will need to install the dependencies:

pip install langchain langchain-mcp-adapters langchain-openai python-dotenv
import asyncio
import os
from langchain.agents import create_agent
from langchain_mcp_adapters.client import MultiServerMCPClient
from langchain_mcp_adapters.tools import load_mcp_tools
from langchain_openai import ChatOpenAI
from dotenv import load_dotenv

# Load environment variables (QISKIT_IBM_TOKEN, OPENAI_API_KEY, etc.)
load_dotenv()

async def main():
    # Configure MCP client
    mcp_client = MultiServerMCPClient({
        "qiskit-code-assistant": {
            "transport": "stdio",
            "command": "qiskit-code-assistant-mcp-server",
            "args": [],
            "env": {
                "QISKIT_IBM_TOKEN": os.getenv("QISKIT_IBM_TOKEN", ""),
            },
        }
    })

    # Use persistent session for efficient tool calls
    async with mcp_client.session("qiskit-code-assistant") as session:
        tools = await load_mcp_tools(session)

        # Create agent with LLM
        llm = ChatOpenAI(model="gpt-5.2", temperature=0)
        agent = create_agent(llm, tools)

        # Run a query
        response = await agent.ainvoke("Write a quantum circuit that creates a Bell state")
        print(response)

asyncio.run(main())

For more LLM providers (Anthropic, Google, Ollama, Watsonx) and detailed examples including Jupyter notebooks, see the examples/ directory.

Testing and debugging the server

Note: to launch the MCP inspector you will need to have node and npm

  1. From a terminal, go into the cloned repo directory

  2. Switch to the virtual environment

    source .venv/bin/activate
    
  3. Run the MCP Inspector:

    npx @modelcontextprotocol/inspector uv run qiskit-code-assistant-mcp-server
    
  4. Open your browser to the URL shown in the console message e.g.,

    MCP Inspector is up and running at http://localhost:5173
    

Testing

This project includes comprehensive unit and integration tests.

Running Tests

Quick test run:

./run_tests.sh

Manual test commands:

# Install test dependencies
uv sync --group dev --group test

# Run all tests
uv run pytest

# Run only unit tests
uv run pytest -m "not integration"

# Run only integration tests
uv run pytest -m "integration"

# Run tests with coverage
uv run pytest --cov=src --cov-report=html

# Run specific test file
uv run pytest tests/test_qca.py -v

Test Structure

  • tests/test_qca.py - Unit tests for QCA functions
  • tests/test_utils.py - Unit tests for utility functions
  • tests/test_constants.py - Unit tests for configuration
  • tests/test_sync.py - Unit tests for synchronous execution
  • tests/test_integration.py - Integration tests
  • tests/conftest.py - Test fixtures and configuration

Test Coverage

The test suite covers:

  • ✅ All QCA API interactions
  • ✅ Model selection and availability guardrail
  • ✅ Error handling and validation
  • ✅ HTTP client management
  • ✅ Synchronous execution (.sync methods)
  • ✅ Configuration validation
  • ✅ Integration scenarios
  • ✅ Resource and tool handlers

Resources

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

qiskit_code_assistant_mcp_server-0.3.0.tar.gz (121.2 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

File details

Details for the file qiskit_code_assistant_mcp_server-0.3.0.tar.gz.

File metadata

File hashes

Hashes for qiskit_code_assistant_mcp_server-0.3.0.tar.gz
Algorithm Hash digest
SHA256 2c5043abe88358653430485efe2b454bbd68db3a4c0eef7a9b531a91da2ff213
MD5 c2cef4610701577a0a41f7f8473a1651
BLAKE2b-256 04b99a5d81b2d96403451cdb0a60f0390b47c55b38144b9d134781e69cd6d937

See more details on using hashes here.

Provenance

The following attestation bundles were made for qiskit_code_assistant_mcp_server-0.3.0.tar.gz:

Publisher: publish-pypi.yml on Qiskit/mcp-servers

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file qiskit_code_assistant_mcp_server-0.3.0-py3-none-any.whl.

File metadata

File hashes

Hashes for qiskit_code_assistant_mcp_server-0.3.0-py3-none-any.whl
Algorithm Hash digest
SHA256 e420bcb57405fce13c0a10f7b9037ca65facec1c385ef241f325051ca98eaf53
MD5 6cbdd2cad955cf864cf5ecefac5ddd26
BLAKE2b-256 bab91fdeb2cafa80bb33546b91a1b1e0942f817c67924509c95202f9a625f3f6

See more details on using hashes here.

Provenance

The following attestation bundles were made for qiskit_code_assistant_mcp_server-0.3.0-py3-none-any.whl:

Publisher: publish-pypi.yml on Qiskit/mcp-servers

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page