This commit is contained in:
Timothy J. Baek 2024-09-16 07:14:17 +02:00
parent 40d7e5089d
commit db0c576f48

View File

@ -81,7 +81,7 @@ async def query_memory(
request: Request, form_data: QueryMemoryForm, user=Depends(get_verified_user) request: Request, form_data: QueryMemoryForm, user=Depends(get_verified_user)
): ):
results = VECTOR_DB_CLIENT.search( results = VECTOR_DB_CLIENT.search(
name=f"user-memory-{user.id}", collection_name=f"user-memory-{user.id}",
vectors=[request.app.state.EMBEDDING_FUNCTION(form_data.content)], vectors=[request.app.state.EMBEDDING_FUNCTION(form_data.content)],
limit=form_data.k, limit=form_data.k,
) )