gpt-pr 0.2.1__py3-none-any.whl → 0.7.2__py3-none-any.whl

gpt_pr-0.2.1.dist-info/METADATA

@@ -1,49 +0,0 @@
-Metadata-Version: 2.1
-Name: gpt-pr
-Version: 0.2.1
-Summary: Automate your GitHub workflow with GPT-PR: an OpenAI powered library for streamlined PR generation.
-Home-page: http://github.com/alissonperez/gpt-pr
-Author: Alisson R. Perez
-Author-email: alissonperez@outlook.com
-License: MIT
-Requires-Python: >=3.7
-Requires-Dist: cffi ==1.15.1
-Requires-Dist: cryptography ==42.0.5
-Requires-Dist: fire ==0.6.0
-Requires-Dist: pycparser ==2.21
-Requires-Dist: wcwidth ==0.2.13
-Requires-Dist: charset-normalizer ==3.3.2 ; python_full_version >= "3.7.0"
-Requires-Dist: prompt-toolkit ==3.0.43 ; python_full_version >= "3.7.0"
-Requires-Dist: openai ==1.14.0 ; python_full_version >= "3.7.1"
-Requires-Dist: exceptiongroup ==1.2.0 ; python_version < "3.11"
-Requires-Dist: cached-property ==1.5.2 ; python_version < "3.8"
-Requires-Dist: typing-extensions ==4.7.1 ; python_version < "3.8"
-Requires-Dist: importlib-metadata ==6.7.0 ; python_version == "3.7"
-Requires-Dist: six ==1.16.0 ; python_version >= "2.7" and python_version not in "3.0, 3.1, 3.2"
-Requires-Dist: deprecated ==1.2.14 ; python_version >= "2.7" and python_version not in "3.0, 3.1, 3.2, 3.3"
-Requires-Dist: idna ==3.6 ; python_version >= "3.5"
-Requires-Dist: certifi ==2024.2.2 ; python_version >= "3.6"
-Requires-Dist: distro ==1.9.0 ; python_version >= "3.6"
-Requires-Dist: pynacl ==1.5.0 ; python_version >= "3.6"
-Requires-Dist: wrapt ==1.16.0 ; python_version >= "3.6"
-Requires-Dist: annotated-types ==0.5.0 ; python_version >= "3.7"
-Requires-Dist: anyio ==3.7.1 ; python_version >= "3.7"
-Requires-Dist: gitdb ==4.0.11 ; python_version >= "3.7"
-Requires-Dist: gitpython ==3.1.42 ; python_version >= "3.7"
-Requires-Dist: h11 ==0.14.0 ; python_version >= "3.7"
-Requires-Dist: httpcore ==0.17.3 ; python_version >= "3.7"
-Requires-Dist: httpx ==0.24.1 ; python_version >= "3.7"
-Requires-Dist: pydantic ==2.5.3 ; python_version >= "3.7"
-Requires-Dist: pydantic-core ==2.14.6 ; python_version >= "3.7"
-Requires-Dist: pygithub ==2.2.0 ; python_version >= "3.7"
-Requires-Dist: pyjwt[crypto] ==2.8.0 ; python_version >= "3.7"
-Requires-Dist: requests ==2.31.0 ; python_version >= "3.7"
-Requires-Dist: smmap ==5.0.1 ; python_version >= "3.7"
-Requires-Dist: sniffio ==1.3.1 ; python_version >= "3.7"
-Requires-Dist: termcolor ==2.3.0 ; python_version >= "3.7"
-Requires-Dist: tqdm ==4.66.2 ; python_version >= "3.7"
-Requires-Dist: urllib3 ==2.0.7 ; python_version >= "3.7"
-Requires-Dist: zipp ==3.15.0 ; python_version >= "3.7"
-Requires-Dist: inquirerpy ==0.3.4 ; python_version >= "3.7" and python_version < "4.0"
-Requires-Dist: pfzy ==0.3.4 ; python_version >= "3.7" and python_version < "4.0"
-

gpt_pr-0.2.1.dist-info/RECORD

@@ -1,13 +0,0 @@
-gptpr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-gptpr/consolecolor.py,sha256=_JmBMNjIflWMlgP2VkCWu6uQLR9oHBy52uV3TRJJgF4,800
-gptpr/gh.py,sha256=aunn_8wi4pGoGkjdqyA2MZVAAxfItpnnmDCp20-i-WA,786
-gptpr/gitutil.py,sha256=NBD3iRnbFEPRU47w7c5TowwtZieDYkU4zybvv0PoOU0,5783
-gptpr/main.py,sha256=rkalqLcc1Nh5WH51w7ayEMIYNoScrRBNVYl3KLuZFdY,1270
-gptpr/prdata.py,sha256=y4VodgCiSaARPXzJHsg32-cqrR_nZqyyGSLij6ee-oo,5846
-gptpr/test_prdata.py,sha256=rSJ-yqOdw-iYdBWyqnA2SXbdrhT8KgIkRTTf9SY1S1g,474
-gptpr/version.py,sha256=HfjVOrpTnmZ-xVFCYSVmX50EXaBQeJteUHG-PD6iQs8,22
-gpt_pr-0.2.1.dist-info/METADATA,sha256=rJ-4clCl3pJh99vscyuuVQsLt4XQzqpgf4K67YYMZJY,2637
-gpt_pr-0.2.1.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-gpt_pr-0.2.1.dist-info/entry_points.txt,sha256=aXCkyNdoUHfSJXVhRKHj8m09twDfcDmY7xC66u5N3hE,43
-gpt_pr-0.2.1.dist-info/top_level.txt,sha256=DZcbzlsjh4BD8njGcvhOeCZ83U_oYWgCn0w8qx5--04,6
-gpt_pr-0.2.1.dist-info/RECORD,,

gpt_pr-0.2.1.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-gpt-pr = gptpr.main:main

gpt_pr-0.2.1.dist-info/top_level.txt

@@ -1 +0,0 @@
-gptpr

gptpr/gh.py

@@ -1,27 +0,0 @@
-import os
-from github import Github
-from InquirerPy import inquirer
-
-GH_TOKEN = os.environ.get('GH_TOKEN')
-
-if not GH_TOKEN:
-    print("Please set GH_TOKEN environment variable")
-    exit(1)
-
-gh = Github(GH_TOKEN)
-
-
-def create_pr(pr_data, yield_confirmation):
-    repo = gh.get_repo(
-        f'{pr_data.branch_info.owner}/{pr_data.branch_info.repo}')
-
-    pr_confirmation = yield_confirmation or inquirer.confirm(
-        message="Create GitHub PR?",
-        default=True).execute()
-
-    if pr_confirmation:
-        pr = repo.create_pull(title=pr_data.title, body=pr_data.body,
-                              head=pr_data.branch_info.branch, base=pr_data.branch_info.base_branch)
-        print("Pull request created successfully: ", pr.html_url)
-    else:
-        print('cancelling...')

gptpr/main.py

@@ -1,52 +0,0 @@
-import fire
-from InquirerPy import inquirer
-
-from gptpr.gitutil import get_branch_info
-from gptpr.gh import create_pr
-from gptpr.prdata import get_pr_data
-from gptpr.version import __version__
-
-
-def run(base_branch='main', yield_confirmation=False, version=False):
-    '''
-    Create Pull Requests from current branch with base branch (default 'main' branch)
-    '''
-
-    if version:
-        print('Current version:', __version__)
-        return
-
-    branch_info = get_branch_info(base_branch, yield_confirmation)
-
-    if not branch_info:
-        exit(0)
-
-    pr_data = None
-    generate_pr_data = True
-    while generate_pr_data:
-        pr_data = get_pr_data(branch_info)
-        print('')
-        print('#########################################')
-        print(pr_data.to_display())
-        print('#########################################')
-        print('')
-
-        if yield_confirmation:
-            break
-
-        generate_pr_data = not inquirer.confirm(
-            message="Create PR with this? If 'no', let's try again...",
-            default=True).execute()
-
-        if generate_pr_data:
-            print('Generating another PR data...')
-
-    create_pr(pr_data, yield_confirmation)
-
-
-def main():
-    fire.Fire(run)
-
-
-if __name__ == '__main__':
-    main()

gptpr/prdata.py

