potato-agent 0.25.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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

use pyo3::exceptions::PyRuntimeError;
use pyo3::PyErr;
use thiserror::Error;
use tracing::error;

#[derive(Error, Debug)]
pub enum AgentError {
    #[error("Error: {0}")]
    Error(String),

    #[error("Failed to downcast Python object: {0}")]
    DowncastError(String),

    #[error("Failed to get environment variable: {0}")]
    EnvVarError(#[from] std::env::VarError),

    #[error("Failed to extract client: {0}")]
    ClientExtractionError(String),

    #[error("No ready tasks found but pending tasks remain. Possible circular dependency.")]
    NoTaskFoundError,

    #[error("Failed to create runtime: {0}")]
    CreateRuntimeError(#[source] std::io::Error),

    #[error(transparent)]
    SerializationError(#[from] serde_json::Error),

    #[error(transparent)]
    UtilError(#[from] potato_util::UtilError),

    #[error("Invalid output type: {0}")]
    InvalidOutputType(String),

    #[error("Failed to create tool: {0}")]
    ToolCreationError(String),

    #[error("Invalid tool definition")]
    InvalidToolDefinitionError,

    #[error("{0}")]
    InvalidInput(String),

    #[error("Provider mismatch: prompt provider {0}, agent provider {1}")]
    ProviderMismatch(String, String),

    #[error(transparent)]
    ProviderError(#[from] potato_provider::error::ProviderError),

    #[error("No provider specified for Agent")]
    MissingProviderError,

    #[error(transparent)]
    TypeError(#[from] potato_type::TypeError),

    #[error(transparent)]
    StdIoError(#[from] std::io::Error),

    #[error(transparent)]
    StoreError(#[from] crate::agents::store::StoreError),

    #[error("Not supported: {0}")]
    NotSupportedError(String),

    #[error("Output validation error: {0}")]
    ValidationError(String),

    #[error("Circular agent call detected: agent '{0}' is already in the call stack")]
    CircularAgentCall(String),

    #[error("Agent call disallowed by policy: agent '{0}'")]
    DisallowedAgentCall(String),

    #[error("Sub-agent calls are disallowed by policy")]
    SubAgentCallsDisallowed,

    #[error("Sub-agent needs user input: {0}")]
    SubAgentNeedsInput(String),

    #[error("Callback aborted run: {0}")]
    CallbackAbort(String),

    #[error("agent exceeded maximum iterations ({0})")]
    MaxIterationsExceeded(u32),

    #[error("internal lock was poisoned")]
    LockPoisoned,
}

impl<'a, 'py> From<pyo3::CastError<'a, 'py>> for AgentError {
    fn from(err: pyo3::CastError<'a, 'py>) -> Self {
        AgentError::DowncastError(err.to_string())
    }
}

impl From<AgentError> for PyErr {
    fn from(err: AgentError) -> PyErr {
        let msg = err.to_string();
        error!("{}", msg);
        PyRuntimeError::new_err(msg)
    }
}

impl From<PyErr> for AgentError {
    fn from(err: PyErr) -> Self {
        AgentError::Error(err.to_string())
    }
}