Fix mypy problems in new waeviate client (#40330)

The new waeviate client (4.6.5) has more typing information and it causes Waeviate Hook to fail mypy check. This PR fixes it.
apache · Jun 19, 2024 · d048122 · d048122
1 parent 9d0f768
commit d048122
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/airflow/providers/weaviate/hooks/weaviate.py b/airflow/providers/weaviate/hooks/weaviate.py
@@ -120,7 +120,9 @@ def get_conn(self) -> WeaviateClient:
         additional_headers = extras.pop("additional_headers", {})
         scope = extras.get("scope", None) or extras.get("oidc_scope", None)
         if api_key:
-            auth_client_secret = AuthApiKey(api_key)
+            auth_client_secret: AuthApiKey | AuthBearerToken | AuthClientCredentials | AuthClientPassword = (
+                AuthApiKey(api_key)
+            )
         elif access_token:
             auth_client_secret = AuthBearerToken(
                 access_token, expires_in=expires_in, refresh_token=refresh_token