feat(database): add vector database with ConfigType enum

2026-03-12 12:25:32 +08:00 · 2026-02-03 17:25:24 +08:00
parent cf83c09165
commit 9efdbb3327
6 changed files with 592 additions and 72 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,8 +7,11 @@ requires-python = ">=3.10"
 dependencies = [
    "accelerate>=1.12.0",
    "datasets>=4.5.0",
-    "h5py>=3.15.1",
    "httpx[socks]>=0.28.1",
+    "lancedb>=0.27.1",
+    "polars[database,numpy,pandas,pydantic]>=1.37.1",
+    "pyarrow>=23.0.0",
+    "pydantic>=2.12.5",
    "torch>=2.10.0",
    "torchvision>=0.25.0",
    "transformers>=5.0.0",