better handling of namespace

lightrag/namespace.py

@@ -0,0 +1,25 @@
+from typing import Iterable
+
+
+class NameSpace:
+    KV_STORE_FULL_DOCS = "full_docs"
+    KV_STORE_TEXT_CHUNKS = "text_chunks"
+    KV_STORE_LLM_RESPONSE_CACHE = "llm_response_cache"
+
+    VECTOR_STORE_ENTITIES = "entities"
+    VECTOR_STORE_RELATIONSHIPS = "relationships"
+    VECTOR_STORE_CHUNKS = "chunks"
+
+    GRAPH_STORE_CHUNK_ENTITY_RELATION = "chunk_entity_relation"
+
+    DOC_STATUS = "doc_status"
+
+
+def make_namespace(prefix: str, base_namespace: str):
+    return prefix + base_namespace
+
+
+def is_namespace(namespace: str, base_namespace: str | Iterable[str]):
+    if isinstance(base_namespace, str):
+        return namespace.endswith(base_namespace)
+    return any(is_namespace(namespace, ns) for ns in base_namespace)