add support for azure openai api (#359)

Author: iaalm

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",