diff --git a/astropylibrarian/algolia/client.py b/astropylibrarian/algolia/client.py index 7a1f48a..1cc448c 100644 --- a/astropylibrarian/algolia/client.py +++ b/astropylibrarian/algolia/client.py @@ -67,10 +67,10 @@ class AlgoliaIndex(BaseAlgoliaIndex): Name of the Algolia index. """ - async def __aenter__(self) -> SearchClient: + async def __aenter__(self) -> "AlgoliaIndex": self._logger.debug("Opening algolia client") self.algolia_client = SearchClient(self.app_id, self._key) - return self.algolia_client + return self async def __aexit__( self,