anubhav77 commited on
Commit
cef520e
·
1 Parent(s): 59b904b
Files changed (3) hide show
  1. .gitignore +2 -1
  2. src/chroma_intf.py +1 -1
  3. src/main.py +2 -2
.gitignore CHANGED
@@ -13,4 +13,5 @@ src/baseInfra/__pycache__/*.pyc
13
  src/*.old.py
14
  src/toolSelector.py
15
  test/__pycache__/*.pyc
16
- src/baseInfra/memoryManager.py
 
 
13
  src/*.old.py
14
  src/toolSelector.py
15
  test/__pycache__/*.pyc
16
+ src/baseInfra/memoryManager.py
17
+ models/bge-large-en-v1.5/**
src/chroma_intf.py CHANGED
@@ -23,7 +23,7 @@ embedding = HuggingFaceBgeEmbeddings(
23
 
24
  persist_directory = 'db'
25
  try:
26
- dbh.restoreFolder("db")
27
  except:
28
  print("Probably folder doesn't exist as it is brand new setup")
29
  docs = [
 
23
 
24
  persist_directory = 'db'
25
  try:
26
+ dbh.restoreFolder("./db")
27
  except:
28
  print("Probably folder doesn't exist as it is brand new setup")
29
  docs = [
src/main.py CHANGED
@@ -58,11 +58,11 @@ async def add_text_document(inStr: str,metadata: dict ) -> Any:
58
  @app.get(api_base+"/persist")
59
  async def persist_db():
60
  persist()
61
- return await dbh.backupFolder("db")
62
 
63
  @app.get(api_base+"/reset")
64
  async def reset_db():
65
- return await dbh.restoreFolder("db")
66
 
67
  print(__name__)
68
 
 
58
  @app.get(api_base+"/persist")
59
  async def persist_db():
60
  persist()
61
+ return await dbh.backupFolder("./db")
62
 
63
  @app.get(api_base+"/reset")
64
  async def reset_db():
65
+ return await dbh.restoreFolder("./db")
66
 
67
  print(__name__)
68