Merge branch 'main' into add-role-to-reflection-with-llm

microsoft · Apr 29, 2024 · eff19ac · eff19ac
2 parents 04b0597 + 11a4342
commit eff19ac
Show file tree

Hide file tree

Showing 260 changed files with 17,807 additions and 2,710 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -46,7 +46,7 @@ jobs:
           python -m pip install --upgrade pip wheel
           pip install -e .[cosmosdb]
           python -c "import autogen"
-          pip install pytest mock
+          pip install pytest-cov>=5 mock
       - name: Install optional dependencies for code executors
         # code executors and udfs auto skip without deps, so only run for python 3.11
         if: matrix.python-version == '3.11'
@@ -71,12 +71,11 @@ jobs:
         if: matrix.python-version == '3.10'
         run: |
           pip install -e .[test,redis,websockets]
-          coverage run -a -m pytest test --ignore=test/agentchat/contrib --skip-openai --durations=10 --durations-min=1.0
-          coverage xml
+          pytest test --ignore=test/agentchat/contrib --skip-openai --durations=10 --durations-min=1.0
       - name: Test with Cosmos DB
         run: |
           pip install -e .[test,cosmosdb]
-          coverage run -a -m pytest test/cache/test_cosmos_db_cache.py --skip-openai --durations=10 --durations-min=1.0
+          pytest test/cache/test_cosmos_db_cache.py --skip-openai --durations=10 --durations-min=1.0
       - name: Upload coverage to Codecov
         if: matrix.python-version == '3.10'
         uses: codecov/codecov-action@v3

diff --git a/.github/workflows/contrib-openai.yml b/.github/workflows/contrib-openai.yml
@@ -5,14 +5,15 @@ name: OpenAI4ContribTests
 
 on:
   pull_request:
-    branches: ['main']
+    branches: ["main"]
     paths:
-      - 'autogen/**'
-      - 'test/agentchat/contrib/**'
-      - '.github/workflows/contrib-openai.yml'
-      - 'setup.py'
-permissions: {}
-    # actions: read
+      - "autogen/**"
+      - "test/agentchat/contrib/**"
+      - ".github/workflows/contrib-openai.yml"
+      - "setup.py"
+permissions:
+  {}
+  # actions: read
   # checks: read
   # contents: read
   # deployments: read
@@ -24,6 +25,21 @@ jobs:
         python-version: ["3.10"]
     runs-on: ${{ matrix.os }}
     environment: openai1
+    services:
+      pgvector:
+        image: ankane/pgvector
+        env:
+          POSTGRES_DB: postgres
+          POSTGRES_USER: postgres
+          POSTGRES_PASSWORD: ${{ secrets.POSTGRES_PASSWORD }}
+          POSTGRES_HOST_AUTH_METHOD: trust
+        options: >-
+          --health-cmd pg_isready
+          --health-interval 10s
+          --health-timeout 5s
+          --health-retries 5
+        ports:
+          - 5432:5432
     steps:
       # checkout to pr branch
       - name: Checkout
@@ -40,30 +56,25 @@ jobs:
           python -m pip install --upgrade pip wheel
           pip install -e .
           python -c "import autogen"
-          pip install coverage pytest-asyncio
-      - name: Install PostgreSQL
-        run: |
-          sudo apt install postgresql -y
-      - name: Start PostgreSQL service
-        run: sudo service postgresql start
+          pip install pytest-cov>=5 pytest-asyncio
       - name: Install packages for test when needed
         run: |
           pip install docker
-          pip install -e .[retrievechat-qdrant,retrievechat-pgvector]
+          pip install -e .[retrievechat,retrievechat-qdrant,retrievechat-pgvector]
       - name: Coverage
         env:
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
           AZURE_OPENAI_API_KEY: ${{ secrets.AZURE_OPENAI_API_KEY }}
           AZURE_OPENAI_API_BASE: ${{ secrets.AZURE_OPENAI_API_BASE }}
           OAI_CONFIG_LIST: ${{ secrets.OAI_CONFIG_LIST }}
         run: |
-          coverage run -a -m pytest test/agentchat/contrib/test_retrievechat.py::test_retrievechat test/agentchat/contrib/test_qdrant_retrievechat.py::test_retrievechat test/agentchat/contrib/test_pgvector_retrievechat.py::test_retrievechat
-          coverage xml
+          pytest test/agentchat/contrib/retrievechat/ test/agentchat/contrib/retrievechat
       - name: Upload coverage to Codecov
         uses: codecov/codecov-action@v3
         with:
           file: ./coverage.xml
           flags: unittests
+
   CompressionTest:
     strategy:
       matrix:
@@ -87,7 +98,7 @@ jobs:
           python -m pip install --upgrade pip wheel
           pip install -e .
           python -c "import autogen"
-          pip install coverage pytest-asyncio
+          pip install pytest-cov>=5 pytest-asyncio
       - name: Install packages for test when needed
         run: |
           pip install docker
