Skip to main content
Glama
Shougakusei

AI Pull Request Generator

by Shougakusei
README.md
# 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.