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:

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

await task_planning(task_file="path/to/task.txt", plan_file="path/to/plan.txt")

2. Code Generation

await code_for_plan(plan_file="path/to/plan.txt", code_file="path/to/code.py")

3. 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.