add ai_functions

Author: maltsev-dev

src/ai_functions/ai_func_architect.rs

@@ -0,0 +1,46 @@
+use auto_gpt_agent::function_to_string as ai;
+
+
+#[ai]
+pub fn print_project_scope(_project_description: &str) {
+    /// Input: Takes in a user request to build a website project description
+    /// Function: Converts user request into JSON response of information items required for a website build.
+    /// Important: At least one of the bool results must be true
+    /// Output: Prints an object response in the following format:
+    ///   {
+    ///     "is_crud_required": bool, // true if site needs CRUD functionality
+    ///     "is_user_login_and_logout": bool // true if site needs users to be able to log in and log out
+    ///     "is_external_urls_required": bool // true if site needs to fetch data from third part providers
+    ///   }
+    /// Example 1:
+    ///   user_request = "I need a full stack website that accepts users and gets stock price data"
+    ///   prints:
+    ///   {
+    ///     "is_crud_required": true
+    ///     "is_user_login_and_logout": true
+    ///     "is_external_urls_required": bool true
+    ///   }
+    /// Example 2:
+    ///   user_request = "I need a simple TODO app"
+    ///   prints:
+    ///   {
+    ///     "is_crud_required": true
+    ///     "is_user_login_and_logout": false
+    ///     "is_external_urls_required": bool false
+    ///   }
+    println!(OUTPUT);
+}
+
+#[ai]
+pub fn print_site_urls(_project_description: &str) {
+    /// Input: Takes in a project description of a website build
+    /// Function: Outputs a list of external public API endpoints that should be used in the building of the website
+    /// Important: Only selects url endpoint(s) which do not require any API Keys at all
+    /// Output: Prints a list response of external urls in the following format:
+    /// ["url1", "url2", "url3", ...]
+    /// Example:
+    ///   website_team_spec = "website_purpose: Some("\"Provides Crypto Price Data from Binance and Kraken\"",)"
+    ///   prints:
+    /// ["https://api.binance.com/api/v3/exchangeInfo", "https://api.binance.com/api/v3/klines?symbol=BTCUSDT&interval=1d"]
+    println!(OUTPUT);
+}

src/ai_functions/ai_func_backend.rs

