gpt-pr 0.6.1__py3-none-any.whl → 0.7.1__py3-none-any.whl

gpt_pr-0.7.1.dist-info/RECORD

@@ -0,0 +1,17 @@
+gpt_pr/__init__.py,sha256=Bs14snCQfkRZPxYfLq-haC5mArtZpKcML3Sa9y9LRYs,72
+gpt_pr/checkversion.py,sha256=fY1lLlFDEKY9EFHgMNkxZjPA0WmY2czydIaaIvgt7uc,2428
+gpt_pr/config.py,sha256=jOr95rErnk-MByWtC92bpgY2dyVc6bbrjaVoP61oO-s,2822
+gpt_pr/consolecolor.py,sha256=_JmBMNjIflWMlgP2VkCWu6uQLR9oHBy52uV3TRJJgF4,800
+gpt_pr/gh.py,sha256=JwFRFjwYqO3O3equVJs1FAFTFpG__qEhSkr_-2C3GrI,1238
+gpt_pr/gitutil.py,sha256=yqvpw_pq7Bw_uOcNa4eaUVIFu6YhrK6cjd-74Obtqfk,5996
+gpt_pr/gpt.py,sha256=CSAX9Z6FQOLXOzbLMe_Opqtc3ruDAKTTk7cPqc6Blh0,263
+gpt_pr/main.py,sha256=zl6qeIoKPX-zT-Mt9sGbCQusz5ejoLAdiKLV4MCFpWA,2850
+gpt_pr/prdata.py,sha256=aWPf0cdsdF2EVSEjFqHqZrq1GzPkEPc6jrySiLt6iuU,7583
+gpt_pr/test_checkversion.py,sha256=slw1Tn4H_2QPrKSQfupi5LMIJsDa1fnSU1v5grGX3Ow,4227
+gpt_pr/test_config.py,sha256=J5CSEj8JbKw_57E0IQit5LgmrsCmboe9li2esT2F5pg,3577
+gpt_pr/test_gh.py,sha256=WfZ0GC4C04hk0g0raIDEPODD74AUJcY4BNT3wAYStcQ,1772
+gpt_pr/test_prdata.py,sha256=TF7FZ_PW-NPj3YvC7ICz5NSVQ98JRVWX1lyHfQv3Eow,499
+gpt_pr-0.7.1.dist-info/METADATA,sha256=k8WML4miusXDgbjhfWuqcwfKo9tY081XGpCWJj3dA2M,9082
+gpt_pr-0.7.1.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+gpt_pr-0.7.1.dist-info/entry_points.txt,sha256=EJAujhCl6OFb9AmBGWRXd5VamvyCObYVtByaGU-sLs4,80
+gpt_pr-0.7.1.dist-info/RECORD,,

{gpt_pr-0.6.1.dist-info → gpt_pr-0.7.1.dist-info}/WHEEL

@@ -1,5 +1,4 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.42.0)
+Generator: poetry-core 2.2.1
 Root-Is-Purelib: true
 Tag: py3-none-any
-

gpt_pr-0.7.1.dist-info/entry_points.txt

@@ -0,0 +1,4 @@
+[console_scripts]
+gpt-pr=gpt_pr.main:main
+gpt-pr-config=gpt_pr.main:run_config
+

gpt_pr-0.6.1.dist-info/METADATA

@@ -1,49 +0,0 @@
-Metadata-Version: 2.1
-Name: gpt-pr
-Version: 0.6.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 ==43.0.1
-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.2 ; python_version < "3.11"
-Requires-Dist: cached-property ==1.5.2 ; 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: certifi ==2024.8.30 ; python_version >= "3.6"
-Requires-Dist: distro ==1.9.0 ; python_version >= "3.6"
-Requires-Dist: idna ==3.8 ; 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.5 ; python_version >= "3.7"
-Requires-Dist: typing-extensions ==4.7.1 ; 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.6.1.dist-info/RECORD

