fix: chroma pysqlite version (#350)

2023-07-27 09:42:27 +02:00
parent 8c91b75b98
commit 079e35b205
3 changed files with 34 additions and 2 deletions
--- a/embedchain/vectordb/chroma_db.py
+++ b/embedchain/vectordb/chroma_db.py
@@ -1,6 +1,12 @@
 import logging

-import chromadb
+try:
+    import chromadb
+except RuntimeError:
+    from embedchain.utils import use_pysqlite3
+
+    use_pysqlite3()
+    import chromadb
 from chromadb.config import Settings

 from embedchain.vectordb.base_vector_db import BaseVectorDB