openai-simple-vectorstore 0.1.0__py3-none-any.whl

openai_simple_vectorstore/__init__.py

@@ -0,0 +1,60 @@
+from .base import *
+from .schemas import *
+from .settings import *
+from .utils import *
+from .version import *
+
+
+def __getattr__(name):
+    # 顶层便捷导出：RedisVectorStore / MilvusVectorStore / RedisVLEngine / MilvusEngine
+    # 及其它类型。backend 相关模块在首次访问时才被导入，避免强制依赖未安装的驱动。
+    if name in ("RedisVLEngine", "RedisVectorStore"):
+        from .engines import redis_vectorstore as _m
+        return getattr(_m, name)
+    if name in ("MilvusEngine", "MilvusVectorStore"):
+        from .engines import milvus_vectorstore as _m
+        return getattr(_m, name)
+    if name in ("MilvusLiteEngine", "MilvusLiteVectorStore"):
+        from .engines import milvus_lite_vectorstore as _m
+        return getattr(_m, name)
+    if name in ("PgVectorEngine", "PgVectorVectorStore"):
+        from .engines import pgvector_vectorstore as _m
+        return getattr(_m, name)
+    if name in ("ElasticsearchEngine", "ElasticsearchVectorStore"):
+        from .engines import elasticsearch_vectorstore as _m
+        return getattr(_m, name)
+    if name in ("SqliteVecEngine", "SqliteVecVectorStore"):
+        from .engines import sqlite_vec_vectorstore as _m
+        return getattr(_m, name)
+    if name in ("create_vector_store", "get_vector_store_class", "register_vector_store"):
+        from .registry import _FACTORY_FUNCTIONS
+        return _FACTORY_FUNCTIONS[name]
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+def _load_default_registry():
+    from .registry import register_vector_store
+
+    def _try_register(name, module_name, attr_name):
+        try:
+            module = __import__(
+                "openai_simple_vectorstore.engines." + module_name,
+                fromlist=[attr_name],
+            )
+            cls = getattr(module, attr_name)
+            register_vector_store(name, cls)
+        except Exception as exc:  # pragma: no cover - 取决于环境
+            import logging
+            logging.getLogger(__name__).debug(
+                "vector store %r not registered: %s", name, exc
+            )
+
+    _try_register("redis", "redis_vectorstore", "RedisVectorStore")
+    _try_register("milvus", "milvus_vectorstore", "MilvusVectorStore")
+    _try_register("milvus-lite", "milvus_lite_vectorstore", "MilvusLiteVectorStore")
+    _try_register("pgvector", "pgvector_vectorstore", "PgVectorVectorStore")
+    _try_register("elasticsearch", "elasticsearch_vectorstore", "ElasticsearchVectorStore")
+    _try_register("sqlite-vec", "sqlite_vec_vectorstore", "SqliteVecVectorStore")
+
+
+_load_default_registry()