Integuru v0

这是首个通过逆向工程平台内部 API 来生成集成代码的 AI Agent。

Integuru v0 实战演示

Integuru in action

Integuru v0 的功能

你可以使用 create_har.py 生成包含所有浏览器网络请求的文件以及 cookies 文件，并编写一段描述浏览器中触发动作的提示词（prompt）。该 Agent 会输出可运行的 Python 代码，通过调用平台的内部端点来执行所需的操作。

工作原理

假设我们要下载水电费账单：

Agent 识别出下载水电费账单的请求。例如，请求 URL 可能如下所示：
```
https://www.example.com/utility-bills?accountId=123&userId=456
```
它识别出请求中依赖于其他请求的部分。上述 URL 包含需要从其他请求中获取的动态部分（accountId 和 userId）。
```
accountId=123 userId=456
```
它查找提供这些参数的请求，并将下载请求与它们建立依赖关系。它还将这些请求附加到原始请求中，以构建依赖图。
```
GET https://www.example.com/get_account_id
GET https://www.example.com/get_user_id
```
此过程不断重复，直到正在检查的请求不依赖于任何其他请求，且仅需要身份验证 cookies。
Agent 从没有出边的节点（请求）开始向上遍历图，直到到达主节点，同时将每个节点转换为可运行的函数。

特性

生成请求依赖图，以完成执行所需操作的最终请求。
允许输入变量（例如，选择下载文档的年份）。目前仅支持图形生成阶段。代码生成的输入变量功能即将推出！
生成调用图中所有请求的代码，以执行所需操作。

安装设置

设置你的 OpenAI API Keys 并添加 OPENAI_API_KEY 环境变量。（我们建议使用能够调用至少与 OpenAI o1-mini 同等能力模型的账户。如果能使用与 OpenAI o1-preview 相当的模型则更为理想。）
通过 poetry 安装 Python 依赖：
```
poetry install
```
打开 poetry shell：
```
poetry shell
```

在 Jupyter 中注册 Poetry 虚拟环境：

poetry run ipython kernel install --user --name=integuru

运行以下命令以启动浏览器：
```
poetry run python create_har.py
```
登录你的目标平台并执行所需操作（例如下载水电费账单）。
运行 Integuru：
```
poetry run integuru --prompt "download utility bills" --model 
```
你也可以通过 Jupyter Notebook main.ipynb 来运行。

建议使用 gpt-4o 作为模型进行图形生成，因为它支持函数调用。如果用户的 OpenAI 账户可用，Integuru 将自动切换到 o1-preview 进行代码生成。

使用方法

在完成项目设置后，你可以使用 Integuru 来分析和逆向外部平台的 API 请求。只需提供相应的 .har 文件以及描述你想要触发的操作的提示词即可。

poetry run integuru --help
Usage: integuru [OPTIONS]

Options:
  --model TEXT                    The LLM model to use (default is gpt-4o)
  --prompt TEXT                   The prompt for the model  [required]
  --har-path TEXT                 The HAR file path (default is
                                  ./network_requests.har)
  --cookie-path TEXT              The cookie file path (default is
                                  ./cookies.json)
  --max_steps INTEGER             The max_steps (default is 20)
  --input_variables ...
                                  Input variables in the format key value
  --generate-code                 Whether to generate the full integration
                                  code
  --help                          Show this message and exit.

运行单元测试

要使用 pytest 运行单元测试，请使用以下命令：

poetry run pytest

持续集成 (CI) 工作流

本仓库包含一个使用 GitHub Actions 的 CI 工作流。工作流定义在 .github/workflows/ci.yml 文件中，并在每次推送到 main 分支或提交 Pull Request 时触发。工作流执行以下步骤：

检出代码。
设置 Python 3.12。
使用 poetry 安装依赖。
使用 pytest 运行测试。

关于双重认证 (2FA) 的说明

当目标网站使用双重认证 (2FA) 时，工作流保持不变。请确保你完成了 2FA 流程，并在 2FA 完成后获取 cookies/auth tokens/session tokens。这些 token 将用于后续的工作流中。

Integuru-AI/Integuru