elicit_sqlx 0.10.0

Elicitation-enabled sqlx type wrappers — newtypes with JsonSchema and MCP reflect methods
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

//! `SqlxContext` — [`PluginContext`] carrying a shared database pool.

use elicitation::PluginContext;
use sqlx::AnyPool;
use tracing::instrument;

/// Holds a shared [`AnyPool`] for database tool calls.
pub struct SqlxContext {
    db: AnyPool,
}

impl SqlxContext {
    /// Creates a new context from an existing pool.
    pub fn new(db: AnyPool) -> Self {
        Self { db }
    }

    /// Returns a reference to the underlying pool.
    pub fn pool(&self) -> &AnyPool {
        &self.db
    }
}

impl PluginContext for SqlxContext {}

/// Connects to a database and produces a [`SqlxContext`].
///
/// Installs the Any driver if not already done.
#[instrument]
pub async fn connect(database_url: &str) -> Result<SqlxContext, sqlx::Error> {
    sqlx::any::install_default_drivers();
    let pool = AnyPool::connect(database_url).await?;
    Ok(SqlxContext::new(pool))
}