@@ -1,17 +0,0 @@
-gptpr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-gptpr/checkversion.py,sha256=PmbGiYJelecs1Fv2aAZwdr6wn-qDC826nsYI4A6LLvk,2389
-gptpr/config.py,sha256=jOr95rErnk-MByWtC92bpgY2dyVc6bbrjaVoP61oO-s,2822
-gptpr/consolecolor.py,sha256=_JmBMNjIflWMlgP2VkCWu6uQLR9oHBy52uV3TRJJgF4,800
-gptpr/gh.py,sha256=E03xg8UHfZlHJAY-aAYphpgS3xl-FZ2dmI5oBC2XdoA,1136
-gptpr/gitutil.py,sha256=4if59fnTmP01dinmkrhE6jUnDNoNFUx3_m_J5iLMWsU,5967
-gptpr/main.py,sha256=gen_8YXYAUJU0BpzULZ28j9Br6Y9PTqTr9JEm4nwsJI,2694
-gptpr/prdata.py,sha256=Uls18CtkxWlflfHKrB62hT21MTUVabf8JgJ6CxOOccQ,7053
-gptpr/test_checkversion.py,sha256=WtJ3v4MMkFG0Kob0R1wi_nwVhcQFd4mXtKCKZHajEhM,4266
-gptpr/test_config.py,sha256=iL3b-Ypv1TaMQTFb6mrDKw8fPio3WBpMcaFkWbiLdNQ,3369
-gptpr/test_prdata.py,sha256=rSJ-yqOdw-iYdBWyqnA2SXbdrhT8KgIkRTTf9SY1S1g,474
-gptpr/version.py,sha256=baAcEjLSYFIeNZF51tOMmA_zAMhN8HvKael-UU-Ruec,22
-gpt_pr-0.6.1.dist-info/METADATA,sha256=74diVzLL8Bo-aMbH3brqCX-q1KE1NSwmatmZzl9cp-s,2639
-gpt_pr-0.6.1.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-gpt_pr-0.6.1.dist-info/entry_points.txt,sha256=WhcbcQXqo5-IGliYWiYMhop4-Wm7bcH2ljFKLWrmO7c,81
-gpt_pr-0.6.1.dist-info/top_level.txt,sha256=DZcbzlsjh4BD8njGcvhOeCZ83U_oYWgCn0w8qx5--04,6
-gpt_pr-0.6.1.dist-info/RECORD,,

gpt_pr-0.6.1.dist-info/entry_points.txt

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

gpt_pr-0.6.1.dist-info/top_level.txt

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

gptpr/gh.py

@@ -1,37 +0,0 @@
-import os
-from github import Github
-from InquirerPy import inquirer
-from gptpr.config import config, config_command_example, CONFIG_README_SECTION
-
-
-def _get_gh_token():
-    gh_token = config.get_user_config('GH_TOKEN')
-    if not gh_token:
-        gh_token = os.environ.get('GH_TOKEN')
-
-    if not gh_token:
-        print('Please set "gh_token" config. Just run:',
-              config_command_example('gh_token', '[my gh token]'),
-              'more about at', CONFIG_README_SECTION)
-        exit(1)
-
-    return gh_token
-
-
-gh = Github(_get_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.create_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,102 +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__
-from gptpr.config import config, config_command_example, CONFIG_README_SECTION
-from gptpr import consolecolor as cc
-from gptpr.checkversion import check_for_updates
-
-
-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 set_config(name, value):
-    name = name.upper()
-    config.set_user_config(name, value)
-    config.persist()
-
-    print('Config value', cc.bold(name), 'set to', cc.yellow(value))
-
-
-def get_config(name):
-    upper_name = name.upper()
-    print('Config value', cc.bold(name), '=', cc.yellow(config.get_user_config(upper_name)))
-
-
-def reset_config(name):
-    upper_name = name.upper()
-    config.reset_user_config(upper_name)
-    print('Config value', cc.bold(name), '=', cc.yellow(config.get_user_config(upper_name)))
-
-
-def print_config():
-    print('Config values at', cc.yellow(config.get_filepath()))
-    print('')
-    print('To set values, just run:', cc.yellow(config_command_example('[config name]', '[value]')))
-    print('More about at', cc.yellow(CONFIG_README_SECTION))
-    print('')
-    current_section = None
-    for section, option, value in config.all_values():
-        if current_section != section:
-            print('')
-            current_section = section
-
-        print(f'[{cc.bold(section)}]', option, '=', cc.yellow(value))
-
-
-def main():
-    check_for_updates()
-
-    fire.Fire(run)
-
-
-def run_config():
-    check_for_updates()
-
-    fire.Fire({
-        'set': set_config,
-        'get': get_config,
-        'print': print_config,
-        'reset': reset_config
-    })
-
-
-if __name__ == '__main__':
-    main()

gptpr/prdata.py

@@ -1,199 +0,0 @@
-from dataclasses import dataclass
-import json
-import os
-from openai import OpenAI
-
-from gptpr.gitutil import BranchInfo
-from gptpr.config import config, CONFIG_PROJECT_REPO_URL
-import gptpr.consolecolor as cc
-
-TOKENIZER_RATIO = 4
-
-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
-
-
-def _get_open_ai_key():
-    api_key = config.get_user_config('OPENAI_API_KEY')
-
-    if not api_key:
-        api_key = os.environ.get('OPENAI_API_KEY')
-
-    if not api_key:
-        print('Please set "openai_api_key" config, just run:',
-              cc.yellow('gpt-pr-config set openai_api_key [open ai key]'))
-        exit(1)
-
-    return api_key
-
-
-def _count_tokens(text):
-    return len(text.split(' '))
-
-
-@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.create_body()}',
-        ])
-
-    def create_body(self):
-        body = self.body
-
-        if config.get_user_config('ADD_TOOL_SIGNATURE') == 'true':
-            pr_signature = f'Generated by [GPT-PR]({CONFIG_PROJECT_REPO_URL})'
-            body += '\n\n---\n\n' + pr_signature
-
-        return 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):
-    client = OpenAI(api_key=_get_open_ai_key())
-
-    messages = _get_messages(branch_info)
-
-    openai_model = config.get_user_config('OPENAI_MODEL')
-    print('Using OpenAI model:', cc.yellow(openai_model))
-
-    chat_completion = client.chat.completions.create(
-        messages=messages,
-        model=openai_model,
-        functions=functions,
-        function_call={'name': 'create_pr'},
-        temperature=0,
-        max_tokens=1000,
-        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 _get_messages(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' + '\n'.join(branch_info.commits)})
-
-    messages.append({'role': 'user', 'content': 'PR template:\n' + _get_pr_template()})
-
-    joined_messages = '\n'.join([m['content'] for m in messages])
-    current_total_tokens = _count_tokens(joined_messages)
-
-    input_max_tokens = int(config.get_user_config('INPUT_MAX_TOKENS'))
-
-    if current_total_tokens > input_max_tokens:
-        exp_message = (f'Length of {current_total_tokens} tokens for basic prompt '
-                       f'(description and commits) is greater than max tokens {input_max_tokens} '
-                       '(config \'input_max_tokens\')')
-        raise Exception(exp_message)
-
-    total_tokens_with_diff = current_total_tokens + _count_tokens(branch_info.diff)
-    if total_tokens_with_diff > input_max_tokens:
-        print_msg = (f'Length git changes with diff is too big (total is {total_tokens_with_diff}, '
-                     f'\'input_max_tokens\' config is {input_max_tokens}).')
-        print(print_msg, cc.red('Skipping changes diff content...'))
-    else:
-        messages.append({'role': 'user', 'content': 'Diff changes:\n' + branch_info.diff})
-
-    return messages
-
-
-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.6.1"