GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-06 20:32:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

change: update error logs

Browse Source
This commit is contained in:
lilong.129
2025-06-30 16:57:41 +08:00
parent c3c2b42d34
commit 1cdf14b85a
2 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
mcphost/host.go
View File

@@ -15,13 +15,14 @@ import (
mcpp "github.com/cloudwego/eino-ext/components/tool/mcp"
"github.com/cloudwego/eino/components/tool"
"github.com/cloudwego/eino/schema"
"github.com/httprunner/httprunner/v5/internal/version"
"github.com/httprunner/httprunner/v5/uixt"
"github.com/httprunner/httprunner/v5/uixt/ai"
"github.com/mark3labs/mcp-go/client"
"github.com/mark3labs/mcp-go/mcp"
"github.com/pkg/errors"
"github.com/rs/zerolog/log"
"github.com/httprunner/httprunner/v5/internal/version"
"github.com/httprunner/httprunner/v5/uixt"
"github.com/httprunner/httprunner/v5/uixt/ai"
)
// MCPHost manages MCP server connections and tools
@@ -71,7 +72,7 @@ func NewMCPHost(configPath string, withUIXT bool) (*MCPHost, error) {
// Initialize MCP servers
if err := host.InitServers(ctx); err != nil {
cancel()
return nil, fmt.Errorf("failed to initialize MCP servers: %w", err)
return nil, errors.Wrapf(err, "failed to initialize MCP servers")
}
return host, nil
@@ -95,7 +96,7 @@ func (h *MCPHost) InitServers(ctx context.Context) error {
}
if err := h.connectToServer(ctx, name, server.Config); err != nil {
return fmt.Errorf("failed to connect to server %s: %w", name, err)
return errors.Wrapf(err, "failed to connect to server %s", name)
}
}
return nil
@@ -118,7 +119,7 @@ func (h *MCPHost) connectToServer(ctx context.Context, serverName string, config
// Close existing connection if any
if existing, exists := h.connections[serverName]; exists {
if err := existing.Client.Close(); err != nil {
return fmt.Errorf("failed to close existing connection: %w", err)
return errors.Wrapf(err, "failed to close existing connection")
}
delete(h.connections, serverName)
}
@@ -153,7 +154,7 @@ func (h *MCPHost) connectToServer(ctx context.Context, serverName string, config
}
if err != nil {
return fmt.Errorf("failed to create client: %w", err)
return errors.Wrapf(err, "failed to create client")
}
// initialize client with timeout

9
server/main.go
View File

@@ -9,11 +9,11 @@ import (
"syscall"
"time"
"github.com/httprunner/httprunner/v5/mcphost"
"github.com/httprunner/httprunner/v5/uixt"
"github.com/gin-gonic/gin"
"github.com/rs/zerolog/log"
"github.com/httprunner/httprunner/v5/mcphost"
"github.com/httprunner/httprunner/v5/uixt"
)
func NewRouter() *Router {
@@ -32,7 +32,8 @@ type Router struct {
func (r *Router) InitMCPHost(configPath string) error {
mcpHost, err := mcphost.NewMCPHost(configPath, true)
if err != nil {
log.Error().Err(err).Msg("init MCP host failed")
log.Error().Err(err).Str("configPath", configPath).
Msg("init MCP host failed")
return err
}
r.mcpHost = mcpHost