Struct MockAgentProvider 

pub struct MockAgentProvider { /* private fields */ }

An AgentProvider backed by a closure.

Examples

use ironflow_core::provider::AgentOutput;
use ironflow_engine::testing::MockAgentProvider;
use serde_json::json;

let provider = MockAgentProvider::new(|cfg| {
    assert!(cfg.prompt.contains("review"));
    Ok(AgentOutput::new(json!({"score": 9})))
});

Implementations

impl MockAgentProvider

pub fn new( f: impl Fn(&AgentConfig) -> Result<AgentOutput, AgentError> + Send + Sync + 'static, ) -> Self

Answer every invocation with f.

Examples

use ironflow_core::provider::AgentOutput;
use ironflow_engine::testing::MockAgentProvider;
use serde_json::json;

let provider = MockAgentProvider::new(|_cfg| Ok(AgentOutput::new(json!("done"))));

Trait Implementations

impl AgentProvider for MockAgentProvider

fn invoke<'a>(&'a self, config: &'a AgentConfig) -> InvokeFuture<'a>

Execute a single agent invocation with the given configuration.

fn invoke_with_logs<'a>( &'a self, config: &'a AgentConfig, log_sink: Arc<dyn LogSink>, ) -> Pin<Box<dyn Future<Output = Result<AgentOutput, AgentError>> + Send + 'a>>

Execute an agent invocation with real-time log streaming.

fn release_run<'a>( &'a self, run_id: &'a str, ) -> Pin<Box<dyn Future<Output = Result<(), AgentError>> + Send + 'a>>

Stop whatever a previous execution of the run run_id left running outside the worker process, before the run executes again.

impl Debug for MockAgentProvider

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

Formats the value using the given formatter.