OpenAI cli client script with streaming output

.zshrc

export OPENAI_API_KEY=sk-xxxxxxx # your API key
ask(){
 ask_openai $1 | openai_parse_stream -
 echo ''
}

ask_openai(){
  curl -N \
  -H "Accept: text/event-stream" \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer $OPENAI_API_KEY" \
  -d "{
    \"model\": \"text-davinci-003\",
    \"prompt\": \"$1\",
    \"temperature\": 0.7,
    \"max_tokens\": 256,
    \"top_p\": 1,
    \"frequency_penalty\": 0,
    \"presence_penalty\": 0,
    \"stream\": true
  }" https://api.openai.com/v1/completions \
  -s
}

openai_parse_stream(){
  cut -c 7- | sed 's/\[DONE\]//' | jq --stream -r -j 'fromstream(1|truncate_stream(inputs))[0].text'
}

useage.sh

$ ask 'What is jq library?'


jq is a lightweight and flexible command-line JSON processor. It is used for transforming and filtering JSON data. It can also be used to parse and inspect JSON data, making it a powerful tool for working with web APIs and other JSON data sources.%

I don't know how can I remove first two lines of \n without interrupting the stream.
It also seems slower than the actual stream, only output is visible few lines at a time.

If anyone know better solution, please let me know in the comment!