#!/bin/bash

role="$1"
shift

if [ -n "$1" ]; then
	aichat --role "$role" "$@"
else 
	session_ts="$(date +%s)"
	while true; do
		read -e -r -p "> " input
		aichat --save-session --session "$session_ts" --role "$role" "$input"
	done
fi