mirror of
https://github.com/mudler/LocalAI.git
synced 2026-09-29 01:25:03 -04:00
fix(failover): stop the preload loop from blocking the scheduler goroutine
applyFailoverWarmTargets ran on the manager's single scheduler goroutine (Sync -> Tick), so a slow or hung PreloadModelByName call froze probing and fail-back for every chain. Keep the watchdog pin synchronous but run the preload loop in its own goroutine. Adds a seam (preloadModelByName) so a unit test can substitute a blocking loader and assert the callback still returns promptly. Assisted-by: Claude:claude-opus-5-5 Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
1 parent
5ea35393fa
commit
6ab7a982f7
2 files changed
+60
-4
No files matched your search
@@ -5,13 +5,28 @@ import (
|
||||
"github.com/mudler/xlog"
|
||||
)
|
||||
|
||||
// preloadModelByName is a seam over backend.PreloadModelByName so tests can
|
||||
// substitute a controllable loader instead of touching real models/disk.
|
||||
var preloadModelByName = backend.PreloadModelByName
|
||||
|
||||
// applyFailoverWarmTargets pins warm failover targets in the watchdog and
|
||||
// loads them, so a switch does not wait for a cold load.
|
||||
//
|
||||
// SyncPinnedModelsToWatchdog runs synchronously: it is a cheap in-memory
|
||||
// update, and the pin must land before the watchdog can evict a target that
|
||||
// is about to become (or stay) a chain's active path. Preloading is not
|
||||
// cheap — it can download or load a multi-GB model — and this callback runs
|
||||
// on the failover manager's single scheduler goroutine (Sync, called from
|
||||
// Tick, called from Run), before that tick's probes fire. A slow or hung
|
||||
// load here would freeze probing and fail-back for every chain, so it runs
|
||||
// in its own goroutine instead of blocking the scheduler loop.
|
||||
func (a *Application) applyFailoverWarmTargets(warm []string) {
|
||||
a.SyncPinnedModelsToWatchdog()
|
||||
for _, name := range warm {
|
||||
if _, err := backend.PreloadModelByName(a.ApplicationConfig().Context, a.ModelConfigLoader(), a.ModelLoader(), a.ApplicationConfig(), name); err != nil {
|
||||
xlog.Warn("failover: could not preload warm target", "model", name, "error", err)
|
||||
go func() {
|
||||
for _, name := range warm {
|
||||
if _, err := preloadModelByName(a.ApplicationConfig().Context, a.ModelConfigLoader(), a.ModelLoader(), a.ApplicationConfig(), name); err != nil {
|
||||
xlog.Warn("failover: could not preload warm target", "model", name, "error", err)
|
||||
}
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
@@ -0,0 +1,41 @@
|
||||
package application
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"github.com/mudler/LocalAI/core/config"
|
||||
"github.com/mudler/LocalAI/pkg/model"
|
||||
. "github.com/onsi/ginkgo/v2"
|
||||
. "github.com/onsi/gomega"
|
||||
)
|
||||
|
||||
var _ = Describe("applyFailoverWarmTargets", func() {
|
||||
It("returns promptly even when the preload call blocks", func() {
|
||||
// Guards against a regression to a synchronous preload loop: onWarm
|
||||
// runs on the failover manager's single scheduler goroutine, so a
|
||||
// blocking loader here must not block the caller.
|
||||
started := make(chan struct{})
|
||||
release := make(chan struct{})
|
||||
orig := preloadModelByName
|
||||
preloadModelByName = func(ctx context.Context, cl *config.ModelConfigLoader, ml *model.ModelLoader, appConfig *config.ApplicationConfig, name string) ([]string, error) {
|
||||
close(started)
|
||||
<-release // never released within the test's timeout
|
||||
return nil, nil
|
||||
}
|
||||
DeferCleanup(func() { preloadModelByName = orig })
|
||||
DeferCleanup(func() { close(release) })
|
||||
|
||||
app := &Application{applicationConfig: &config.ApplicationConfig{Context: context.Background()}}
|
||||
|
||||
callReturned := make(chan struct{})
|
||||
go func() {
|
||||
defer GinkgoRecover()
|
||||
app.applyFailoverWarmTargets([]string{"warm-a"})
|
||||
close(callReturned)
|
||||
}()
|
||||
|
||||
Eventually(callReturned, time.Second).Should(BeClosed(), "applyFailoverWarmTargets must not wait on the preload goroutine")
|
||||
Eventually(started, time.Second).Should(BeClosed(), "the preload goroutine should still run in the background")
|
||||
})
|
||||
})
|
||||
Reference in new issue
Block a user