semi stable

pr_agent/tools/pr_similar_issue.py

@@ -0,0 +1,77 @@
+import copy
+import json
+import logging
+from typing import List, Tuple
+
+from jinja2 import Environment, StrictUndefined
+
+from pr_agent.algo.ai_handler import AiHandler
+from pr_agent.algo.pr_processing import get_pr_diff, retry_with_fallback_models
+from pr_agent.algo.token_handler import TokenHandler
+from pr_agent.algo.utils import load_yaml
+from pr_agent.config_loader import get_settings
+from pr_agent.git_providers import get_git_provider
+from pr_agent.git_providers.git_provider import get_main_pr_language
+
+
+class PRSimilarIssue:
+    def __init__(self, pr_url: str, issue_url: str, args: list = None):
+        load_data_from_local = True
+        if not load_data_from_local:
+            self.git_provider = get_git_provider()()
+            repo_name, issue_number = self.git_provider._parse_issue_url(issue_url.split('=')[-1])
+            self.git_provider.repo = repo_name
+            self.git_provider.repo_obj = self.git_provider.github_client.get_repo(repo_name)
+            repo_obj = self.git_provider.repo_obj
+
+            def _process_issue(issue):
+                header = body = issue_str = comments_str = ""
+                if issue.pull_request:
+                    return header, body, issue_str, comments_str
+                header = issue.title
+                body = issue.body
+                comments_obj = list(issue.get_comments())
+                comments_str = ""
+                for i, comment in enumerate(comments_obj):
+                    comments_str += f"comment {i}:\n{comment.body}\n\n\n"
+                issue_str = f"Issue Header: \"{header}\"\n\nIssue Body:\n{body}"
+                return header, body, issue_str, comments_str
+
+            main_issue = repo_obj.get_issue(issue_number)
+            assert not main_issue.pull_request
+            _, _, main_issue_str, main_comments_str = _process_issue(main_issue)
+
+            issues_str_list = []
+            comments_str_list = []
+            issues = list(repo_obj.get_issues(state='all')) # 'open', 'closed', 'all'
+            for i, issue in enumerate(issues):
+                if issue.url == main_issue.url:
+                    continue
+                if issue.pull_request:
+                    continue
+                _, _, issue_str, comments_str = _process_issue(issue)
+                issues_str_list.append(issue_str)
+                comments_str_list.append(comments_str)
+
+            json_output = {}
+            json_output['main_issue'] = {}
+            json_output['main_issue']['issue'] = main_issue_str
+            json_output['main_issue']['comment'] = main_comments_str
+            json_output['issues'] = {}
+            for i in range(len(issues_str_list)):
+                json_output['issues'][f'issue_{i}'] = {}
+                json_output['issues'][f'issue_{i}']['issue'] = issues_str_list[i]
+                json_output['issues'][f'issue_{i}'][f'comments'] = comments_str_list[i]
+
+            jsonFile = open("/Users/talrid/Desktop/issues_data.json", "w")
+            jsonFile.write(json.dumps(json_output))
+            jsonFile.close()
+        else:
+            jsonFile = open("/Users/talrid/Desktop/issues_data.json", "r")
+            json_output=json.loads(jsonFile.read())
+
+            from langchain.document_loaders import TextLoader
+            from langchain.text_splitter import CharacterTextSplitter
+            text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
+
+            aaa=3