xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-12 16:50:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #382 from Codium-ai/tr/similar_issue_fix

Browse Source 
Enhancements and Error Handling in Similar Issue Tool
This commit is contained in:
mrT23
2023-10-17 09:49:35 -07:00
committed by GitHub
parent bce2262d4e d41fe0cf79
commit fe284a8f91
1 changed files with 20 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
pr_agent/tools/pr_similar_issue.py
View File

@ -1,3 +1,4 @@
import time
from enum import Enum from enum import Enum
from typing import List from typing import List
@ -46,6 +47,13 @@ class PRSimilarIssue:
# check if index exists, and if repo is already indexed # check if index exists, and if repo is already indexed
run_from_scratch = False run_from_scratch = False
if run_from_scratch: # for debugging
if not index_name in pinecone.list_indexes():
get_logger().info('Removing index...')
pinecone.init(api_key=api_key, environment=environment)
pinecone.delete_index(index_name)
get_logger().info('Done')
upsert = True upsert = True
pinecone.init(api_key=api_key, environment=environment) pinecone.init(api_key=api_key, environment=environment)
if not index_name in pinecone.list_indexes(): if not index_name in pinecone.list_indexes():
@ -116,7 +124,16 @@ class PRSimilarIssue:
relevant_comment_number_list = [] relevant_comment_number_list = []
score_list = [] score_list = []
for r in res['matches']: for r in res['matches']:
issue_number = int(r["id"].split('.')[0].split('_')[-1]) # skip example issue
if 'example_issue_' in r["id"]:
continue
try:
issue_number = int(r["id"].split('.')[0].split('_')[-1])
except:
get_logger().debug(f"Failed to parse issue number from {r['id']}")
continue
if original_issue_number == issue_number: if original_issue_number == issue_number:
continue continue
if issue_number not in relevant_issues_number_list: if issue_number not in relevant_issues_number_list:
@ -237,6 +254,7 @@ class PRSimilarIssue:
if not upsert: if not upsert:
get_logger().info('Creating index from scratch...') get_logger().info('Creating index from scratch...')
ds.to_pinecone_index(self.index_name, api_key=api_key, environment=environment) ds.to_pinecone_index(self.index_name, api_key=api_key, environment=environment)
time.sleep(15) # wait for pinecone to finalize indexing before querying
else: else:
get_logger().info('Upserting index...') get_logger().info('Upserting index...')
namespace = "" namespace = ""
@ -244,6 +262,7 @@ class PRSimilarIssue:
concurrency: int = 10 concurrency: int = 10
pinecone.init(api_key=api_key, environment=environment) pinecone.init(api_key=api_key, environment=environment)
ds._upsert_to_index(self.index_name, namespace, batch_size, concurrency) ds._upsert_to_index(self.index_name, namespace, batch_size, concurrency)
time.sleep(5) # wait for pinecone to finalize upserting before querying
get_logger().info('Done') get_logger().info('Done')