agent-block 0.10.0

Lua-first Agent Runtime built on AgentMesh
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

//! `std.sql` thin adapter.
//!
//! Bridge implementation moved to the `mlua-batteries` crate
//! (`mlua_batteries::sql`).  This module only resolves the host's
//! environment-driven SQL configuration into a
//! [`mlua_batteries::sql::SqlConfig`] before delegating to
//! [`mlua_batteries::sql::register_with`], then layers the agent-block
//! Lua tool helpers (`sql_tools.lua`) on top.
//!
//! See `bridge/config.rs` for the ENV → config mapping.

use std::sync::Arc;

use mlua::prelude::*;
use mlua_batteries::sql::SqlConfig;

use crate::host::HostContext;

pub fn register(lua: &Lua, ctx: &HostContext) -> LuaResult<()> {
    let cfg = SqlConfig {
        query_timeout: super::config::sql_query_timeout(),
    };
    mlua_batteries::sql::register_with(
        lua,
        Arc::clone(&ctx.sql_conn),
        Arc::clone(&ctx.sql_interrupt),
        cfg,
    )?;

    // Load std.sql.register_tools (LLM-facing helper; requires `tool` global).
    lua.load(include_str!("sql_tools.lua"))
        .set_name("std.sql.register_tools")
        .exec()?;

    Ok(())
}