potato-agent 0.26.0

Potato brands
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use crate::agents::{
    error::AgentError, run_context::ResumeContext, session::SessionState, types::AgentResponse,
};
use async_trait::async_trait;
use std::fmt::Debug;

/// Final result of a completed agent run.
#[derive(Debug, Clone)]
pub struct AgentRunResult {
    pub final_response: AgentResponse,
    pub iterations: u32,
    pub completion_reason: String,
    /// Combined text from all parallel agents (set by `CollectAll` merge strategy).
    pub combined_text: Option<String>,
}

/// Outcome of calling `AgentRunner::run()`.
#[derive(Debug)]
pub enum AgentRunOutcome {
    /// The run completed — final answer available.
    Complete(Box<AgentRunResult>),
    /// The agent is paused and needs user input before continuing.
    NeedsInput {
        question: String,
        resume_context: ResumeContext,
    },
}

impl AgentRunOutcome {
    pub fn complete(result: AgentRunResult) -> Self {
        Self::Complete(Box::new(result))
    }
}

/// Core trait implemented by all runnable agents (Agent, SequentialAgent, ParallelAgent, AgentTool).
#[async_trait]
pub trait AgentRunner: Send + Sync + Debug {
    fn id(&self) -> &str;

    async fn run(
        &self,
        input: &str,
        session: &mut SessionState,
    ) -> Result<AgentRunOutcome, AgentError>;

    /// Resume a previously paused run with the user's answer.
    async fn resume(
        &self,
        user_answer: &str,
        ctx: ResumeContext,
        session: &mut SessionState,
    ) -> Result<AgentRunOutcome, AgentError>;
}