Struct ToolUsePlugin 

pub struct ToolUsePlugin { /* private fields */ }

Tool use plugin

Implementations

impl ToolUsePlugin

pub fn new(registry: Arc<ToolRegistry>) -> Self

Create a new tool use plugin

pub fn with_config( registry: Arc<ToolRegistry>, config: ToolUsePluginConfig, ) -> Self

Create with custom configuration

Trait Implementations

impl Debug for ToolUsePlugin

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Plugin for ToolUsePlugin

fn name(&self) -> &str

Plugin name

fn enforce(&self) -> PluginPhase

Plugin execution phase

fn transform_params<'life0, 'life1, 'async_trait>( &'life0 self, params: TextParams, _ctx: &'life1 RequestContext, ) -> Pin<Box<dyn Future<Output = Result<TextParams, AiError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Transform request parameters

fn transform_result<'life0, 'life1, 'async_trait>( &'life0 self, result: TextResult, _ctx: &'life1 RequestContext, ) -> Pin<Box<dyn Future<Output = Result<TextResult, AiError>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Transform result

fn resolve_model<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, _model_id: &'life1 str, _ctx: &'life2 RequestContext, ) -> Pin<Box<dyn Future<Output = Result<Option<String>, AiError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, Self: 'async_trait,

Resolve model ID

fn load_template<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, _template_name: &'life1 str, _ctx: &'life2 RequestContext, ) -> Pin<Box<dyn Future<Output = Result<Option<Vec<Message>>, AiError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, Self: 'async_trait,

Load template

fn on_request_start<'life0, 'life1, 'async_trait>( &'life0 self, _ctx: &'life1 RequestContext, ) -> Pin<Box<dyn Future<Output = Result<(), AiError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, Self: 'async_trait,

Hook called when a request starts

fn on_request_end<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, _ctx: &'life1 RequestContext, _result: &'life2 TextResult, ) -> Pin<Box<dyn Future<Output = Result<(), AiError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, Self: 'async_trait,

Hook called when a request ends successfully

fn on_error<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, _error: &'life1 AiError, _ctx: &'life2 RequestContext, ) -> Pin<Box<dyn Future<Output = Result<(), AiError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, Self: 'async_trait,

Hook called when an error occurs

fn transform_stream( &self, stream: Box<dyn Stream<Item = Result<TextChunk, AiError>> + Unpin + Send>, ) -> Box<dyn Stream<Item = Result<TextChunk, AiError>> + Unpin + Send>

Transform text stream