initial vectordb storage

microsoft · Oct 23, 2024 · 7da80e9 · 7da80e9
1 parent fb49453
commit 7da80e9
Show file tree

Hide file tree

Showing 8 changed files with 2,199 additions and 14 deletions.
diff --git a/python/packages/autogen-ext/pyproject.toml b/python/packages/autogen-ext/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "autogen-ext"
-version = "0.4.0.dev2"
+version = "0.4.0dev1"
 license = {file = "LICENSE-CODE"}
 description = "AutoGen extensions library"
 readme = "README.md"
@@ -15,7 +15,7 @@ classifiers = [
     "Operating System :: OS Independent",
 ]
 dependencies = [
-    "autogen-core==0.4.0.dev2",
+    "autogen-core==0.4.0dev1",
 ]
 
 
@@ -27,6 +27,7 @@ langchain = ["langchain_core~= 0.3.3"]
 azure = ["azure-core", "azure-identity"]
 docker = ["docker~=7.0"]
 openai = ["openai>=1.3"]
+chromadb = ["chromadb~=0.5.15", "sentence-transformers"]
 
 [tool.hatch.build.targets.wheel]
 packages = ["src/autogen_ext"]
@@ -56,4 +57,4 @@ test = "pytest -n auto"
 [tool.mypy]
 [[tool.mypy.overrides]]
 module = "docker.*"
-ignore_missing_imports = true
+ignore_missing_imports = true
diff --git a/python/packages/autogen-ext/src/autogen_ext/storage/__init__.py b/python/packages/autogen-ext/src/autogen_ext/storage/__init__.py
@@ -0,0 +1,4 @@
+from ._chromadb import ChromaVectorDB
+from ._factory import VectorDBFactory
+
+__all__ = ["ChromaVectorDB", "VectorDBFactory"]