feat: add Milvus as a choice of vectorstore. (#1389)

* unit tests passed * add tests * final format linting needed * pre pr done * add description to code * codespell keep origin * final edits
Sinaptik-AI · Oct 16, 2024 · 2bc45ef · 2bc45ef
1 parent df06bec
commit 2bc45ef
Show file tree

Hide file tree

Showing 5 changed files with 700 additions and 322 deletions.
diff --git a/pandasai/ee/vectorstores/__init__.py b/pandasai/ee/vectorstores/__init__.py
@@ -4,6 +4,7 @@
 
 from .chroma import ChromaDB
 from .lanceDB import LanceDB
+from .milvus import Milvus
 from .qdrant import Qdrant
 
-__all__ = ["ChromaDB", "Qdrant", "LanceDB"]
+__all__ = ["ChromaDB", "Qdrant", "LanceDB", "Milvus"]