@@ -98,8 +109,7 @@ jobs:
           AZURE_OPENAI_API_BASE: ${{ secrets.AZURE_OPENAI_API_BASE }}
           OAI_CONFIG_LIST: ${{ secrets.OAI_CONFIG_LIST }}
         run: |
-          coverage run -a -m pytest test/agentchat/contrib/test_compressible_agent.py
-          coverage xml
+          pytest test/agentchat/contrib/test_compressible_agent.py
       - name: Upload coverage to Codecov
         uses: codecov/codecov-action@v3
         with:
@@ -128,7 +138,7 @@ jobs:
           python -m pip install --upgrade pip wheel
           pip install -e .
           python -c "import autogen"
-          pip install coverage pytest-asyncio
+          pip install pytest-cov>=5 pytest-asyncio
       - name: Install packages for test when needed
         run: |
           pip install docker
@@ -139,8 +149,7 @@ jobs:
           AZURE_OPENAI_API_BASE: ${{ secrets.AZURE_OPENAI_API_BASE }}
           OAI_CONFIG_LIST: ${{ secrets.OAI_CONFIG_LIST }}
         run: |
-          coverage run -a -m pytest test/agentchat/contrib/test_gpt_assistant.py
-          coverage xml
+          pytest test/agentchat/contrib/test_gpt_assistant.py
       - name: Upload coverage to Codecov
         uses: codecov/codecov-action@v3
         with:
@@ -169,16 +178,15 @@ jobs:
           python -m pip install --upgrade pip wheel
           pip install -e .[teachable]
           python -c "import autogen"
-          pip install coverage pytest
+          pip install pytest-cov>=5
       - name: Coverage
         env:
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
           AZURE_OPENAI_API_KEY: ${{ secrets.AZURE_OPENAI_API_KEY }}
           AZURE_OPENAI_API_BASE: ${{ secrets.AZURE_OPENAI_API_BASE }}
           OAI_CONFIG_LIST: ${{ secrets.OAI_CONFIG_LIST }}
         run: |
-          coverage run -a -m pytest test/agentchat/contrib/capabilities/test_teachable_agent.py
-          coverage xml
+          pytest test/agentchat/contrib/capabilities/test_teachable_agent.py
       - name: Upload coverage to Codecov
         uses: codecov/codecov-action@v3
         with:
@@ -187,8 +195,8 @@ jobs:
   AgentBuilder:
     strategy:
       matrix:
-        os: [ ubuntu-latest ]
-        python-version: [ "3.11" ]
+        os: [ubuntu-latest]
+        python-version: ["3.11"]
     runs-on: ${{ matrix.os }}
     environment: openai1
     steps:
@@ -207,7 +215,7 @@ jobs:
           python -m pip install --upgrade pip wheel
           pip install -e .
           python -c "import autogen"
-          pip install coverage pytest-asyncio
+          pip install pytest-cov>=5 pytest-asyncio
       - name: Install packages for test when needed
         run: |
           pip install -e .[autobuild]
@@ -218,8 +226,7 @@ jobs:
           AZURE_OPENAI_API_BASE: ${{ secrets.AZURE_OPENAI_API_BASE }}
           OAI_CONFIG_LIST: ${{ secrets.OAI_CONFIG_LIST }}
         run: |
-          coverage run -a -m pytest test/agentchat/contrib/test_agent_builder.py
-          coverage xml
+          pytest test/agentchat/contrib/test_agent_builder.py
       - name: Upload coverage to Codecov
         uses: codecov/codecov-action@v3
         with:
@@ -248,7 +255,7 @@ jobs:
           python -m pip install --upgrade pip wheel
           pip install -e .[websurfer]
           python -c "import autogen"
-          pip install coverage pytest
+          pip install pytest-cov>=5
       - name: Coverage
         env:
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
@@ -257,93 +264,90 @@ jobs:
           OAI_CONFIG_LIST: ${{ secrets.OAI_CONFIG_LIST }}
           BING_API_KEY: ${{ secrets.BING_API_KEY }}
         run: |
-          coverage run -a -m pytest test/agentchat/contrib/test_web_surfer.py
-          coverage xml
+          pytest test/agentchat/contrib/test_web_surfer.py
       - name: Upload coverage to Codecov
         uses: codecov/codecov-action@v3
         with:
           file: ./coverage.xml
           flags: unittests
   ContextHandling:
-      strategy:
-        matrix:
-          os: [ubuntu-latest]
-          python-version: ["3.11"]
-      runs-on: ${{ matrix.os }}
-      environment: openai1
-      steps:
-        # checkout to pr branch
-        - name: Checkout
-          uses: actions/checkout@v4
-          with:
-            ref: ${{ github.event.pull_request.head.sha }}
-        - name: Set up Python ${{ matrix.python-version }}
-          uses: actions/setup-python@v5
-          with:
-            python-version: ${{ matrix.python-version }}
-        - name: Install packages and dependencies
-          run: |
-            docker --version
-            python -m pip install --upgrade pip wheel
-            pip install -e .
-            python -c "import autogen"
-            pip install coverage pytest
-        - name: Coverage
-          env:
-            OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
-            AZURE_OPENAI_API_KEY: ${{ secrets.AZURE_OPENAI_API_KEY }}
-            AZURE_OPENAI_API_BASE: ${{ secrets.AZURE_OPENAI_API_BASE }}
-            OAI_CONFIG_LIST: ${{ secrets.OAI_CONFIG_LIST }}
-            BING_API_KEY: ${{ secrets.BING_API_KEY }}
-          run: |
-            coverage run -a -m pytest test/agentchat/contrib/capabilities/test_context_handling.py
-            coverage xml
-        - name: Upload coverage to Codecov
-          uses: codecov/codecov-action@v3
-          with:
-            file: ./coverage.xml
-            flags: unittests
+    strategy:
+      matrix:
+        os: [ubuntu-latest]
+        python-version: ["3.11"]
+    runs-on: ${{ matrix.os }}
+    environment: openai1
+    steps:
+      # checkout to pr branch
+      - name: Checkout
+        uses: actions/checkout@v4
+        with:
+          ref: ${{ github.event.pull_request.head.sha }}
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Install packages and dependencies
+        run: |
+          docker --version
+          python -m pip install --upgrade pip wheel
+          pip install -e .
+          python -c "import autogen"
+          pip install pytest-cov>=5
+      - name: Coverage
+        env:
+          OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+          AZURE_OPENAI_API_KEY: ${{ secrets.AZURE_OPENAI_API_KEY }}
+          AZURE_OPENAI_API_BASE: ${{ secrets.AZURE_OPENAI_API_BASE }}
+          OAI_CONFIG_LIST: ${{ secrets.OAI_CONFIG_LIST }}
+          BING_API_KEY: ${{ secrets.BING_API_KEY }}
+        run: |
+          pytest test/agentchat/contrib/capabilities/test_context_handling.py
+      - name: Upload coverage to Codecov
+        uses: codecov/codecov-action@v3
+        with:
+          file: ./coverage.xml
+          flags: unittests
   ImageGen:
-      strategy:
-        matrix:
-          os: [ubuntu-latest]
-          python-version: ["3.12"]
-      runs-on: ${{ matrix.os }}
-      environment: openai1
-      steps:
-        # checkout to pr branch
-        - name: Checkout
-          uses: actions/checkout@v4
-          with:
-            ref: ${{ github.event.pull_request.head.sha }}
-        - name: Set up Python ${{ matrix.python-version }}
-          uses: actions/setup-python@v5
-          with:
-            python-version: ${{ matrix.python-version }}
-        - name: Install packages and dependencies
-          run: |
-            docker --version
-            python -m pip install --upgrade pip wheel
-            pip install -e .[lmm]
-            python -c "import autogen"
-            pip install coverage pytest
-        - name: Coverage
-          env:
-            OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
-          run: |
-            coverage run -a -m pytest test/agentchat/contrib/capabilities/test_image_generation_capability.py
-            coverage xml
-        - name: Upload coverage to Codecov
-          uses: codecov/codecov-action@v3
-          with:
-            file: ./coverage.xml
-            flags: unittests
+    strategy:
+      matrix:
+        os: [ubuntu-latest]
+        python-version: ["3.12"]
+    runs-on: ${{ matrix.os }}
+    environment: openai1
+    steps:
+      # checkout to pr branch
+      - name: Checkout
+        uses: actions/checkout@v4
+        with:
+          ref: ${{ github.event.pull_request.head.sha }}
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Install packages and dependencies
+        run: |
+          docker --version
+          python -m pip install --upgrade pip wheel
+          pip install -e .[lmm]
+          python -c "import autogen"
+          pip install pytest-cov>=5
+      - name: Coverage
+        env:
+          OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+        run: |
+          pytest test/agentchat/contrib/capabilities/test_image_generation_capability.py
+      - name: Upload coverage to Codecov
+        uses: codecov/codecov-action@v3
+        with:
+          file: ./coverage.xml
+          flags: unittests
 
   AgentOptimizer:
     strategy:
       matrix:
-        os: [ ubuntu-latest ]
-        python-version: [ "3.11" ]
+        os: [ubuntu-latest]
+        python-version: ["3.11"]
     runs-on: ${{ matrix.os }}
     environment: openai1
     steps:
@@ -362,16 +366,15 @@ jobs:
           python -m pip install --upgrade pip wheel
           pip install -e .
           python -c "import autogen"
-          pip install coverage pytest
+          pip install pytest-cov>=5
       - name: Coverage
         env:
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
           AZURE_OPENAI_API_KEY: ${{ secrets.AZURE_OPENAI_API_KEY }}
           AZURE_OPENAI_API_BASE: ${{ secrets.AZURE_OPENAI_API_BASE }}
           OAI_CONFIG_LIST: ${{ secrets.OAI_CONFIG_LIST }}
         run: |
-          coverage run -a -m pytest test/agentchat/contrib/test_agent_optimizer.py
-          coverage xml
+          pytest test/agentchat/contrib/test_agent_optimizer.py
       - name: Upload coverage to Codecov
         uses: codecov/codecov-action@v3
         with: