azure oai (microsoft#920)

* azure oai * price update in notebook * text Davinci * pytorch-lightning version * trigger action in merge queue * types * doc check in mege group
whiskyboy · Feb 16, 2023 · a8a89c2 · a8a89c2
1 parent d62ffda
commit a8a89c2
Show file tree

Hide file tree

Showing 6 changed files with 312 additions and 597 deletions.
diff --git a/.github/workflows/deploy-website.yml b/.github/workflows/deploy-website.yml
@@ -14,6 +14,8 @@ on:
       - 'website/*'
       - '.github/workflows/deploy-website.yml'
   workflow_dispatch:
+  merge_group:
+    types: [checks_requested]
 
 jobs:
   checks:

diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
@@ -20,6 +20,8 @@ on:
       - 'notebook/**'
       - '.github/workflows/python-package.yml'
       - 'setup.py'
+  merge_group:
+    types: [checks_requested]
 
 jobs:
   build:

diff --git a/flaml/integrations/oai/completion.py b/flaml/integrations/oai/completion.py
@@ -45,8 +45,8 @@ class Completion:
         "text-ada-001": 0.0004,
         "text-babbage-001": 0.0005,
         "text-curie-001": 0.002,
-        "code-cushman-001": 0.002,  # TODO: update when available
-        "code-davinci-002": 0.02,  # TODO: update when available
+        "code-cushman-001": 0.024,
+        "code-davinci-002": 0.1,
         "text-davinci-002": 0.02,
         "text-davinci-003": 0.02,
     }