feat: add auto_generate macro and minimal_llm_function

Author: bregydoc

examples/over_process.rs

@@ -0,0 +1,12 @@
+use auto_rust::auto_generate;
+
+fn main() {
+    let a = auto_generate!("a random number between 1 and 10");
+
+    print_type_of(&a);
+    println!("{}", a);
+}
+
+fn print_type_of<T>(_: &T) {
+    println!("{}", std::any::type_name::<T>())
+}

src/generator.rs

@@ -54,3 +54,11 @@ pub fn generate_body_function_from_head(
 
     Ok(implementation)
 }
+
+pub fn minimal_llm_function(input: String) -> String {
+    let system_message = "Your task is respond with a string with double quote".to_string();
+
+    let res = open_ai_chat_completions(system_message, input).unwrap();
+
+    res.choices.first().unwrap().to_owned().message.content
+}

src/lib.rs

@@ -7,43 +7,29 @@ use dotenv::dotenv;
 use proc_macro::TokenStream;
 use syn::{ItemFn, __private::ToTokens};
 
-use crate::generator::generate_body_function_from_head;
+use crate::generator::{generate_body_function_from_head, minimal_llm_function};
 
+/// This macro gets an input like "String, "This is a llm generated function"" and returns a function that returns a String
 #[proc_macro]
-pub fn implement(_item: TokenStream) -> TokenStream {
-    // TODO: Evaluate the use of dotenv in this crate
+pub fn auto_generate(item: TokenStream) -> TokenStream {
     dotenv().ok();
 
-    let implemented_fn = generate_body_function_from_head(_item.to_string(), None).unwrap();
+    let res = minimal_llm_function(item.to_string());
+    // println!("{:?}", res);
 
-    // println!("{}", implemented_fn);
-
-    implemented_fn.parse().unwrap()
+    res.parse().unwrap()
 }
 
 #[proc_macro_attribute]
 pub fn auto_implement(args: TokenStream, input: TokenStream) -> TokenStream {
-    // env::vars()
-    //     .map(|v| format!("{}: {}\n", v.0, v.1))
-    //     .for_each(|v| print!("{}", v));
-
-    // let this_file = file!();
-    // column!();
-
-    // println!("This file: {}", this_file);
-
     let ast: ItemFn = syn::parse(input).expect("Failed to parse input as a function");
 
     let context = args.to_string();
 
-    // println!("Context: {}", context);
-
     let mut prompt_input = String::new();
 
     let fn_header = ast.sig.to_token_stream().to_string();
 
-    // println!("Function header: {}", fn_header);
-
     for attr in ast.attrs {
         let data = attr.to_token_stream().to_string();
 
@@ -53,27 +39,9 @@ pub fn auto_implement(args: TokenStream, input: TokenStream) -> TokenStream {
 
     prompt_input.push_str(&fn_header);
 
-    // println!("Information extracted: {:?}", prompt_input);
-
     dotenv().ok();
 
     let implemented_fn = generate_body_function_from_head(prompt_input, Some(context)).unwrap();
 
-    // println!("\n{}\n", implemented_fn);
-
-    // #[allow(long_running_const_eval)]
-
-    // loop {
-    // let mut line = String::new();
-    // let b1 = std::io::stdin().read_line(&mut line).unwrap();
-    // println!("no of bytes read , {}", b1);
-    // }
-
-    // static line = String::new();
-    // println!("Enter your name :");
-    //
-    // println!("Hello , {}", line);
-    // println!("no of bytes read , {}", b1);
-
     implemented_fn.parse().unwrap()
 }