Skip to content

Commit

Permalink
fix bug
Browse files Browse the repository at this point in the history
  • Loading branch information
dannyl1u committed Nov 22, 2024
1 parent 6748c20 commit e869744
Show file tree
Hide file tree
Showing 2 changed files with 24 additions and 17 deletions.
39 changes: 23 additions & 16 deletions src/issue_handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,24 +14,31 @@ def handle_new_issue(
full_issue = f"{issue_title} {issue_body}"
similar_issue = query_similar_issue(full_issue, repo_id)

if similar_issue and similar_issue["distance"] < 1 - SIMILARITY_THRESHOLD:
comment_text = f"Closed due to high similarity with issue #{similar_issue['issue_number']} with title '{similar_issue['title']}'"
leave_comment(installation_id, repo_full_name, issue_number, comment_text)
close_issue(installation_id, repo_full_name, issue_number)
logger.info(
f"The new issue #{issue_number} with title '{issue_title}' is most similar to existing issue #{similar_issue['issue_number']} with title '{similar_issue['title']}', with a cosine similarity of {1 - similar_issue['distance']:.2f}."
)
elif similar_issue and similar_issue["distance"] < 1 - (SIMILARITY_THRESHOLD * 0.5):
comment_text = f"Possibly related to issue #{similar_issue['issue_number']} with title '{similar_issue['title']}'"
leave_comment(installation_id, repo_full_name, issue_number, comment_text)
logger.info(
f"The new issue #{issue_number} with title '{issue_title}' is possibly similar to existing issue #{similar_issue['issue_number']} with title '{similar_issue['title']}', with a cosine similarity of {1 - similar_issue['distance']:.2f}."
)
if similar_issue:
if similar_issue["distance"] < 1 - SIMILARITY_THRESHOLD:
comment_text = f"Closed due to high similarity with issue #{similar_issue['issue_number']} with title '{similar_issue['title']}'"
leave_comment(installation_id, repo_full_name, issue_number, comment_text)
close_issue(installation_id, repo_full_name, issue_number)
logger.info(
f"The new issue #{issue_number} with title '{issue_title}' is most similar to existing issue #{similar_issue['issue_number']} with title '{similar_issue['title']}', with a cosine similarity of {1 - similar_issue['distance']:.2f}."
)
elif similar_issue["distance"] < 1 - (SIMILARITY_THRESHOLD * 0.5):
comment_text = f"Possibly related to issue #{similar_issue['issue_number']} with title '{similar_issue['title']}'"
leave_comment(installation_id, repo_full_name, issue_number, comment_text)
logger.info(
f"The new issue #{issue_number} with title '{issue_title}' is possibly similar to existing issue #{similar_issue['issue_number']} with title '{similar_issue['title']}', with a cosine similarity of {1 - similar_issue['distance']:.2f}."
)
else:
comment_text = f"Most likely a new issue, most similar issue: #{similar_issue['issue_number']} with title '{similar_issue['title']}'"
leave_comment(installation_id, repo_full_name, issue_number, comment_text)
logger.info(
f"The new issue #{issue_number} with title '{issue_title}' is not similar enough to close, most similar: #{similar_issue['issue_number']} with title '{similar_issue['title']}', with a cosine similarity of {1 - similar_issue['distance']:.2f}"
)
else:
comment_text = f"Most likely a new issue, most similar issue: #{similar_issue['issue_number']} with title '{similar_issue['title']}'"
comment_text = "No similar issues found. This seems to be a new issue."
leave_comment(installation_id, repo_full_name, issue_number, comment_text)
logger.info(
f"The new issue #{issue_number} with title '{issue_title}' is not similar enough to close, most similar: #{similar_issue['issue_number']} with title '{similar_issue['title']}', with a cosine similarity of {1 - similar_issue['distance']:.2f}"
f"The new issue #{issue_number} with title '{issue_title}' has no similar issues in the database."
)

add_issue_to_chroma(full_issue, issue_number, issue_title, repo_id)
add_issue_to_chroma(full_issue, issue_number, issue_title, repo_id)
2 changes: 1 addition & 1 deletion src/vector_db.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ def remove_issues_from_chroma(repo_id):
collection.delete(ids=results["ids"])


def add_issues_to_chroma(issues):
def add_issues_to_chroma(issues, repo_id):
for issue in issues:
issue_number = issue["number"]
issue_title = issue["title"]
Expand Down

0 comments on commit e869744

Please sign in to comment.