diff --git a/openai/cli.py b/openai/cli.py index 91daa1c5ff..e1bf3eac06 100644 --- a/openai/cli.py +++ b/openai/cli.py @@ -123,6 +123,7 @@ def create(cls, args): resp = openai.ChatCompletion.create( # Required model=args.model, + engine=args.engine, messages=messages, # Optional n=args.n, @@ -714,12 +715,6 @@ def help(args): req = sub.add_argument_group("required arguments") opt = sub.add_argument_group("optional arguments") - req.add_argument( - "-m", - "--model", - help="The model to use.", - required=True, - ) req.add_argument( "-g", "--message", @@ -729,6 +724,19 @@ def help(args): help="A message in `{role} {content}` format. Use this argument multiple times to add multiple messages.", required=True, ) + + group = opt.add_mutually_exclusive_group() + group.add_argument( + "-e", + "--engine", + help="The engine to use. See https://learn.microsoft.com/en-us/azure/cognitive-services/openai/chatgpt-quickstart?pivots=programming-language-python for more about what engines are available.", + ) + group.add_argument( + "-m", + "--model", + help="The model to use.", + ) + opt.add_argument( "-n", "--n",