feat: clear memory

This commit is contained in:
Timothy J. Baek
2024-05-19 09:26:24 -07:00
parent 8f83f51c68
commit 6139d775ef
4 changed files with 75 additions and 5 deletions
+20 -1
View File
@@ -108,7 +108,26 @@ async def reset_memory_from_vector_db(
############################
# DeleteUserById
# DeleteMemoriesByUserId
############################
@router.delete("/user", response_model=bool)
async def delete_memory_by_user_id(user=Depends(get_verified_user)):
result = Memories.delete_memories_by_user_id(user.id)
if result:
try:
CHROMA_CLIENT.delete_collection(f"user-memory-{user.id}")
except Exception as e:
log.error(e)
return True
return False
############################
# DeleteMemoryById
############################