ashishbangwal commited on
Commit
1362320
·
1 Parent(s): 6ce472c

chroma-db fix

Browse files
Files changed (2) hide show
  1. components/Database.py +5 -0
  2. requirements.txt +1 -0
components/Database.py CHANGED
@@ -2,6 +2,11 @@
2
  Contain Wrapper Class for ChormaDB client, that can process and store documents and retrive document chunks.
3
  """
4
 
 
 
 
 
 
5
  from io import BytesIO
6
  from typing import List
7
  from typing_extensions import Literal
 
2
  Contain Wrapper Class for ChormaDB client, that can process and store documents and retrive document chunks.
3
  """
4
 
5
+ # for chromaDB
6
+ __import__('pysqlite3')
7
+ import sys
8
+ sys.modules['sqlite3'] = sys.modules.pop('pysqlite3')
9
+
10
  from io import BytesIO
11
  from typing import List
12
  from typing_extensions import Literal
requirements.txt CHANGED
@@ -76,6 +76,7 @@ PyMuPDF==1.24.9
76
  PyMuPDFb==1.24.9
77
  PyPika==0.48.9
78
  pyproject_hooks==1.1.0
 
79
  python-dateutil==2.9.0.post0
80
  python-docx==1.1.2
81
  python-dotenv==1.0.1
 
76
  PyMuPDFb==1.24.9
77
  PyPika==0.48.9
78
  pyproject_hooks==1.1.0
79
+ pysqlite3-binary==0.5.3
80
  python-dateutil==2.9.0.post0
81
  python-docx==1.1.2
82
  python-dotenv==1.0.1