From 738f9856a49490902421030f385afbd4efc42852 Mon Sep 17 00:00:00 2001 From: abishlal Date: Sat, 21 Jun 2025 17:28:59 +0530 Subject: [PATCH] feat: add WorkItemTrackingClient to Azure DevOps provider and update client return type Signed-off-by: abishlal --- pr_agent/git_providers/azuredevops_provider.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pr_agent/git_providers/azuredevops_provider.py b/pr_agent/git_providers/azuredevops_provider.py index 1f08f4b8..3ce70c94 100644 --- a/pr_agent/git_providers/azuredevops_provider.py +++ b/pr_agent/git_providers/azuredevops_provider.py @@ -22,6 +22,7 @@ try: from azure.devops.connection import Connection # noinspection PyUnresolvedReferences from azure.devops.released.git import (Comment, CommentThread, GitPullRequest, GitVersionDescriptor, GitClient, CommentThreadContext, CommentPosition) + from azure.devops.released.work_item_tracking import WorkItemTrackingClient # noinspection PyUnresolvedReferences from azure.identity import DefaultAzureCredential from msrest.authentication import BasicAuthentication @@ -566,7 +567,7 @@ class AzureDevopsProvider(GitProvider): return workspace_slug, repo_slug, pr_number @staticmethod - def _get_azure_devops_client() -> GitClient: + def _get_azure_devops_client() -> Tuple[GitClient, WorkItemTrackingClient]: org = get_settings().azure_devops.get("org", None) pat = get_settings().azure_devops.get("pat", None) @@ -654,7 +655,7 @@ class AzureDevopsProvider(GitProvider): get_logger().exception(f"Failed to get linked work items, error: {e}") return [] - def get_work_items(self, work_item_ids: int) -> list: + def get_work_items(self, work_item_ids: list) -> list: """ Get work items by their IDs. """