nix/home/program/bash/module/Ask.nix

{ ... }: {
	text = ''
		# Ask general AI.
		# Usage: ask <QUERY>
		function ask() {
			curl http://localhost:11434/api/generate -d "{
				\"model\":\"''${OLLAMA_MODEL}\",
				\"raw\":true,
				\"prompt\":\"''${*}\"
			}" 2> /dev/null | parallel -j1 -- "printf '%s\n' {} | jq -r .response | sed -e 's/^$/\+\+\+/' | tr -d '\n' | sed -e 's/\+\+\+/\n/'"
			echo
		}

		# Specify ask model.
		function ask_model() {
			export OLLAMA_MODEL="''${1}"
		}

		function _complete_ask_model() {
			local IFS=$'\n'
			local models=($(ollama list | sed -e "1d" | cut -f1))
			_autocomplete_first ''${models[@]}
		}

		complete -F _complete_ask_model ask_model
	'';
}