refac: memory handling

This commit is contained in:
Timothy Jaeryang Baek committed 2025-07-20 15:00:24 +04:00
1 parent 521d223f05
commit 40ebf8cd62
2 files changed
+17 -4

No files matched your search

+4
View File
@@ -82,6 +82,10 @@ class QueryMemoryForm(BaseModel):
async def query_memory(
request: Request, form_data: QueryMemoryForm, user=Depends(get_verified_user)
):
memories = Memories.get_memories_by_user_id(user.id)
if not memories:
raise HTTPException(status_code=404, detail="No memories found for user")
results = VECTOR_DB_CLIENT.search(
collection_name=f"user-memory-{user.id}",
vectors=[request.app.state.EMBEDDING_FUNCTION(form_data.content, user=user)],