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

{gpt_pr-0.5.0.dist-info → gpt_pr-0.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gpt-pr
-Version: 0.5.0
+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
@@ -8,7 +8,7 @@ Author-email: alissonperez@outlook.com
 License: MIT
 Requires-Python: >=3.7
 Requires-Dist: cffi ==1.15.1
-Requires-Dist: cryptography ==43.0.0
+Requires-Dist: cryptography ==43.0.1
 Requires-Dist: fire ==0.6.0
 Requires-Dist: pycparser ==2.21
 Requires-Dist: wcwidth ==0.2.13
@@ -20,9 +20,9 @@ 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: idna ==3.7 ; python_version >= "3.5"
-Requires-Dist: certifi ==2024.7.4 ; python_version >= "3.6"
+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"
@@ -40,7 +40,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.4 ; 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"

gpt_pr-0.6.1.dist-info/RECORD

@@ -0,0 +1,17 @@
+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,,

gptpr/config.py

@@ -7,7 +7,8 @@ def config_command_example(name, value_sample):
     return f'gpt-pr-config set {name} {value_sample}'
 
 
-CONFIG_README_SECTION = 'https://github.com/alissonperez/gpt-pr?tab=readme-ov-file#configuration'
+CONFIG_PROJECT_REPO_URL = 'https://github.com/alissonperez/gpt-pr'
+CONFIG_README_SECTION = f'{CONFIG_PROJECT_REPO_URL}?tab=readme-ov-file#configuration'
 
 
 class Config:
@@ -15,6 +16,9 @@ class Config:
     config_filename = '.gpt-pr.ini'
 
     _default_config = {
+        # Amenities
+        'ADD_TOOL_SIGNATURE': 'true',  # Add GPT-PR signature to PRs
+
         # Github
         'GH_TOKEN': '',
 

gptpr/gh.py

@@ -30,7 +30,7 @@ def create_pr(pr_data, yield_confirmation):
         default=True).execute()
 
     if pr_confirmation:
-        pr = repo.create_pull(title=pr_data.title, body=pr_data.body,
+        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:

gptpr/gitutil.py

@@ -183,10 +183,15 @@ def _get_stats(repo, base_branch, branch):
             continue
 
         line = line.split('\t')
+
+        # Binary files will not have stats (just "-")
+        added = int(line[0]) if line[0] and line[0].isdigit() else 0
+        removed = int(line[1]) if line[1] and line[1].isdigit() else 0
+
         files_changed.append(FileChange(
             file_path=line[2],
-            lines_added=int(line[0]),
-            lines_removed=int(line[1])
+            lines_added=added,
+            lines_removed=removed
         ))
 
     return files_changed

gptpr/prdata.py

@@ -4,7 +4,7 @@ import os
 from openai import OpenAI
 
 from gptpr.gitutil import BranchInfo
-from gptpr.config import config
+from gptpr.config import config, CONFIG_PROJECT_REPO_URL
 import gptpr.consolecolor as cc
 
 TOKENIZER_RATIO = 4
@@ -67,9 +67,18 @@ class PrData():
             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}',
+            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 = [
     {

gptpr/test_config.py

@@ -87,10 +87,12 @@ def test_all_values(temp_config):
     all_values = config.all_values()
 
     assert all_values == [
+        ('DEFAULT', 'add_tool_signature', 'true'),
         ('DEFAULT', 'gh_token', ''),
         ('DEFAULT', 'input_max_tokens', '15000'),
         ('DEFAULT', 'openai_model', 'gpt-4o-mini'),
         ('DEFAULT', 'openai_api_key', ''),
+        ('user', 'add_tool_signature', 'true'),
         ('user', 'gh_token', ''),
         ('user', 'input_max_tokens', '15000'),
         ('user', 'openai_model', 'gpt-4o-mini'),

gptpr/version.py

@@ -1 +1 @@
-__version__ = "0.5.0"
+__version__ = "0.6.1"

gpt_pr-0.5.0.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=XwHwveyVmeFoHqHpwZXI2BF1Ac3nT1hnEpclmP2TYhQ,2677
-gptpr/consolecolor.py,sha256=_JmBMNjIflWMlgP2VkCWu6uQLR9oHBy52uV3TRJJgF4,800
-gptpr/gh.py,sha256=uSWY_TzbrAM00neOBkyfV5vxDO4FzMtIrs-Zczp-Tck,1127
-gptpr/gitutil.py,sha256=NBD3iRnbFEPRU47w7c5TowwtZieDYkU4zybvv0PoOU0,5783
-gptpr/main.py,sha256=gen_8YXYAUJU0BpzULZ28j9Br6Y9PTqTr9JEm4nwsJI,2694
-gptpr/prdata.py,sha256=h3RvlfGcAXWUSxy6wfVqem4uxk53gxpvSrmj7xxuYUc,6749
-gptpr/test_checkversion.py,sha256=WtJ3v4MMkFG0Kob0R1wi_nwVhcQFd4mXtKCKZHajEhM,4266
-gptpr/test_config.py,sha256=ucqbj2mbee4fDK1JwXVY_FfnPXMeRA3VMM6m4-x7OTU,3270
-gptpr/test_prdata.py,sha256=rSJ-yqOdw-iYdBWyqnA2SXbdrhT8KgIkRTTf9SY1S1g,474
-gptpr/version.py,sha256=LBK46heutvn3KmsCrKIYu8RQikbfnjZaj2xFrXaeCzQ,22
-gpt_pr-0.5.0.dist-info/METADATA,sha256=Dt_i-Of-R8hHlBDFWGm2rzWij5diWPVE6cT2FKeXd1k,2638
-gpt_pr-0.5.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-gpt_pr-0.5.0.dist-info/entry_points.txt,sha256=WhcbcQXqo5-IGliYWiYMhop4-Wm7bcH2ljFKLWrmO7c,81
-gpt_pr-0.5.0.dist-info/top_level.txt,sha256=DZcbzlsjh4BD8njGcvhOeCZ83U_oYWgCn0w8qx5--04,6
-gpt_pr-0.5.0.dist-info/RECORD,,