JairoDanielMT commited on
Commit
279cdfd
·
verified ·
1 Parent(s): 9753f1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -0
app.py CHANGED
@@ -168,6 +168,31 @@ async def download_vectorstore(
168
  except Exception as e:
169
  raise HTTPException(status_code=500, detail=str(e))
170
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
171
 
172
  if __name__ == "__main__":
173
  import os
 
168
  except Exception as e:
169
  raise HTTPException(status_code=500, detail=str(e))
170
 
171
+ @app.get("/vectorstore/list", tags=["List VectorStores"])
172
+ async def list_vectorstores():
173
+ try:
174
+ vectorstores = [
175
+ f for f in os.listdir(path_db) if os.path.isdir(os.path.join(path_db, f))
176
+ ]
177
+ return {"vectorstores": str(vectorstores)}
178
+ except Exception as e:
179
+ raise HTTPException(status_code=500, detail=str(e))
180
+
181
+
182
+ @app.get("/vectorstore/exists", tags=["Exists VectorStore"])
183
+ async def exists_vectorstore(name: str):
184
+ try:
185
+ vectorstores = [
186
+ f for f in os.listdir(path_db) if os.path.isdir(os.path.join(path_db, f))
187
+ ]
188
+ for vectorstore in vectorstores:
189
+ if name in vectorstore:
190
+ return {"exists": True}
191
+ return {"exists": False}
192
+ except Exception as e:
193
+ raise HTTPException(status_code=500, detail=str(e))
194
+
195
+
196
 
197
  if __name__ == "__main__":
198
  import os