opensearch-mcp-server 2.0.1__tar.gz → 2.0.3__tar.gz

{opensearch_mcp_server-2.0.1 → opensearch_mcp_server-2.0.3}/.gitignore

@@ -10,3 +10,7 @@ __pycache__
 
 # Configuration and Credentials
 .env
+
+# Spring AI
+.gradle/
+build

{opensearch_mcp_server-2.0.1 → opensearch_mcp_server-2.0.3}/Makefile

@@ -11,6 +11,8 @@ release:
 		data["project"]["version"] = "$(version:v%=%)"; \
 		tomli_w.dump(data, open("pyproject.toml", "wb"))'
 	@git add pyproject.toml
+	@uv sync
+	@git add uv.lock
 	@git commit -m "release: update version to $(version:v%=%)"
 	@git push origin main
 	@git tag $(version)

{opensearch_mcp_server-2.0.1 → opensearch_mcp_server-2.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensearch-mcp-server
-Version: 2.0.1
+Version: 2.0.3
 Summary: MCP Server for interacting with Elasticsearch and OpenSearch
 License:                                  Apache License
                                    Version 2.0, January 2004
@@ -205,14 +205,15 @@ License:                                  Apache License
            limitations under the License.
 License-File: LICENSE
 Requires-Python: >=3.10
-Requires-Dist: anthropic>=0.49.0
-Requires-Dist: elasticsearch>=8.0.0
-Requires-Dist: fastmcp>=0.4.0
-Requires-Dist: mcp>=1.0.0
-Requires-Dist: opensearch-py>=2.0.0
-Requires-Dist: python-dotenv>=1.0.0
-Requires-Dist: tomli-w>=1.2.0
-Requires-Dist: tomli>=2.2.1
+Requires-Dist: anthropic==0.49.0
+Requires-Dist: elasticsearch==8.17.2
+Requires-Dist: fastmcp==0.4.1
+Requires-Dist: mcp==1.6.0
+Requires-Dist: opensearch-py==2.8.0
+Requires-Dist: python-dotenv==1.1.0
+Requires-Dist: toml==0.10.2
+Requires-Dist: tomli-w==1.2.0
+Requires-Dist: tomli==2.2.1
 Description-Content-Type: text/markdown
 
 # Elasticsearch/OpenSearch MCP Server
@@ -231,6 +232,10 @@ https://github.com/user-attachments/assets/f7409e31-fac4-4321-9c94-b0ff2ea7ff15
 
 ## Features
 
+### General Operations
+
+- `general_api_request`: Perform a general HTTP API request. Use this tool for any Elasticsearch/OpenSearch API that does not have a dedicated tool.
+
 ### Index Operations
 
 - `list_indices`: List all indices.

{opensearch_mcp_server-2.0.1 → opensearch_mcp_server-2.0.3}/README.md

@@ -14,6 +14,10 @@ https://github.com/user-attachments/assets/f7409e31-fac4-4321-9c94-b0ff2ea7ff15
 
 ## Features
 
+### General Operations
+
+- `general_api_request`: Perform a general HTTP API request. Use this tool for any Elasticsearch/OpenSearch API that does not have a dedicated tool.
+
 ### Index Operations
 
 - `list_indices`: List all indices.

{opensearch_mcp_server-2.0.1 → opensearch_mcp_server-2.0.3}/pyproject.toml

@@ -1,18 +1,19 @@
 [project]
 name = "opensearch-mcp-server"
-version = "2.0.1"
+version = "2.0.3"
 description = "MCP Server for interacting with Elasticsearch and OpenSearch"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "elasticsearch>=8.0.0",
-    "opensearch-py>=2.0.0",
-    "mcp>=1.0.0",
-    "python-dotenv>=1.0.0",
-    "fastmcp>=0.4.0",
-    "anthropic>=0.49.0",
-    "tomli>=2.2.1",
-    "tomli-w>=1.2.0",
+    "elasticsearch==8.17.2",
+    "opensearch-py==2.8.0",
+    "mcp==1.6.0",
+    "python-dotenv==1.1.0",
+    "fastmcp==0.4.1",
+    "anthropic==0.49.0",
+    "tomli==2.2.1",
+    "tomli-w==1.2.0",
+    "toml==0.10.2",
 ]
 
 [project.license]

{opensearch_mcp_server-2.0.1 → opensearch_mcp_server-2.0.3}/src/clients/base.py

@@ -3,6 +3,10 @@ import logging
 import warnings
 from typing import Dict
 
