itachi-ai commited on
Commit
4c04658
·
verified ·
1 Parent(s): 2913302

updated vectorizer

Browse files
Files changed (1) hide show
  1. vectorize.py +1 -1
vectorize.py CHANGED
@@ -38,7 +38,7 @@ class VectorDataBase():
38
  return 'Some issues'
39
  @staticmethod
40
  def add_collection_database(doc):
41
- collection.insert_one(
42
  {
43
  'text': doc.page_content,
44
  'embedding': embeddings.embed_query(doc.page_content),
 
38
  return 'Some issues'
39
  @staticmethod
40
  def add_collection_database(doc):
41
+ self.db_collection.insert_one(
42
  {
43
  'text': doc.page_content,
44
  'embedding': embeddings.embed_query(doc.page_content),