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

Add risks

Browse Source 
Signed-off-by: Bryan Frimin <bryan@getprobo.com>
This commit is contained in:
Bryan Frimin
2025-11-23 20:58:23 +01:00
parent 0efecd232e
commit fa97a8e9c6
9 changed files with 666 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -17,6 +17,10 @@ type ResolverInterface interface {
UpdateVendorTool(ctx context.Context, req *mcp.CallToolRequest, input *types.UpdateVendorInput) (*mcp.CallToolResult, types.UpdateVendorOutput, error)
GetPeopleTool(ctx context.Context, req *mcp.CallToolRequest, input *types.GetPeopleInput) (*mcp.CallToolResult, types.GetPeopleOutput, error)
AddPeopleTool(ctx context.Context, req *mcp.CallToolRequest, input *types.AddPeopleInput) (*mcp.CallToolResult, types.AddPeopleOutput, 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)
UpdateRiskTool(ctx context.Context, req *mcp.CallToolRequest, input *types.UpdateRiskInput) (*mcp.CallToolResult, types.UpdateRiskOutput, error)
}
// New creates a new MCP server instance with all handlers registered.
@@ -106,4 +110,44 @@ func registerToolHandlers(server *mcp.Server, resolver ResolverInterface) {
},
resolver.AddPeopleTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "listRisks",
Description: "List all risks for the organization",
InputSchema: types.ListRisksToolInputSchema,
OutputSchema: types.ListRisksToolOutputSchema,
},
resolver.ListRisksTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "getRisk",
Description: "Get a risk by ID",
InputSchema: types.GetRiskToolInputSchema,
OutputSchema: types.GetRiskToolOutputSchema,
},
resolver.GetRiskTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "addRisk",
Description: "Add a new risk to the organization",
InputSchema: types.AddRiskToolInputSchema,
OutputSchema: types.AddRiskToolOutputSchema,
},
resolver.AddRiskTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "updateRisk",
Description: "Update an existing risk",
InputSchema: types.UpdateRiskToolInputSchema,
OutputSchema: types.UpdateRiskToolOutputSchema,
},
resolver.UpdateRiskTool,
)
}