diff --git a/python/semantic_kernel/connectors/memory/chroma/chroma_memory_store.py b/python/semantic_kernel/connectors/memory/chroma/chroma_memory_store.py index b0d8734449df..f6c1853b2cce 100644 --- a/python/semantic_kernel/connectors/memory/chroma/chroma_memory_store.py +++ b/python/semantic_kernel/connectors/memory/chroma/chroma_memory_store.py @@ -181,7 +181,7 @@ async def upsert_batch(self, collection_name: str, records: list[MemoryRecord]) # upsert is checking collection existence return [await self.upsert(collection_name, record) for record in records] - async def get(self, collection_name: str, key: str, with_embedding: bool) -> MemoryRecord: + async def get(self, collection_name: str, key: str, with_embedding: bool = False) -> MemoryRecord: """Gets a record. Args: @@ -200,7 +200,12 @@ async def get(self, collection_name: str, key: str, with_embedding: bool) -> Mem f"Record with key '{key}' does not exist in collection '{collection_name}'" ) from exc - async def get_batch(self, collection_name: str, keys: list[str], with_embeddings: bool) -> list[MemoryRecord]: + async def get_batch( + self, + collection_name: str, + keys: list[str], + with_embeddings: bool = False + ) -> list[MemoryRecord]: """Gets a batch of records. Args: