From d3c7dcc40712f5b28f13205aa8ff5220c511dfaf Mon Sep 17 00:00:00 2001 From: mrT23 Date: Mon, 28 Aug 2023 20:21:29 +0300 Subject: [PATCH] AZURE_DEVOPS_AVAILABLE --- pr_agent/git_providers/azuredevops_provider.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/pr_agent/git_providers/azuredevops_provider.py b/pr_agent/git_providers/azuredevops_provider.py index 3669bae6..71ae0947 100644 --- a/pr_agent/git_providers/azuredevops_provider.py +++ b/pr_agent/git_providers/azuredevops_provider.py @@ -5,10 +5,13 @@ from urllib.parse import urlparse import os -from msrest.authentication import BasicAuthentication -from azure.devops.connection import Connection - -from azure.devops.v7_1.git.models import Comment, CommentThread, GitVersionDescriptor, GitPullRequest +AZURE_DEVOPS_AVAILABLE = True +try: + from msrest.authentication import BasicAuthentication + from azure.devops.connection import Connection + from azure.devops.v7_1.git.models import Comment, CommentThread, GitVersionDescriptor, GitPullRequest +except ImportError: + AZURE_DEVOPS_AVAILABLE = False from ..algo.pr_processing import clip_tokens from ..config_loader import get_settings @@ -19,6 +22,8 @@ from .git_provider import EDIT_TYPE, FilePatchInfo class AzureDevopsProvider: def __init__(self, pr_url: Optional[str] = None, incremental: Optional[bool] = False): + if not AZURE_DEVOPS_AVAILABLE: + raise ImportError("Azure DevOps provider is not available. Please install the required dependencies.") self.azure_devops_client = self._get_azure_devops_client()