①快速上手

安装指南

CAI支持多种操作系统和平台。请根据您的环境选择合适的安装方法。

OS X 安装

确保Python 3.12在PATH中

# 安装Homebrew (如果没有安装)
/bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
# 安装Git和Python 3.12
brew install git python@3.12
# 安装CAI框架
pip install cai-framework

Ubuntu 安装

推荐使用虚拟环境避免依赖冲突

# 更新包管理器
sudo apt update
# 安装必要的依赖
sudo apt-get install -y git python3-pip python3.12-venv
# 创建虚拟环境
python3.12 -m venv cai-env
source cai-env/bin/activate
# 安装CAI框架
pip install cai-framework

Windows WSL 安装

Windows用户建议使用WSL2环境

# 在WSL2中执行 (推荐Ubuntu 20.04/22.04)
# 安装依赖
sudo apt update
sudo apt-get install -y git python3-pip python3.12-venv
# 创建并激活虚拟环境
python3.12 -m venv cai-env
source cai-env/bin/activate
# 安装CAI框架
pip install cai-framework

Android 安装

需要root权限来编译Python

# 通过UserLand安装Kali Linux
# 下载并安装UserLand应用
# 在Kali中安装依赖
sudo apt update
sudo apt install -y python3 python3-pip git
# 编译Python 3.12 (如果需要)
wget https://www.python.org/ftp/python/3.12.0/Python-3.12.0.tgz
tar -xzf Python-3.12.0.tgz
cd Python-3.12.0
./configure --enable-optimizations
make -j4
sudo make altinstall
# 安装CAI框架
pip3.12 install cai-framework

首次启动

环境配置

配置API密钥，选择合适的AI模型，设置Agent的工作模式

基本测试

运行简单的测试任务，验证框架安装正确且能正常工作

功能探索

了解不同Agent的能力，测试工具集成，体验多Agent协作

快速验证安装

# 验证安装
python -c "import cai; print('CAI安装成功!')"

# 检查可用Agent
python -c "from cai import list_agents; print(list_agents())"

# 运行测试
cai --test

基本使用

通过以下示例了解CAI的基本使用方式，从简单配置到复杂的多Agent协作。

1. 基本配置

配置环境变量和API密钥

import os
# 配置OpenAI API密钥
os.environ["OPENAI_API_KEY"] = "your-api-key-here"

# 或者配置其他模型提供商
os.environ["ANTHROPIC_API_KEY"] = "your-anthropic-key"

2. 运行简单Agent

启动一个红队Agent进行渗透测试

from cai import Agent

# 创建红队Agent
redteam_agent = Agent(
    model="openai/gpt-4",
    system_prompt="你是一个专业的渗透测试专家..."
)

# 运行Agent
result = redteam_agent.run("测试目标网站的安全性")
print(result.output)

3. 多Agent协作

使用多个Agent协同工作

from cai import MultiAgent

# 创建多Agent系统
multi_agent = MultiAgent([
    Agent("recon_agent", model="openai/gpt-4"),
    Agent("exploit_agent", model="anthropic/claude-3"),
    Agent("report_agent", model="openai/gpt-4")
])

# 编排任务
result = multi_agent.orchestrate("完整的渗透测试流程")

4. 添加自定义工具

集成自定义安全工具

from cai import Tool

class CustomNmapTool(Tool):
    name = "nmap_scanner"
    description = "端口扫描和漏洞检测"
    
    def run(self, target: str) -> dict:
        # 实现nmap扫描逻辑
        return {"open_ports": [80, 443], "vulnerabilities": []}

# 添加工具到Agent
agent.add_tool(CustomNmapTool())