mcp-sql 0.2.0

MCP server that lets LLMs query PostgreSQL, SQLite, and MySQL databases
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

use rmcp::model::ErrorData;

#[derive(Debug, thiserror::Error)]
pub enum McpSqlError {
    #[error("Database error: {0}")]
    Database(#[from] sqlx::Error),

    #[error("Write operation rejected: {0}")]
    ReadOnly(String),

    #[error("Database not found: {0}")]
    DatabaseNotFound(String),

    #[error("Ambiguous database: multiple databases connected, specify the 'database' parameter")]
    AmbiguousDatabase,

    #[error("Invalid SQL: {0}")]
    InvalidSql(String),

    #[error("Query timed out after {0} seconds")]
    QueryTimeout(u64),

    #[error("{0}")]
    Other(String),
}

impl McpSqlError {
    pub fn to_mcp_error(&self) -> ErrorData {
        match self {
            McpSqlError::ReadOnly(_) | McpSqlError::InvalidSql(_) => {
                ErrorData::invalid_params(self.to_string(), None)
            }
            McpSqlError::DatabaseNotFound(_) | McpSqlError::AmbiguousDatabase => {
                ErrorData::invalid_params(self.to_string(), None)
            }
            McpSqlError::QueryTimeout(_) => {
                ErrorData::internal_error(self.to_string(), None)
            }
            McpSqlError::Database(_) | McpSqlError::Other(_) => {
                ErrorData::internal_error(self.to_string(), None)
            }
        }
    }
}