Skip to main content

harn_vm/
lib.rs

1#![allow(clippy::result_large_err, clippy::cloned_ref_to_slice_refs)]
2
3pub mod bridge;
4pub mod checkpoint;
5mod chunk;
6mod compiler;
7pub mod events;
8mod http;
9pub mod llm;
10pub mod llm_config;
11pub mod mcp;
12pub mod mcp_server;
13pub mod metadata;
14pub mod orchestration;
15pub mod schema;
16pub mod stdlib;
17pub mod stdlib_modules;
18pub mod store;
19pub mod tracing;
20pub mod value;
21mod vm;
22
23pub use checkpoint::register_checkpoint_builtins;
24pub use chunk::*;
25pub use compiler::*;
26pub use http::{register_http_builtins, reset_http_state};
27pub use llm::register_llm_builtins;
28pub use mcp::{
29    connect_mcp_server, connect_mcp_server_from_json, connect_mcp_server_from_spec,
30    register_mcp_builtins,
31};
32pub use mcp_server::{
33    take_mcp_serve_prompts, take_mcp_serve_registry, take_mcp_serve_resource_templates,
34    take_mcp_serve_resources, tool_registry_to_mcp_tools, McpServer,
35};
36pub use metadata::{register_metadata_builtins, register_scan_builtins};
37pub use stdlib::{
38    register_agent_stdlib, register_core_stdlib, register_io_stdlib, register_vm_stdlib,
39};
40pub use store::register_store_builtins;
41pub use value::*;
42pub use vm::*;
43
44/// Reset all thread-local state that can leak between test runs.
45/// Call this before each test execution for proper isolation.
46pub fn reset_thread_local_state() {
47    llm::reset_llm_state();
48    http::reset_http_state();
49    stdlib::reset_stdlib_state();
50    events::reset_event_sinks();
51}