consistency_level added to CosmosClient

d7e45d06 · Oleksandr Barabash · 183de997 · d7e45d06
Commit d7e45d06 authored Aug 30, 2022 by Oleksandr Barabash
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletions

cosmos_client.py utils/cosmos_client.py +4 -1

No files found.
--- a/utils/cosmos_client.py
+++ b/utils/cosmos_client.py
@@ -51,7 +51,10 @@ class CosmosClient:
    def __init__(self, host: str, client_id: str):
        self.executor = futures.ThreadPoolExecutor()
        mgmt_credentials = ManagedIdentityCredential(client_id=client_id)
-        self.client = cosmos_client.CosmosClient(host, mgmt_credentials)
+        self.client = cosmos_client.CosmosClient(
+            host, mgmt_credentials,
+            consistency_level=documents.ConsistencyLevel.Strong
+        )
        # self.client = cosmos_client.CosmosClient(host,
        #                                          dict(masterKey=master_key))