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

Add audit mcp tools

Browse Source 
Signed-off-by: Bryan Frimin <bryan@getprobo.com>
This commit is contained in:
Bryan Frimin
2025-11-24 09:46:17 +01:00
parent 2d722700d6
commit d8abb230b8
5 changed files with 560 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -49,6 +49,10 @@ type ResolverInterface interface {
GetContinualImprovementTool(ctx context.Context, req *mcp.CallToolRequest, input *types.GetContinualImprovementInput) (*mcp.CallToolResult, types.GetContinualImprovementOutput, error)
AddContinualImprovementTool(ctx context.Context, req *mcp.CallToolRequest, input *types.AddContinualImprovementInput) (*mcp.CallToolResult, types.AddContinualImprovementOutput, error)
UpdateContinualImprovementTool(ctx context.Context, req *mcp.CallToolRequest, input *types.UpdateContinualImprovementInput) (*mcp.CallToolResult, types.UpdateContinualImprovementOutput, error)
ListAuditsTool(ctx context.Context, req *mcp.CallToolRequest, input *types.ListAuditsInput) (*mcp.CallToolResult, types.ListAuditsOutput, error)
GetAuditTool(ctx context.Context, req *mcp.CallToolRequest, input *types.GetAuditInput) (*mcp.CallToolResult, types.GetAuditOutput, error)
AddAuditTool(ctx context.Context, req *mcp.CallToolRequest, input *types.AddAuditInput) (*mcp.CallToolResult, types.AddAuditOutput, error)
UpdateAuditTool(ctx context.Context, req *mcp.CallToolRequest, input *types.UpdateAuditInput) (*mcp.CallToolResult, types.UpdateAuditOutput, error)
}
// New creates a new MCP server instance with all handlers registered.
@@ -458,4 +462,44 @@ func registerToolHandlers(server *mcp.Server, resolver ResolverInterface) {
},
resolver.UpdateContinualImprovementTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "listAudits",
Description: "List all audits for the organization",
InputSchema: types.ListAuditsToolInputSchema,
OutputSchema: types.ListAuditsToolOutputSchema,
},
resolver.ListAuditsTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "getAudit",
Description: "Get an audit by ID",
InputSchema: types.GetAuditToolInputSchema,
OutputSchema: types.GetAuditToolOutputSchema,
},
resolver.GetAuditTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "addAudit",
Description: "Add a new audit to the organization",
InputSchema: types.AddAuditToolInputSchema,
OutputSchema: types.AddAuditToolOutputSchema,
},
resolver.AddAuditTool,
)
mcp.AddTool(
server,
&mcp.Tool{
Name: "updateAudit",
Description: "Update an existing audit",
InputSchema: types.UpdateAuditToolInputSchema,
OutputSchema: types.UpdateAuditToolOutputSchema,
},
resolver.UpdateAuditTool,
)
}