rohit/probo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add delete vendor tool on mcp

Browse Source 
Signed-off-by: Sacha Al Himdani <sacha@getprobo.com>
This commit is contained in:
Sacha Al Himdani
2026-03-03 15:10:07 +01:00
parent c50c6f55b2
commit e0de1548d6
4 changed files with 70 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
pkg/server/api/mcp/v1/server/server.go
View File

@@ -23,6 +23,7 @@ type ResolverInterface interface {
UpdateVendorTool(ctx context.Context, req *mcp.CallToolRequest, input *types.UpdateVendorInput) (*mcp.CallToolResult, types.UpdateVendorOutput, error)
ListVendorRiskAssessmentsTool(ctx context.Context, req *mcp.CallToolRequest, input *types.ListVendorRiskAssessmentsInput) (*mcp.CallToolResult, types.ListVendorRiskAssessmentsOutput, error)
AddVendorRiskAssessmentTool(ctx context.Context, req *mcp.CallToolRequest, input *types.AddVendorRiskAssessmentInput) (*mcp.CallToolResult, types.AddVendorRiskAssessmentOutput, error)
DeleteVendorTool(ctx context.Context, req *mcp.CallToolRequest, input *types.DeleteVendorInput) (*mcp.CallToolResult, types.DeleteVendorOutput, error)
ListRisksTool(ctx context.Context, req *mcp.CallToolRequest, input *types.ListRisksInput) (*mcp.CallToolResult, types.ListRisksOutput, error)
GetRiskTool(ctx context.Context, req *mcp.CallToolRequest, input *types.GetRiskInput) (*mcp.CallToolResult, types.GetRiskOutput, error)
AddRiskTool(ctx context.Context, req *mcp.CallToolRequest, input *types.AddRiskInput) (*mcp.CallToolResult, types.AddRiskOutput, error)
@@ -307,6 +308,19 @@ func registerToolHandlers(server *mcp.Server, resolver ResolverInterface) {
},
resolver.AddVendorRiskAssessmentTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "deleteVendor",
Description: "Delete a vendor",
InputSchema: types.DeleteVendorToolInputSchema,
OutputSchema: types.DeleteVendorToolOutputSchema,
Annotations: &mcp.ToolAnnotations{
DestructiveHint: boolPtr(true),
},
},
resolver.DeleteVendorTool,
)
mcp.AddTool(
server,
&mcp.Tool{