From bd301285fabc4586650a5842a2ebe8c5046ede62 Mon Sep 17 00:00:00 2001
From: eladkal <45845474+eladkal@users.noreply.github.com>
Date: Sun, 3 Oct 2021 21:56:11 +0300
Subject: [PATCH] update azure cosmos to latest version

---
 airflow/providers/microsoft/azure/hooks/azure_cosmos.py | 6 +++---
 setup.py                                                | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/airflow/providers/microsoft/azure/hooks/azure_cosmos.py b/airflow/providers/microsoft/azure/hooks/azure_cosmos.py
index a0188db2c2732..b75d75bfbfa06 100644
--- a/airflow/providers/microsoft/azure/hooks/azure_cosmos.py
+++ b/airflow/providers/microsoft/azure/hooks/azure_cosmos.py
@@ -27,7 +27,7 @@
 from typing import Any, Dict, Optional
 
 from azure.cosmos.cosmos_client import CosmosClient
-from azure.cosmos.errors import HTTPFailure
+from azure.cosmos.exceptions import CosmosHttpResponseError
 
 from airflow.exceptions import AirflowBadRequest
 from airflow.hooks.base import BaseHook
@@ -307,7 +307,7 @@ def get_document(
                     document_id,
                 )
             )
-        except HTTPFailure:
+        except CosmosHttpResponseError:
             return None
 
     def get_documents(
@@ -334,7 +334,7 @@ def get_documents(
             )
 
             return list(result_iterable)
-        except HTTPFailure:
+        except CosmosHttpResponseError:
             return None
 
 
diff --git a/setup.py b/setup.py
index 8c04a23f40c6e..76e4b1d2629b2 100644
--- a/setup.py
+++ b/setup.py
@@ -201,7 +201,7 @@ def write_version(filename: str = os.path.join(*[my_dir, "airflow", "git_version
 ]
 azure = [
     'azure-batch>=8.0.0',
-    'azure-cosmos>=3.0.1,<4',
+    'azure-cosmos>=4.0.0,<5',
     'azure-datalake-store>=0.0.45',
     'azure-identity>=1.3.1',
     'azure-keyvault>=4.1.0',