xtekky/gpt4free
1
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-11-26 09:57:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix offset_line in copilot

Browse Source
This commit is contained in:
Heiner Lohaus 2024-01-20 18:54:32 +01:00
parent 8787db385e
commit de993f539c
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
etc/tool/copilot.py
View File

@ -108,8 +108,8 @@ def analyze_code(pull: PullRequest, diff: str)-> list:
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)
if line.startswith('\\') or line.startswith('diff') and changed_lines:
prompt = create_prompt(changed_lines, pull, current_file_path)
response = get_ai_response(prompt)
for review in response.get('reviews', []):
review['path'] = current_file_path
@ -117,11 +117,12 @@ def analyze_code(pull: PullRequest, diff: str)-> list:
changed_lines = []
current_file_path = None
elif not line.startswith('-'):
changed_lines.append(line)
changed_lines.append(f"{offset_line}:{line}")
offset_line += 1
return comments
def create_prompt(changed_lines: list, pull: PullRequest, file_path: str, offset_line: int):
def create_prompt(changed_lines: list, pull: PullRequest, file_path: str):
"""
Creates a prompt for the g4f model.
@ -132,8 +133,7 @@ def create_prompt(changed_lines: list, pull: PullRequest, file_path: str, offset
Returns:
str: The generated prompt.
"""
code = "\n".join([f"{offset_line+idx}:{line}" for idx, line in enumerate(changed_lines)])
print("Code:", code)
code = "\n".join(changed_lines)
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}