From c6633793be200a3468722fce702d86747869daf9 Mon Sep 17 00:00:00 2001 From: jacktempo7 <134408853+jacktempo7@users.noreply.github.com> Date: Tue, 27 Jun 2023 09:14:50 -0400 Subject: [PATCH] T10H-85 - VectorBase change for namespace allocation (#449) * T10H-85 - VectorBase change for namespace allocation Signed-off-by: Jack McKnight * adding namespace kwargs Signed-off-by: Jack McKnight --------- Signed-off-by: Jack McKnight --- gptcache/manager/vector_data/manager.py | 2 ++ gptcache/manager/vector_data/redis_vectorstore.py | 4 +++- 2 files changed, 5 insertions(+), 1 deletion(-) 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: