2447 fix pgvector tests and notebook #4171

Workflow file for this run

.github/workflows/contrib-openai.yml at 3ff9972

	# This workflow will install Python dependencies and run tests
	# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions

	name: OpenAI4ContribTests

	on:
	pull_request:
	branches: ['main']
	paths:
	- 'autogen/**'
	- 'test/agentchat/contrib/**'
	- '.github/workflows/contrib-openai.yml'
	- 'setup.py'
	permissions: {}
	# actions: read
	# checks: read
	# contents: read
	# deployments: read
	jobs:
	RetrieveChatTest:
	strategy:
	matrix:
	os: [ubuntu-latest]
	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
	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-asyncio
	- name: Install packages for test when needed
	run: \|
	pip install docker
	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 -k test_retrievechat test/agentchat/contrib/retrievechat
	coverage xml
	- name: Upload coverage to Codecov
	uses: codecov/codecov-action@v3
	with:
	file: ./coverage.xml
	flags: unittests

	CompressionTest:
	strategy:
	matrix:
	os: [ubuntu-latest]
	python-version: ["3.9"]
	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-asyncio
	- name: Install packages for test when needed
	run: \|
	pip install docker
	- 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_compressible_agent.py
	coverage xml
	- name: Upload coverage to Codecov
	uses: codecov/codecov-action@v3
	with:
	file: ./coverage.xml
	flags: unittests
	GPTAssistantAgent:
	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-asyncio
	- name: Install packages for test when needed
	run: \|
	pip install docker
	- 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_gpt_assistant.py
	coverage xml
	- name: Upload coverage to Codecov
	uses: codecov/codecov-action@v3
	with:
	file: ./coverage.xml
	flags: unittests
	TeachableAgent:
	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 .[teachable]
	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 }}
	run: \|
	coverage run -a -m pytest test/agentchat/contrib/capabilities/test_teachable_agent.py
	coverage xml
	- name: Upload coverage to Codecov
	uses: codecov/codecov-action@v3
	with:
	file: ./coverage.xml
	flags: unittests
	AgentBuilder:
	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-asyncio
	- name: Install packages for test when needed
	run: \|
	pip install -e .[autobuild]
	- 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_builder.py
	coverage xml
	- name: Upload coverage to Codecov
	uses: codecov/codecov-action@v3
	with:
	file: ./coverage.xml
	flags: unittests
	WebSurfer:
	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 .[websurfer]
	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/test_web_surfer.py
	coverage xml
	- 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
	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

	AgentOptimizer:
	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 }}
	run: \|
	coverage run -a -m pytest test/agentchat/contrib/test_agent_optimizer.py
	coverage xml
	- name: Upload coverage to Codecov
	uses: codecov/codecov-action@v3
	with:
	file: ./coverage.xml
	flags: unittests

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

2447 fix pgvector tests and notebook #4171

Workflow file

2447 fix pgvector tests and notebook #4171

Jobs

Run details

Workflow file for this run