GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-09-05 15:37:35 +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
mcphost/host.go
+8 -7
View File
@@ -15,13 +15,14 @@ import (
mcpp "github.com/cloudwego/eino-ext/components/tool/mcp" mcpp "github.com/cloudwego/eino-ext/components/tool/mcp"
"github.com/cloudwego/eino/components/tool" "github.com/cloudwego/eino/components/tool"
"github.com/cloudwego/eino/schema" "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/client"
"github.com/mark3labs/mcp-go/mcp" "github.com/mark3labs/mcp-go/mcp"
"github.com/pkg/errors" "github.com/pkg/errors"
"github.com/rs/zerolog/log" "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 // MCPHost manages MCP server connections and tools
@@ -71,7 +72,7 @@ func NewMCPHost(configPath string, withUIXT bool) (*MCPHost, error) {
// Initialize MCP servers // Initialize MCP servers
if err := host.InitServers(ctx); err != nil { if err := host.InitServers(ctx); err != nil {
cancel() 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 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 { 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 return nil
@@ -118,7 +119,7 @@ func (h *MCPHost) connectToServer(ctx context.Context, serverName string, config
// Close existing connection if any // Close existing connection if any
if existing, exists := h.connections[serverName]; exists { if existing, exists := h.connections[serverName]; exists {
if err := existing.Client.Close(); err != nil { 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) delete(h.connections, serverName)
} }
@@ -153,7 +154,7 @@ func (h *MCPHost) connectToServer(ctx context.Context, serverName string, config
} }
if err != nil { if err != nil {
return fmt.Errorf("failed to create client: %w", err) return errors.Wrapf(err, "failed to create client")
} }
// initialize client with timeout // initialize client with timeout
server/main.go
+5 -4
View File
@@ -9,11 +9,11 @@ import (
"syscall" "syscall"
"time" "time"
"github.com/httprunner/httprunner/v5/mcphost"
"github.com/httprunner/httprunner/v5/uixt"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"github.com/rs/zerolog/log" "github.com/rs/zerolog/log"
"github.com/httprunner/httprunner/v5/mcphost"
"github.com/httprunner/httprunner/v5/uixt"
) )
func NewRouter() *Router { func NewRouter() *Router {
@@ -32,7 +32,8 @@ type Router struct {
func (r *Router) InitMCPHost(configPath string) error { func (r *Router) InitMCPHost(configPath string) error {
mcpHost, err := mcphost.NewMCPHost(configPath, true) mcpHost, err := mcphost.NewMCPHost(configPath, true)
if err != nil { 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 return err
} }
r.mcpHost = mcpHost r.mcpHost = mcpHost