async DefaultAzureCredential incompatible with AsyncTokenCredential per pyright

[kristapratico]

Pyright reports a type error when the async DefaultAzureCredential is used anywhere where an AsyncTokenCredential is expected:

  C:\Users\krpratic\azure-sdk-for-python\sdk\cognitivelanguage\azure-ai-language-conversations\samples\async\sample_authentication_async.py:57:66 - error: Argument of type "DefaultAzureCredential" cannot be assigned to parameter "credential" of type "AzureKeyCredential | AsyncTokenCredential" in function "__init__"
    Type "DefaultAzureCredential" cannot be assigned to type "AzureKeyCredential | AsyncTokenCredential"
      "DefaultAzureCredential" is incompatible with "AzureKeyCredential"
      "DefaultAzureCredential" is incompatible with protocol "AsyncTokenCredential"
        "__aenter__" is an incompatible type
          Type "() -> Coroutine[Any, Any, DefaultAzureCredential]" cannot be assigned to type "() -> Coroutine[Any, Any, None]"
            Function return type "Coroutine[Any, Any, DefaultAzureCredential]" is incompatible with type "Coroutine[Any, Any, None]"
        "__aexit__" is an incompatible type
          Type "(*args: Unknown) -> Coroutine[Any, Any, None]" cannot be assigned to type "(exc_type: Unknown, exc_value: Unknown, traceback: Unknown) -> Coroutine[Any, Any, None]"

This is due to the difference in how the Protocol is typed vs. how it is typed in identity:

azure-sdk-for-python/sdk/core/azure-core/azure/core/credentials_async.py

Lines 28 to 35 in ac0672f

	async def close(self) -> None:
	pass
	async def __aenter__(self):
	pass
	async def __aexit__(self, exc_type, exc_value, traceback) -> None:
	pass

azure-sdk-for-python/sdk/identity/azure-identity/azure/identity/aio/_internal/__init__.py

Lines 11 to 20 in ac0672f

	class AsyncContextManager(abc.ABC):
	@abc.abstractmethod
	async def close(self):
	pass
	async def __aenter__(self):
	return self
	async def __aexit__(self, *args):
	await self.close()

Note that this doesn't occur with the sync credential since __enter__ and __exit__ are not described by TokenCredential.

[kristapratico]

I think it would be less disruptive to update AsyncTokenCredential (although I'm not sure if this would be considered breaking):

 async def close(self) -> None: 
     pass 
  
 async def __aenter__(self) -> typing.Self: 
     pass 
  
 async def __aexit__(self, *args: Any) -> None: 
     pass 

[xiangyan99]

@mccoyp is this something you are working on? (I want to avoid dup the work)

[laiapat]

@xiangyan99 Looks like pyright is more strict than mypy, and in a similar boat as PyCharm's type checker. #25788 should resolve this.

When I last spoke with Laurent, he feels that we should update our type signatures to correctly implement the TokenCredential protocols (even when some of the kwargs can't be used, like tenant_id with ManagedIdentityCredential). We would just need to clearly document when these kwargs aren't being used.