wip

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