mirror of
https://github.com/mudler/LocalAI.git
synced 2026-10-09 22:54:42 -04:00
mudler-agent
cd604b5c5b
Fence model load jobs by generation, lease them on the database clock, bound the work on the worker with operations and a process-group watchdog, and add one stop path with a load-cancel API. See the pull request for the design, the rolling upgrade notes and the test evidence. Assisted-by: Claude Code:claude-sonnet-5-5 Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
65 lines
3.0 KiB
Go
65 lines
3.0 KiB
Go
package schema
|
|
|
|
import "time"
|
|
|
|
// ModelLoadingStatus describes a cold load that is still in progress. In
|
|
// distributed mode a model can take tens of minutes to stage onto a worker,
|
|
// which is far longer than a request may be held; a caller that runs out of
|
|
// wait budget gets this instead of an anonymous hang or a misleading error.
|
|
type ModelLoadingStatus struct {
|
|
// JobID names the load attempt. A cancel must quote it: it is the
|
|
// precondition that keeps a cancel from hitting a replacement attempt.
|
|
JobID string `json:"job_id,omitempty"`
|
|
// LeaseExpiresIn is the seconds left on the owner's lease. It is negative
|
|
// when the lease already ran out, which means the owner is gone.
|
|
LeaseExpiresIn *int `json:"lease_expires_in,omitempty"`
|
|
// CancelRequested is true when an administrator cancelled the attempt.
|
|
CancelRequested bool `json:"cancel_requested,omitempty"`
|
|
// LastError is the cause of a failed attempt.
|
|
LastError string `json:"last_error,omitempty"`
|
|
// Stopping is true while the remote work of a failed attempt is not yet
|
|
// confirmed ended. StopDeadline is when the model is released regardless.
|
|
Stopping bool `json:"stopping,omitempty"`
|
|
StopDeadline *time.Time `json:"stop_deadline,omitempty"`
|
|
// RetryAfter is the seconds until a new load may start, for a failed attempt.
|
|
RetryAfter int `json:"retry_after,omitempty"`
|
|
Model string `json:"model"`
|
|
State string `json:"state"`
|
|
Node string `json:"node,omitempty"`
|
|
Progress float64 `json:"progress"`
|
|
BytesSent int64 `json:"bytes_sent"`
|
|
TotalBytes int64 `json:"total_bytes"`
|
|
FileIndex int `json:"file_index"`
|
|
TotalFiles int `json:"total_files"`
|
|
// ETASeconds is omitted rather than guessed until enough bytes have moved
|
|
// for the observed rate to mean anything. A confidently wrong ETA on a
|
|
// twenty-minute wait is worse than none.
|
|
ETASeconds int `json:"eta_seconds,omitempty"`
|
|
}
|
|
|
|
// ModelLoadingResponse is the 503 body served while a model is still loading.
|
|
// The `error` envelope keeps OpenAI-client compatibility; `loading` is additive,
|
|
// so existing clients ignore it and load-aware ones can render real progress.
|
|
type ModelLoadingResponse struct {
|
|
Error *APIError `json:"error,omitempty"`
|
|
Loading *ModelLoadingStatus `json:"loading,omitempty"`
|
|
}
|
|
|
|
// ModelLoadCancelRequest is the body of POST /api/models/{id}/load-cancel.
|
|
// JobID is the exact attempt to cancel, as load-status reports it.
|
|
type ModelLoadCancelRequest struct {
|
|
JobID string `json:"job_id"`
|
|
}
|
|
|
|
// ModelLoadCancelResponse reports what a cancel did. State is "stopped" (the
|
|
// worker confirmed the work ended), "stopping" (the cancel is recorded and the
|
|
// stop is pending; RetryAfter says when the model is released regardless) or
|
|
// "gone" (no such load exists any more). CurrentJobID is set on a conflict.
|
|
type ModelLoadCancelResponse struct {
|
|
Model string `json:"model"`
|
|
JobID string `json:"job_id"`
|
|
State string `json:"state"`
|
|
RetryAfter int `json:"retry_after,omitempty"`
|
|
CurrentJobID string `json:"current_job_id,omitempty"`
|
|
}
|