2020-12-04 09:40:27 -08:00
|
|
|
import os
|
|
|
|
|
2023-11-06 13:33:51 -08:00
|
|
|
import setuptools
|
|
|
|
|
2020-12-04 09:40:27 -08:00
|
|
|
here = os.path.abspath(os.path.dirname(__file__))
|
|
|
|
|
2022-01-02 20:12:34 -05:00
|
|
|
with open("README.md", "r", encoding="UTF-8") as fh:
|
2020-12-04 09:40:27 -08:00
|
|
|
long_description = fh.read()
|
|
|
|
|
|
|
|
# Get the code version
|
|
|
|
version = {}
|
2023-09-16 10:57:57 +00:00
|
|
|
with open(os.path.join(here, "autogen/version.py")) as fp:
|
2020-12-04 09:40:27 -08:00
|
|
|
exec(fp.read(), version)
|
|
|
|
__version__ = version["__version__"]
|
|
|
|
|
|
|
|
install_requires = [
|
2024-04-24 01:26:06 +08:00
|
|
|
"openai>=1.3",
|
2023-09-16 10:57:57 +00:00
|
|
|
"diskcache",
|
|
|
|
"termcolor",
|
2023-09-16 15:30:28 +00:00
|
|
|
"flaml",
|
2024-03-12 11:05:53 -04:00
|
|
|
# numpy is installed by flaml, but we want to pin the version to below 2.x (see https://github.com/microsoft/autogen/issues/1960)
|
|
|
|
"numpy>=1.17.0,<2",
|
2023-10-05 05:45:11 +01:00
|
|
|
"python-dotenv",
|
2023-11-03 21:01:49 -07:00
|
|
|
"tiktoken",
|
2024-02-07 17:15:07 -05:00
|
|
|
# Disallowing 2.6.0 can be removed when this is fixed https://github.com/pydantic/pydantic/issues/8705
|
|
|
|
"pydantic>=1.10,<3,!=2.6.0", # could be both V1 and V2
|
2024-01-18 19:03:49 +02:00
|
|
|
"docker",
|
2021-09-10 16:39:16 -07:00
|
|
|
]
|
2020-12-04 09:40:27 -08:00
|
|
|
|
2024-03-20 03:01:16 +01:00
|
|
|
jupyter_executor = [
|
|
|
|
"jupyter-kernel-gateway",
|
|
|
|
"websocket-client",
|
|
|
|
"requests",
|
|
|
|
"jupyter-client>=8.6.0",
|
|
|
|
"ipykernel>=6.29.0",
|
|
|
|
]
|
|
|
|
|
2024-04-21 02:32:57 +08:00
|
|
|
rag = ["sentence_transformers", "pypdf", "ipython", "beautifulsoup4", "markdownify"]
|
|
|
|
|
2020-12-04 09:40:27 -08:00
|
|
|
setuptools.setup(
|
2023-09-19 17:50:43 +00:00
|
|
|
name="pyautogen",
|
2020-12-04 09:40:27 -08:00
|
|
|
version=__version__,
|
2023-09-16 10:57:57 +00:00
|
|
|
author="AutoGen",
|
2023-09-16 16:41:15 +00:00
|
|
|
author_email="auto-gen@outlook.com",
|
2023-09-16 10:57:57 +00:00
|
|
|
description="Enabling Next-Gen LLM Applications via Multi-Agent Conversation Framework",
|
2020-12-04 09:40:27 -08:00
|
|
|
long_description=long_description,
|
|
|
|
long_description_content_type="text/markdown",
|
2023-09-16 10:57:57 +00:00
|
|
|
url="https://github.com/microsoft/autogen",
|
2023-09-19 18:26:50 +00:00
|
|
|
packages=setuptools.find_packages(include=["autogen*"], exclude=["test"]),
|
2020-12-04 09:40:27 -08:00
|
|
|
install_requires=install_requires,
|
|
|
|
extras_require={
|
|
|
|
"test": [
|
|
|
|
"coverage>=5.3",
|
2023-10-03 19:52:50 +02:00
|
|
|
"ipykernel",
|
2022-12-24 00:18:49 +08:00
|
|
|
"nbconvert",
|
|
|
|
"nbformat",
|
2023-10-03 19:52:50 +02:00
|
|
|
"pre-commit",
|
|
|
|
"pytest-asyncio",
|
2024-01-28 01:52:18 +01:00
|
|
|
"pytest>=6.1.1,<8",
|
2024-03-27 19:45:17 -04:00
|
|
|
"pandas",
|
2021-04-06 11:37:52 -07:00
|
|
|
],
|
2023-09-16 15:30:28 +00:00
|
|
|
"blendsearch": ["flaml[blendsearch]"],
|
2023-09-16 10:57:57 +00:00
|
|
|
"mathchat": ["sympy", "pydantic==1.10.9", "wolframalpha"],
|
2024-04-21 02:32:57 +08:00
|
|
|
"retrievechat": ["chromadb"] + rag,
|
|
|
|
"retrievechat-pgvector": ["pgvector>=0.2.5", "psycopg>=3.1.18"] + rag,
|
|
|
|
"retrievechat-qdrant": ["qdrant_client[fastembed]"] + rag,
|
2024-01-07 02:23:23 +09:00
|
|
|
"autobuild": ["chromadb", "sentence-transformers", "huggingface-hub"],
|
2023-10-20 19:27:10 -07:00
|
|
|
"teachable": ["chromadb"],
|
2023-11-06 13:33:51 -08:00
|
|
|
"lmm": ["replicate", "pillow"],
|
2024-02-06 14:13:18 +11:00
|
|
|
"graph": ["networkx", "matplotlib"],
|
2024-04-16 17:24:07 -07:00
|
|
|
"gemini": ["google-generativeai>=0.5,<1", "pillow", "pydantic"],
|
2024-01-21 19:43:15 -08:00
|
|
|
"websurfer": ["beautifulsoup4", "markdownify", "pdfminer.six", "pathvalidate"],
|
2024-01-20 09:06:29 -08:00
|
|
|
"redis": ["redis"],
|
2024-04-11 12:37:24 -04:00
|
|
|
"cosmosdb": ["azure-cosmos>=4.2.0"],
|
2024-03-26 23:39:55 +01:00
|
|
|
"websockets": ["websockets>=12.0,<13"],
|
2024-03-20 03:01:16 +01:00
|
|
|
"jupyter-executor": jupyter_executor,
|
2024-03-26 23:39:55 +01:00
|
|
|
"types": ["mypy==1.9.0", "pytest>=6.1.1,<8"] + jupyter_executor,
|
2020-12-04 09:40:27 -08:00
|
|
|
},
|
|
|
|
classifiers=[
|
|
|
|
"Programming Language :: Python :: 3",
|
|
|
|
"License :: OSI Approved :: MIT License",
|
2021-04-08 09:29:55 -07:00
|
|
|
"Operating System :: OS Independent",
|
2020-12-04 09:40:27 -08:00
|
|
|
],
|
2024-01-17 23:15:06 -08:00
|
|
|
python_requires=">=3.8,<3.13",
|
2020-12-04 09:40:27 -08:00
|
|
|
)
|