Struct QueryEngine 

pub struct QueryEngine { /* private fields */ }

Implementations

impl QueryEngine

pub fn new(config: QueryEngineConfig) -> Self

pub fn register_tool<F>(&mut self, name: String, executor: F)

where F: Fn(Value, &ToolContext) -> Pin<Box<dyn Future<Output = Result<ToolResult, AgentError>> + Send>> + Send + Sync + 'static,

Register a tool executor

pub fn set_messages(&mut self, messages: Vec<Message>)

Set initial messages (for continuing a conversation)

pub async fn execute_tool( &mut self, name: &str, input: Value, ) -> Result<ToolResult, AgentError>

Execute a tool by name

pub fn set_hook_registry(&self, registry: HookRegistry)

Set the hook registry

pub fn set_event_callback<F>(&mut self, callback: F)

where F: Fn(AgentEvent) + Send + Sync + 'static,

Set the event callback for agent events (tool start/complete/error, thinking, done)

pub fn get_turn_count(&self) -> u32

pub fn get_usage(&self) -> TokenUsage

Get total token usage from all API calls

pub fn get_messages(&self) -> Vec<Message>

pub async fn submit_message( &mut self, prompt: &str, ) -> Result<(String, ExitReason), AgentError>