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

View File

@@ -96,6 +96,15 @@ class MemoriesTable:
except:
return False
def delete_memories_by_user_id(self, user_id: str) -> bool:
try:
query = Memory.delete().where(Memory.user_id == user_id)
query.execute()
return True
except:
return False
def delete_memory_by_id_and_user_id(self, id: str, user_id: str) -> bool:
try:
query = Memory.delete().where(Memory.id == id, Memory.user_id == user_id)

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
############################