fix(client): add plugin client typing to the opensearch client

Signed-off-by: florian <[email protected]>
opensearch-project · Jul 19, 2023 · 00b0e34 · 00b0e34
1 parent 4dba35d
commit 00b0e34
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git 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
@@ -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: ...