diff options
Diffstat (limited to 'etc/tool/copilot.py')
| -rw-r--r-- | etc/tool/copilot.py | 226 |
1 files changed, 160 insertions, 66 deletions
diff --git a/etc/tool/copilot.py b/etc/tool/copilot.py index ed1fdf16..646da43b 100644 --- a/etc/tool/copilot.py +++ b/etc/tool/copilot.py @@ -20,6 +20,16 @@ GITHUB_REPOSITORY = os.getenv('GITHUB_REPOSITORY') G4F_PROVIDER = os.getenv('G4F_PROVIDER') G4F_MODEL = os.getenv('G4F_MODEL') or g4f.models.gpt_4 +def get_github_token(): + token = os.getenv('GITHUB_TOKEN') + if not token: + raise ValueError("GITHUB_TOKEN environment variable is not set") + print(f"Token length: {len(token)}") + print(f"Token (masked): {'*' * (len(token) - 4) + token[-4:]}") + if len(token) != 40 or not token.isalnum(): + raise ValueError("GITHUB_TOKEN appears to be invalid (should be 40 alphanumeric characters)") + return token + def get_pr_details(github: Github) -> PullRequest: """ Retrieves the details of the pull request from GitHub. @@ -30,15 +40,24 @@ def get_pr_details(github: Github) -> PullRequest: Returns: PullRequest: An object representing the pull request. """ - with open('./pr_number', 'r') as file: - pr_number = file.read().strip() + pr_number = os.getenv('PR_NUMBER') if not pr_number: - return - - repo = github.get_repo(GITHUB_REPOSITORY) - pull = repo.get_pull(int(pr_number)) + print("PR_NUMBER environment variable is not set.") + return None - return pull + try: + print(f"Attempting to get repo: {GITHUB_REPOSITORY}") + repo = github.get_repo(GITHUB_REPOSITORY) + print(f"Successfully got repo: {repo.full_name}") + + print(f"Attempting to get pull request: {pr_number}") + pull = repo.get_pull(int(pr_number)) + print(f"Successfully got pull request: #{pull.number}") + + return pull + except Exception as e: + print(f"Error in get_pr_details: {e}") + return None def get_diff(diff_url: str) -> str: """ @@ -99,15 +118,36 @@ def get_ai_response(prompt: str, as_json: bool = True) -> Union[dict, str]: Returns: Union[dict, str]: The parsed response from g4f, either as a dictionary or a string. """ - response = g4f.ChatCompletion.create( - G4F_MODEL, - [{'role': 'user', 'content': prompt}], - G4F_PROVIDER, - ignore_stream_and_auth=True - ) - return read_json(response) if as_json else read_text(response) - -def analyze_code(pull: PullRequest, diff: str)-> list[dict]: + max_retries = 5 + providers = [None, 'Chatgpt4Online', 'OpenaiChat', 'Bing', 'Ai4Chat', 'NexraChatGPT'] + + for provider in providers: + for _ in range(max_retries): + try: + response = g4f.chat.completions.create( + G4F_MODEL, + [{'role': 'user', 'content': prompt}], + provider, + ignore_stream_and_auth=True + ) + if as_json: + parsed_response = read_json(response) + if parsed_response and 'reviews' in parsed_response: + return parsed_response + else: + parsed_response = read_text(response) + if parsed_response.strip(): + return parsed_response + except Exception as e: + print(f"Error with provider {provider}: {e}") + + # If all retries and providers fail, return a default response + if as_json: + return {"reviews": []} + else: + return "AI Code Review: Unable to generate a detailed response. Please review the changes manually." + +def analyze_code(pull: PullRequest, diff: str) -> list[dict]: """ Analyzes the code changes in the pull request. @@ -123,28 +163,34 @@ def analyze_code(pull: PullRequest, diff: str)-> list[dict]: current_file_path = None offset_line = 0 - for line in diff.split('\n'): - if line.startswith('+++ b/'): - current_file_path = line[6:] - changed_lines = [] - 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')) and changed_lines: - prompt = create_analyze_prompt(changed_lines, pull, current_file_path) - response = get_ai_response(prompt) - for review in response.get('reviews', []): - review['path'] = current_file_path - comments.append(review) - current_file_path = None - elif line.startswith('-'): - changed_lines.append(line) - else: - changed_lines.append(f"{offset_line}:{line}") - offset_line += 1 - + try: + for line in diff.split('\n'): + if line.startswith('+++ b/'): + current_file_path = line[6:] + changed_lines = [] + 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')) and changed_lines: + prompt = create_analyze_prompt(changed_lines, pull, current_file_path) + response = get_ai_response(prompt) + for review in response.get('reviews', []): + review['path'] = current_file_path + comments.append(review) + current_file_path = None + elif line.startswith('-'): + changed_lines.append(line) + else: + changed_lines.append(f"{offset_line}:{line}") + offset_line += 1 + except Exception as e: + print(f"Error in analyze_code: {e}") + + if not comments: + print("No comments generated by analyze_code") + return comments def create_analyze_prompt(changed_lines: list[str], pull: PullRequest, file_path: str): @@ -194,57 +240,105 @@ def create_review_prompt(pull: PullRequest, diff: str): Returns: str: The generated prompt for review. """ + description = pull.body if pull.body else "No description provided." return f"""Your task is to review a pull request. Instructions: - Write in name of g4f copilot. Don't use placeholder. - Write the review in GitHub Markdown format. - Thank the author for contributing to the project. +- If no issues are found, still provide a brief summary of the changes. -Pull request author: {pull.user.name} -Pull request title: {pull.title} +Pull request author: {pull.user.name or "Unknown"} +Pull request title: {pull.title or "Untitled Pull Request"} Pull request description: --- -{pull.body} +{description} --- Diff: ```diff {diff} ``` + +Please provide a comprehensive review of the changes, highlighting any potential issues or improvements, or summarizing the changes if no issues are found. """ def main(): try: - github = Github(GITHUB_TOKEN) + github_token = get_github_token() + except ValueError as e: + print(f"Error: {str(e)}") + return + + if not GITHUB_REPOSITORY or not os.getenv('PR_NUMBER'): + print("Error: GITHUB_REPOSITORY or PR_NUMBER environment variables are not set.") + return + + print(f"GITHUB_REPOSITORY: {GITHUB_REPOSITORY}") + print(f"PR_NUMBER: {os.getenv('PR_NUMBER')}") + print("GITHUB_TOKEN is set") + + try: + github = Github(github_token) + + # Test GitHub connection + print("Testing GitHub connection...") + try: + user = github.get_user() + print(f"Successfully authenticated as: {user.login}") + except Exception as e: + print(f"Error authenticating: {str(e)}") + print(f"Error type: {type(e).__name__}") + print(f"Error args: {e.args}") + return + + # If connection is successful, proceed with PR details pull = get_pr_details(github) if not pull: - print(f"No PR number found") - exit() + print(f"No PR number found or invalid PR number") + return + print(f"Successfully fetched PR #{pull.number}") if pull.get_reviews().totalCount > 0 or pull.get_issue_comments().totalCount > 0: print(f"Has already a review") - exit() + return + diff = get_diff(pull.diff_url) + review = "AI Code Review: Unable to generate a detailed response." + comments = [] + + try: + review = get_ai_response(create_review_prompt(pull, diff), False) + comments = analyze_code(pull, diff) + except Exception as analysis_error: + print(f"Error during analysis: {analysis_error}") + review += f" Error during analysis: {str(analysis_error)[:200]}" + + print("Comments:", comments) + + review_body = review if review and review.strip() else "AI Code Review" + if not review_body.strip(): + review_body = "AI Code Review: No specific issues found." + + try: + if comments: + pull.create_review(body=review_body, comments=comments, event='COMMENT') + else: + pull.create_review(body=review_body, event='COMMENT') + print("Review posted successfully") + except Exception as post_error: + print(f"Error posting review: {post_error}") + error_message = f"AI Code Review: An error occurred while posting the review. Error: {str(post_error)[:200]}. Please review the changes manually." + pull.create_issue_comment(body=error_message) + except Exception as e: - print(f"Error get details: {e.__class__.__name__}: {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: - if comments: - pull.create_review(body=review, comments=comments) - else: - pull.create_issue_comment(body=review) - except Exception as e: - print(f"Error posting review: {e}") - exit(1) + print(f"Unexpected error in main: {e.__class__.__name__}: {e}") + try: + if 'pull' in locals(): + error_message = f"AI Code Review: An error occurred while processing this pull request. Error: {str(e)[:200]}. Please review the changes manually." + pull.create_issue_comment(body=error_message) + else: + print("Unable to post error message: Pull request object not available") + except Exception as post_error: + print(f"Failed to post error message to pull request: {post_error}") if __name__ == "__main__": main() |