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

Add vendor risk assessment to mcp

Browse Source 
Signed-off-by: Sacha Al Himdani <sacha@getprobo.com>
This commit is contained in:
Sacha Al Himdani
2026-02-25 13:38:36 +01:00
parent 214672b6b5
commit ff62eb4a80
5 changed files with 361 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -21,6 +21,8 @@ type ResolverInterface interface {
RemoveUserTool(ctx context.Context, req *mcp.CallToolRequest, input *types.RemoveUserInput) (*mcp.CallToolResult, types.RemoveUserOutput, error)
AddVendorTool(ctx context.Context, req *mcp.CallToolRequest, input *types.AddVendorInput) (*mcp.CallToolResult, types.AddVendorOutput, error)
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)
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)
@@ -282,6 +284,30 @@ func registerToolHandlers(server *mcp.Server, resolver ResolverInterface) {
},
resolver.UpdateVendorTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "listVendorRiskAssessments",
Description: "List all risk assessments for a vendor",
InputSchema: types.ListVendorRiskAssessmentsToolInputSchema,
OutputSchema: types.ListVendorRiskAssessmentsToolOutputSchema,
Annotations: &mcp.ToolAnnotations{
ReadOnlyHint: true,
IdempotentHint: true,
},
},
resolver.ListVendorRiskAssessmentsTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "addVendorRiskAssessment",
Description: "Add a new risk assessment for a vendor",
InputSchema: types.AddVendorRiskAssessmentToolInputSchema,
OutputSchema: types.AddVendorRiskAssessmentToolOutputSchema,
},
resolver.AddVendorRiskAssessmentTool,
)
mcp.AddTool(
server,
&mcp.Tool{