ci: add session handling to agent

Author: rahulguptajss

integration/test/tools_test.go

@@ -154,7 +154,7 @@ func (a *Agent) callMCPTool(ctx context.Context, toolName string, arguments map[
 		Name:      toolName,
 		Arguments: arguments,
 	})
-	if err != nil && strings.Contains(err.Error(), "connection closed") {
+	if err != nil && errors.Is(err, mcp.ErrConnectionClosed) {
 		slog.Warn("MCP session dropped, reconnecting", slog.Any("error", err))
 		if reconnErr := a.reconnect(ctx); reconnErr != nil {
 			return "", fmt.Errorf("failed to call tool: %w (reconnect failed: %w)", err, reconnErr)