mirror of
https://github.com/httprunner/httprunner.git
synced 2026-06-26 01:51:29 +08:00
35 lines
680 B
Go
35 lines
680 B
Go
package server
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type ToolRequest struct {
|
|
ServerName string `json:"mcp_server"`
|
|
ToolName string `json:"tool_name"`
|
|
Args map[string]interface{} `json:"args"`
|
|
}
|
|
|
|
func (r *Router) invokeToolHandler(c *gin.Context) {
|
|
if r.mcpHub == nil {
|
|
RenderError(c, errors.New("mcp hub not initialized"))
|
|
return
|
|
}
|
|
|
|
var req ToolRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
RenderError(c, err)
|
|
return
|
|
}
|
|
|
|
result, err := r.mcpHub.InvokeTool(c.Request.Context(),
|
|
req.ServerName, req.ToolName, req.Args)
|
|
if err != nil {
|
|
RenderError(c, err)
|
|
return
|
|
}
|
|
RenderSuccess(c, result)
|
|
}
|