feat: new example and second api is working

Author: bregydoc

examples/complex.rs

@@ -7,6 +7,6 @@ fn is_email(input: String) -> bool {
 }
 
 fn main() {
-    let result = is_email("bregy@minsky.cc".to_string());
+    let result = is_email("bregyminsky.cc".to_string());
     println!("result: {}", result);
 }

examples/pi.rs

@@ -0,0 +1,12 @@
+use auto_rust::auto_implement;
+
+#[auto_implement]
+#[doc = "This function calculates pi with a robust and very fast algorithm."]
+fn calculate_pi_with_n_decimals(n: u32) -> f64 {
+    todo!()
+}
+
+fn main() {
+    let result = calculate_pi_with_n_decimals(10);
+    println!("pi: {}", result);
+}

src/generator.rs

@@ -4,7 +4,7 @@ use proc_macro::TokenStream;
 
 use crate::api::open_ai_chat_completions;
 
-pub fn generate_body_function_from_head(head: TokenStream) -> Result<String, Box<dyn Error>> {
+pub fn generate_body_function_from_head(head: String) -> Result<String, Box<dyn Error>> {
     let system_message = "You are an AI code assistant trained on the GPT-4 architecture. Your task is to generate Rust function body implementations based only on the provided function signatures. When the user provides a function signature using the command '/complete', your response must be the plain text function body, without any explanations, formatting, or code blocks. Do not include the function signature, function name, or any other information in your response. Triple backticks (```) and function signatures are strictly prohibited in your response. Responding with any prohibited content will result in a penalty. 
     example 1:
     INPUT: /complete fn my_ip() -> String

src/lib.rs

@@ -16,7 +16,7 @@ pub fn implement(_item: TokenStream) -> TokenStream {
     // TODO: Evaluate the use of dotenv in this crate
     dotenv().ok();
 
-    let implemented_fn = generate_body_function_from_head(_item).unwrap();
+    let implemented_fn = generate_body_function_from_head(_item.to_string()).unwrap();
 
     println!("{}", implemented_fn);
 
@@ -31,7 +31,7 @@ pub fn auto_implement(args: TokenStream, input: TokenStream) -> TokenStream {
 
     // Search for the information within the attributes.
 
-    let mut target_info = String::new();
+    let mut prompt_input = String::new();
 
     let fn_header = ast.sig.to_token_stream().to_string();
 
@@ -48,11 +48,22 @@ pub fn auto_implement(args: TokenStream, input: TokenStream) -> TokenStream {
         //     break;
         //     // }
         // }
-        println!("{}", data)
+        println!("{}", data);
+
+        prompt_input.push_str(&data);
+        prompt_input.push('\n');
         // }
     }
 
-    println!("Information extracted: {:?}", target_info);
+    prompt_input.push_str(&fn_header);
+
+    println!("Information extracted: {:?}", prompt_input);
+
+    dotenv().ok();
+
+    let implemented_fn = generate_body_function_from_head(prompt_input).unwrap();
+
+    println!("{}", implemented_fn);
 
-    input
+    implemented_fn.parse().unwrap()
 }