AI Pull Request Generator

一个使用 FastMCP 框架实现的自动化拉取请求创建和代码生成的 AI 工具。

特性

• 使用 AI (GPT-4) 进行任务规划
• 基于计划的代码生成
• 自动化 GitHub 拉取请求创建
• 与 FastMCP 框架集成

要求

• Python 3.8+
• 具有仓库访问权限的 GitHub token
• OpenAI API 密钥
• 必需的 Python 包（参见 requirements.txt）

设置

1. 安装依赖项：
   bash
   pip install -r requirements.txt

2. 设置环境变量：
   创建一个 .env 文件，并包含以下变量：

GITHUB_TOKEN=your_github_token
API_KEY=your_openai_api_key
API_BASE=your_api_base_url

使用方法

该工具提供了三个主要功能：

1. 任务规划
   python
   await task_planning(task_file="path/to/task.txt", plan_file="path/to/plan.txt")

2. 代码生成
   python
   await code_for_plan(plan_file="path/to/plan.txt", code_file="path/to/code.py")

3. 创建拉取请求
   python
   await create_github_pull_request(
   repository="owner/repo",
   branch_name="feature-branch",
   file_path="path/to/file.py",
   file_content="your code",
   commit_message="feat: add new feature",
   title="Add new feature",
   body="Description of changes",
   base_branch="main"
   )

开发

该工具使用 FastMCP 框架进行通信并作为服务器运行。要本地运行：

bash
python pull-request.py

许可证

MIT 许可证

贡献

欢迎贡献！请随时提交拉取请求。