opensearch-mcp-server 2.0.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- opensearch_mcp_server-2.0.0.dist-info/METADATA +395 -0
- opensearch_mcp_server-2.0.0.dist-info/RECORD +22 -0
- opensearch_mcp_server-2.0.0.dist-info/WHEEL +4 -0
- opensearch_mcp_server-2.0.0.dist-info/entry_points.txt +3 -0
- opensearch_mcp_server-2.0.0.dist-info/licenses/LICENSE +201 -0
- src/__init__.py +6 -0
- src/clients/__init__.py +42 -0
- src/clients/base.py +54 -0
- src/clients/common/__init__.py +6 -0
- src/clients/common/alias.py +20 -0
- src/clients/common/client.py +25 -0
- src/clients/common/cluster.py +12 -0
- src/clients/common/document.py +37 -0
- src/clients/common/index.py +20 -0
- src/clients/exceptions.py +69 -0
- src/server.py +71 -0
- src/tools/__init__.py +13 -0
- src/tools/alias.py +46 -0
- src/tools/cluster.py +17 -0
- src/tools/document.py +64 -0
- src/tools/index.py +44 -0
- src/tools/register.py +41 -0
@@ -0,0 +1,395 @@
|
|
1
|
+
Metadata-Version: 2.4
|
2
|
+
Name: opensearch-mcp-server
|
3
|
+
Version: 2.0.0
|
4
|
+
Summary: MCP Server for interacting with Elasticsearch and OpenSearch
|
5
|
+
License: Apache License
|
6
|
+
Version 2.0, January 2004
|
7
|
+
http://www.apache.org/licenses/
|
8
|
+
|
9
|
+
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
10
|
+
|
11
|
+
1. Definitions.
|
12
|
+
|
13
|
+
"License" shall mean the terms and conditions for use, reproduction,
|
14
|
+
and distribution as defined by Sections 1 through 9 of this document.
|
15
|
+
|
16
|
+
"Licensor" shall mean the copyright owner or entity authorized by
|
17
|
+
the copyright owner that is granting the License.
|
18
|
+
|
19
|
+
"Legal Entity" shall mean the union of the acting entity and all
|
20
|
+
other entities that control, are controlled by, or are under common
|
21
|
+
control with that entity. For the purposes of this definition,
|
22
|
+
"control" means (i) the power, direct or indirect, to cause the
|
23
|
+
direction or management of such entity, whether by contract or
|
24
|
+
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
25
|
+
outstanding shares, or (iii) beneficial ownership of such entity.
|
26
|
+
|
27
|
+
"You" (or "Your") shall mean an individual or Legal Entity
|
28
|
+
exercising permissions granted by this License.
|
29
|
+
|
30
|
+
"Source" form shall mean the preferred form for making modifications,
|
31
|
+
including but not limited to software source code, documentation
|
32
|
+
source, and configuration files.
|
33
|
+
|
34
|
+
"Object" form shall mean any form resulting from mechanical
|
35
|
+
transformation or translation of a Source form, including but
|
36
|
+
not limited to compiled object code, generated documentation,
|
37
|
+
and conversions to other media types.
|
38
|
+
|
39
|
+
"Work" shall mean the work of authorship, whether in Source or
|
40
|
+
Object form, made available under the License, as indicated by a
|
41
|
+
copyright notice that is included in or attached to the work
|
42
|
+
(an example is provided in the Appendix below).
|
43
|
+
|
44
|
+
"Derivative Works" shall mean any work, whether in Source or Object
|
45
|
+
form, that is based on (or derived from) the Work and for which the
|
46
|
+
editorial revisions, annotations, elaborations, or other modifications
|
47
|
+
represent, as a whole, an original work of authorship. For the purposes
|
48
|
+
of this License, Derivative Works shall not include works that remain
|
49
|
+
separable from, or merely link (or bind by name) to the interfaces of,
|
50
|
+
the Work and Derivative Works thereof.
|
51
|
+
|
52
|
+
"Contribution" shall mean any work of authorship, including
|
53
|
+
the original version of the Work and any modifications or additions
|
54
|
+
to that Work or Derivative Works thereof, that is intentionally
|
55
|
+
submitted to Licensor for inclusion in the Work by the copyright owner
|
56
|
+
or by an individual or Legal Entity authorized to submit on behalf of
|
57
|
+
the copyright owner. For the purposes of this definition, "submitted"
|
58
|
+
means any form of electronic, verbal, or written communication sent
|
59
|
+
to the Licensor or its representatives, including but not limited to
|
60
|
+
communication on electronic mailing lists, source code control systems,
|
61
|
+
and issue tracking systems that are managed by, or on behalf of, the
|
62
|
+
Licensor for the purpose of discussing and improving the Work, but
|
63
|
+
excluding communication that is conspicuously marked or otherwise
|
64
|
+
designated in writing by the copyright owner as "Not a Contribution."
|
65
|
+
|
66
|
+
"Contributor" shall mean Licensor and any individual or Legal Entity
|
67
|
+
on behalf of whom a Contribution has been received by Licensor and
|
68
|
+
subsequently incorporated within the Work.
|
69
|
+
|
70
|
+
2. Grant of Copyright License. Subject to the terms and conditions of
|
71
|
+
this License, each Contributor hereby grants to You a perpetual,
|
72
|
+
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
73
|
+
copyright license to reproduce, prepare Derivative Works of,
|
74
|
+
publicly display, publicly perform, sublicense, and distribute the
|
75
|
+
Work and such Derivative Works in Source or Object form.
|
76
|
+
|
77
|
+
3. Grant of Patent License. Subject to the terms and conditions of
|
78
|
+
this License, each Contributor hereby grants to You a perpetual,
|
79
|
+
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
80
|
+
(except as stated in this section) patent license to make, have made,
|
81
|
+
use, offer to sell, sell, import, and otherwise transfer the Work,
|
82
|
+
where such license applies only to those patent claims licensable
|
83
|
+
by such Contributor that are necessarily infringed by their
|
84
|
+
Contribution(s) alone or by combination of their Contribution(s)
|
85
|
+
with the Work to which such Contribution(s) was submitted. If You
|
86
|
+
institute patent litigation against any entity (including a
|
87
|
+
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
88
|
+
or a Contribution incorporated within the Work constitutes direct
|
89
|
+
or contributory patent infringement, then any patent licenses
|
90
|
+
granted to You under this License for that Work shall terminate
|
91
|
+
as of the date such litigation is filed.
|
92
|
+
|
93
|
+
4. Redistribution. You may reproduce and distribute copies of the
|
94
|
+
Work or Derivative Works thereof in any medium, with or without
|
95
|
+
modifications, and in Source or Object form, provided that You
|
96
|
+
meet the following conditions:
|
97
|
+
|
98
|
+
(a) You must give any other recipients of the Work or
|
99
|
+
Derivative Works a copy of this License; and
|
100
|
+
|
101
|
+
(b) You must cause any modified files to carry prominent notices
|
102
|
+
stating that You changed the files; and
|
103
|
+
|
104
|
+
(c) You must retain, in the Source form of any Derivative Works
|
105
|
+
that You distribute, all copyright, patent, trademark, and
|
106
|
+
attribution notices from the Source form of the Work,
|
107
|
+
excluding those notices that do not pertain to any part of
|
108
|
+
the Derivative Works; and
|
109
|
+
|
110
|
+
(d) If the Work includes a "NOTICE" text file as part of its
|
111
|
+
distribution, then any Derivative Works that You distribute must
|
112
|
+
include a readable copy of the attribution notices contained
|
113
|
+
within such NOTICE file, excluding those notices that do not
|
114
|
+
pertain to any part of the Derivative Works, in at least one
|
115
|
+
of the following places: within a NOTICE text file distributed
|
116
|
+
as part of the Derivative Works; within the Source form or
|
117
|
+
documentation, if provided along with the Derivative Works; or,
|
118
|
+
within a display generated by the Derivative Works, if and
|
119
|
+
wherever such third-party notices normally appear. The contents
|
120
|
+
of the NOTICE file are for informational purposes only and
|
121
|
+
do not modify the License. You may add Your own attribution
|
122
|
+
notices within Derivative Works that You distribute, alongside
|
123
|
+
or as an addendum to the NOTICE text from the Work, provided
|
124
|
+
that such additional attribution notices cannot be construed
|
125
|
+
as modifying the License.
|
126
|
+
|
127
|
+
You may add Your own copyright statement to Your modifications and
|
128
|
+
may provide additional or different license terms and conditions
|
129
|
+
for use, reproduction, or distribution of Your modifications, or
|
130
|
+
for any such Derivative Works as a whole, provided Your use,
|
131
|
+
reproduction, and distribution of the Work otherwise complies with
|
132
|
+
the conditions stated in this License.
|
133
|
+
|
134
|
+
5. Submission of Contributions. Unless You explicitly state otherwise,
|
135
|
+
any Contribution intentionally submitted for inclusion in the Work
|
136
|
+
by You to the Licensor shall be under the terms and conditions of
|
137
|
+
this License, without any additional terms or conditions.
|
138
|
+
Notwithstanding the above, nothing herein shall supersede or modify
|
139
|
+
the terms of any separate license agreement you may have executed
|
140
|
+
with Licensor regarding such Contributions.
|
141
|
+
|
142
|
+
6. Trademarks. This License does not grant permission to use the trade
|
143
|
+
names, trademarks, service marks, or product names of the Licensor,
|
144
|
+
except as required for reasonable and customary use in describing the
|
145
|
+
origin of the Work and reproducing the content of the NOTICE file.
|
146
|
+
|
147
|
+
7. Disclaimer of Warranty. Unless required by applicable law or
|
148
|
+
agreed to in writing, Licensor provides the Work (and each
|
149
|
+
Contributor provides its Contributions) on an "AS IS" BASIS,
|
150
|
+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
151
|
+
implied, including, without limitation, any warranties or conditions
|
152
|
+
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
153
|
+
PARTICULAR PURPOSE. You are solely responsible for determining the
|
154
|
+
appropriateness of using or redistributing the Work and assume any
|
155
|
+
risks associated with Your exercise of permissions under this License.
|
156
|
+
|
157
|
+
8. Limitation of Liability. In no event and under no legal theory,
|
158
|
+
whether in tort (including negligence), contract, or otherwise,
|
159
|
+
unless required by applicable law (such as deliberate and grossly
|
160
|
+
negligent acts) or agreed to in writing, shall any Contributor be
|
161
|
+
liable to You for damages, including any direct, indirect, special,
|
162
|
+
incidental, or consequential damages of any character arising as a
|
163
|
+
result of this License or out of the use or inability to use the
|
164
|
+
Work (including but not limited to damages for loss of goodwill,
|
165
|
+
work stoppage, computer failure or malfunction, or any and all
|
166
|
+
other commercial damages or losses), even if such Contributor
|
167
|
+
has been advised of the possibility of such damages.
|
168
|
+
|
169
|
+
9. Accepting Warranty or Additional Liability. While redistributing
|
170
|
+
the Work or Derivative Works thereof, You may choose to offer,
|
171
|
+
and charge a fee for, acceptance of support, warranty, indemnity,
|
172
|
+
or other liability obligations and/or rights consistent with this
|
173
|
+
License. However, in accepting such obligations, You may act only
|
174
|
+
on Your own behalf and on Your sole responsibility, not on behalf
|
175
|
+
of any other Contributor, and only if You agree to indemnify,
|
176
|
+
defend, and hold each Contributor harmless for any liability
|
177
|
+
incurred by, or claims asserted against, such Contributor by reason
|
178
|
+
of your accepting any such warranty or additional liability.
|
179
|
+
|
180
|
+
END OF TERMS AND CONDITIONS
|
181
|
+
|
182
|
+
APPENDIX: How to apply the Apache License to your work.
|
183
|
+
|
184
|
+
To apply the Apache License to your work, attach the following
|
185
|
+
boilerplate notice, with the fields enclosed by brackets "[]"
|
186
|
+
replaced with your own identifying information. (Don't include
|
187
|
+
the brackets!) The text should be enclosed in the appropriate
|
188
|
+
comment syntax for the file format. We also recommend that a
|
189
|
+
file or class name and description of purpose be included on the
|
190
|
+
same "printed page" as the copyright notice for easier
|
191
|
+
identification within third-party archives.
|
192
|
+
|
193
|
+
Copyright [2025] [cr7258]
|
194
|
+
|
195
|
+
Licensed under the Apache License, Version 2.0 (the "License");
|
196
|
+
you may not use this file except in compliance with the License.
|
197
|
+
You may obtain a copy of the License at
|
198
|
+
|
199
|
+
http://www.apache.org/licenses/LICENSE-2.0
|
200
|
+
|
201
|
+
Unless required by applicable law or agreed to in writing, software
|
202
|
+
distributed under the License is distributed on an "AS IS" BASIS,
|
203
|
+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
204
|
+
See the License for the specific language governing permissions and
|
205
|
+
limitations under the License.
|
206
|
+
License-File: LICENSE
|
207
|
+
Requires-Python: >=3.10
|
208
|
+
Requires-Dist: anthropic>=0.49.0
|
209
|
+
Requires-Dist: elasticsearch>=8.0.0
|
210
|
+
Requires-Dist: fastmcp>=0.4.0
|
211
|
+
Requires-Dist: mcp>=1.0.0
|
212
|
+
Requires-Dist: opensearch-py>=2.0.0
|
213
|
+
Requires-Dist: python-dotenv>=1.0.0
|
214
|
+
Requires-Dist: tomli-w>=1.2.0
|
215
|
+
Requires-Dist: tomli>=2.2.1
|
216
|
+
Description-Content-Type: text/markdown
|
217
|
+
|
218
|
+
# Elasticsearch/OpenSearch MCP Server
|
219
|
+
|
220
|
+
[](https://smithery.ai/server/elasticsearch-mcp-server)
|
221
|
+
|
222
|
+
## Overview
|
223
|
+
|
224
|
+
A Model Context Protocol (MCP) server implementation that provides Elasticsearch and OpenSearch interaction. This server enables searching documents, analyzing indices, and managing cluster through a set of tools.
|
225
|
+
|
226
|
+
<a href="https://glama.ai/mcp/servers/b3po3delex"><img width="380" height="200" src="https://glama.ai/mcp/servers/b3po3delex/badge" alt="Elasticsearch MCP Server" /></a>
|
227
|
+
|
228
|
+
## Demo
|
229
|
+
|
230
|
+
https://github.com/user-attachments/assets/f7409e31-fac4-4321-9c94-b0ff2ea7ff15
|
231
|
+
|
232
|
+
## Features
|
233
|
+
|
234
|
+
### Index Operations
|
235
|
+
|
236
|
+
- `list_indices`: List all indices.
|
237
|
+
- `get_index`: Returns information (mappings, settings, aliases) about one or more indices.
|
238
|
+
- `create_index`: Create a new index.
|
239
|
+
- `delete_index`: Delete an index.
|
240
|
+
|
241
|
+
### Document Operations
|
242
|
+
|
243
|
+
- `search_documents`: Search for documents.
|
244
|
+
- `index_document`: Creates or updates a document in the index.
|
245
|
+
- `get_document`: Get a document by ID.
|
246
|
+
- `delete_document`: Delete a document by ID.
|
247
|
+
- `delete_by_query`: Deletes documents matching the provided query.
|
248
|
+
|
249
|
+
### Cluster Operations
|
250
|
+
|
251
|
+
- `get_cluster_health`: Returns basic information about the health of the cluster.
|
252
|
+
- `get_cluster_stats`: Returns high-level overview of cluster statistics.
|
253
|
+
|
254
|
+
### Alias Operations
|
255
|
+
|
256
|
+
- `list_aliases`: List all aliases.
|
257
|
+
- `get_alias`: Get alias information for a specific index.
|
258
|
+
- `put_alias`: Create or update an alias for a specific index.
|
259
|
+
- `delete_alias`: Delete an alias for a specific index.
|
260
|
+
|
261
|
+
## Configure Environment Variables
|
262
|
+
|
263
|
+
Copy the `.env.example` file to `.env` and update the values accordingly.
|
264
|
+
|
265
|
+
## Start Elasticsearch/OpenSearch Cluster
|
266
|
+
|
267
|
+
Start the Elasticsearch/OpenSearch cluster using Docker Compose:
|
268
|
+
|
269
|
+
```bash
|
270
|
+
# For Elasticsearch
|
271
|
+
docker-compose -f docker-compose-elasticsearch.yml up -d
|
272
|
+
|
273
|
+
# For OpenSearch
|
274
|
+
docker-compose -f docker-compose-opensearch.yml up -d
|
275
|
+
```
|
276
|
+
|
277
|
+
The default Elasticsearch username is `elastic` and password is `test123`. The default OpenSearch username is `admin` and password is `admin`.
|
278
|
+
|
279
|
+
You can access Kibana/OpenSearch Dashboards from http://localhost:5601.
|
280
|
+
|
281
|
+
## Usage with Claude Desktop
|
282
|
+
|
283
|
+
### Option 1: Installing via Smithery
|
284
|
+
|
285
|
+
To install Elasticsearch Server for Claude Desktop automatically via [Smithery](https://smithery.ai/server/elasticsearch-mcp-server):
|
286
|
+
|
287
|
+
```bash
|
288
|
+
npx -y @smithery/cli install elasticsearch-mcp-server --client claude
|
289
|
+
```
|
290
|
+
|
291
|
+
### Option 2: Using uvx
|
292
|
+
|
293
|
+
Using `uvx` will automatically install the package from PyPI, no need to clone the repository locally. Add the following configuration to Claude Desktop's config file `claude_desktop_config.json`.
|
294
|
+
|
295
|
+
```json
|
296
|
+
// For Elasticsearch
|
297
|
+
{
|
298
|
+
"mcpServers": {
|
299
|
+
"elasticsearch-mcp-server": {
|
300
|
+
"command": "uvx",
|
301
|
+
"args": [
|
302
|
+
"elasticsearch-mcp-server"
|
303
|
+
],
|
304
|
+
"env": {
|
305
|
+
"ELASTICSEARCH_HOST": "https://localhost:9200",
|
306
|
+
"ELASTICSEARCH_USERNAME": "elastic",
|
307
|
+
"ELASTICSEARCH_PASSWORD": "test123"
|
308
|
+
}
|
309
|
+
}
|
310
|
+
}
|
311
|
+
}
|
312
|
+
|
313
|
+
// For OpenSearch
|
314
|
+
{
|
315
|
+
"mcpServers": {
|
316
|
+
"opensearch-mcp-server": {
|
317
|
+
"command": "uvx",
|
318
|
+
"args": [
|
319
|
+
"opensearch-mcp-server"
|
320
|
+
],
|
321
|
+
"env": {
|
322
|
+
"OPENSEARCH_HOST": "https://localhost:9200",
|
323
|
+
"OPENSEARCH_USERNAME": "admin",
|
324
|
+
"OPENSEARCH_PASSWORD": "admin"
|
325
|
+
}
|
326
|
+
}
|
327
|
+
}
|
328
|
+
}
|
329
|
+
```
|
330
|
+
|
331
|
+
### Option 3: Using uv with local development
|
332
|
+
|
333
|
+
Using `uv` requires cloning the repository locally and specifying the path to the source code. Add the following configuration to Claude Desktop's config file `claude_desktop_config.json`.
|
334
|
+
|
335
|
+
```json
|
336
|
+
// For Elasticsearch
|
337
|
+
{
|
338
|
+
"mcpServers": {
|
339
|
+
"elasticsearch-mcp-server": {
|
340
|
+
"command": "uv",
|
341
|
+
"args": [
|
342
|
+
"--directory",
|
343
|
+
"path/to/src/elasticsearch_mcp_server",
|
344
|
+
"run",
|
345
|
+
"elasticsearch-mcp-server"
|
346
|
+
],
|
347
|
+
"env": {
|
348
|
+
"ELASTICSEARCH_HOST": "https://localhost:9200",
|
349
|
+
"ELASTICSEARCH_USERNAME": "elastic",
|
350
|
+
"ELASTICSEARCH_PASSWORD": "test123"
|
351
|
+
}
|
352
|
+
}
|
353
|
+
}
|
354
|
+
}
|
355
|
+
|
356
|
+
// For OpenSearch
|
357
|
+
{
|
358
|
+
"mcpServers": {
|
359
|
+
"opensearch-mcp-server": {
|
360
|
+
"command": "uv",
|
361
|
+
"args": [
|
362
|
+
"--directory",
|
363
|
+
"path/to/src/elasticsearch_mcp_server",
|
364
|
+
"run",
|
365
|
+
"opensearch-mcp-server"
|
366
|
+
],
|
367
|
+
"env": {
|
368
|
+
"OPENSEARCH_HOST": "https://localhost:9200",
|
369
|
+
"OPENSEARCH_USERNAME": "admin",
|
370
|
+
"OPENSEARCH_PASSWORD": "admin"
|
371
|
+
}
|
372
|
+
}
|
373
|
+
}
|
374
|
+
}
|
375
|
+
```
|
376
|
+
|
377
|
+
- On macOS: `~/Library/Application Support/Claude/claude_desktop_config.json`
|
378
|
+
- On Windows: `%APPDATA%/Claude/claude_desktop_config.json`
|
379
|
+
|
380
|
+
Restart Claude Desktop to load the new MCP server.
|
381
|
+
|
382
|
+
Now you can interact with your Elasticsearch/OpenSearch cluster through Claude using natural language commands like:
|
383
|
+
- "List all indices in the cluster"
|
384
|
+
- "How old is the student Bob?"
|
385
|
+
- "Show me the cluster health status"
|
386
|
+
|
387
|
+
## Usage with Anthropic MCP Client
|
388
|
+
|
389
|
+
```python
|
390
|
+
uv run mcp_client/client.py src/server.py
|
391
|
+
```
|
392
|
+
|
393
|
+
## License
|
394
|
+
|
395
|
+
This project is licensed under the Apache License Version 2.0 - see the [LICENSE](LICENSE) file for details.
|
@@ -0,0 +1,22 @@
|
|
1
|
+
src/__init__.py,sha256=aNKcThftSLh9IjOTA-UUpoRzIm4R0WwXKGAzykwecmU,211
|
2
|
+
src/server.py,sha256=BtNr3oKiRK1rbnocFIKzwpukY87_AylzKh_ffwP6Zr8,2153
|
3
|
+
src/clients/__init__.py,sha256=3UezAt9422S-7BvMiCo2Y9pmATVutorwsIQXP_g_CkA,1221
|
4
|
+
src/clients/base.py,sha256=vTe4I62ruO2bEeSAag-2B5fJOWJDXllKbo8qA0h4VkM,2160
|
5
|
+
src/clients/exceptions.py,sha256=NYF3KVw-9aAgCViin5OuBI6miMEPS5QsdHx6bcis1wc,2493
|
6
|
+
src/clients/common/__init__.py,sha256=VgvgxFpESn2wAuJmH0XM_Ej2izI7dxK7QJe9wG4fmW0,211
|
7
|
+
src/clients/common/alias.py,sha256=rB53TSld5x2vZyDNAwyEdnh1KWUXMSD7h5fSv_ubR2Q,759
|
8
|
+
src/clients/common/client.py,sha256=GD-V97Dj8c2XGUYt58vqCggkk-k8WJePPgngBDWO5aM,973
|
9
|
+
src/clients/common/cluster.py,sha256=pd5BVpqqDU6Lck3K704eEdhgFgzt9NstotWQLyG9zFM,401
|
10
|
+
src/clients/common/document.py,sha256=ZzZiXDf_UhlN2FCmqW3drVjIZ07kGYP9yth_sgsJGPc,1623
|
11
|
+
src/clients/common/index.py,sha256=vyH5iXlJe5JLcDK6fhCCEPN-tgGm6zP5ilwGPdWCXbY,776
|
12
|
+
src/tools/__init__.py,sha256=qiMRNUKlpNGnqbrkgBHGa7El1Pw_0a2qUJnuk_Q8LtY,324
|
13
|
+
src/tools/alias.py,sha256=p9TD4gXkGRGWHTYfvCg7G2hj-Uch9jwDXJNUY1hSD0Y,1376
|
14
|
+
src/tools/cluster.py,sha256=XRAG-uxdfrieYX1ov_cBb66IYXaa8OoSWCUfXvNauy0,587
|
15
|
+
src/tools/document.py,sha256=XZTVuk4di9VEwWMZN7jyDVnzoOiXkb4FBrXF44sVXTs,2052
|
16
|
+
src/tools/index.py,sha256=7KNPtElTFelkjRSvdMqPBx9nx_9Uk01OnTMeVo7YeCs,1345
|
17
|
+
src/tools/register.py,sha256=wrG2P6-YPW77bTg1j_ELp8omWRYsJjFeOHUy_unHe6Y,1344
|
18
|
+
opensearch_mcp_server-2.0.0.dist-info/METADATA,sha256=0A6zv4UFDA609Uom3Dl2cv8MtuZPyTurw2jEPcmR_-E,18509
|
19
|
+
opensearch_mcp_server-2.0.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
|
20
|
+
opensearch_mcp_server-2.0.0.dist-info/entry_points.txt,sha256=ImfJnUwMpQUBmu-1MeBG_P0dwamfXKQh82mBKW7tWNY,138
|
21
|
+
opensearch_mcp_server-2.0.0.dist-info/licenses/LICENSE,sha256=DBsjuP5FR51d9kaUdXlVBuBv3cQ_I_adq9gefYQ9FcY,11339
|
22
|
+
opensearch_mcp_server-2.0.0.dist-info/RECORD,,
|
@@ -0,0 +1,201 @@
|
|
1
|
+
Apache License
|
2
|
+
Version 2.0, January 2004
|
3
|
+
http://www.apache.org/licenses/
|
4
|
+
|
5
|
+
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
6
|
+
|
7
|
+
1. Definitions.
|
8
|
+
|
9
|
+
"License" shall mean the terms and conditions for use, reproduction,
|
10
|
+
and distribution as defined by Sections 1 through 9 of this document.
|
11
|
+
|
12
|
+
"Licensor" shall mean the copyright owner or entity authorized by
|
13
|
+
the copyright owner that is granting the License.
|
14
|
+
|
15
|
+
"Legal Entity" shall mean the union of the acting entity and all
|
16
|
+
other entities that control, are controlled by, or are under common
|
17
|
+
control with that entity. For the purposes of this definition,
|
18
|
+
"control" means (i) the power, direct or indirect, to cause the
|
19
|
+
direction or management of such entity, whether by contract or
|
20
|
+
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
21
|
+
outstanding shares, or (iii) beneficial ownership of such entity.
|
22
|
+
|
23
|
+
"You" (or "Your") shall mean an individual or Legal Entity
|
24
|
+
exercising permissions granted by this License.
|
25
|
+
|
26
|
+
"Source" form shall mean the preferred form for making modifications,
|
27
|
+
including but not limited to software source code, documentation
|
28
|
+
source, and configuration files.
|
29
|
+
|
30
|
+
"Object" form shall mean any form resulting from mechanical
|
31
|
+
transformation or translation of a Source form, including but
|
32
|
+
not limited to compiled object code, generated documentation,
|
33
|
+
and conversions to other media types.
|
34
|
+
|
35
|
+
"Work" shall mean the work of authorship, whether in Source or
|
36
|
+
Object form, made available under the License, as indicated by a
|
37
|
+
copyright notice that is included in or attached to the work
|
38
|
+
(an example is provided in the Appendix below).
|
39
|
+
|
40
|
+
"Derivative Works" shall mean any work, whether in Source or Object
|
41
|
+
form, that is based on (or derived from) the Work and for which the
|
42
|
+
editorial revisions, annotations, elaborations, or other modifications
|
43
|
+
represent, as a whole, an original work of authorship. For the purposes
|
44
|
+
of this License, Derivative Works shall not include works that remain
|
45
|
+
separable from, or merely link (or bind by name) to the interfaces of,
|
46
|
+
the Work and Derivative Works thereof.
|
47
|
+
|
48
|
+
"Contribution" shall mean any work of authorship, including
|
49
|
+
the original version of the Work and any modifications or additions
|
50
|
+
to that Work or Derivative Works thereof, that is intentionally
|
51
|
+
submitted to Licensor for inclusion in the Work by the copyright owner
|
52
|
+
or by an individual or Legal Entity authorized to submit on behalf of
|
53
|
+
the copyright owner. For the purposes of this definition, "submitted"
|
54
|
+
means any form of electronic, verbal, or written communication sent
|
55
|
+
to the Licensor or its representatives, including but not limited to
|
56
|
+
communication on electronic mailing lists, source code control systems,
|
57
|
+
and issue tracking systems that are managed by, or on behalf of, the
|
58
|
+
Licensor for the purpose of discussing and improving the Work, but
|
59
|
+
excluding communication that is conspicuously marked or otherwise
|
60
|
+
designated in writing by the copyright owner as "Not a Contribution."
|
61
|
+
|
62
|
+
"Contributor" shall mean Licensor and any individual or Legal Entity
|
63
|
+
on behalf of whom a Contribution has been received by Licensor and
|
64
|
+
subsequently incorporated within the Work.
|
65
|
+
|
66
|
+
2. Grant of Copyright License. Subject to the terms and conditions of
|
67
|
+
this License, each Contributor hereby grants to You a perpetual,
|
68
|
+
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
69
|
+
copyright license to reproduce, prepare Derivative Works of,
|
70
|
+
publicly display, publicly perform, sublicense, and distribute the
|
71
|
+
Work and such Derivative Works in Source or Object form.
|
72
|
+
|
73
|
+
3. Grant of Patent License. Subject to the terms and conditions of
|
74
|
+
this License, each Contributor hereby grants to You a perpetual,
|
75
|
+
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
76
|
+
(except as stated in this section) patent license to make, have made,
|
77
|
+
use, offer to sell, sell, import, and otherwise transfer the Work,
|
78
|
+
where such license applies only to those patent claims licensable
|
79
|
+
by such Contributor that are necessarily infringed by their
|
80
|
+
Contribution(s) alone or by combination of their Contribution(s)
|
81
|
+
with the Work to which such Contribution(s) was submitted. If You
|
82
|
+
institute patent litigation against any entity (including a
|
83
|
+
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
84
|
+
or a Contribution incorporated within the Work constitutes direct
|
85
|
+
or contributory patent infringement, then any patent licenses
|
86
|
+
granted to You under this License for that Work shall terminate
|
87
|
+
as of the date such litigation is filed.
|
88
|
+
|
89
|
+
4. Redistribution. You may reproduce and distribute copies of the
|
90
|
+
Work or Derivative Works thereof in any medium, with or without
|
91
|
+
modifications, and in Source or Object form, provided that You
|
92
|
+
meet the following conditions:
|
93
|
+
|
94
|
+
(a) You must give any other recipients of the Work or
|
95
|
+
Derivative Works a copy of this License; and
|
96
|
+
|
97
|
+
(b) You must cause any modified files to carry prominent notices
|
98
|
+
stating that You changed the files; and
|
99
|
+
|
100
|
+
(c) You must retain, in the Source form of any Derivative Works
|
101
|
+
that You distribute, all copyright, patent, trademark, and
|
102
|
+
attribution notices from the Source form of the Work,
|
103
|
+
excluding those notices that do not pertain to any part of
|
104
|
+
the Derivative Works; and
|
105
|
+
|
106
|
+
(d) If the Work includes a "NOTICE" text file as part of its
|
107
|
+
distribution, then any Derivative Works that You distribute must
|
108
|
+
include a readable copy of the attribution notices contained
|
109
|
+
within such NOTICE file, excluding those notices that do not
|
110
|
+
pertain to any part of the Derivative Works, in at least one
|
111
|
+
of the following places: within a NOTICE text file distributed
|
112
|
+
as part of the Derivative Works; within the Source form or
|
113
|
+
documentation, if provided along with the Derivative Works; or,
|
114
|
+
within a display generated by the Derivative Works, if and
|
115
|
+
wherever such third-party notices normally appear. The contents
|
116
|
+
of the NOTICE file are for informational purposes only and
|
117
|
+
do not modify the License. You may add Your own attribution
|
118
|
+
notices within Derivative Works that You distribute, alongside
|
119
|
+
or as an addendum to the NOTICE text from the Work, provided
|
120
|
+
that such additional attribution notices cannot be construed
|
121
|
+
as modifying the License.
|
122
|
+
|
123
|
+
You may add Your own copyright statement to Your modifications and
|
124
|
+
may provide additional or different license terms and conditions
|
125
|
+
for use, reproduction, or distribution of Your modifications, or
|
126
|
+
for any such Derivative Works as a whole, provided Your use,
|
127
|
+
reproduction, and distribution of the Work otherwise complies with
|
128
|
+
the conditions stated in this License.
|
129
|
+
|
130
|
+
5. Submission of Contributions. Unless You explicitly state otherwise,
|
131
|
+
any Contribution intentionally submitted for inclusion in the Work
|
132
|
+
by You to the Licensor shall be under the terms and conditions of
|
133
|
+
this License, without any additional terms or conditions.
|
134
|
+
Notwithstanding the above, nothing herein shall supersede or modify
|
135
|
+
the terms of any separate license agreement you may have executed
|
136
|
+
with Licensor regarding such Contributions.
|
137
|
+
|
138
|
+
6. Trademarks. This License does not grant permission to use the trade
|
139
|
+
names, trademarks, service marks, or product names of the Licensor,
|
140
|
+
except as required for reasonable and customary use in describing the
|
141
|
+
origin of the Work and reproducing the content of the NOTICE file.
|
142
|
+
|
143
|
+
7. Disclaimer of Warranty. Unless required by applicable law or
|
144
|
+
agreed to in writing, Licensor provides the Work (and each
|
145
|
+
Contributor provides its Contributions) on an "AS IS" BASIS,
|
146
|
+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
147
|
+
implied, including, without limitation, any warranties or conditions
|
148
|
+
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
149
|
+
PARTICULAR PURPOSE. You are solely responsible for determining the
|
150
|
+
appropriateness of using or redistributing the Work and assume any
|
151
|
+
risks associated with Your exercise of permissions under this License.
|
152
|
+
|
153
|
+
8. Limitation of Liability. In no event and under no legal theory,
|
154
|
+
whether in tort (including negligence), contract, or otherwise,
|
155
|
+
unless required by applicable law (such as deliberate and grossly
|
156
|
+
negligent acts) or agreed to in writing, shall any Contributor be
|
157
|
+
liable to You for damages, including any direct, indirect, special,
|
158
|
+
incidental, or consequential damages of any character arising as a
|
159
|
+
result of this License or out of the use or inability to use the
|
160
|
+
Work (including but not limited to damages for loss of goodwill,
|
161
|
+
work stoppage, computer failure or malfunction, or any and all
|
162
|
+
other commercial damages or losses), even if such Contributor
|
163
|
+
has been advised of the possibility of such damages.
|
164
|
+
|
165
|
+
9. Accepting Warranty or Additional Liability. While redistributing
|
166
|
+
the Work or Derivative Works thereof, You may choose to offer,
|
167
|
+
and charge a fee for, acceptance of support, warranty, indemnity,
|
168
|
+
or other liability obligations and/or rights consistent with this
|
169
|
+
License. However, in accepting such obligations, You may act only
|
170
|
+
on Your own behalf and on Your sole responsibility, not on behalf
|
171
|
+
of any other Contributor, and only if You agree to indemnify,
|
172
|
+
defend, and hold each Contributor harmless for any liability
|
173
|
+
incurred by, or claims asserted against, such Contributor by reason
|
174
|
+
of your accepting any such warranty or additional liability.
|
175
|
+
|
176
|
+
END OF TERMS AND CONDITIONS
|
177
|
+
|
178
|
+
APPENDIX: How to apply the Apache License to your work.
|
179
|
+
|
180
|
+
To apply the Apache License to your work, attach the following
|
181
|
+
boilerplate notice, with the fields enclosed by brackets "[]"
|
182
|
+
replaced with your own identifying information. (Don't include
|
183
|
+
the brackets!) The text should be enclosed in the appropriate
|
184
|
+
comment syntax for the file format. We also recommend that a
|
185
|
+
file or class name and description of purpose be included on the
|
186
|
+
same "printed page" as the copyright notice for easier
|
187
|
+
identification within third-party archives.
|
188
|
+
|
189
|
+
Copyright [2025] [cr7258]
|
190
|
+
|
191
|
+
Licensed under the Apache License, Version 2.0 (the "License");
|
192
|
+
you may not use this file except in compliance with the License.
|
193
|
+
You may obtain a copy of the License at
|
194
|
+
|
195
|
+
http://www.apache.org/licenses/LICENSE-2.0
|
196
|
+
|
197
|
+
Unless required by applicable law or agreed to in writing, software
|
198
|
+
distributed under the License is distributed on an "AS IS" BASIS,
|
199
|
+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
200
|
+
See the License for the specific language governing permissions and
|
201
|
+
limitations under the License.
|
src/__init__.py
ADDED
src/clients/__init__.py
ADDED
@@ -0,0 +1,42 @@
|
|
1
|
+
import os
|
2
|
+
|
3
|
+
from dotenv import load_dotenv
|
4
|
+
|
5
|
+
from src.clients.common.client import SearchClient
|
6
|
+
from src.clients.exceptions import handle_search_exceptions
|
7
|
+
|
8
|
+
def create_search_client(engine_type: str) -> SearchClient:
|
9
|
+
"""
|
10
|
+
Create a search client for the specified engine type.
|
11
|
+
|
12
|
+
Args:
|
13
|
+
engine_type: Type of search engine to use ("elasticsearch" or "opensearch")
|
14
|
+
|
15
|
+
Returns:
|
16
|
+
A search client instance
|
17
|
+
"""
|
18
|
+
# Load configuration from environment variables
|
19
|
+
load_dotenv()
|
20
|
+
|
21
|
+
# Get configuration from environment variables
|
22
|
+
prefix = engine_type.upper()
|
23
|
+
hosts_str = os.environ.get(f"{prefix}_HOSTS", "https://localhost:9200")
|
24
|
+
hosts = [host.strip() for host in hosts_str.split(",")]
|
25
|
+
username = os.environ.get(f"{prefix}_USERNAME")
|
26
|
+
password = os.environ.get(f"{prefix}_PASSWORD")
|
27
|
+
verify_certs = os.environ.get(f"{prefix}_VERIFY_CERTS", "false").lower() == "true"
|
28
|
+
|
29
|
+
config = {
|
30
|
+
"hosts": hosts,
|
31
|
+
"username": username,
|
32
|
+
"password": password,
|
33
|
+
"verify_certs": verify_certs
|
34
|
+
}
|
35
|
+
|
36
|
+
return SearchClient(config, engine_type)
|
37
|
+
|
38
|
+
__all__ = [
|
39
|
+
'create_search_client',
|
40
|
+
'handle_search_exceptions',
|
41
|
+
'SearchClient',
|
42
|
+
]
|
src/clients/base.py
ADDED
@@ -0,0 +1,54 @@
|
|
1
|
+
from abc import ABC
|
2
|
+
import logging
|
3
|
+
import warnings
|
4
|
+
from typing import Dict
|
5
|
+
|
6
|
+
class SearchClientBase(ABC):
|
7
|
+
def __init__(self, config: Dict, engine_type: str):
|
8
|
+
"""
|
9
|
+
Initialize the search client.
|
10
|
+
|
11
|
+
Args:
|
12
|
+
config: Configuration dictionary with connection parameters
|
13
|
+
engine_type: Type of search engine to use ("elasticsearch" or "opensearch")
|
14
|
+
"""
|
15
|
+
self.logger = logging.getLogger()
|
16
|
+
self.config = config
|
17
|
+
self.engine_type = engine_type
|
18
|
+
|
19
|
+
# Extract common configuration
|
20
|
+
hosts = config.get("hosts")
|
21
|
+
username = config.get("username")
|
22
|
+
password = config.get("password")
|
23
|
+
verify_certs = config.get("verify_certs", False)
|
24
|
+
|
25
|
+
# Disable insecure request warnings if verify_certs is False
|
26
|
+
if not verify_certs:
|
27
|
+
warnings.filterwarnings("ignore", message=".*verify_certs=False is insecure.*")
|
28
|
+
warnings.filterwarnings("ignore", message=".*Unverified HTTPS request is being made to host.*")
|
29
|
+
|
30
|
+
try:
|
31
|
+
import urllib3
|
32
|
+
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
|
33
|
+
except ImportError:
|
34
|
+
pass
|
35
|
+
|
36
|
+
# Initialize client based on engine type
|
37
|
+
if engine_type == "elasticsearch":
|
38
|
+
from elasticsearch import Elasticsearch
|
39
|
+
self.client = Elasticsearch(
|
40
|
+
hosts=hosts,
|
41
|
+
basic_auth=(username, password) if username and password else None,
|
42
|
+
verify_certs=verify_certs
|
43
|
+
)
|
44
|
+
self.logger.info(f"Elasticsearch client initialized with hosts: {hosts}")
|
45
|
+
elif engine_type == "opensearch":
|
46
|
+
from opensearchpy import OpenSearch
|
47
|
+
self.client = OpenSearch(
|
48
|
+
hosts=hosts,
|
49
|
+
http_auth=(username, password) if username and password else None,
|
50
|
+
verify_certs=verify_certs
|
51
|
+
)
|
52
|
+
self.logger.info(f"OpenSearch client initialized with hosts: {hosts}")
|
53
|
+
else:
|
54
|
+
raise ValueError(f"Unsupported engine type: {engine_type}")
|
@@ -0,0 +1,20 @@
|
|
1
|
+
from typing import Dict
|
2
|
+
|
3
|
+
from src.clients.base import SearchClientBase
|
4
|
+
|
5
|
+
class AliasClient(SearchClientBase):
|
6
|
+
def list_aliases(self) -> Dict:
|
7
|
+
"""Get all aliases."""
|
8
|
+
return self.client.cat.aliases()
|
9
|
+
|
10
|
+
def get_alias(self, index: str) -> Dict:
|
11
|
+
"""Get aliases for the specified index."""
|
12
|
+
return self.client.indices.get_alias(index=index)
|
13
|
+
|
14
|
+
def put_alias(self, index: str, name: str, body: Dict) -> Dict:
|
15
|
+
"""Creates or updates an alias."""
|
16
|
+
return self.client.indices.put_alias(index=index, name=name, body=body)
|
17
|
+
|
18
|
+
def delete_alias(self, index: str, name: str) -> Dict:
|
19
|
+
"""Delete an alias for the specified index."""
|
20
|
+
return self.client.indices.delete_alias(index=index, name=name)
|
@@ -0,0 +1,25 @@
|
|
1
|
+
from typing import Dict
|
2
|
+
|
3
|
+
from src.clients.common.alias import AliasClient
|
4
|
+
from src.clients.common.cluster import ClusterClient
|
5
|
+
from src.clients.common.document import DocumentClient
|
6
|
+
from src.clients.common.index import IndexClient
|
7
|
+
|
8
|
+
class SearchClient(IndexClient, DocumentClient, ClusterClient, AliasClient):
|
9
|
+
"""
|
10
|
+
Unified search client that combines all search functionality.
|
11
|
+
|
12
|
+
This class uses multiple inheritance to combine all specialized client implementations
|
13
|
+
(index, document, cluster, alias) into a single unified client.
|
14
|
+
"""
|
15
|
+
|
16
|
+
def __init__(self, config: Dict, engine_type: str):
|
17
|
+
"""
|
18
|
+
Initialize the search client.
|
19
|
+
|
20
|
+
Args:
|
21
|
+
config: Configuration dictionary with connection parameters
|
22
|
+
engine_type: Type of search engine to use ("elasticsearch" or "opensearch")
|
23
|
+
"""
|
24
|
+
super().__init__(config, engine_type)
|
25
|
+
self.logger.info(f"Initialized the {engine_type} client")
|
@@ -0,0 +1,12 @@
|
|
1
|
+
from typing import Dict
|
2
|
+
|
3
|
+
from src.clients.base import SearchClientBase
|
4
|
+
|
5
|
+
class ClusterClient(SearchClientBase):
|
6
|
+
def get_cluster_health(self) -> Dict:
|
7
|
+
"""Get cluster health information from OpenSearch."""
|
8
|
+
return self.client.cluster.health()
|
9
|
+
|
10
|
+
def get_cluster_stats(self) -> Dict:
|
11
|
+
"""Get cluster statistics from OpenSearch."""
|
12
|
+
return self.client.cluster.stats()
|
@@ -0,0 +1,37 @@
|
|
1
|
+
from typing import Dict, Optional
|
2
|
+
|
3
|
+
from src.clients.base import SearchClientBase
|
4
|
+
|
5
|
+
class DocumentClient(SearchClientBase):
|
6
|
+
def search_documents(self, index: str, body: Dict) -> Dict:
|
7
|
+
"""Search for documents in the index."""
|
8
|
+
return self.client.search(index=index, body=body)
|
9
|
+
|
10
|
+
def index_document(self, index: str, document: Dict, id: Optional[str] = None) -> Dict:
|
11
|
+
"""Creates a new document in the index."""
|
12
|
+
# Handle parameter name differences between Elasticsearch and OpenSearch
|
13
|
+
if self.engine_type == "elasticsearch":
|
14
|
+
# For Elasticsearch: index(index, document, id=None, ...)
|
15
|
+
if id is not None:
|
16
|
+
return self.client.index(index=index, document=document, id=id)
|
17
|
+
else:
|
18
|
+
return self.client.index(index=index, document=document)
|
19
|
+
else:
|
20
|
+
# For OpenSearch: index(index, body, id=None, ...)
|
21
|
+
if id is not None:
|
22
|
+
return self.client.index(index=index, body=document, id=id)
|
23
|
+
else:
|
24
|
+
return self.client.index(index=index, body=document)
|
25
|
+
|
26
|
+
def get_document(self, index: str, id: str) -> Dict:
|
27
|
+
"""Get a document by ID."""
|
28
|
+
return self.client.get(index=index, id=id)
|
29
|
+
|
30
|
+
def delete_document(self, index: str, id: str) -> Dict:
|
31
|
+
"""Removes a document from the index."""
|
32
|
+
return self.client.delete(index=index, id=id)
|
33
|
+
|
34
|
+
def delete_by_query(self, index: str, body: Dict) -> Dict:
|
35
|
+
"""Deletes documents matching the provided query."""
|
36
|
+
return self.client.delete_by_query(index=index, body=body)
|
37
|
+
|
@@ -0,0 +1,20 @@
|
|
1
|
+
from typing import Dict, Optional
|
2
|
+
|
3
|
+
from src.clients.base import SearchClientBase
|
4
|
+
|
5
|
+
class IndexClient(SearchClientBase):
|
6
|
+
def list_indices(self) -> Dict:
|
7
|
+
"""List all indices."""
|
8
|
+
return self.client.cat.indices()
|
9
|
+
|
10
|
+
def get_index(self, index: str) -> Dict:
|
11
|
+
"""Returns information (mappings, settings, aliases) about one or more indices."""
|
12
|
+
return self.client.indices.get(index=index)
|
13
|
+
|
14
|
+
def create_index(self, index: str, body: Optional[Dict] = None) -> Dict:
|
15
|
+
"""Creates an index with optional settings and mappings."""
|
16
|
+
return self.client.indices.create(index=index, body=body)
|
17
|
+
|
18
|
+
def delete_index(self, index: str) -> Dict:
|
19
|
+
"""Delete an index."""
|
20
|
+
return self.client.indices.delete(index=index)
|
@@ -0,0 +1,69 @@
|
|
1
|
+
import functools
|
2
|
+
import logging
|
3
|
+
from typing import TypeVar, Callable
|
4
|
+
|
5
|
+
from fastmcp import FastMCP
|
6
|
+
from mcp.types import TextContent
|
7
|
+
|
8
|
+
T = TypeVar('T')
|
9
|
+
|
10
|
+
def handle_search_exceptions(func: Callable[..., T]) -> Callable[..., list[TextContent]]:
|
11
|
+
"""
|
12
|
+
Decorator to handle exceptions in search client operations.
|
13
|
+
|
14
|
+
Args:
|
15
|
+
func: The function to decorate
|
16
|
+
|
17
|
+
Returns:
|
18
|
+
Decorated function that handles exceptions
|
19
|
+
"""
|
20
|
+
@functools.wraps(func)
|
21
|
+
def wrapper(*args, **kwargs):
|
22
|
+
logger = logging.getLogger()
|
23
|
+
try:
|
24
|
+
return func(*args, **kwargs)
|
25
|
+
except Exception as e:
|
26
|
+
logger.error(f"Unexpected error in {func.__name__}: {e}")
|
27
|
+
return [TextContent(type="text", text=f"Unexpected error in {func.__name__}: {str(e)}")]
|
28
|
+
|
29
|
+
return wrapper
|
30
|
+
|
31
|
+
def with_exception_handling(tool_instance: object, mcp: FastMCP) -> None:
|
32
|
+
"""
|
33
|
+
Register tools from a tool instance with automatic exception handling applied to all tools.
|
34
|
+
|
35
|
+
This function temporarily replaces mcp.tool with a wrapped version that automatically
|
36
|
+
applies the handle_search_exceptions decorator to all registered tool methods.
|
37
|
+
|
38
|
+
Args:
|
39
|
+
tool_instance: The tool instance that has a register_tools method
|
40
|
+
mcp: The FastMCP instance used for tool registration
|
41
|
+
"""
|
42
|
+
# Save the original tool method
|
43
|
+
original_tool = mcp.tool
|
44
|
+
|
45
|
+
@functools.wraps(original_tool)
|
46
|
+
def wrapped_tool(*args, **kwargs):
|
47
|
+
# Get the original decorator
|
48
|
+
decorator = original_tool(*args, **kwargs)
|
49
|
+
|
50
|
+
# Return a new decorator that applies both the exception handler and original decorator
|
51
|
+
def combined_decorator(func):
|
52
|
+
# First apply the exception handling decorator
|
53
|
+
wrapped_func = handle_search_exceptions(func)
|
54
|
+
# Then apply the original mcp.tool decorator
|
55
|
+
return decorator(wrapped_func)
|
56
|
+
|
57
|
+
return combined_decorator
|
58
|
+
|
59
|
+
try:
|
60
|
+
# Temporarily replace mcp.tool with our wrapped version
|
61
|
+
mcp.tool = wrapped_tool
|
62
|
+
|
63
|
+
# Call the registration method on the tool instance
|
64
|
+
tool_instance.register_tools(mcp)
|
65
|
+
finally:
|
66
|
+
# Restore the original mcp.tool to avoid affecting other code that might use mcp.tool
|
67
|
+
# This ensures that our modification is isolated to just this tool registration
|
68
|
+
# and prevents multiple nested decorators if register_all_tools is called multiple times
|
69
|
+
mcp.tool = original_tool
|
src/server.py
ADDED
@@ -0,0 +1,71 @@
|
|
1
|
+
import logging
|
2
|
+
import sys
|
3
|
+
|
4
|
+
from fastmcp import FastMCP
|
5
|
+
|
6
|
+
from src.clients import create_search_client
|
7
|
+
from src.tools.alias import AliasTools
|
8
|
+
from src.tools.cluster import ClusterTools
|
9
|
+
from src.tools.document import DocumentTools
|
10
|
+
from src.tools.index import IndexTools
|
11
|
+
from src.tools.register import ToolsRegister
|
12
|
+
|
13
|
+
class SearchMCPServer:
|
14
|
+
def __init__(self, engine_type):
|
15
|
+
# Set engine type
|
16
|
+
self.engine_type = engine_type
|
17
|
+
self.name = f"{self.engine_type}_mcp_server"
|
18
|
+
self.mcp = FastMCP(self.name)
|
19
|
+
self.logger = logging.getLogger()
|
20
|
+
self.logger.info(f"Initializing {self.name}...")
|
21
|
+
|
22
|
+
# Create the corresponding search client
|
23
|
+
self.search_client = create_search_client(self.engine_type)
|
24
|
+
|
25
|
+
# Initialize tools
|
26
|
+
self._register_tools()
|
27
|
+
|
28
|
+
def _register_tools(self):
|
29
|
+
"""Register all MCP tools."""
|
30
|
+
# Create a tools register
|
31
|
+
register = ToolsRegister(self.logger, self.search_client, self.mcp)
|
32
|
+
|
33
|
+
# Define all tool classes to register
|
34
|
+
tool_classes = [
|
35
|
+
IndexTools,
|
36
|
+
DocumentTools,
|
37
|
+
ClusterTools,
|
38
|
+
AliasTools
|
39
|
+
]
|
40
|
+
# Register all tools
|
41
|
+
register.register_all_tools(tool_classes)
|
42
|
+
|
43
|
+
def run(self):
|
44
|
+
"""Run the MCP server."""
|
45
|
+
self.mcp.run()
|
46
|
+
|
47
|
+
def run_search_server(engine_type):
|
48
|
+
"""Run search server with specified engine type."""
|
49
|
+
server = SearchMCPServer(engine_type=engine_type)
|
50
|
+
server.run()
|
51
|
+
|
52
|
+
def elasticsearch_mcp_server():
|
53
|
+
"""Entry point for Elasticsearch MCP server."""
|
54
|
+
run_search_server(engine_type="elasticsearch")
|
55
|
+
|
56
|
+
def opensearch_mcp_server():
|
57
|
+
"""Entry point for OpenSearch MCP server."""
|
58
|
+
run_search_server(engine_type="opensearch")
|
59
|
+
|
60
|
+
if __name__ == "__main__":
|
61
|
+
# Default to Elasticsearch
|
62
|
+
engine_type = "elasticsearch"
|
63
|
+
|
64
|
+
# If command line arguments are provided, use the first argument as the engine type
|
65
|
+
if len(sys.argv) > 1:
|
66
|
+
engine_type = sys.argv[1].lower()
|
67
|
+
|
68
|
+
if engine_type == "opensearch":
|
69
|
+
opensearch_mcp_server()
|
70
|
+
else:
|
71
|
+
elasticsearch_mcp_server()
|
src/tools/__init__.py
ADDED
@@ -0,0 +1,13 @@
|
|
1
|
+
from src.tools.index import IndexTools
|
2
|
+
from src.tools.document import DocumentTools
|
3
|
+
from src.tools.cluster import ClusterTools
|
4
|
+
from src.tools.alias import AliasTools
|
5
|
+
from src.tools.register import ToolsRegister
|
6
|
+
|
7
|
+
__all__ = [
|
8
|
+
'IndexTools',
|
9
|
+
'DocumentTools',
|
10
|
+
'ClusterTools',
|
11
|
+
'AliasTools',
|
12
|
+
'ToolsRegister',
|
13
|
+
]
|
src/tools/alias.py
ADDED
@@ -0,0 +1,46 @@
|
|
1
|
+
from typing import Dict, List
|
2
|
+
|
3
|
+
from fastmcp import FastMCP
|
4
|
+
|
5
|
+
class AliasTools:
|
6
|
+
def __init__(self, search_client):
|
7
|
+
self.search_client = search_client
|
8
|
+
def register_tools(self, mcp: FastMCP):
|
9
|
+
@mcp.tool()
|
10
|
+
def list_aliases() -> List[Dict]:
|
11
|
+
"""List all aliases."""
|
12
|
+
return self.search_client.list_aliases()
|
13
|
+
|
14
|
+
@mcp.tool()
|
15
|
+
def get_alias(index: str) -> Dict:
|
16
|
+
"""
|
17
|
+
Get alias information for a specific index.
|
18
|
+
|
19
|
+
Args:
|
20
|
+
index: Name of the index
|
21
|
+
"""
|
22
|
+
return self.search_client.get_alias(index=index)
|
23
|
+
|
24
|
+
@mcp.tool()
|
25
|
+
def put_alias(index: str, name: str, body: Dict) -> Dict:
|
26
|
+
"""
|
27
|
+
Create or update an alias for a specific index.
|
28
|
+
|
29
|
+
Args:
|
30
|
+
index: Name of the index
|
31
|
+
name: Name of the alias
|
32
|
+
body: Alias configuration
|
33
|
+
"""
|
34
|
+
return self.search_client.put_alias(index=index, name=name, body=body)
|
35
|
+
|
36
|
+
@mcp.tool()
|
37
|
+
def delete_alias(index: str, name: str) -> Dict:
|
38
|
+
"""
|
39
|
+
Delete an alias for a specific index.
|
40
|
+
|
41
|
+
Args:
|
42
|
+
index: Name of the index
|
43
|
+
name: Name of the alias
|
44
|
+
"""
|
45
|
+
return self.search_client.delete_alias(index=index, name=name)
|
46
|
+
|
src/tools/cluster.py
ADDED
@@ -0,0 +1,17 @@
|
|
1
|
+
from typing import Dict
|
2
|
+
|
3
|
+
from fastmcp import FastMCP
|
4
|
+
|
5
|
+
class ClusterTools:
|
6
|
+
def __init__(self, search_client):
|
7
|
+
self.search_client = search_client
|
8
|
+
def register_tools(self, mcp: FastMCP):
|
9
|
+
@mcp.tool()
|
10
|
+
def get_cluster_health() -> Dict:
|
11
|
+
"""Returns basic information about the health of the cluster."""
|
12
|
+
return self.search_client.get_cluster_health()
|
13
|
+
|
14
|
+
@mcp.tool()
|
15
|
+
def get_cluster_stats() -> Dict:
|
16
|
+
"""Returns high-level overview of cluster statistics."""
|
17
|
+
return self.search_client.get_cluster_stats()
|
src/tools/document.py
ADDED
@@ -0,0 +1,64 @@
|
|
1
|
+
from typing import Dict, Optional
|
2
|
+
|
3
|
+
from fastmcp import FastMCP
|
4
|
+
|
5
|
+
class DocumentTools:
|
6
|
+
def __init__(self, search_client):
|
7
|
+
self.search_client = search_client
|
8
|
+
|
9
|
+
def register_tools(self, mcp: FastMCP):
|
10
|
+
@mcp.tool()
|
11
|
+
def search_documents(index: str, body: Dict) -> Dict:
|
12
|
+
"""
|
13
|
+
Search for documents.
|
14
|
+
|
15
|
+
Args:
|
16
|
+
index: Name of the index
|
17
|
+
body: Search query
|
18
|
+
"""
|
19
|
+
return self.search_client.search_documents(index=index, body=body)
|
20
|
+
|
21
|
+
@mcp.tool()
|
22
|
+
def index_document(index: str, document: Dict, id: Optional[str] = None) -> Dict:
|
23
|
+
"""
|
24
|
+
Creates or updates a document in the index.
|
25
|
+
|
26
|
+
Args:
|
27
|
+
index: Name of the index
|
28
|
+
document: Document data
|
29
|
+
id: Optional document ID
|
30
|
+
"""
|
31
|
+
return self.search_client.index_document(index=index, id=id, document=document)
|
32
|
+
|
33
|
+
@mcp.tool()
|
34
|
+
def get_document(index: str, id: str) -> Dict:
|
35
|
+
"""
|
36
|
+
Get a document by ID.
|
37
|
+
|
38
|
+
Args:
|
39
|
+
index: Name of the index
|
40
|
+
id: Document ID
|
41
|
+
"""
|
42
|
+
return self.search_client.get_document(index=index, id=id)
|
43
|
+
|
44
|
+
@mcp.tool()
|
45
|
+
def delete_document(index: str, id: str) -> Dict:
|
46
|
+
"""
|
47
|
+
Delete a document by ID.
|
48
|
+
|
49
|
+
Args:
|
50
|
+
index: Name of the index
|
51
|
+
id: Document ID
|
52
|
+
"""
|
53
|
+
return self.search_client.delete_document(index=index, id=id)
|
54
|
+
|
55
|
+
@mcp.tool()
|
56
|
+
def delete_by_query(index: str, body: Dict) -> Dict:
|
57
|
+
"""
|
58
|
+
Deletes documents matching the provided query.
|
59
|
+
|
60
|
+
Args:
|
61
|
+
index: Name of the index
|
62
|
+
body: Query to match documents for deletion
|
63
|
+
"""
|
64
|
+
return self.search_client.delete_by_query(index=index, body=body)
|
src/tools/index.py
ADDED
@@ -0,0 +1,44 @@
|
|
1
|
+
from typing import Dict, Optional, List
|
2
|
+
|
3
|
+
from fastmcp import FastMCP
|
4
|
+
|
5
|
+
class IndexTools:
|
6
|
+
def __init__(self, search_client):
|
7
|
+
self.search_client = search_client
|
8
|
+
|
9
|
+
def register_tools(self, mcp: FastMCP):
|
10
|
+
@mcp.tool()
|
11
|
+
def list_indices() -> List[Dict]:
|
12
|
+
"""List all indices."""
|
13
|
+
return self.search_client.list_indices()
|
14
|
+
|
15
|
+
@mcp.tool()
|
16
|
+
def get_index(index: str) -> Dict:
|
17
|
+
"""
|
18
|
+
Returns information (mappings, settings, aliases) about one or more indices.
|
19
|
+
|
20
|
+
Args:
|
21
|
+
index: Name of the index
|
22
|
+
"""
|
23
|
+
return self.search_client.get_index(index=index)
|
24
|
+
|
25
|
+
@mcp.tool()
|
26
|
+
def create_index(index: str, body: Optional[Dict] = None) -> Dict:
|
27
|
+
"""
|
28
|
+
Create a new index.
|
29
|
+
|
30
|
+
Args:
|
31
|
+
index: Name of the index
|
32
|
+
body: Optional index configuration including mappings and settings
|
33
|
+
"""
|
34
|
+
return self.search_client.create_index(index=index, body=body)
|
35
|
+
|
36
|
+
@mcp.tool()
|
37
|
+
def delete_index(index: str) -> Dict:
|
38
|
+
"""
|
39
|
+
Delete an index.
|
40
|
+
|
41
|
+
Args:
|
42
|
+
index: Name of the index
|
43
|
+
"""
|
44
|
+
return self.search_client.delete_index(index=index)
|
src/tools/register.py
ADDED
@@ -0,0 +1,41 @@
|
|
1
|
+
import logging
|
2
|
+
from typing import List, Type
|
3
|
+
|
4
|
+
from fastmcp import FastMCP
|
5
|
+
|
6
|
+
from src.clients import SearchClient
|
7
|
+
from src.clients.exceptions import with_exception_handling
|
8
|
+
|
9
|
+
class ToolsRegister:
|
10
|
+
"""Class to handle registration of MCP tools."""
|
11
|
+
|
12
|
+
def __init__(self, logger: logging.Logger, search_client: SearchClient, mcp: FastMCP):
|
13
|
+
"""
|
14
|
+
Initialize the tools register.
|
15
|
+
|
16
|
+
Args:
|
17
|
+
logger: Logger instance
|
18
|
+
search_client: Search client instance
|
19
|
+
mcp: FastMCP instance
|
20
|
+
"""
|
21
|
+
self.logger = logger
|
22
|
+
self.search_client = search_client
|
23
|
+
self.mcp = mcp
|
24
|
+
|
25
|
+
def register_all_tools(self, tool_classes: List[Type]):
|
26
|
+
"""
|
27
|
+
Register all tools with the MCP server.
|
28
|
+
|
29
|
+
Args:
|
30
|
+
tool_classes: List of tool classes to register
|
31
|
+
"""
|
32
|
+
for tool_class in tool_classes:
|
33
|
+
self.logger.info(f"Registering tools from {tool_class.__name__}")
|
34
|
+
tool_instance = tool_class(self.search_client)
|
35
|
+
|
36
|
+
# Set logger and client attributes
|
37
|
+
tool_instance.logger = self.logger
|
38
|
+
tool_instance.search_client = self.search_client
|
39
|
+
|
40
|
+
# Register tools with automatic exception handling
|
41
|
+
with_exception_handling(tool_instance, self.mcp)
|