@@ -0,0 +1,114 @@
+use auto_gpt_agent::function_to_string as ai;
+
+#[ai]
+pub fn print_backend_webserver_code(_project_description: &str) {
+    /// INPUT: Takes in a PROJECT_DESCRIPTION and CODE_TEMPLATE for a website backend build
+    /// IMPORTANT: The backend code is ONLY an example. If the Project Description requires it, make as many changes as you like.
+    /// IMPORTANT: You do not need to follow the backend code exactly. Write functions that make sense for the users request if required.
+    /// FUNCTION: Takes an existing set of code marked as CODE_TEMPLATE and updates or re-writes it to work for the purpose in the PROJECT_DESCRIPTION
+    /// IMPORTANT: The following libraries are already installed
+    ///   reqwest, serde, serde_json, tokio, actix-web, async-trait, actix_cors
+    /// No other external libraries should be used. Write functions that fit with the description from the PROJECT_DESCRIPTION
+    /// OUTPUT: Print ONLY the code, nothing else. This function ONLY prints code.
+    println!(OUTPUT);
+}
+
+#[ai]
+pub fn print_improved_webserver_code(_project_description: &str) {
+    /// INPUT: Takes in a PROJECT_DESCRIPTION and CODE_TEMPLATE for a website backend build
+    /// FUNCTION: Performs the following tasks:
+    ///   1. Removes any bugs in the code and adds minor additional functionality
+    ///   2. Makes sure everything requested in the spec from a backend standpoint was followed. If not, add the feature. No code should be implemented later. Everything should be written now.
+    ///   3. ONLY writes the code. No commentary.
+    /// IMPORTANT: The following libraries are already installed. Does not use ANY libraries other than what was provided in the template
+    ///   reqwest, serde, serde_json, tokio, actix-web, async-trait
+
+    println!(OUTPUT);
+}
+
+#[ai]
+pub fn print_fixed_code(_project_description: &str) {
+    /// INPUT: Takes in Rust BROKEN_CODE and the ERROR_BUGS found
+    /// FUNCTION: Removes bugs from code
+    /// IMPORTANT: Only prints out the new and improved code. No commentary or anything else
+    println!(OUTPUT);
+}
+
+
+#[ai]
+pub fn print_rest_api_endpoints(_project_description: &str) {
+    /// INPUT: Takes in Rust webserver CODE_INPUT based on actix-web
+    /// FUNCTION: Prints out the JSON schema for url endpoints and their respective types
+    /// LOGIC: Script analyses all code and can categorize into the following object keys:
+    ///   "route": This represents the url path of the endpoint
+    ///   "is_route_dynamic": if a route has curly braces in it such as {symbol} or {id} as an example, then this will be set to true
+    ///   "method": This represents the method being called
+    ///   "request_body": This represents the body of a post method request
+    ///   "response": This represents the output based upon the structs in the code and understanding the functions
+    /// IMPORTANT: Only prints out the JSON schema. No commentary or anything else.
+    /// MUST READ: All keys are strings. Even bool should be wrapped in double quotes as "bool"
+    /// EXAMPLE:
+    /// INPUT_CODE:
+    /// ...
+    /// pub struct Item {
+    ///   pub id: u64,
+    ///   pub name: String,
+    ///   pub completed: bool,
+    /// }
+    /// pub struct User {
+    ///   pub id: u64,
+    ///   pub username: String,
+    ///   pub password: String,
+    /// }
+    /// ...
+    /// HttpServer::new(move || {
+    ///   App::new()
+    ///       .app_data(data.clone())
+    ///       .route("/item", web::post().to(create_item))
+    ///       .route("/item/{id}", web::get().to(read_item))
+    ///       .route("/item/{id}", web::put().to(update_item))
+    ///       .route("/item/{id}", web::delete().to(delete_item))
+    ///       .route("/signup", web::post().to(signup))
+    ///       .route("/crypto", web::get().to(crypto))
+    /// PRINTS JSON FORMATTED OUTPUT:
+    /// [
+    ///   {
+    ///     "route": "/item/{id}",
+    ///     "is_route_dynamic": "true",
+    ///     "method": "get"
+    ///     "request_body": "None",
+    ///     "response": {
+    ///       "id": "number",
+    ///       "name": "string",
+    ///       "completed": "bool",
+    ///     }
+    ///   },
+    ///   {
+    ///     "route": "/item",
+    ///     "is_route_dynamic": "false",
+    ///     "method": "post",
+    ///     "request_body": {
+    ///       "id": "number",
+    ///       "name": "string",
+    ///       "completed": "bool",
+    ///     },
+    ///     "response": "None"
+    ///   },
+    ///   {
+    ///     "route": "/item/{id}",
+    ///     "is_route_dynamic": "true",
+    ///     "method": "delete",
+    ///     "request_body": "None",
+    ///     "response": "None"
+    ///   },
+    ///   {
+    ///     "route": "/crypto",
+    ///     "is_route_dynamic": "false",
+    ///     "method": "get",
+    ///     "request_body": "None",
+    ///     "response": "not_provided"
+    ///   },
+    ///   ... // etc
+    /// ]
+    println!(OUTPUT);
+}

src/ai_functions/ai_func_manager.rs

@@ -0,0 +1,15 @@
+use auto_gpt_agent::function_to_string as ai;
+
+#[ai]
+pub fn convert_user_input_to_goal(_user_request: &str) {
+    /// Input: Takes in a user request
+    /// Function: Converts user request into a short summarized goal
+    /// Output: Prints goal. All outputs start with "build a website that ..."
+    /// Example 1:
+    ///   user_request = "I need a website that lets users login and logout. It needs to look fancy and accept payments."
+    ///   OUTPUT = "build a website that handles users logging in and logging out and accepts payments"
+    /// Example 2:
+    ///   user_request = "Create something that stores crypto price data in a database using supabase and retrieves prices on the frontend."
+    ///   OUTPUT = "build a website that fetches and stores crypto price data within a supabase setup including a frontend UI to fetch the data."
+    println!(OUTPUT)
+}

src/ai_functions/mod.rs

@@ -0,0 +1,3 @@
+pub mod ai_func_architect;
+pub mod ai_func_backend;
+pub mod ai_func_manager;