diff --git a/arch-user/bin/ai b/arch-user/bin/ai
index 9442ad9..ab8c7c2 100755
--- a/arch-user/bin/ai
+++ b/arch-user/bin/ai
@@ -1,14 +1,21 @@
 #!/bin/bash
 
+session_ts="$(date +%s)"
 role="$1"
 shift
 
-session_ts="$(date +%s)"
+work() {
+	aichat --save-session --session "$session_ts" --role "$role" "$*"
+}
+
 if [ -n "$1" ]; then
-	aichat --save-session --session "$session_ts" --role "$role" "$@"
+	work "$@"
 fi
 
 while true; do
 	read -e -r -p "> " input
-	aichat --save-session --session "$session_ts" --role "$role" "$input"
+	if [ -z "$input" ]; then
+		input="$(rlwrap cat)"
+	fi
+	work "$input"
 done
diff --git a/arch-user/config/aichat/config.yaml.tmpl b/arch-user/config/aichat/config.yaml.tmpl
index 1d20836..cacbf79 100644
--- a/arch-user/config/aichat/config.yaml.tmpl
+++ b/arch-user/config/aichat/config.yaml.tmpl
@@ -3,6 +3,7 @@
 
 compress_threshold: 100000
 summarize_prompt: 'Summarize the discussion in 5000 words or less to use as a prompt for future context.'
+save_session: false
 
 clients: