saya-cli 0.2.0

Database-aware AI agent for the terminal: full-screen TUI, schema discovery, and bounded read-only SQL over PostgreSQL, MySQL, DuckDB, and Snowflake.
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

//! Blocking helpers that execute a command and RETURN a renderable event,
//! instead of printing to stdout (which would corrupt the alternate screen).
//! The TUI renders the returned event into the transcript.

use crate::config::runtime::RuntimeConfig;
use crate::render::TerminalEvent;
use saya_connectors::{ConnectorOptions, build_connector_with_prompt};
use saya_types::QueryRequest;

/// Runs a raw SQL query against `profile_name` and returns the result (or an
/// error) as a `TerminalEvent`. Read-only + row-limit enforcement happen inside
/// the connector's `execute`. Never prompts (the TUI owns the screen).
pub(crate) async fn run_sql(
    runtime: &RuntimeConfig,
    profile_name: Option<&str>,
    sql: &str,
) -> TerminalEvent {
    let Some(name) = profile_name else {
        return TerminalEvent::Error {
            message: "No active profile. Use /connect <profile> first.".into(),
        };
    };
    let profile = match runtime.named_profile(name) {
        Ok(profile) => profile,
        Err(error) => {
            return TerminalEvent::Error {
                message: error.to_string(),
            };
        }
    };
    let settings = ConnectorOptions {
        query_timeout_seconds: runtime.resolved.query_timeout_seconds,
        ..Default::default()
    };
    let connector =
        match build_connector_with_prompt(profile, &runtime.secret_resolver(), settings, false)
            .await
        {
            Ok(connector) => connector,
            Err(error) => {
                return TerminalEvent::Error {
                    message: error.to_string(),
                };
            }
        };
    if let Err(error) = connector.connect().await {
        return TerminalEvent::Error {
            message: error.to_string(),
        };
    }
    match connector
        .execute(QueryRequest::new(
            sql.to_string(),
            runtime.resolved.max_rows,
        ))
        .await
    {
        Ok(result) => TerminalEvent::QueryResult { result },
        Err(error) => TerminalEvent::Error {
            message: error.to_string(),
        },
    }
}