T10H-85 - VectorBase change for namespace allocation (#449)

* T10H-85 - VectorBase change for namespace allocation

Signed-off-by: Jack McKnight <jack@tempoclaims.com>

* adding namespace kwargs

Signed-off-by: Jack McKnight <jack@tempoclaims.com>

---------

Signed-off-by: Jack McKnight <jack@tempoclaims.com>

Author: jacktempo7

gptcache/manager/vector_data/manager.py

@@ -203,6 +203,7 @@ def get(name, **kwargs):
             port = kwargs.get("port", "6379")
             user = kwargs.get("user")
             password = kwargs.get("password")
+            namespace = kwargs.get("namespace")
             dimension = kwargs.get("dimension", DIMENSION)
             collection_name = kwargs.get("collection_name", COLLECTION_NAME)
 
@@ -211,6 +212,7 @@ def get(name, **kwargs):
                 port=port,
                 username=user,
                 password=password,
+                namespace=namespace,
                 dimension=dimension,
                 collection_name=collection_name,
                 top_k=top_k,

gptcache/manager/vector_data/redis_vectorstore.py

@@ -49,14 +49,16 @@ def __init__(
         dimension: int = 0,
         collection_name: str = "gptcache",
         top_k: int = 1,
+        namespace: str = "",
     ):
         self._client = Redis(
             host=host, port=int(port), username=username, password=password
         )
         self.top_k = top_k
         self.dimension = dimension
         self.collection_name = collection_name
-        self.doc_prefix = "doc:"
+        self.namespace = namespace
+        self.doc_prefix = f"{self.namespace}doc:"  # Prefix with the specified namespace
         self._create_collection(collection_name)
 
     def _check_index_exists(self, index_name: str) -> bool: