Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Improve vector store API #1201

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
malatewang merged 24 commits into from
Apr 2, 2026
Merged

Improve vector store API #1201

Changes from 1 commit
Commits
Show all changes
24 commits
Select commit Hold shift + click to select a range
260a2bf
Improve vector query return value
edwinyyyu Mar 9, 2026
a28a1b2
Update APIs
edwinyyyu Mar 10, 2026
6e4f200
Add partition key to support efficient native multi-tenancy
edwinyyyu Mar 13, 2026
fd313c5
Hoist partition key concept up -- now namespaces and names
edwinyyyu Mar 14, 2026
8d6f304
Revert "Hoist partition key concept up -- now namespaces and names"
edwinyyyu Mar 14, 2026
19aec8e
Reapply "Hoist partition key concept up -- now namespaces and names"
edwinyyyu Mar 16, 2026
b51a62a
Define concurrency guarantees
edwinyyyu Mar 16, 2026
12a9146
Revert two commits
edwinyyyu Mar 16, 2026
a12e15c
Revert "Revert two commits"
edwinyyyu Mar 17, 2026
a6257f6
Update concurrency guarantees
edwinyyyu Mar 17, 2026
e3d77e5
Update lifecycle APIs
edwinyyyu Mar 17, 2026
aef722a
Remove unnecessary error
edwinyyyu Mar 17, 2026
a4b1615
Add new error
edwinyyyu Mar 17, 2026
491afd4
Add collection configuration data model
edwinyyyu Mar 18, 2026
74da1c4
Rename error for symmetry and simplicity
edwinyyyu Mar 18, 2026
3be75a1
Guarantee serialization order
edwinyyyu Mar 18, 2026
0b27fef
Support deserialization
edwinyyyu Mar 18, 2026
031aa6e
Fix return type
edwinyyyu Mar 18, 2026
bb18f3b
Don't confuse {} and None
edwinyyyu Mar 18, 2026
d6ff783
Add naming constraints
edwinyyyu Mar 18, 2026
1dbb8ab
Validate identifiers
edwinyyyu Mar 18, 2026
dc52d85
Add timezone utilities
edwinyyyu Mar 27, 2026
63dfe14
Define behavior for empty Record attributes
edwinyyyu Apr 1, 2026
5108d37
Merge branch 'main' into batch_vectors
malatewang Apr 2, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Add partition key to support efficient native multi-tenancy 
Signed-off-by: Edwin Yu <[email protected]>
  • Loading branch information
@edwinyyyu
edwinyyyu committed Apr 2, 2026
commit 6e4f2006bb71edb4339e4d528b81d6dee589c4d1
12 changes: 12 additions & 0 deletions packages/server/src/memmachine_server/common/vector_store/vector_store.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,12 +33,15 @@ async def shutdown(self) -> None:
async def upsert(
self,
*,
partition_key: str,
Copy link
Copy Markdown
Contributor

@malatewang malatewang Mar 14, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why do we need partition here?

Copy link
Copy Markdown
Contributor Author

@edwinyyyu edwinyyyu Mar 18, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Fixed.

records: Iterable[Record],
) -> None:
"""
Upsert records in the collection.

Args:
partition_key (str):
The key of the partition to which the records belong.
records (Iterable[Record]):
Iterable of records to upsert.

Expand All @@ -49,6 +52,7 @@ async def upsert(
async def query(
self,
*,
partition_key: str,
query_vectors: Iterable[Sequence[float]],
score_threshold: float | None = None,
limit: int | None = None,
Expand All @@ -60,6 +64,8 @@ async def query(
Query for records matching the criteria by query vectors.

Args:
partition_key (str):
The key of the partition to which the records belong.
query_vectors (Iterable[Sequence[float]]):
The vectors to compare against.
score_threshold (float | None):
Expand Down Expand Up @@ -91,6 +97,7 @@ async def query(
async def get(
self,
*,
partition_key: str,
record_uuids: Iterable[UUID],
return_vector: bool = False,
return_properties: bool = True,
Expand All @@ -99,6 +106,8 @@ async def get(
Get records from the collection by their UUIDs.

Args:
partition_key (str):
The key of the partition to which the records belong.
record_uuids (Iterable[UUID]):
Iterable of UUIDs of the records to retrieve.
return_vector (bool):
Expand All @@ -120,12 +129,15 @@ async def get(
async def delete(
self,
*,
partition_key: str,
record_uuids: Iterable[UUID],
) -> None:
"""
Delete records from the collection by their UUIDs.

Args:
partition_key (str):
The key of the partition to which the records belong.
record_uuids (Iterable[UUID]):
Iterable of UUIDs of the records to delete.

Expand Down