diff --git a/gptcache/manager/vector_data/manager.py b/gptcache/manager/vector_data/manager.py index 9320ae71..2bcb0a36 100644 --- a/gptcache/manager/vector_data/manager.py +++ b/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, diff --git a/gptcache/manager/vector_data/redis_vectorstore.py b/gptcache/manager/vector_data/redis_vectorstore.py index f0dbdde5..aef512a5 100644 --- a/gptcache/manager/vector_data/redis_vectorstore.py +++ b/gptcache/manager/vector_data/redis_vectorstore.py @@ -49,6 +49,7 @@ 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 @@ -56,7 +57,8 @@ def __init__( 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: