mirror of
https://github.com/mostlygeek/llama-swap.git
synced 2026-06-09 06:46:34 +02:00
Benson Wong
02e015fa49
This is a huge backend change that essentially started with rewriting the concurrency handling for processes and blew up to a refactor of the entire application. In short these are the improvements: **Better state and life cycle management:** Life cycle management of processes has always been the trickiest part of the code. Juggling mutex locks between multiple locations to reduce race conditions was complex. Too complex for my feeble brain to build a simple mental model around as llama-swap gained more features. All of that has been refactored. Most of the locks are gone, replaced with a single run() that owns all state changes. There is one place to start from now to understand and extend routing logic. The improved life cycle management makes it easier to implement more complex swap optimization strategies in the future like #727. **Collation of requests:** llama-swap previously handled requests and swapping in the order they came in. For example requests for models in this order ABCABC would result in 5 swaps. Now those requests are handled in this order AABBCC. The result is less time waiting for swap under a high churn request queue. This fixes #588 #612. A possible future enhancement is to support a starvation parameter so swap can be forced when models have been waiting too long. **Shared base implementation for groups and swap matrix:** During the refactor it became clear that much of the swapping logic was shared between these two implementations. That is not surprising considering the swap matrix was added many moons after groups. Now they share a common base and their specific swap strategies are implemented into the swapPlanner interface. Requests for bespoke or specific swapping scenarios is a common theme in the issues. Now users can implement whatever bespoke and weird swapping strategy they want in their own fork. Just ask your agent of choice to implement swapPlanner. I'll still remaining more conservative on what actually lands in core llama-swap and will continue to evaluate PRs if the changes is good for everyone or just one specific use case. **AI / Agentic Disclosure:** I paid very close attention to the low level swap concurrency design and implementation. It's important to keep that essential part reliable, boring and no surprises. Backwards compatibility was also maintained, even the one way non-exclusive group model loading behaviour that people have rightly pointed out be a weird design decision. With the underlying swap core done the web server, api and UI sitting on top were largely ported over with Claude Code and Opus 4.7 in multiple phases. If you're curious I kept the changes in docs/newrouter-todo.md. I did several passes to make sure things weren't left behind. However, even frontier LLMs at the time of this PR still make small decisions that don't make a lot of sense. They get shit wrong all the time, just in small subtle way. That said, there's likely to be some new bugs introduced with this massive refactor. I'm fairly confident that there's no major architectural flaws that would cause goal seeking agents to make dumb, ugly code decisions. For a little while the legacy llama-swap will be available under cmd/legacy/llama-swap. The plan is to eventually delete that entry point as well as the proxy package. On a bit of a personal note, this PR is exciting and a bit sad for me. I hand wrote much of the original code and this PR ultimately replaces much of it. While the old code served as a good reference for the agent to implement the new stuff it still a bit sad to eventually delete it all.
83 lines
1.9 KiB
Go
83 lines
1.9 KiB
Go
package process
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/mostlygeek/llama-swap/internal/config"
|
|
"github.com/mostlygeek/llama-swap/internal/event"
|
|
"github.com/mostlygeek/llama-swap/internal/shared"
|
|
)
|
|
|
|
func TestProcessCommand_EmitsStateChangeEvents(t *testing.T) {
|
|
skipIfNoSimpleResponder(t)
|
|
|
|
var mu sync.Mutex
|
|
var transitions []shared.ProcessStateChangeEvent
|
|
cancel := event.On(func(e shared.ProcessStateChangeEvent) {
|
|
if e.ProcessName != t.Name() {
|
|
return
|
|
}
|
|
mu.Lock()
|
|
transitions = append(transitions, e)
|
|
mu.Unlock()
|
|
})
|
|
defer cancel()
|
|
|
|
cmd, port := simpleResponderCmd(t, "-silent", "-respond hello")
|
|
p := newProcessCommand(t, config.ModelConfig{
|
|
Cmd: cmd,
|
|
Proxy: fmt.Sprintf("http://127.0.0.1:%d", port),
|
|
CheckEndpoint: "/health",
|
|
HealthCheckTimeout: 10,
|
|
})
|
|
|
|
runErr := runAsync(t, p)
|
|
if err := p.Stop(testStopTimeout); err != nil {
|
|
t.Fatalf("Stop: %v", err)
|
|
}
|
|
<-runErr
|
|
|
|
// Events are delivered asynchronously; give the dispatcher a moment.
|
|
deadline := time.Now().Add(time.Second)
|
|
for time.Now().Before(deadline) {
|
|
mu.Lock()
|
|
n := len(transitions)
|
|
mu.Unlock()
|
|
if n >= 4 {
|
|
break
|
|
}
|
|
time.Sleep(testPollInterval)
|
|
}
|
|
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
|
|
for _, e := range transitions {
|
|
if e.OldState == e.NewState {
|
|
t.Errorf("emitted no-op transition: %s -> %s", e.OldState, e.NewState)
|
|
}
|
|
}
|
|
|
|
want := []string{
|
|
string(StateStopped) + "->" + string(StateStarting),
|
|
string(StateStarting) + "->" + string(StateReady),
|
|
string(StateReady) + "->" + string(StateStopping),
|
|
string(StateStopping) + "->" + string(StateStopped),
|
|
}
|
|
got := make([]string, len(transitions))
|
|
for i, e := range transitions {
|
|
got[i] = e.OldState + "->" + e.NewState
|
|
}
|
|
if len(got) != len(want) {
|
|
t.Fatalf("transitions = %v, want %v", got, want)
|
|
}
|
|
for i := range want {
|
|
if got[i] != want[i] {
|
|
t.Fatalf("transitions = %v, want %v", got, want)
|
|
}
|
|
}
|
|
}
|