From 9cf3c7902ec559f01109102a528cfb92119508ba Mon Sep 17 00:00:00 2001
From: Sindhu Balasubramanian <sindhuba@microsoft.com>
Date: Mon, 15 Sep 2025 11:27:58 -0700
Subject: [PATCH] Add flags for PPAF

---
 src/Common/CosmosClient.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/Common/CosmosClient.ts b/src/Common/CosmosClient.ts
index 54444d09e..fd73d908b 100644
--- a/src/Common/CosmosClient.ts
+++ b/src/Common/CosmosClient.ts
@@ -207,6 +207,8 @@ export function client(): Cosmos.CosmosClient {
     defaultHeaders: _defaultHeaders,
     connectionPolicy: {
       enableEndpointDiscovery: !userContext.selectedRegionalEndpoint,
+      enablePartitionLevelFailover: true,
+      enablePartitionLevelCircuitBreaker: true,
       retryOptions: {
         maxRetryAttemptCount: LocalStorageUtility.getEntryNumber(StorageKey.RetryAttempts),
         fixedRetryIntervalInMilliseconds: LocalStorageUtility.getEntryNumber(StorageKey.RetryInterval),