Update APIs

Signed-off-by: Edwin Yu <[email protected]>
MemMachine · malatewang · Apr 2, 2026 · Mar 9, 2026 · Mar 10, 2026 · Mar 13, 2026
commit a28a1b2ea8bb1d03dc6ecaaaedab4880287dbf02
diff --git a/packages/server/src/memmachine_server/common/vector_store/vector_store.py b/packages/server/src/memmachine_server/common/vector_store/vector_store.py
@@ -19,6 +19,16 @@
 class Collection(ABC):
     """Collection in a vector store."""
 
+    @abstractmethod
+    async def startup(self) -> None:
+        """Startup."""
+        raise NotImplementedError
+
+    @abstractmethod
+    async def shutdown(self) -> None:
+        """Shutdown."""
+        raise NotImplementedError
+
     @abstractmethod
     async def upsert(
         self,
@@ -43,7 +53,7 @@ async def query(
         score_threshold: float | None = None,
         limit: int | None = None,
         property_filter: FilterExpr | None = None,
-        return_vector: bool = True,
+        return_vector: bool = False,
         return_properties: bool = True,
     ) -> Iterable[QueryResult]:
         """
@@ -82,7 +92,7 @@ async def get(
         self,
         *,
         record_uuids: Iterable[UUID],
-        return_vector: bool = True,
+        return_vector: bool = False,
         return_properties: bool = True,
     ) -> Iterable[Record]:
         """
@@ -164,7 +174,7 @@ async def create_collection(
         raise NotImplementedError
 
     @abstractmethod
-    async def get_collection(self, collection_name: str) -> Collection:
+    async def get_collection(self, collection_name: str) -> Collection | None:
         """
         Get a collection from the vector store.
 
@@ -173,8 +183,8 @@ async def get_collection(self, collection_name: str) -> Collection:
                 Name of the collection to get.
 
         Returns:
-            Collection:
-                The requested collection.
+            Collection | None:
+                The requested collection, or None if it does not exist.
 
         """
         raise NotImplementedError