Mistral Codestral

🚀 Mistral Codestral Code Service Station Server

A MCP server implemented for the Mistral Codestral API, offering code completion, error fixing, and test generation capabilities.

🚀 Quick Start

This server is designed to provide code completion, error fixing, and test generation services. To get started, follow the installation steps below and then you can start using the server.

✨ Features

• Code Completion: Utilizes Mistral's Codestral model for code completion.
• Error Detection and Fixing: Supports detection and fixing of code errors.
• Test Code Generation: Automatically generates test code.
• Multi - language Support: Supports multiple programming languages.
• Rate Limiting and Error Handling: Implements rate limiting and comprehensive error handling.
• Resource Access: Provides access to code files.
• Fill - in - the - Middle (FIM) Completion: Supports FIM completion.
• Input Validation: Uses Zod schema for thorough input validation.

📦 Installation

1. Clone this repository.
2. Install dependencies:

   npm install
   

3. Copy .env.example to .env and add your Mistral API key:

   cp .env.example .env
   

4. Build the project:

   npm run build
   

💻 Usage Examples

Basic Usage

Start the server:

npm start

Configuration

Add your Mistral API key to the .env file:

MISTRAL_API_KEY=Your_API_Key_Here

Available Tools

code_completion

Process code for completion, fixing, or test generation: Parameters:

• code (string, required): The code to be processed.
• language (string, optional): The programming language.
• task (string, required): One of the following:
  • "complete": Code completion.
  • "fix": Error fixing.
  • "test": Test generation.
  • "fim": Fill - in - the - Middle completion.

📚 Documentation

API Details

The server supports two main Mistral models:

• codestral-latest: The default model for code - related tasks.
• codestral-mamba-latest: An alternative model using the Mamba architecture.

Features include:

• Automatic Rate Limiting: Minimum 100ms between requests.
• API Timeout: 30 - second timeout for API calls.
• Error Handling: Comprehensive error handling for API responses.
• Response Validation: Uses Zod schema for response validation.
• Code Block Extraction: Automatically extracts code blocks from responses.
• Debug Logging: Logs API interactions for debugging.

Development Information

Run in development mode with auto - reload:

npm run dev

Run tests:

npm test

Error Handling

The server implements comprehensive error handling:

• Captures and formats API errors.
• Implements rate limiting.
• Uses Zod schema for input validation.
• Propagates errors to the client.
• Provides specific error messages for common issues:
  • Authentication failure.
  • Request exceeded limits.
  • Server error.
  • Invalid input validation.

Resource Access

The server provides access to code files via the file://code resource URI, supporting integration with workspace files.

📄 License

This project is licensed under the MIT license.