1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
|
import sys
from pathlib import Path
sys.path.append(str(Path(__file__).parent.parent.parent))
import g4f
import json
import os
import re
import requests
from typing import Union
from github import Github
from github.PullRequest import PullRequest
g4f.debug.logging = True
g4f.debug.version_check = False
GITHUB_TOKEN = os.getenv('GITHUB_TOKEN')
G4F_PROVIDER = os.getenv('G4F_PROVIDER') or g4f.Provider.OpenaiChat
G4F_MODEL = os.getenv('G4F_MODEL') or g4f.models.gpt_4
def get_pr_details(github: Github) -> PullRequest:
"""
Rteurns the details of the pull request from GitHub.
Returns:
PullRequest: A PullRequest instance.
"""
with open(os.getenv('GITHUB_EVENT_PATH', ''), 'r') as file:
data = json.load(file)
repo = github.get_repo(f"{data['repository']['owner']['login']}/{data['repository']['name']}")
pull = repo.get_pull(data['number'])
return pull
def get_diff(diff_url: str) -> str:
"""
Fetches the diff of the pull request.
Args:
pull (PullRequest): Pull request.
Returns:
str or None: The diff of the pull request or None if not available.
"""
response = requests.get(diff_url)
response.raise_for_status()
return response.text
def read_json(text: str) -> dict:
match = re.search(r"```(json|)\n(?P<code>[\S\s]+?)\n```", text)
if match:
text = match.group("code")
try:
return json.loads(text.strip())
except json.JSONDecodeError:
print("No valid json:", text)
return {}
def read_text(text: str) -> dict:
match = re.search(r"```(markdown|)\n(?P<text>[\S\s]+?)\n```", text)
if match:
return match.group("text")
return text
def get_ai_response(prompt, as_json: bool = True) -> Union[dict, str]:
"""
Gets a response from g4f API based on the prompt.
Args:
prompt (str): The prompt to send to g4f.
Returns:
dict: The parsed response from g4f.
"""
response = g4f.ChatCompletion.create(
G4F_MODEL,
[{'role': 'user', 'content': prompt}],
G4F_PROVIDER,
ignore_stream_and_auth=True
)
if as_json:
return read_json(response)
return read_text(response)
def analyze_code(pull: PullRequest, diff: str)-> list:
"""
Analyzes the code changes in the pull request.
Args:
diff (str): The diff of the pull request.
pr_details (dict): Details of the pull request.
Returns:
list: List of comments generated by the analysis.
"""
comments = []
changed_lines = []
current_file_path = None
offset_line = 0
for line in diff.split('\n'):
if line.startswith('+++ b/'):
current_file_path = line[6:]
elif line.startswith('@@'):
match = re.search(r'\+([0-9]+?),', line)
if match:
offset_line = int(match.group(1))
elif current_file_path:
if line.startswith('\\') or line.startswith('diff'):
prompt = create_prompt(changed_lines, pull, current_file_path, offset_line)
response = get_ai_response(prompt)
for review in response.get('reviews', []):
review['path'] = current_file_path
comments.append(review)
changed_lines = []
current_file_path = None
elif not line.startswith('-'):
changed_lines.append(line)
return comments
def create_prompt(changed_lines: list, pull: PullRequest, file_path: str, offset_line: int):
"""
Creates a prompt for the g4f model.
Args:
diff (str): The line of code to analyze.
pr_details (dict): Details of the pull request.
Returns:
str: The generated prompt.
"""
code = "\n".join([f"{offset_line+idx}:{line}" for idx, line in enumerate(changed_lines)])
print("Code:", code)
example = '{"reviews": [{"line": <line_number>, "body": "<review comment>"}]}'
return f"""Your task is to review pull requests. Instructions:
- Provide the response in following JSON format: {example}
- Do not give positive comments or compliments.
- Provide comments and suggestions ONLY if there is something to improve, otherwise "reviews" should be an empty array.
- Write the comment in GitHub Markdown format.
- Use the given description only for the overall context and only comment the code.
- IMPORTANT: NEVER suggest adding comments to the code.
Review the following code diff in the file "{file_path}" and take the pull request title and description into account when writing the response.
Pull request title: {pull.title}
Pull request description:
---
{pull.body}
---
Each line is prefixed by its number. Code to review:
```
{code}
```
"""
def create_review_prompt(pull: PullRequest, diff: str):
"""
Creates a prompt to create a review.
Args:
diff (str): The line of code to analyze.
Returns:
str: The generated prompt.
"""
return f"""Your task is to review a pull request. Instructions:
- Your name / you are copilot.
- Write the review in GitHub Markdown format.
- Thank the author for contributing to the project.
- Point out that you might leave a few comments on the files.
Pull request author: {pull.user.name}
Pull request title: {pull.title}
Pull request description:
---
{pull.body}
---
Diff:
```diff
{diff}
```
"""
def main():
try:
github = Github(GITHUB_TOKEN)
pull = get_pr_details(github)
diff = get_diff(pull.diff_url)
except Exception as e:
print(f"Error get details: {e}")
exit(1)
try:
review = get_ai_response(create_review_prompt(pull, diff), False)
except Exception as e:
print(f"Error create review: {e}")
exit(1)
try:
comments = analyze_code(pull, diff)
except Exception as e:
print(f"Error analyze: {e}")
exit(1)
print("Comments:", comments)
try:
pull.create_review(body=review, comments=comments)
except Exception as e:
print(f"Error posting review: {e}")
exit(1)
if __name__ == "__main__":
main()
|
