Motivation

Improve compatibility with OpenAI API.

Modification

prompt of /v1/completions is

a string, array of strings, array of tokens, or array of token arrays.

https://platform.openai.com/docs/api-reference/completions/create#completions-create-prompt

Support array of tokens and array of token arrays.

Checklist

• 1. Ensure pre-commit pre-commit run --all-files or other linting tools are used to fix potential lint issues.
• 2. Confirm that modifications are covered by complete unit tests. If not, please add more unit tests for correctness.
• 3. Modify documentation as needed, such as docstrings or example tutorials.