mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-24 09:20:52 +00:00
a3547f82c4
* Replace `assert`s in the `conversable_agent` module with `if-log-raise`. * Use a `logger` object in the `code_utils` module. * Replace use of `assert` with `if-log-raise` in the `code_utils` module. * Replace use of `assert` in the `math_utils` module with `if-not-raise`. * Replace `assert` with `if` in the `oai.completion` module. * Replace `assert` in the `retrieve_utils` module with an if statement. * Add missing `not`. * Blacken `completion.py`. * Test `generate_reply` and `a_generate_reply` raise an assertion error when there are neither `messages` nor a `sender`. * Test `execute_code` raises an `AssertionError` when neither code nor filename is provided. * Test `split_text_to_chunks` raises when passed an invalid chunk mode. * * Add `tiktoken` and `chromadb` to test dependencies as they're used in the `test_retrieve_utils` module. * Sort the test requirements alphabetically.
66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
import setuptools
|
|
import os
|
|
|
|
here = os.path.abspath(os.path.dirname(__file__))
|
|
|
|
with open("README.md", "r", encoding="UTF-8") as fh:
|
|
long_description = fh.read()
|
|
|
|
|
|
# Get the code version
|
|
version = {}
|
|
with open(os.path.join(here, "autogen/version.py")) as fp:
|
|
exec(fp.read(), version)
|
|
__version__ = version["__version__"]
|
|
|
|
install_requires = [
|
|
"openai",
|
|
"diskcache",
|
|
"termcolor",
|
|
"flaml",
|
|
]
|
|
|
|
|
|
setuptools.setup(
|
|
name="pyautogen",
|
|
version=__version__,
|
|
author="AutoGen",
|
|
author_email="auto-gen@outlook.com",
|
|
description="Enabling Next-Gen LLM Applications via Multi-Agent Conversation Framework",
|
|
long_description=long_description,
|
|
long_description_content_type="text/markdown",
|
|
url="https://github.com/microsoft/autogen",
|
|
packages=setuptools.find_packages(include=["autogen*"], exclude=["test"]),
|
|
# package_data={
|
|
# "autogen.default": ["*/*.json"],
|
|
# },
|
|
# include_package_data=True,
|
|
install_requires=install_requires,
|
|
extras_require={
|
|
"test": [
|
|
"chromadb",
|
|
"coverage>=5.3",
|
|
"datasets",
|
|
"ipykernel",
|
|
"nbconvert",
|
|
"nbformat",
|
|
"pre-commit",
|
|
"pydantic==1.10.9",
|
|
"pytest-asyncio",
|
|
"pytest>=6.1.1",
|
|
"sympy",
|
|
"tiktoken",
|
|
"wolframalpha",
|
|
],
|
|
"blendsearch": ["flaml[blendsearch]"],
|
|
"mathchat": ["sympy", "pydantic==1.10.9", "wolframalpha"],
|
|
"retrievechat": ["chromadb", "tiktoken", "sentence_transformers", "pypdf"],
|
|
},
|
|
classifiers=[
|
|
"Programming Language :: Python :: 3",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Operating System :: OS Independent",
|
|
],
|
|
python_requires=">=3.8",
|
|
)
|