From 00b0e34312c1041bcc8ff34013281607440caf46 Mon Sep 17 00:00:00 2001 From: florian Date: Wed, 19 Jul 2023 18:11:25 +0200 Subject: [PATCH] fix(client): add plugin client typing to the opensearch client Signed-off-by: florian --- opensearchpy/_async/client/__init__.pyi | 8 ++++++++ opensearchpy/client/__init__.pyi | 8 ++++++++ 2 files changed, 16 insertions(+) diff --git a/opensearchpy/_async/client/__init__.pyi b/opensearchpy/_async/client/__init__.pyi index f81c978b..58040d88 100644 --- a/opensearchpy/_async/client/__init__.pyi +++ b/opensearchpy/_async/client/__init__.pyi @@ -30,6 +30,9 @@ from __future__ import unicode_literals import logging from typing import Any, Collection, MutableMapping, Optional, Tuple, Type, Union +from ..plugins.alerting import AlertingClient +from ..plugins.index_management import IndexManagementClient +from ..plugins.security import SecurityClient from ..transport import AsyncTransport from .cat import CatClient from .cluster import ClusterClient @@ -56,6 +59,11 @@ class AsyncOpenSearch(object): remote: RemoteClient snapshot: SnapshotClient tasks: TasksClient + + alerting: AlertingClient + security: SecurityClient + index_management: IndexManagementClient + def __init__( self, hosts: Any = ..., diff --git a/opensearchpy/client/__init__.pyi b/opensearchpy/client/__init__.pyi index 3357ca1c..9d16b01a 100644 --- a/opensearchpy/client/__init__.pyi +++ b/opensearchpy/client/__init__.pyi @@ -30,6 +30,9 @@ from __future__ import unicode_literals import logging from typing import Any, Collection, MutableMapping, Optional, Tuple, Type, Union +from ..plugins.alerting import AlertingClient +from ..plugins.index_management import IndexManagementClient +from ..plugins.security import SecurityClient from ..transport import Transport from .cat import CatClient from .cluster import ClusterClient @@ -56,6 +59,11 @@ class OpenSearch(object): remote: RemoteClient snapshot: SnapshotClient tasks: TasksClient + + alerting: AlertingClient + security: SecurityClient + index_management: IndexManagementClient + def __init__( self, hosts: Any = ..., transport_class: Type[Transport] = ..., **kwargs: Any ) -> None: ...