@@ -1,161 +0,0 @@
-from dataclasses import dataclass
-import json
-import os
-from openai import OpenAI
-
-from gptpr.gitutil import BranchInfo
-import gptpr.consolecolor as cc
-
-TOKENIZER_RATIO = 4
-MAX_TOKENS = 6000
-
-DEFAULT_PR_TEMPLATE = ('### Ref. [Link]\n\n## What was done?\n[Fill here]\n\n'
-                       '## How was it done?\n[Fill here]\n\n'
-                       '## How was it tested?\n[Fill here with test information from diff content or commits]')
-
-
-def _get_pr_template():
-    pr_template = DEFAULT_PR_TEMPLATE
-
-    try:
-        github_dir = os.path.join(os.getcwd(), '.github')
-        github_files = os.listdir(github_dir)
-        pr_template_file = [f for f in github_files if f.lower().startswith('pull_request_template')][0]
-        pr_template_file_path = os.path.join(github_dir, pr_template_file)
-
-        with open(pr_template_file_path, 'r') as f:
-            local_pr_template = f.read()
-
-            if local_pr_template.strip() != '':
-                print('Found PR template at:', pr_template_file_path)
-                pr_template = local_pr_template
-            else:
-                print('Empty PR template at:', pr_template_file_path, 'using default template.')
-    except Exception:
-        print('PR template not found in .github dir. Using default template.')
-
-    return pr_template
-
-
-@dataclass
-class PrData():
-    branch_info: BranchInfo
-    title: str
-    body: str
-
-    def to_display(self):
-        return '\n'.join([
-            f'{cc.bold("Repository")}: {cc.yellow(self.branch_info.owner)}/{cc.yellow(self.branch_info.repo)}',
-            f'{cc.bold("Title")}: {cc.yellow(self.title)}',
-            f'{cc.bold("Branch name")}: {cc.yellow(self.branch_info.branch)}',
-            f'{cc.bold("Base branch")}: {cc.yellow(self.branch_info.base_branch)}',
-            f'{cc.bold("PR Description")}:\n{self.body}',
-        ])
-
-
-functions = [
-    {
-        "name": "create_pr",
-        "description": "Creates a Github Pull Request",
-        "parameters": {
-            "type": "object",
-            "properties": {
-                "title": {
-                    "type": "string",
-                    "description": "PR title, following angular commit convention"
-                },
-                "description": {
-                    "type": "string",
-                    "description": "PR description"
-                },
-            },
-            "required": ["title", "description"]
-        },
-    }
-]
-
-
-def get_pr_data(branch_info):
-    system_content = ('You are a development assistant designed to craft Git pull requests '
-                      'by incorporating information from main and secondary commits, diff changes, '
-                      'and adhering to a provided PR template. Your output includes a complete PR '
-                      'template with all necessary details and a suitable PR title. In the '
-                      'PR description, detail the work accomplished, the methodology employed, '
-                      'including testing procedures, and list significant changes in bullet points '
-                      'if they are extensive. Avoid incorporating diff content directly into '
-                      'the PR description.')
-
-    messages = [
-        {'role': 'system', 'content': system_content},
-    ]
-
-    if len(branch_info.highlight_commits) > 0:
-        messages.append({'role': 'user', 'content': 'main commits: ' + '\n'.join(branch_info.highlight_commits)})
-        messages.append({'role': 'user', 'content': 'secondary commits: ' + '\n'.join(branch_info.commits)})
-    else:
-        messages.append({'role': 'user', 'content': 'git commits: ' + '\n'.join(branch_info.commits)})
-
-    messages.append({'role': 'user', 'content': 'PR template:\n' + _get_pr_template()})
-
-    current_total_length = sum([len(m['content']) for m in messages])
-
-    if current_total_length / TOKENIZER_RATIO > MAX_TOKENS:
-        raise Exception(f'Current total length {current_total_length} is greater than max tokens {MAX_TOKENS}')
-
-    total_length_with_diff = current_total_length + len(branch_info.diff)
-    if total_length_with_diff / TOKENIZER_RATIO > MAX_TOKENS:
-        print('Total content length (with diff) is too big.', cc.red('Skipping diff content...'))
-    else:
-        messages.append({'role': 'user', 'content': 'Diff changes:\n' + branch_info.diff})
-
-    openai_api_key = os.environ.get('OPENAI_API_KEY')
-
-    if not openai_api_key:
-        print("Please set OPENAI_API_KEY environment variable.")
-        exit(1)
-
-    client = OpenAI(api_key=openai_api_key)
-
-    chat_completion = client.chat.completions.create(
-        messages=messages,
-        model='gpt-4-0613',
-        functions=functions,
-        function_call={'name': 'create_pr'},
-        temperature=0,
-        max_tokens=512,
-        top_p=1,
-        frequency_penalty=0,
-        presence_penalty=0
-    )
-
-    arguments = _parse_json(chat_completion.choices[0].message.function_call.arguments)
-
-    return PrData(
-        branch_info=branch_info,
-        title=arguments['title'],
-        body=arguments['description']
-    )
-
-
-def _parse_json(content):
-    '''
-    A bit of a hack to parse the json content from the chat completion
-    Sometimes it returns a string with invalid json content (line breaks) that
-    makes it hard to parse.
-    example:
-
-    content = '{\n"title": "feat(dependencies): pin dependencies versions",\n"description":
-                "### Ref. [Link]\n\n## What was done? ..."\n}'
-    '''
-
-    try:
-        content = content.replace('{\n"title":', '{"title":')
-        content = content.replace(',\n"description":', ',"description":')
-        content = content.replace('\n}', '}')
-        content = content.replace('\n', '\\n')
-
-        return json.loads(content)
-    except Exception as e:
-        print('Error to decode message:', e)
-        print('Content:', content)
-        raise e

gptpr/test_prdata.py

@@ -1,13 +0,0 @@
-from gptpr.prdata import _parse_json
-
-
-def test_parse_json():
-    content = ('{\n"title": "feat(dependencies): pin dependencies '
-               'versions",\n"description": "### Ref. [Link]\n\n## What was done? ..."\n}')
-
-    expected = {
-        'title': 'feat(dependencies): pin dependencies versions',
-        'description': '### Ref. [Link]\n\n## What was done? ...'
-    }
-
-    assert _parse_json(content) == expected, "The function did not return the expected output."

gptpr/version.py

@@ -1 +0,0 @@
-__version__ = "0.2.1"