Skip to content

Commit b33aefc

Browse files
Michael Christensenim-michaelc
authored andcommitted
Implement generic exception handling messages
1 parent 861ae66 commit b33aefc

File tree

1 file changed

+6
-6
lines changed
  • src/amazon-keyspaces-mcp-server/awslabs/amazon_keyspaces_mcp_server

1 file changed

+6
-6
lines changed

src/amazon-keyspaces-mcp-server/awslabs/amazon_keyspaces_mcp_server/server.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,7 @@ async def _handle_list_keyspaces(self, ctx: Optional[Any] = None) -> str:
217217
return formatted_text
218218
except Exception as e:
219219
logger.error(f'Error listing keyspaces: {str(e)}')
220-
raise SchemaError(f'Error listing keyspaces: {str(e)}') from e
220+
raise SchemaError('Unable to retrieve keyspace information') from e
221221

222222
async def _handle_list_tables(self, keyspace: str, ctx: Optional[Context] = None) -> str:
223223
"""Handle the listTables tool."""
@@ -246,7 +246,7 @@ async def _handle_list_tables(self, keyspace: str, ctx: Optional[Context] = None
246246
return formatted_text
247247
except Exception as e:
248248
logger.error(f'Error listing tables: {str(e)}')
249-
raise SchemaError(f'Error listing tables: {str(e)}') from e
249+
raise SchemaError('Unable to retrieve table information') from e
250250

251251
async def _handle_describe_keyspace(
252252
self, keyspace: str, ctx: Optional[Context] = None
@@ -290,7 +290,7 @@ async def _handle_describe_keyspace(
290290
return formatted_text
291291
except Exception as e:
292292
logger.error(f'Error describing keyspace: {str(e)}')
293-
raise SchemaError(f'Error describing keyspace: {str(e)}') from e
293+
raise SchemaError('Unable to retrieve keyspace details') from e
294294

295295
async def _handle_describe_table(
296296
self, keyspace: str, table: str, ctx: Optional[Context] = None
@@ -359,7 +359,7 @@ async def _handle_describe_table(
359359
return formatted_text
360360
except Exception as e:
361361
logger.error(f'Error describing table: {str(e)}')
362-
raise SchemaError(f'Error describing table: {str(e)}') from e
362+
raise SchemaError('Unable to retrieve table details') from e
363363

364364
async def _handle_execute_query(
365365
self, keyspace: str, query: str, ctx: Optional[Context] = None
@@ -434,7 +434,7 @@ async def _handle_execute_query(
434434
raise
435435
except Exception as e:
436436
logger.error(f'Error executing query: {str(e)}')
437-
raise QueryExecutionError(f'Error executing query: {str(e)}') from e
437+
raise QueryExecutionError('Unable to execute query') from e
438438

439439
async def _handle_analyze_query_performance(
440440
self, keyspace: str, query: str, ctx: Optional[Context] = None
@@ -473,7 +473,7 @@ async def _handle_analyze_query_performance(
473473
raise
474474
except Exception as e:
475475
logger.error(f'Error analyzing query: {str(e)}')
476-
raise QueryExecutionError(f'Error analyzing query: {str(e)}') from e
476+
raise QueryExecutionError('Unable to analyze query') from e
477477

478478

479479
def main():

0 commit comments

Comments
 (0)