From 3e085c11d6411a2e53dcfccd7d970a63a98cd62c Mon Sep 17 00:00:00 2001
From: Shoham Elias <shohame@amazon.com>
Date: Wed, 18 Dec 2024 16:04:51 +0200
Subject: [PATCH] add misssing files

Signed-off-by: Shoham Elias <shohame@amazon.com>
---
 python/python/glide/__init__.py | 4 ++++
 python/python/tests/conftest.py | 6 ++++--
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/python/python/glide/__init__.py b/python/python/glide/__init__.py
index 8f6ceac47b..f2ecc3da4e 100644
--- a/python/python/glide/__init__.py
+++ b/python/python/glide/__init__.py
@@ -111,6 +111,8 @@
     TTransaction,
 )
 from glide.config import (
+    AdvancedGlideClientConfiguration,
+    AdvancedGlideClusterClientConfiguration,
     BackoffStrategy,
     GlideClientConfiguration,
     GlideClusterClientConfiguration,
@@ -176,6 +178,8 @@
     "TGlideClient",
     "TTransaction",
     # Config
+    "AdvancedGlideClientConfiguration",
+    "AdvancedGlideClusterClientConfiguration",
     "GlideClientConfiguration",
     "GlideClusterClientConfiguration",
     "BackoffStrategy",
diff --git a/python/python/tests/conftest.py b/python/python/tests/conftest.py
index f20e2d9cbb..bc8e7e2529 100644
--- a/python/python/tests/conftest.py
+++ b/python/python/tests/conftest.py
@@ -5,6 +5,8 @@
 
 import pytest
 from glide.config import (
+    AdvancedGlideClientConfiguration,
+    AdvancedGlideClusterClientConfiguration,
     GlideClientConfiguration,
     GlideClusterClientConfiguration,
     NodeAddress,
@@ -261,7 +263,7 @@ async def create_client(
         assert database_id == 0
         k = min(3, len(pytest.valkey_cluster.nodes_addr))
         seed_nodes = random.sample(pytest.valkey_cluster.nodes_addr, k=k)
-        cluster_config = GlideClusterClientConfiguration(
+        cluster_config = AdvancedGlideClusterClientConfiguration(
             addresses=seed_nodes if addresses is None else addresses,
             use_tls=use_tls,
             credentials=credentials,
@@ -277,7 +279,7 @@ async def create_client(
         return await GlideClusterClient.create(cluster_config)
     else:
         assert type(pytest.standalone_cluster) is ValkeyCluster
-        config = GlideClientConfiguration(
+        config = AdvancedGlideClientConfiguration(
             addresses=(
                 pytest.standalone_cluster.nodes_addr if addresses is None else addresses
             ),