anubhav77 commited on
Commit
bc7d2c2
·
1 Parent(s): 4caf01e
Files changed (1) hide show
  1. src/chromaIntf.py +3 -0
src/chromaIntf.py CHANGED
@@ -29,6 +29,7 @@ class ChromaIntf():
29
  persist_docs_directory = "docs"
30
  try:
31
  dbh.restoreFolder("db")
 
32
  except:
33
  print("Probably folder doesn't exist as it is brand new setup")
34
  docs = [
@@ -118,6 +119,8 @@ class ChromaIntf():
118
 
119
  docs = [
120
  Document(page_content=inStr, metadata=metadata)]
 
 
121
  try:
122
  return await self.vectorstore.add_documents(docs,ids=[metadata.ID])
123
  except:
 
29
  persist_docs_directory = "docs"
30
  try:
31
  dbh.restoreFolder("db")
32
+ dbh.restoreFolder("docs")
33
  except:
34
  print("Probably folder doesn't exist as it is brand new setup")
35
  docs = [
 
119
 
120
  docs = [
121
  Document(page_content=inStr, metadata=metadata)]
122
+ with open("./docs/"+metadata['ID'],"w") as fd:
123
+ fd.write(inStr)
124
  try:
125
  return await self.vectorstore.add_documents(docs,ids=[metadata.ID])
126
  except: