Fixing builds

Author: Aayush Kataria

agent_memory_toolkit/chat.py

@@ -179,9 +179,7 @@ def _ensure_async_client(self) -> Any:
 
                 token_provider = get_bearer_token_provider(self._credential, _TOKEN_SCOPE)
             else:
-                token_provider = _make_sync_token_provider_for_async(
-                    self._credential, _TOKEN_SCOPE
-                )
+                token_provider = _make_sync_token_provider_for_async(self._credential, _TOKEN_SCOPE)
 
             self._async_client = AsyncAzureOpenAI(
                 api_version=self._api_version,

tests/unit/test_chat.py

@@ -302,6 +302,7 @@ def __init__(self, **kwargs):
             captured.update(kwargs)
 
     import sys
+
     fake_openai = MagicMock()
     fake_openai.AsyncAzureOpenAI = FakeAsyncAzureOpenAI
     monkeypatch.setitem(sys.modules, "openai", fake_openai)