Heat stress is one of the primary environmental factors that harm both the productivity and health of buffaloes. The current study was conducted to estimate the threshold of temperature humidity index (THI)1 and genetic features for milk yield of first-lactation Mehsana buffaloes using an univariate repeatability test-day model. The data included 130,475 first lactation test-day milk yield (FLTDMY) records of 13,887 Mehsana buffaloes and the daily temperature and humidity. The statistical model included herd test day as fixed effects, days-in-milk (DIM) classes, age of the animal, as well as random factors such as the additive genetic effect (AGE) of animal in general conditions (intercept), AGE of the buffaloes subjected to heat stress (slope), permanent environmental effect of animal in general conditions (intercept), permanent environmental effect of animal under heat stress conditions (slope) and random residual effect. It was expected that the general effects and the heat-tolerance effects would be correlated, represented by the present investigation's repeatability models. The variance components of FLTDMY in the present study were computed using the REML method. The threshold for THI was 78. At the THI below the threshold, the heritability estimated for the FLTDMY trait was 0.29, and the additive genetic variance (AGV) for heat stress conditions was 0. At THI of 83, AGV for heat stress conditions was highest for FLTDMY. The genetic correlation of general AGE to heat-tolerant AGE was −0.40. The results indicated that a consistent selection for milk production, avoiding the thermal tolerance, may diminish the thermal tolerance capacity of Mehsana buffaloes.