README.md•1.55 kB
# 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
1. Install dependencies:
```bash
pip install -r requirements.txt
```
2. Set up environment variables:
Create a `.env` file 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:
1. Task Planning
```python
await task_planning(task_file="path/to/task.txt", plan_file="path/to/plan.txt")
```
2. Code Generation
```python
await code_for_plan(plan_file="path/to/plan.txt", code_file="path/to/code.py")
```
3. Pull Request Creation
```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"
)
```
## Development
The tool uses FastMCP framework for communication and runs as a server. To run locally:
```bash
python pull-request.py
```
## License
MIT License
## Contributing
Contributions are welcome! Please feel free to submit a pull request.