apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-23 20:21:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Fix MCP delete tool params

Browse source
This commit is contained in:
Cyber MacGeddon 2025-07-16 23:08:20 +01:00
parent a32875647b
commit 7ed484f222
1 changed files with 16 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

32
trustgraph-cli/scripts/tg-delete-mcp-tool
View file

@ -2,7 +2,7 @@
""" """
Deletes MCP (Model Control Protocol) tools from the TrustGraph system. Deletes MCP (Model Control Protocol) tools from the TrustGraph system.
Removes MCP tool configurations by name from the 'mcp' configuration group. Removes MCP tool configurations by ID from the 'mcp' configuration group.
""" """
import argparse import argparse
@ -14,7 +14,7 @@ default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/')
def delete_mcp_tool( def delete_mcp_tool(
url : str, url : str,
name : str, id : str,
): ):
api = Api(url).config() api = Api(url).config()
@ -22,28 +22,28 @@ def delete_mcp_tool(
# Check if the tool exists first # Check if the tool exists first
try: try:
values = api.get([ values = api.get([
ConfigKey(type="mcp", key=name) ConfigKey(type="mcp", key=id)
]) ])
if not values or not values[0].value: if not values or not values[0].value:
print(f"MCP tool '{name}' not found.") print(f"MCP tool '{id}' not found.")
return False return False
except Exception as e: except Exception as e:
print(f"MCP tool '{name}' not found.") print(f"MCP tool '{id}' not found.")
return False return False
# Delete the MCP tool configuration from the 'mcp' group # Delete the MCP tool configuration from the 'mcp' group
try: try:
api.delete([ api.delete([
ConfigKey(type="mcp", key=name) ConfigKey(type="mcp", key=id)
]) ])
print(f"MCP tool '{name}' deleted successfully.") print(f"MCP tool '{id}' deleted successfully.")
return True return True
except Exception as e: except Exception as e:
print(f"Error deleting MCP tool '{name}': {e}") print(f"Error deleting MCP tool '{id}': {e}")
return False return False
def main(): def main():
@ -56,9 +56,9 @@ def main():
Once deleted, the tool will no longer be available for use. Once deleted, the tool will no longer be available for use.
Examples: Examples:
%(prog)s --name weather %(prog)s --id weather
%(prog)s --name calculator %(prog)s --id calculator
%(prog)s --api-url http://localhost:9000/ --name file-reader %(prog)s --api-url http://localhost:9000/ --id file-reader
''').strip(), ''').strip(),
formatter_class=argparse.RawDescriptionHelpFormatter formatter_class=argparse.RawDescriptionHelpFormatter
) )
@ -70,21 +70,21 @@ def main():
) )
parser.add_argument( parser.add_argument(
'--name', '--id',
required=True, required=True,
help='MCP tool name to delete', help='MCP tool ID to delete',
) )
args = parser.parse_args() args = parser.parse_args()
try: try:
if not args.name: if not args.id:
raise RuntimeError("Must specify --name for MCP tool to delete") raise RuntimeError("Must specify --id for MCP tool to delete")
delete_mcp_tool( delete_mcp_tool(
url=args.api_url, url=args.api_url,
name=args.name id=args.id
) )
except Exception as e: except Exception as e: