feat:Context-Aware Queries

Author: MostlyKIGuess

src/main.rs

@@ -110,7 +110,16 @@ fn main() {
 }
 
 fn process_query(query: &str, api_key: &str, config_dir: &PathBuf, history_path: &str) {
-    match api::get_gemini_response(query, api_key, config_dir) {
+
+    let recent_history = memory::get_recent_history(history_path, 3);
+
+    let context_query = if !recent_history.is_empty() {
+        format!("Previous conversations for context:\n{}\n\nNew query: {}", recent_history, query)
+    } else{
+        query.to_string()
+    };
+
+    match api::get_gemini_response(&context_query, api_key, config_dir) {
         Ok(response) => {
             let formatted_response = utils::render_response(&response);
             println!("{}", formatted_response);

src/memory.rs

@@ -27,3 +27,20 @@ pub fn load_chat(file_path: &str) -> Vec<ChatLog> {
         .filter_map(|line| serde_json::from_str::<ChatLog>(&line.unwrap()).ok())
         .collect()
 }
+
+pub fn get_recent_history(history_path: &str, limit: usize) -> String{
+    let chat_history = load_chat(history_path);
+    if chat_history.is_empty() {
+        return String::new();
+    }
+    let recent_chats = chat_history.iter().rev().take(limit).rev();
+    
+    let mut result = String::new();
+    for (idx, entry) in recent_chats.enumerate() {
+        result.push_str(&format!("--- Previous Conversation #{} ---\n", idx + 1));
+        result.push_str(&format!("User: {}\n", entry.user));
+        result.push_str(&format!("AI: {}\n\n", entry.bot));
+    }
+    
+    result
+}