Skip to main content

可选依赖项

LLM 缓存

要在 Redis 中使用 LLM 缓存,您需要安装带有 redis 选项的 Python 包:

pip install "pyautogen[redis]"

有关详细信息,请参阅 LLM 缓存

IPython 代码执行器

要使用 IPython 代码执行器,您需要安装 jupyter-clientipykernel 包:

pip install "pyautogen[ipython]"

要使用 IPython 代码执行器:

from autogen import UserProxyAgent

proxy = UserProxyAgent(name="proxy", code_execution_config={"executor": "ipython-embedded"})

blendsearch

pyautogen<0.2 提供了一种经济高效的超参数优化技术 EcoOptiGen 用于调整大型语言模型。请安装 [blendsearch] 选项以使用它。

pip install "pyautogen[blendsearch]<0.2"

示例笔记本:

为代码生成进行优化

为数学进行优化

retrievechat

pyautogen 支持检索增强生成任务,例如问答和代码生成,使用 RAG 代理。请安装 [retrievechat] 选项以在 ChromaDB 中使用它。

pip install "pyautogen[retrievechat]"

或者,pyautogen 还支持可以替代 ChromaDB 安装的 PGVector 和 Qdrant。

pip install "pyautogen[retrievechat-pgvector]"
pip install "pyautogen[retrievechat-qdrant]"

RetrieveChat 可以处理各种类型的文档。默认情况下,它可以处理纯文本和 PDF 文件,包括 'txt'、'json'、'csv'、'tsv'、'md'、'html'、'htm'、'rtf'、'rst'、'jsonl'、'log'、'xml'、'yaml'、'yml' 和 'pdf' 等格式。如果您安装了 unstructured (pip install "unstructured[all-docs]"),还将支持其他文档类型,如 'docx'、'doc'、'odt'、'pptx'、'ppt'、'xlsx'、'eml'、'msg'、'epub'。

您可以使用 autogen.retrieve_utils.TEXT_FORMATS 查找所有支持的文档类型的列表。

示例笔记本:

使用检索增强代理进行自动代码生成和问答

使用检索增强生成进行群聊(5 个群成员代理和 1 个管理者代理)

使用基于 Qdrant 的检索增强代理进行自动代码生成和问答

Teachability

要使用 Teachability,请使用 [teachable] 选项安装 AutoGen。

pip install "pyautogen[teachable]"

示例笔记本:与可教授代理人聊天

大型多模态模型(LMM)代理人

我们提供了多模态可对话代理人和 LLaVA 代理人。请使用 [lmm] 选项进行安装。

pip install "pyautogen[lmm]"

示例笔记本:

LLaVA 代理人

mathchat

pyautogen<0.2 提供了一个用于解决数学问题的实验性代理人。请使用 [mathchat] 选项进行安装。

pip install "pyautogen[mathchat]<0.2"

示例笔记本:

使用 MathChat 解决数学问题

图表

要在 GroupChat 中使用图表,特别是用于图表可视化,请使用 [graph] 选项安装 AutoGen。

pip install "pyautogen[graph]"

示例笔记本:有限状态机图表以设置发言者转换约束

处理长文本上下文

AutoGen 使用 LLMLingua 库进行文本压缩,以支持处理长文本上下文。要启用此功能,请使用 [long-context] 选项安装 AutoGen:

pip install "pyautogen[long-context]"