elasticsearch-mcp-server 2.0.14__py3-none-any.whl → 2.0.16__py3-none-any.whl

{elasticsearch_mcp_server-2.0.14.dist-info → elasticsearch_mcp_server-2.0.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: elasticsearch-mcp-server
-Version: 2.0.14
+Version: 2.0.16
 Summary: MCP Server for interacting with Elasticsearch and OpenSearch
 License:                                  Apache License
                                    Version 2.0, January 2004
@@ -210,6 +210,7 @@ Requires-Dist: elasticsearch==8.17.2
 Requires-Dist: fastmcp==2.8.0
 Requires-Dist: mcp==1.9.2
 Requires-Dist: opensearch-py==2.8.0
+Requires-Dist: pydantic<2.12.0,>=2.11.0
 Requires-Dist: python-dotenv==1.1.0
 Requires-Dist: tomli-w==1.2.0
 Requires-Dist: tomli==2.2.1
@@ -248,6 +249,9 @@ https://github.com/user-attachments/assets/f7409e31-fac4-4321-9c94-b0ff2ea7ff15
 - `get_index`: Returns information (mappings, settings, aliases) about one or more indices.
 - `create_index`: Create a new index.
 - `delete_index`: Delete an index.
+- `create_data_stream`: Create a new data stream (requires matching index template).
+- `get_data_stream`: Get information about one or more data streams.
+- `delete_data_stream`: Delete one or more data streams and their backing indices.
 
 ### Document Operations
 
@@ -372,7 +376,7 @@ Using `uv` requires cloning the repository locally and specifying the path to th
       "command": "uv",
       "args": [
         "--directory",
-        "path/to/src/elasticsearch_mcp_server",
+        "path/to/elasticsearch-mcp-server",
         "run",
         "elasticsearch-mcp-server"
       ],
@@ -392,7 +396,7 @@ Using `uv` requires cloning the repository locally and specifying the path to th
       "command": "uv",
       "args": [
         "--directory",
-        "path/to/src/elasticsearch_mcp_server",
+        "path/to/elasticsearch-mcp-server",
         "run",
         "elasticsearch-mcp-server"
       ],
@@ -411,7 +415,7 @@ Using `uv` requires cloning the repository locally and specifying the path to th
       "command": "uv",
       "args": [
         "--directory",
-        "path/to/src/elasticsearch_mcp_server",
+        "path/to/elasticsearch-mcp-server",
         "run",
         "opensearch-mcp-server"
       ],
@@ -489,11 +493,7 @@ uv run src/server.py elasticsearch-mcp-server --transport streamable-http --host
 
 ## Compatibility
 
-The MCP server is compatible with Elasticsearch 7.x, 8.x, and 9.x. By default, it uses the Elasticsearch 8.x client (without a suffix). To use the Elasticsearch 7.x client, run the `elasticsearch-mcp-server-es7` variant. For Elasticsearch 9.x, use `elasticsearch-mcp-server-es9`. For example:
-
-```bash
-uvx elasticsearch-mcp-server-es7
-```
+The MCP server is compatible with Elasticsearch 7.x, 8.x, and 9.x. By default, it uses the Elasticsearch 8.x client (without a suffix).
 
 | MCP Server | Elasticsearch |
 | --- | --- |
@@ -502,6 +502,18 @@ uvx elasticsearch-mcp-server-es7
 | elasticsearch-mcp-server-es9 | Elasticsearch 9.x |
 | opensearch-mcp-server | OpenSearch 1.x, 2.x, 3.x |
 
+ To use the Elasticsearch 7.x client, run the `elasticsearch-mcp-server-es7` variant. For Elasticsearch 9.x, use `elasticsearch-mcp-server-es9`. For example:
+
+```bash
+uvx elasticsearch-mcp-server-es7
+```
+
+If you want to run different Elasticsearch variants (e.g., 7.x or 9.x) locally, simply update the `elasticsearch` dependency version in `pyproject.toml`, then start the server with:
+
+```bash
+uv run src/server.py elasticsearch-mcp-server
+```
+
 ## License
 
 This project is licensed under the Apache License Version 2.0 - see the [LICENSE](LICENSE) file for details.

{elasticsearch_mcp_server-2.0.14.dist-info → elasticsearch_mcp_server-2.0.16.dist-info}/RECORD

@@ -1,25 +1,27 @@
 src/__init__.py,sha256=aNKcThftSLh9IjOTA-UUpoRzIm4R0WwXKGAzykwecmU,211
-src/server.py,sha256=BfMAgXTFV9C4waCWRro8Sfl6yXEXr2vJ7YoGY7HhMPY,5072
-src/version.py,sha256=GvRgvOJ3H4RUuTwx8ESkywk1a8KDv2u3vzO_y4-TIlQ,23
+src/server.py,sha256=VDp65H4uiscYlyzrCiRornKknVvj5EmHqAE3YkMWCcI,5342
+src/version.py,sha256=6KKZqs-riZy7B_ZI5UKDLJyHCTX9LRlwPfBN0xcfq_U,23
 src/clients/__init__.py,sha256=MUKeCxvAFzzVfQr4x1seB7cRVLtWAk7pb1-5tLG0T8g,1299
-src/clients/base.py,sha256=tMNdKRsAG1dPHj7famwPvCXCONuT9jML-7FmWV9vXbs,5348
+src/clients/base.py,sha256=8HQ7YCOQXly_abbUpcMGOlz_-ShmJ4nF3yfa5SqT3cU,5535
 src/clients/exceptions.py,sha256=NYF3KVw-9aAgCViin5OuBI6miMEPS5QsdHx6bcis1wc,2493
 src/clients/common/__init__.py,sha256=VgvgxFpESn2wAuJmH0XM_Ej2izI7dxK7QJe9wG4fmW0,211
 src/clients/common/alias.py,sha256=rB53TSld5x2vZyDNAwyEdnh1KWUXMSD7h5fSv_ubR2Q,759
-src/clients/common/client.py,sha256=khRO3osowTvoXnPscefUSuBYMrhOao2urOnRwqiUMbI,1041
+src/clients/common/client.py,sha256=xXsCab5H3360lF4SXSVXiZY-F4XTAMljGIFl-PyNdG0,1119
 src/clients/common/cluster.py,sha256=pd5BVpqqDU6Lck3K704eEdhgFgzt9NstotWQLyG9zFM,401
+src/clients/common/data_stream.py,sha256=2zFCekGxF9f6kxFSRmb8qP8nzONf7IrLzb4vO0bHjfs,741
 src/clients/common/document.py,sha256=ZzZiXDf_UhlN2FCmqW3drVjIZ07kGYP9yth_sgsJGPc,1623
 src/clients/common/general.py,sha256=_1eS11Ha-JC1x1YoSu63zUWXKvYkwvN8RQOUM1fx39M,469
 src/clients/common/index.py,sha256=vyH5iXlJe5JLcDK6fhCCEPN-tgGm6zP5ilwGPdWCXbY,776
 src/tools/__init__.py,sha256=PGsXXpPubVnUbzXJG4keDhgjr0j6EnMNDs0aLqNIb4A,387
 src/tools/alias.py,sha256=p9TD4gXkGRGWHTYfvCg7G2hj-Uch9jwDXJNUY1hSD0Y,1376
 src/tools/cluster.py,sha256=XRAG-uxdfrieYX1ov_cBb66IYXaa8OoSWCUfXvNauy0,587
+src/tools/data_stream.py,sha256=l9VItmvIkkffF2g3kLaRU-4VytzWteorB7vnpvovcdE,1825
 src/tools/document.py,sha256=XZTVuk4di9VEwWMZN7jyDVnzoOiXkb4FBrXF44sVXTs,2052
 src/tools/general.py,sha256=whj1spjIb8SS75h843X6c3RTsrZcSm-66KVLlY7OEh0,817
 src/tools/index.py,sha256=7KNPtElTFelkjRSvdMqPBx9nx_9Uk01OnTMeVo7YeCs,1345
 src/tools/register.py,sha256=wrG2P6-YPW77bTg1j_ELp8omWRYsJjFeOHUy_unHe6Y,1344
-elasticsearch_mcp_server-2.0.14.dist-info/METADATA,sha256=YtZXxlYiuJCcrbywct3J07iXbLQoXCekvRVBjahK-Fc,22745
-elasticsearch_mcp_server-2.0.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-elasticsearch_mcp_server-2.0.14.dist-info/entry_points.txt,sha256=ImfJnUwMpQUBmu-1MeBG_P0dwamfXKQh82mBKW7tWNY,138
-elasticsearch_mcp_server-2.0.14.dist-info/licenses/LICENSE,sha256=DBsjuP5FR51d9kaUdXlVBuBv3cQ_I_adq9gefYQ9FcY,11339
-elasticsearch_mcp_server-2.0.14.dist-info/RECORD,,
+elasticsearch_mcp_server-2.0.16.dist-info/METADATA,sha256=ov7ljqcgU2VQ7MyVbcwCnUIdBvFWvihiCroybqQ-naU,23254
+elasticsearch_mcp_server-2.0.16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+elasticsearch_mcp_server-2.0.16.dist-info/entry_points.txt,sha256=ImfJnUwMpQUBmu-1MeBG_P0dwamfXKQh82mBKW7tWNY,138
+elasticsearch_mcp_server-2.0.16.dist-info/licenses/LICENSE,sha256=DBsjuP5FR51d9kaUdXlVBuBv3cQ_I_adq9gefYQ9FcY,11339
+elasticsearch_mcp_server-2.0.16.dist-info/RECORD,,

src/clients/base.py

@@ -91,6 +91,9 @@ class SearchClientBase(ABC):
         # Check Elasticsearch package version to determine auth parameter name
         try:
             from elasticsearch import __version__ as es_version
+            # Convert version tuple to string format
+            version_str = '.'.join(map(str, es_version))
+            self.logger.info(f"Elasticsearch client version: {version_str}")
             major_version = es_version[0]    
             if major_version >= 8:
                 # ES 8+ uses basic_auth

src/clients/common/client.py

@@ -2,11 +2,12 @@ from typing import Dict
 
 from src.clients.common.alias import AliasClient
 from src.clients.common.cluster import ClusterClient
+from src.clients.common.data_stream import DataStreamClient
 from src.clients.common.document import DocumentClient
 from src.clients.common.general import GeneralClient
 from src.clients.common.index import IndexClient
 
-class SearchClient(IndexClient, DocumentClient, ClusterClient, AliasClient, GeneralClient):
+class SearchClient(IndexClient, DocumentClient, ClusterClient, AliasClient, DataStreamClient, GeneralClient):
     """
     Unified search client that combines all search functionality.
     

src/clients/common/data_stream.py

@@ -0,0 +1,18 @@
+from typing import Dict, Optional
+from src.clients.base import SearchClientBase
+
+class DataStreamClient(SearchClientBase):  
+    def create_data_stream(self, name: str) -> Dict:
+        """Create a new data stream."""
+        return self.client.indices.create_data_stream(name=name)
+    
+    def get_data_stream(self, name: Optional[str] = None) -> Dict:
+        """Get information about one or more data streams."""
+        if name:
+            return self.client.indices.get_data_stream(name=name)
+        else:
+            return self.client.indices.get_data_stream()
+    
+    def delete_data_stream(self, name: str) -> Dict:
+        """Delete one or more data streams."""
+        return self.client.indices.delete_data_stream(name=name)

src/server.py

@@ -7,6 +7,7 @@ from fastmcp import FastMCP
 from src.clients import create_search_client
 from src.tools.alias import AliasTools
 from src.tools.cluster import ClusterTools
+from src.tools.data_stream import DataStreamTools
 from src.tools.document import DocumentTools
 from src.tools.general import GeneralTools
 from src.tools.index import IndexTools
@@ -19,7 +20,13 @@ class SearchMCPServer:
         self.engine_type = engine_type
         self.name = f"{self.engine_type}-mcp-server"
         self.mcp = FastMCP(self.name)
-        self.logger = logging.getLogger()
+        
+        # Configure logging
+        logging.basicConfig(
+            level=logging.INFO,
+            format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+        )
+        self.logger = logging.getLogger(__name__)
         self.logger.info(f"Initializing {self.name}, Version: {VERSION}")
         
         # Create the corresponding search client
@@ -39,6 +46,7 @@ class SearchMCPServer:
             DocumentTools,
             ClusterTools,
             AliasTools,
+            DataStreamTools,
             GeneralTools,
         ]        
         # Register all tools

src/tools/data_stream.py

@@ -0,0 +1,47 @@
+from typing import Dict, Optional
+from fastmcp import FastMCP
+
+class DataStreamTools:    
+    def __init__(self, search_client):
+        self.search_client = search_client
+    
+    def register_tools(self, mcp: FastMCP):
+        """Register data stream tools with the MCP server."""
+        
+        @mcp.tool()
+        def create_data_stream(name: str) -> Dict:
+            """Create a new data stream.
+            
+            This creates a new data stream with the specified name.
+            The data stream must have a matching index template before creation.
+            
+            Args:
+                name: Name of the data stream to create
+            """
+            return self.search_client.create_data_stream(name=name)
+        
+        @mcp.tool()
+        def get_data_stream(name: Optional[str] = None) -> Dict:
+            """Get information about one or more data streams.
+            
+            Retrieves configuration, mappings, settings, and other information
+            about the specified data streams.
+            
+            Args:
+                name: Name of the data stream(s) to retrieve.
+                      Can be a comma-separated list or wildcard pattern.
+                      If not provided, retrieves all data streams.
+            """
+            return self.search_client.get_data_stream(name=name)
+        
+        @mcp.tool()
+        def delete_data_stream(name: str) -> Dict:
+            """Delete one or more data streams.
+            
+            Permanently deletes the specified data streams and all their backing indices.
+            
+            Args:
+                name: Name of the data stream(s) to delete.
+                      Can be a comma-separated list or wildcard pattern.
+            """
+            return self.search_client.delete_data_stream(name=name)

src/version.py

@@ -1 +1 @@
-__version__ = "2.0.14"
+__version__ = "2.0.16"