get_valkey_connection works with shard client as well

(cherry picked from commit 61c0220)

Author: amirreza8002

django_valkey/__init__.py

@@ -2,7 +2,7 @@
 __version__ = ".".join(map(str, VERSION))
 
 
-def get_valkey_connection(alias="default", write=True):
+def get_valkey_connection(alias="default", write=True, key=None):
     """
     Helper used for obtaining a raw valkey client.
     """
@@ -18,4 +18,7 @@ def get_valkey_connection(alias="default", write=True):
     if not hasattr(cache.client, "get_client"):
         raise NotImplementedError(error_message)
 
+    if hasattr(cache.client, "get_server_name"):
+        return cache.client.get_client(key=key)
+
     return cache.client.get_client(write)