AI Pull Request Generator
An AI-powered tool for automating pull request creation and code generation using FastMCP framework.
Features
- Task planning using AI (GPT-4)
- Code generation based on plans
- Automated GitHub pull request creation
- Integration with FastMCP framework
Requirements
- Python 3.8+
- GitHub token with repository access
- OpenAI API key
- Required Python packages (see requirements.txt)
Setup
- Install dependencies:
pip install -r requirements.txt
- Set up environment variables:
Create a
.envfile with the following variables:
GITHUB_TOKEN=your_github_token
API_KEY=your_openai_api_key
API_BASE=your_api_base_url
Usage
The tool provides three main functions:
- Task Planning
await task_planning(task_file="path/to/task.txt", plan_file="path/to/plan.txt")
- Code Generation
await code_for_plan(plan_file="path/to/plan.txt", code_file="path/to/code.py")
- Pull Request Creation
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"
)
Development
The tool uses FastMCP framework for communication and runs as a server. To run locally:
python pull-request.py
License
MIT License
Contributing
Contributions are welcome! Please feel free to submit a pull request.