+from elasticsearch import Elasticsearch
+import httpx
+from opensearchpy import OpenSearch
+
 class SearchClientBase(ABC):
     def __init__(self, config: Dict, engine_type: str):
         """
@@ -35,7 +39,6 @@ class SearchClientBase(ABC):
         
         # Initialize client based on engine type
         if engine_type == "elasticsearch":
-            from elasticsearch import Elasticsearch
             self.client = Elasticsearch(
                 hosts=hosts,
                 basic_auth=(username, password) if username and password else None,
@@ -43,7 +46,6 @@ class SearchClientBase(ABC):
             )
             self.logger.info(f"Elasticsearch client initialized with hosts: {hosts}")
         elif engine_type == "opensearch":
-            from opensearchpy import OpenSearch
             self.client = OpenSearch(
                 hosts=hosts,
                 http_auth=(username, password) if username and password else None,
@@ -52,3 +54,34 @@ class SearchClientBase(ABC):
             self.logger.info(f"OpenSearch client initialized with hosts: {hosts}")
         else:
             raise ValueError(f"Unsupported engine type: {engine_type}")
+
+        # General REST client
+        base_url = hosts[0] if isinstance(hosts, list) else hosts
+        self.general_client = GeneralRestClient(
+            base_url=base_url,
+            username=username,
+            password=password,
+            verify_certs=verify_certs,
+        )
+
+class GeneralRestClient:
+    def __init__(self, base_url: str, username: str, password: str, verify_certs: bool):
+        self.base_url = base_url.rstrip("/")
+        self.auth = (username, password) if username and password else None
+        self.verify_certs = verify_certs
+
+    def request(self, method, path, params=None, body=None):
+        url = f"{self.base_url}/{path.lstrip('/')}"
+        with httpx.Client(verify=self.verify_certs) as client:
+            resp = client.request(
+                method=method.upper(),
+                url=url,
+                params=params,
+                json=body,
+                auth=self.auth
+            )
+            resp.raise_for_status()
+            ct = resp.headers.get("content-type", "")
+            if ct.startswith("application/json"):
+                return resp.json()
+            return resp.text

{opensearch_mcp_server-2.0.1 → opensearch_mcp_server-2.0.3}/src/clients/common/client.py

@@ -3,9 +3,10 @@ from typing import Dict
 from src.clients.common.alias import AliasClient
 from src.clients.common.cluster import ClusterClient
 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):
+class SearchClient(IndexClient, DocumentClient, ClusterClient, AliasClient, GeneralClient):
     """
     Unified search client that combines all search functionality.
     

opensearch_mcp_server-2.0.3/src/clients/common/general.py

@@ -0,0 +1,10 @@
+from typing import Dict, Optional
+
+from src.clients.base import SearchClientBase
+
+class GeneralClient(SearchClientBase):
+    def general_api_request(self, method: str, path: str, params: Optional[Dict] = None, body: Optional[Dict] = None):
+        """Perform a general HTTP API request.
+           Use this tool for any Elasticsearch/OpenSearch API that does not have a dedicated tool.
+        """
+        return self.general_client.request(method, path, params, body)

{opensearch_mcp_server-2.0.1 → opensearch_mcp_server-2.0.3}/src/server.py

@@ -2,11 +2,13 @@ import logging
 import sys
 
 from fastmcp import FastMCP
+import toml
 
 from src.clients import create_search_client
 from src.tools.alias import AliasTools
 from src.tools.cluster import ClusterTools
 from src.tools.document import DocumentTools
+from src.tools.general import GeneralTools
 from src.tools.index import IndexTools
 from src.tools.register import ToolsRegister
 
@@ -17,7 +19,8 @@ class SearchMCPServer:
         self.name = f"{self.engine_type}_mcp_server"
         self.mcp = FastMCP(self.name)
         self.logger = logging.getLogger()
-        self.logger.info(f"Initializing {self.name}...")
+        self.version = toml.load("pyproject.toml")["project"]["version"]
+        self.logger.info(f"Initializing {self.name}, Version: {self.version}")
         
         # Create the corresponding search client
         self.search_client = create_search_client(self.engine_type)
@@ -35,7 +38,8 @@ class SearchMCPServer:
             IndexTools,
             DocumentTools,
             ClusterTools,
-            AliasTools
+            AliasTools,
+            GeneralTools,
         ]        
         # Register all tools
         register.register_all_tools(tool_classes)

{opensearch_mcp_server-2.0.1 → opensearch_mcp_server-2.0.3}/src/tools/__init__.py

@@ -1,13 +1,15 @@
-from src.tools.index import IndexTools
-from src.tools.document import DocumentTools
-from src.tools.cluster import ClusterTools
 from src.tools.alias import AliasTools
+from src.tools.cluster import ClusterTools
+from src.tools.document import DocumentTools
+from src.tools.general import GeneralTools
+from src.tools.index import IndexTools
 from src.tools.register import ToolsRegister
 
 __all__ = [
-    'IndexTools',
-    'DocumentTools',
-    'ClusterTools',
     'AliasTools',
+    'ClusterTools',
+    'DocumentTools',
+    'GeneralTools',
+    'IndexTools',
     'ToolsRegister',
 ]

opensearch_mcp_server-2.0.3/src/tools/general.py

@@ -0,0 +1,20 @@
+from typing import Dict, Optional
+
+from fastmcp import FastMCP
+
+class GeneralTools:
+    def __init__(self, search_client):
+        self.search_client = search_client
+    def register_tools(self, mcp: FastMCP):
+        @mcp.tool()
+        def general_api_request(method: str, path: str, params: Optional[Dict] = None, body: Optional[Dict] = None):
+            """Perform a general HTTP API request.
+               Use this tool for any Elasticsearch/OpenSearch API that does not have a dedicated tool.
+            
+            Args:
+                method: HTTP method (GET, POST, PUT, DELETE, etc.)
+                path: API endpoint path
+                params: Query parameters
+                body: Request body
+            """
+            return self.search_client.general_api_request(method, path, params, body)