diff --git a/setup.py b/setup.py index 2163b4233..b69f05b45 100644 --- a/setup.py +++ b/setup.py @@ -22,6 +22,29 @@ here = Path(__file__).resolve().parent long_description = (here / "README.md").read_text(encoding="utf-8") requirements = (here / "requirements.txt").read_text(encoding="utf-8").splitlines() + +extras_require = { + "playwright": ["playwright>=1.26", "beautifulsoup4"], + "selenium": ["selenium>4", "webdriver_manager", "beautifulsoup4"], + "search-google": ["google-api-python-client==2.94.0"], + "search-ddg": ["duckduckgo-search==3.8.5"], + "pyppeteer": ["pyppeteer>=1.0.2"], + "ocr": ["paddlepaddle==2.4.2", "paddleocr>=2.0.1", "tabulate==0.9.0"], + "test": ["pytest", "pytest-cov", "pytest-asyncio", "pytest-mock"], +} + +extras_require["test"] = [ + *set(i for j in extras_require.values() for i in j), + "pytest", + "pytest-asyncio", + "pytest-cov", + "pytest-mock", + "pytest-html", +] + +extras_require["dev"] = (["pylint~=3.0.3", "black~=23.3.0", "isort~=5.12.0", "pre-commit~=3.6.0"],) + + setup( name="metagpt", version="0.5.2", @@ -36,16 +59,7 @@ setup( packages=find_packages(exclude=["contrib", "docs", "examples", "tests*"]), python_requires=">=3.9", install_requires=requirements, - extras_require={ - "playwright": ["playwright>=1.26", "beautifulsoup4"], - "selenium": ["selenium>4", "webdriver_manager", "beautifulsoup4"], - "search-google": ["google-api-python-client==2.94.0"], - "search-ddg": ["duckduckgo-search==3.8.5"], - "pyppeteer": ["pyppeteer>=1.0.2"], - "ocr": ["paddlepaddle==2.4.2", "paddleocr>=2.0.1", "tabulate==0.9.0"], - "dev": ["pylint~=3.0.3", "black~=23.3.0", "isort~=5.12.0", "pre-commit~=3.6.0"], - "test": ["pytest", "pytest-cov", "pytest-asyncio", "pytest-mock"], - }, + extras_require=extras_require, cmdclass={ "install_mermaid": InstallMermaidCLI, },