mod agent_sessions;
mod agents;
mod concurrency;
mod crypto;
mod datetime;
mod fs;
pub mod host;
mod io;
mod iter;
pub(crate) mod json;
mod logging;
mod math;
mod path;
pub mod process;
mod regex;
mod sets;
mod shapes;
mod skills;
mod strings;
pub mod template;
mod testing;
pub(crate) mod tools;
pub mod tracing;
mod types;
use crate::http::register_http_builtins;
use crate::llm::register_llm_builtins;
use crate::mcp::register_mcp_builtins;
use crate::mcp_server::register_mcp_server_builtins;
use crate::vm::Vm;
pub(crate) use crate::schema::{json_to_vm_value, schema_result_value};
pub(crate) fn set_thread_source_dir(dir: &std::path::Path) {
process::set_thread_source_dir(dir);
}
pub fn register_core_stdlib(vm: &mut Vm) {
types::register_type_builtins(vm);
math::register_math_builtins(vm);
strings::register_string_builtins(vm);
json::register_json_builtins(vm);
datetime::register_datetime_builtins(vm);
regex::register_regex_builtins(vm);
crypto::register_crypto_builtins(vm);
path::register_path_helper_builtins(vm);
sets::register_set_builtins(vm);
iter::register_iter_builtins(vm);
shapes::register_shape_builtins(vm);
testing::register_testing_builtins(vm);
}
pub fn register_io_stdlib(vm: &mut Vm) {
io::register_io_builtins(vm);
host::register_host_builtins(vm);
fs::register_fs_builtins(vm);
process::register_process_builtins(vm);
process::register_path_builtins(vm);
tracing::register_tracing_builtins(vm);
}
pub fn register_agent_stdlib(vm: &mut Vm) {
concurrency::register_concurrency_builtins(vm);
tools::register_tool_builtins(vm);
skills::register_skill_builtins(vm);
agents::register_agent_builtins(vm);
agent_sessions::register_agent_session_builtins(vm);
register_http_builtins(vm);
register_llm_builtins(vm);
register_mcp_builtins(vm);
register_mcp_server_builtins(vm);
}
pub fn register_vm_stdlib(vm: &mut Vm) {
register_core_stdlib(vm);
register_io_stdlib(vm);
register_agent_stdlib(vm);
}
pub fn stdlib_builtin_names() -> Vec<String> {
let mut vm = Vm::new();
register_vm_stdlib(&mut vm);
let tmp = std::path::PathBuf::from("/tmp");
crate::store::register_store_builtins(&mut vm, &tmp);
crate::checkpoint::register_checkpoint_builtins(&mut vm, &tmp, "default");
crate::metadata::register_metadata_builtins(&mut vm, &tmp);
crate::metadata::register_scan_builtins(&mut vm);
let mut names = vm.builtin_names();
for extra in [
"spawn",
"await",
"cancel",
"cancel_graceful",
"is_cancelled",
] {
names.push(extra.to_string());
}
names
}
pub fn reset_stdlib_state() {
logging::reset_logging_state();
process::reset_process_state();
fs::reset_fs_state();
json::reset_json_state();
host::reset_host_state();
agents::records::reset_eval_metrics();
tools::clear_current_tool_registry();
template::reset_prompt_registry();
}