rig-tools 0.45.0

Coding tools for Rig agents: read, write, edit, search and shell, and the file and process helpers they share.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

//! Runs blocking tool work on its own thread, so no async executor thread
//! blocks. Tools that do file, process or long CPU work use it.

use futures::channel::oneshot;
use rig_core::tool::ToolExecutionError;

/// Run `work` on a new thread and await its answer. A panic in `work`
/// becomes an error.
pub async fn blocking<T: Send + 'static>(
    work: impl FnOnce() -> Result<T, ToolExecutionError> + Send + 'static,
) -> Result<T, ToolExecutionError> {
    let (sender, receiver) = oneshot::channel();
    std::thread::Builder::new()
        .name("rig-tool".to_owned())
        .spawn(move || {
            sender.send(work()).ok();
        })
        .map_err(|error| ToolExecutionError::other(format!("could not start the tool: {error}")))?;
    receiver
        .await
        .map_err(|_| ToolExecutionError::other("the tool crashed before it answered"))?
}