Struct Query 

pub struct Query { /* private fields */ }

Internal query handler with control protocol support

Implementations

impl Query

pub fn new( transport: Arc<Mutex<Box<dyn Transport + Send>>>, is_streaming_mode: bool, can_use_tool: Option<Arc<dyn CanUseTool>>, hooks: Option<HashMap<String, Vec<HookMatcher>>>, sdk_mcp_servers: HashMap<String, Arc<dyn Any + Send + Sync>>, ) -> Self

Create a new Query handler

pub async fn register_hook_callback_for_test( &self, callback_id: String, callback: Arc<dyn HookCallback>, )

Test helper to register a hook callback with a known ID

This is intended for E2E tests to inject a callback ID that can be referenced by inbound hook_callback control messages.

pub async fn start(&mut self) -> Result<()>

Start the query handler

pub async fn initialize(&mut self) -> Result<()>

Initialize the control protocol

pub async fn stream_input<S>(&mut self, input_stream: S) -> Result<()>

where S: Stream<Item = JsonValue> + Send + 'static,

Stream input messages to the CLI stdin by converting JSON values to InputMessage

pub async fn receive_messages(&mut self) -> Receiver<Result<Message>>

Receive messages

pub async fn interrupt(&mut self) -> Result<()>

Send interrupt request

pub async fn set_permission_mode(&mut self, mode: &str) -> Result<()>

Set permission mode via control protocol

pub async fn set_model(&mut self, model: Option<String>) -> Result<()>

Set the active model via control protocol

pub async fn close(&mut self) -> Result<()>

Close the query handler

pub fn get_initialization_result(&self) -> Option<&JsonValue>

Get initialization result