typecheck: fix asynccontextmanager return annotations

Signed-off-by: phernandez <[email protected]>

Author: phernandez

src/basic_memory/cli/auth.py

@@ -8,7 +8,8 @@
 import time
 import webbrowser
 from contextlib import asynccontextmanager
-from typing import AsyncContextManager, Callable
+from collections.abc import AsyncIterator, Callable
+from typing import AsyncContextManager
 
 import httpx
 from rich.console import Console
@@ -38,7 +39,7 @@ def __init__(
         self._http_client_factory = http_client_factory
 
     @asynccontextmanager
-    async def _get_http_client(self) -> AsyncContextManager[httpx.AsyncClient]:
+    async def _get_http_client(self) -> AsyncIterator[httpx.AsyncClient]:
         """Create an AsyncClient, optionally via injected factory.
 
         Why: enables reliable tests without monkeypatching httpx internals while

src/basic_memory/cli/commands/cloud/api_client.py

@@ -1,5 +1,6 @@
 """Cloud API client utilities."""
 
+from collections.abc import AsyncIterator
 from typing import Optional
 from contextlib import asynccontextmanager
 from typing import AsyncContextManager, Callable
@@ -58,7 +59,7 @@ async def get_authenticated_headers(auth: CLIAuth | None = None) -> dict[str, st
 
 
 @asynccontextmanager
-async def _default_http_client(timeout: float) -> AsyncContextManager[httpx.AsyncClient]:
+async def _default_http_client(timeout: float) -> AsyncIterator[httpx.AsyncClient]:
     async with httpx.AsyncClient(timeout=timeout) as client:
         yield client