gpt-pr 0.0.3__tar.gz → 0.1.0__tar.gz

{gpt-pr-0.0.3 → gpt-pr-0.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gpt-pr
-Version: 0.0.3
+Version: 0.1.0
 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

{gpt-pr-0.0.3 → gpt-pr-0.1.0}/README.md

@@ -35,9 +35,11 @@ You can install and use GPT-PR in one of two ways. Choose the option that best s
 1. Install the package:
 
 ```bash
-pip install gpt-pr
+pip install -U gpt-pr
 ```
 
+> Note: Use this command to **update** gpt-pr package to the latest version.
+
 2. Export API keys as environment variables ([Authentication & API Keys](#authentication--api-keys)).
 
 3. Inside the Git repository you are working on, ensure you have pushed your branch to origin, then run:

{gpt-pr-0.0.3 → gpt-pr-0.1.0}/gpt_pr.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gpt-pr
-Version: 0.0.3
+Version: 0.1.0
 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

gpt-pr-0.1.0/gptpr/__version__.py

@@ -0,0 +1 @@
+__version__ = '0.1.0'

{gpt-pr-0.0.3 → gpt-pr-0.1.0}/gptpr/prdata.py

@@ -18,9 +18,32 @@ if not OPENAI_API_KEY:
 
 openai.api_key = OPENAI_API_KEY
 
-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]')
+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
@@ -80,7 +103,7 @@ def get_pr_data(branch_info):
     else:
         messages.append({'role': 'user', 'content': 'git commits: ' + '\n'.join(branch_info.commits)})
 
-    messages.append({'role': 'user', 'content': 'PR template:\n' + pr_template})
+    messages.append({'role': 'user', 'content': 'PR template:\n' + get_pr_template()})
 
     current_total_length = sum([len(m['content']) for m in messages])
 

gpt-pr-0.0.3/gptpr/__version__.py

@@ -1 +0,0 @@
-__version__ = '0.0.3'