Add basic time sense

Author: Snehal-Reddy

src/agent/context.rs

@@ -2,6 +2,8 @@
 
 use std::path::Path;
 
+use time::OffsetDateTime;
+
 use crate::llm::{Message, Role};
 use crate::workspace;
 
@@ -21,14 +23,14 @@ pub fn build_messages(
 ) -> Vec<Message> {
     let mut system = String::new();
 
-    // Identity: time, workspace
-    let now_utc = std::time::SystemTime::now()
-        .duration_since(std::time::UNIX_EPOCH)
-        .map(|d| d.as_secs())
-        .unwrap_or(0);
+    // Identity: current date/time (human-readable + timezone) and Unix, workspace
+    let now = OffsetDateTime::now_utc();
+    let now_unix = now.unix_timestamp();
     system.push_str("You are iCrab, a minimal personal AI assistant. ");
-    system.push_str("Current time (Unix): ");
-    system.push_str(&now_utc.to_string());
+    system.push_str("Current time: ");
+    system.push_str(&format!("{} {} {} UTC. ", now.weekday(), now.date(), now.time()));
+    system.push_str("Unix: ");
+    system.push_str(&now_unix.to_string());
     system.push_str(". Workspace: ");
     system.push_str(workspace_path.to_string_lossy().as_ref());
     system.push_str(".\n\n");
@@ -111,3 +113,56 @@ pub fn build_messages(
     });
     messages
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    const WEEKDAYS: &[&str] = &[
+        "Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday",
+    ];
+
+    #[test]
+    fn system_prompt_includes_human_readable_time_and_unix() {
+        let workspace = std::env::temp_dir();
+        let messages = build_messages(
+            &workspace,
+            &[],
+            "",
+            "hello",
+            None,
+            "",
+            &[],
+            None,
+        );
+        let system = &messages[0].content;
+        assert!(
+            system.contains("Current time:"),
+            "system prompt should include 'Current time:'"
+        );
+        assert!(
+            system.contains(" UTC."),
+            "system prompt should include ' UTC.'"
+        );
+        assert!(
+            system.contains("Unix: "),
+            "system prompt should include 'Unix: '"
+        );
+        let has_weekday = WEEKDAYS.iter().any(|w| system.contains(w));
+        assert!(
+            has_weekday,
+            "system prompt should include a weekday (e.g. Wednesday)"
+        );
+        // Unix timestamp should be a positive number after "Unix: "
+        let unix_prefix = "Unix: ";
+        let start = system.find(unix_prefix).unwrap() + unix_prefix.len();
+        let rest = &system[start..];
+        let end = rest.find('.').unwrap_or(rest.len());
+        let unix_str = rest[..end].trim();
+        assert!(
+            unix_str.parse::<u64>().is_ok(),
+            "Unix value should be numeric, got: {}",
+            unix_str
+        );
+    }
+}

src/cron_runner.rs

@@ -82,24 +82,32 @@ mod tests {
     use super::*;
     use crate::tools::cron::{CronStore, Schedule};
 
+    fn unix_now() -> u64 {
+        std::time::SystemTime::now()
+            .duration_since(std::time::UNIX_EPOCH)
+            .map(|d| d.as_secs())
+            .unwrap_or(0)
+    }
+
     #[tokio::test]
     async fn tick_fires_due_direct_job() {
         let dir = std::env::temp_dir().join("icrab_cron_runner_direct");
         let _ = std::fs::remove_dir_all(&dir);
         std::fs::create_dir_all(&dir).unwrap();
         let store = CronStore::empty(&dir);
+        let base = unix_now();
         store
             .add(
                 None,
                 "Reminder".to_string(),
                 JobAction::Direct,
-                Schedule::Once { at_unix: 100 },
+                Schedule::Once { at_unix: base + 60 },
                 12345,
             )
             .unwrap();
         let (inbound_tx, _inbound_rx) = mpsc::channel(8);
         let (outbound_tx, mut outbound_rx) = mpsc::channel(8);
-        tick_once(&store, &inbound_tx, &outbound_tx, 500).await;
+        tick_once(&store, &inbound_tx, &outbound_tx, base + 61).await;
         let msg = outbound_rx.try_recv().unwrap();
         assert_eq!(msg.chat_id, 12345);
         assert_eq!(msg.text, "Reminder");
@@ -116,18 +124,19 @@ mod tests {
         let _ = std::fs::remove_dir_all(&dir);
         std::fs::create_dir_all(&dir).unwrap();
         let store = CronStore::empty(&dir);
+        let base = unix_now();
         store
             .add(
                 None,
                 "Agent task".to_string(),
                 JobAction::Agent,
-                Schedule::Once { at_unix: 100 },
+                Schedule::Once { at_unix: base + 60 },
                 999,
             )
             .unwrap();
         let (inbound_tx, mut inbound_rx) = mpsc::channel(8);
         let (outbound_tx, _outbound_rx) = mpsc::channel(8);
-        tick_once(&store, &inbound_tx, &outbound_tx, 500).await;
+        tick_once(&store, &inbound_tx, &outbound_tx, base + 61).await;
         let msg = inbound_rx.try_recv().unwrap();
         assert_eq!(msg.chat_id, 999);
         assert_eq!(msg.text, "Agent task");
@@ -142,18 +151,19 @@ mod tests {
         let _ = std::fs::remove_dir_all(&dir);
         std::fs::create_dir_all(&dir).unwrap();
         let store = CronStore::empty(&dir);
+        let base = unix_now();
         store
             .add(
                 None,
                 "Later".to_string(),
                 JobAction::Direct,
-                Schedule::Once { at_unix: 99999 },
+                Schedule::Once { at_unix: base + 1000 },
                 1,
             )
             .unwrap();
         let (inbound_tx, _inbound_rx) = mpsc::channel(8);
         let (outbound_tx, mut outbound_rx) = mpsc::channel(8);
-        tick_once(&store, &inbound_tx, &outbound_tx, 500).await;
+        tick_once(&store, &inbound_tx, &outbound_tx, base + 500).await;
         assert!(outbound_rx.try_recv().is_err());
         let _ = std::fs::remove_dir_all(&dir);
     }