autogpt 0.1.15

🦀 A Pure Rust Framework For Building AGIs. 
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
109
110
111
112
113
114
115
116
117
118
119

use crate::agents::agent::AgentGPT;
use crate::agents::architect::ArchitectGPT;
use crate::agents::backend::BackendGPT;
#[cfg(feature = "img")]
use crate::agents::designer::DesignerGPT;
use crate::agents::frontend::FrontendGPT;
#[cfg(feature = "git")]
use crate::agents::git::GitGPT;
use crate::agents::optimizer::OptimizerGPT;
use crate::common::utils::Task;
use crate::traits::agent::Agent;
use crate::traits::functions::{AsyncFunctions, Functions};
use anyhow::Result;

/// Enum representing different types of GPT agents.
#[derive(Debug, Clone)]
pub enum AgentType {
    /// Architect GPT agent.
    Architect(ArchitectGPT),
    /// Backend GPT agent.
    Backend(BackendGPT),
    /// Frontend GPT agent.
    Frontend(FrontendGPT),
    /// Designer GPT agent.
    #[cfg(feature = "img")]
    Designer(DesignerGPT),
    /// Git GPT agent.
    #[cfg(feature = "git")]
    Git(GitGPT),
    /// Optimizer GPT agent.
    Optimize(OptimizerGPT),
}

impl Default for AgentType {
    fn default() -> Self {
        AgentType::Architect(ArchitectGPT::default())
    }
}

impl AgentType {
    /// Asynchronously executes tasks associated with the agent.
    ///
    /// # Arguments
    ///
    /// * `tasks` - A mutable reference to tasks to be executed.
    /// * `execute` - A boolean indicating whether to execute the tasks.
    /// * `max_tries` - Maximum number of attempts to execute tasks.
    ///
    /// # Returns
    ///
    /// (`Result<()>`): Result indicating success or failure of task execution.
    ///
    /// # Errors
    ///
    /// Returns an error if there's a failure in executing tasks.
    ///
    /// # Business Logic
    ///
    /// - Executes tasks associated with the agent based on its type.
    ///
    pub async fn execute(
        &mut self,
        tasks: &mut Task,
        execute: bool,
        browse: bool,
        max_tries: u64,
    ) -> Result<()> {
        match self {
            AgentType::Architect(agent) => agent.execute(tasks, execute, browse, max_tries).await,
            AgentType::Backend(agent) => agent.execute(tasks, execute, browse, max_tries).await,
            AgentType::Frontend(agent) => agent.execute(tasks, execute, browse, max_tries).await,
            #[cfg(feature = "img")]
            AgentType::Designer(agent) => agent.execute(tasks, execute, browse, max_tries).await,
            #[cfg(feature = "git")]
            AgentType::Git(agent) => agent.execute(tasks, execute, browse, max_tries).await,
            AgentType::Optimize(agent) => agent.execute(tasks, execute, browse, max_tries).await,
        }
    }

    /// Retrieves the position of the agent.
    ///
    /// # Returns
    ///
    /// (`String`): The position of the agent.
    ///
    /// # Business Logic
    ///
    /// - Retrieves the position of the agent based on its type.
    pub fn position(&self) -> String {
        match self {
            AgentType::Architect(agent) => agent.get_agent().position().to_string(),
            AgentType::Backend(agent) => agent.get_agent().position().to_string(),
            AgentType::Frontend(agent) => agent.get_agent().position().to_string(),
            AgentType::Optimize(agent) => agent.get_agent().position().to_string(),
            #[cfg(feature = "img")]
            AgentType::Designer(agent) => agent.get_agent().position().to_string(),
            #[cfg(feature = "git")]
            AgentType::Git(agent) => agent.get_agent().position().to_string(),
        }
    }

    /// Retrieves the agent.
    ///
    /// # Returns
    ///
    /// (`AgentGPT`): The agent.
    pub fn get_agent(&self) -> &AgentGPT {
        match self {
            AgentType::Architect(agent) => agent.get_agent(),
            AgentType::Backend(agent) => agent.get_agent(),
            AgentType::Frontend(agent) => agent.get_agent(),
            AgentType::Optimize(agent) => agent.get_agent(),
            #[cfg(feature = "img")]
            AgentType::Designer(agent) => agent.get_agent(),
            #[cfg(feature = "git")]
            AgentType::Git(agent) => agent.get_agent(),
        }
    }
}