feat: added examples to system prompt

Author: bregydoc

Cargo.lock

@@ -8,6 +8,7 @@ proc-macro = true
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+dotenv = "0.15.0"
 reqwest = { version = "0.11.16", features = ["blocking", "json"] }
 serde = "1.0.162"
 serde_derive = "1.0.162"

Cargo.toml

@@ -0,0 +1,8 @@
+use auto_rust::implement;
+
+implement!(fn is_email(input: String) -> bool);
+
+fn main() {
+    let result = is_email("bregy@minsky.cc".to_string());
+    println!("result: {}", result);
+}

examples/complex.rs

@@ -14,9 +14,8 @@ pub fn generate_body_function_from_head(
 
     let body_str = res.choices.first().unwrap().to_owned().message.content;
 
-    // let body = quote! {
-    //     "Hello, world!".into()
-    // };
+    //TODO: improve the prompt to eliminate the need for this
+    let body_str = body_str.trim_matches('`').to_string();
 
     let implementation = format!("{} {{{}}}", head, body_str);
 

examples/basic.rs examples/simple.rsexamples/basic.rs renamed to examples/simple.rs

@@ -4,12 +4,14 @@ mod api;
 mod data;
 mod generator;
 
+use dotenv::dotenv;
 use proc_macro::TokenStream;
 
 use crate::generator::generate_body_function_from_head;
 
 #[proc_macro]
 pub fn implement(_item: TokenStream) -> TokenStream {
+    dotenv().ok();
     // println!("implement: {:?}", _item);
     // let resp = reqwest::blocking::get("https://httpbin.org/ip")
     //     .unwrap()
@@ -18,7 +20,21 @@ pub fn implement(_item: TokenStream) -> TokenStream {
 
     // let ip = resp.get("origin").unwrap().to_owned();
 
-    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.".to_string();
+    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
+    OUTPUT: use std::net::UdpSocket;
+    
+        let udp_socket = UdpSocket::bind(\"0.0.0.0:0\").unwrap();
+        udp_socket.connect(\"8.8.8.8:80\").unwrap();
+        let socket_addr = udp_socket.local_addr().unwrap();
+        let ip_addr = socket_addr.ip();
+        ip_addr.to_string()
+    
+    example 2:
+    INPUT: /complete fn hello_world() -> String
+    OUTPUT: \"Hello World\".to_string()
+".to_string();
     let implemented_fn = generate_body_function_from_head(system_message, _item).unwrap();
 
     println!("{}", implemented_fn);