Sotopia：开放式社交学习环境

新闻

• [2024年5月] Sotopia在ICLR 2024会议上作为聚焦报告进行展示 ⭐！

简介

Sotopia是一个开放式社交学习环境，允许智能体相互交互并与环境互动。该环境旨在成为评估和促进语言智能体社交智能的平台。环境设计为开放式，意味着可以轻松扩展以包含新环境和新智能体。该环境还设计为可扩展的，意味着可以轻松扩展以包含大量智能体和环境。

@inproceedings{zhou2024sotopia,
  title = {SOTOPIA: Interactive Evaluation for Social Intelligence in Language Agents},
  author = {Zhou*, Xuhui and Zhu*, Hao and Mathur, Leena and Zhang, Ruohong and Qi, Zhengyang and Yu, Haofei and Morency, Louis-Philippe and Bisk, Yonatan and Fried, Daniel and Neubig, Graham and Sap, Maarten},
  journal = {ICLR},
  year = {2024},
  url = {https://openreview.net/forum?id=mM7VurbA4r},
}

帮助

更多详情请参阅文档。

开始使用

本地安装

我们建议使用虚拟环境，例如使用anaconda3：conda create -n sotopia python=3.11; conda activate sotopia; curl -sSL https://install.python-poetry.org | python3。

然后： python -m pip install sotopia; sotopia install

这将设置必要的环境变量并下载所需数据。

[!提示] 安装遇到问题？或者暂时不想安装redis？我们正在为您准备一个公共redis服务器。敬请期待！

运行代码需要OpenAI密钥。请将环境变量OPENAI_API_KEY设置为您的密钥。推荐的方法是将密钥添加到conda环境中：

conda env config vars set OPENAI_API_KEY=your_key

简易示例服务器

您可以使用以下默认参数查看一个示例情节：

import asyncio
from sotopia.samplers import UniformSampler
from sotopia.server import run_async_server

asyncio.run(
    run_async_server(
        model_dict={
            "env": "gpt-4",
            "agent1": "gpt-3.5-turbo",
            "agent2": "gpt-3.5-turbo",
        },
        sampler=UniformSampler(),
    )
)

或者运行

python examples/minimalist_demo.py