Skip to content

Commit 10ad064

Browse files
committed
Fix type check for python 3.9
1 parent 2dd50b6 commit 10ad064

1 file changed

Lines changed: 11 additions & 2 deletions

File tree

weaviate/connect/helpers.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,14 @@
66
from deprecation import deprecated as docstring_deprecated
77
from typing_extensions import deprecated as typing_deprecated
88

9-
from weaviate.auth import Auth, AuthCredentials
9+
from weaviate.auth import (
10+
Auth,
11+
AuthCredentials,
12+
_APIKey,
13+
_BearerToken,
14+
_ClientCredentials,
15+
_ClientPassword,
16+
)
1017
from weaviate.client import WeaviateAsyncClient, WeaviateClient
1118
from weaviate.config import AdditionalConfig
1219
from weaviate.connect.base import ConnectionParams, ProtocolParams
@@ -31,7 +38,9 @@ def __parse_auth_credentials(creds: Union[str, AuthCredentials, None]) -> Option
3138
if isinstance(creds, str):
3239
# If the credentials are a string, assume it's an API key.
3340
return Auth.api_key(creds)
34-
elif isinstance(creds, AuthCredentials):
41+
elif isinstance(
42+
creds, (_BearerToken, _ClientPassword, _ClientCredentials, _APIKey)
43+
): # use AuthCredentials after python 3.9 has been removed
3544
# If the credentials are already an AuthCredentials object, return it as is.
3645
return creds
3746
elif creds is None:

0 commit comments

Comments
 (0)