jerpint commited on
Commit
2a23253
·
1 Parent(s): 425d30e

last fixes

Browse files
buster/formatters/prompts.py CHANGED
@@ -18,7 +18,7 @@ class SystemPromptFormatter:
18
  documents_list = matched_documents.content.to_list()
19
  documents_str = ""
20
  for idx, doc in enumerate(documents_list):
21
- documents_str += f"<DOCUMENT> {doc} <\DOCUMENT>"
22
 
23
  # truncate the documents to fit
24
  # TODO: increase to actual token count
 
18
  documents_list = matched_documents.content.to_list()
19
  documents_str = ""
20
  for idx, doc in enumerate(documents_list):
21
+ documents_str += f"<DOCUMENT> {doc} <\\DOCUMENT>"
22
 
23
  # truncate the documents to fit
24
  # TODO: increase to actual token count
buster/retriever/sqlite.py CHANGED
@@ -25,7 +25,6 @@ class SQLiteRetriever(Retriever):
25
  else:
26
  self.db_path = None
27
  self.conn = db_path
28
- schema.initialize_db(self.conn)
29
  schema.setup_db(self.conn)
30
 
31
  def __del__(self):
 
25
  else:
26
  self.db_path = None
27
  self.conn = db_path
 
28
  schema.setup_db(self.conn)
29
 
30
  def __del__(self):