add misssing files

Signed-off-by: Shoham Elias <shohame@amazon.com>
valkey-io · Dec 18, 2024 · 3e085c1 · 3e085c1
1 parent 03ea9e9
commit 3e085c1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git 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
@@ -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
             ),