From 23633ae744f2dd39f765279dd9d3ece805bcdf40 Mon Sep 17 00:00:00 2001
From: Chi Wang <wang.chi@microsoft.com>
Date: Thu, 21 Sep 2023 15:39:52 +0000
Subject: [PATCH] cleanup

---
 .github/workflows/openai.yml          | 6 +++---
 notebook/agentchat_RetrieveChat.ipynb | 6 +++---
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/openai.yml b/.github/workflows/openai.yml
index c91e69fbb..c06ff5e3e 100644
--- a/.github/workflows/openai.yml
+++ b/.github/workflows/openai.yml
@@ -34,7 +34,7 @@ jobs:
         run: |
           docker --version
           python -m pip install --upgrade pip wheel
-          pip install -e .[tune]
+          pip install -e .
           python -c "import autogen"
           pip install coverage pytest-asyncio datasets
       - name: Install packages for test when needed
@@ -46,7 +46,7 @@ jobs:
         run: |
           pip install -e .[mathchat]
       - name: Install packages for RetrieveChat when needed
-        if: matrix.python-version != '3.11'
+        if: matrix.python-version == '3.9'
         run: |
           pip install -e .[retrievechat]
       - name: Coverage
@@ -68,7 +68,7 @@ jobs:
           WOLFRAM_ALPHA_APPID: ${{ secrets.WOLFRAM_ALPHA_APPID }}
           OAI_CONFIG_LIST: ${{ secrets.OAI_CONFIG_LIST }}
         run: |
-          pip install nbconvert nbformat ipykernel
+          pip install -e.[blendsearch] nbconvert nbformat ipykernel
           coverage run -a -m pytest test/test_notebook.py
           coverage xml
           cat "$(pwd)/test/executed_openai_notebook_output.txt"
diff --git a/notebook/agentchat_RetrieveChat.ipynb b/notebook/agentchat_RetrieveChat.ipynb
index bdc994d0f..169e6b1a3 100644
--- a/notebook/agentchat_RetrieveChat.ipynb
+++ b/notebook/agentchat_RetrieveChat.ipynb
@@ -37,9 +37,9 @@
     "\n",
     "## Requirements\n",
     "\n",
-    "AutoGen requires `Python>=3.8`. To run this notebook example, please install the [retrievechat] option and flaml.\n",
+    "AutoGen requires `Python>=3.8`. To run this notebook example, please install the [retrievechat] option.\n",
     "```bash\n",
-    "pip install \"pyautogen[retrievechat]\" flaml\n",
+    "pip install \"pyautogen[retrievechat]\" \"flaml[automl]\"\n",
     "```"
    ]
   },
@@ -49,7 +49,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# %pip install \"pyautogen[retrievechat]~=0.1.0\" flaml"
+    "# %pip install \"pyautogen[retrievechat]~=0.1.1\" \"flaml[automl]\""
    ]
   },
   {