xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-03 20:30:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Azure: refactor publish_code_suggestions() to use azure classes

Browse Source
This commit is contained in:
Thomas De Keulenaer
2025-05-09 13:12:47 +02:00
parent 24a90cab8e
commit 954d61e5dc
1 changed files with 8 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
pr_agent/git_providers/azuredevops_provider.py
View File

@ -21,7 +21,7 @@ try:
# noinspection PyUnresolvedReferences # noinspection PyUnresolvedReferences
from azure.devops.connection import Connection from azure.devops.connection import Connection
# noinspection PyUnresolvedReferences # noinspection PyUnresolvedReferences
from azure.devops.released.git import (Comment, CommentThread, GitPullRequest, GitVersionDescriptor, GitClient, CommentThreadContext) from azure.devops.released.git import (Comment, CommentThread, GitPullRequest, GitVersionDescriptor, GitClient, CommentThreadContext, CommentPosition)
# noinspection PyUnresolvedReferences # noinspection PyUnresolvedReferences
from azure.identity import DefaultAzureCredential from azure.identity import DefaultAzureCredential
from msrest.authentication import BasicAuthentication from msrest.authentication import BasicAuthentication
@ -73,40 +73,13 @@ class AzureDevopsProvider(GitProvider):
f"relevant_lines_start is {relevant_lines_start}") f"relevant_lines_start is {relevant_lines_start}")
continue continue
if relevant_lines_end > relevant_lines_start: thread_context = CommentThreadContext(
post_parameters = { file_path=relevant_file,
"body": body, right_file_start=CommentPosition(offset=1, line=relevant_lines_start),
"path": relevant_file, right_file_end=CommentPosition(offset=1, line=relevant_lines_end))
"line": relevant_lines_end, comment = Comment(content=body, comment_type=1)
"start_line": relevant_lines_start, thread = CommentThread(comments=[comment], thread_context=thread_context)
"start_side": "RIGHT",
}
else: # API is different for single line comments
post_parameters = {
"body": body,
"path": relevant_file,
"line": relevant_lines_start,
"side": "RIGHT",
}
post_parameters_list.append(post_parameters)
if not post_parameters_list:
return False
for post_parameters in post_parameters_list:
try: try:
comment = Comment(content=post_parameters["body"], comment_type=1)
thread = CommentThread(comments=[comment],
thread_context={
"filePath": post_parameters["path"],
"rightFileStart": {
"line": post_parameters["start_line"],
"offset": 1,
},
"rightFileEnd": {
"line": post_parameters["line"],
"offset": 1,
},
})
self.azure_devops_client.create_thread( self.azure_devops_client.create_thread(
comment_thread=thread, comment_thread=thread,
project=self.workspace_slug, project=self.workspace_slug,
@ -114,7 +87,7 @@ class AzureDevopsProvider(GitProvider):
pull_request_id=self.pr_num pull_request_id=self.pr_num
) )
except Exception as e: except Exception as e:
get_logger().warning(f"Azure failed to publish code suggestion, error: {e}") get_logger().error(f"Azure failed to publish code suggestion, error: {e}", suggestion=suggestion)
return True return True
def reply_to_comment_from_comment_id(self, comment_id: int, body: str, is_temporary: bool = False) -> Comment: def reply_to_comment_from_comment_id(self, comment_id: int, body: str, is_temporary: bool = False) -> Comment: