feat(mcp): add scheduling HTTP client support

Assisted-by: Hephaestus:openai/gpt-5.5
Signed-off-by: Owen Adirah <owenadira@gmail.com>

Author: owezzy

pkg/mcp/localaitools/httpapi/client.go

@@ -483,6 +483,49 @@ func (c *Client) ListNodes(ctx context.Context) ([]localaitools.Node, error) {
 	return out, nil
 }
 
+func (c *Client) ListScheduling(ctx context.Context) ([]localaitools.ModelSchedulingConfig, error) {
+	var out []localaitools.ModelSchedulingConfig
+	if err := c.do(ctx, http.MethodGet, routeScheduling, nil, &out); err != nil {
+		if errors.Is(err, ErrHTTPNotFound) {
+			return []localaitools.ModelSchedulingConfig{}, nil
+		}
+		return nil, err
+	}
+	return out, nil
+}
+
+func (c *Client) GetScheduling(ctx context.Context, modelName string) (*localaitools.ModelSchedulingConfig, error) {
+	if modelName == "" {
+		return nil, errors.New("model_name is required")
+	}
+	var out localaitools.ModelSchedulingConfig
+	if err := c.do(ctx, http.MethodGet, routeModelScheduling(modelName), nil, &out); err != nil {
+		if errors.Is(err, ErrHTTPNotFound) {
+			return nil, nil
+		}
+		return nil, err
+	}
+	return &out, nil
+}
+
+func (c *Client) SetScheduling(ctx context.Context, req localaitools.SetSchedulingRequest) (*localaitools.ModelSchedulingConfig, error) {
+	if req.ModelName == "" {
+		return nil, errors.New("model_name is required")
+	}
+	var out localaitools.ModelSchedulingConfig
+	if err := c.do(ctx, http.MethodPost, routeScheduling, req, &out); err != nil {
+		return nil, err
+	}
+	return &out, nil
+}
+
+func (c *Client) DeleteScheduling(ctx context.Context, modelName string) error {
+	if modelName == "" {
+		return errors.New("model_name is required")
+	}
+	return c.do(ctx, http.MethodDelete, routeModelScheduling(modelName), nil, nil)
+}
+
 func (c *Client) SetNodeVRAMBudget(ctx context.Context, nodeID, budget string) error {
 	// PUT with an empty value clears the override server-side (Task 9), so we
 	// use PUT uniformly rather than switching to DELETE for the clear case.

pkg/mcp/localaitools/httpapi/client_test.go

@@ -84,6 +84,34 @@ func fakeLocalAI() *httptest.Server {
 		})
 	})
 
+	mux.HandleFunc("/api/nodes/scheduling", func(w http.ResponseWriter, r *http.Request) {
+		switch r.Method {
+		case http.MethodGet:
+			_ = json.NewEncoder(w).Encode([]map[string]any{{"model_name": "qwen", "min_replicas": 1, "max_replicas": 2}})
+		case http.MethodPost:
+			var body map[string]any
+			if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
+				http.Error(w, err.Error(), http.StatusBadRequest)
+				return
+			}
+			body["id"] = "sched-1"
+			_ = json.NewEncoder(w).Encode(body)
+		default:
+			http.Error(w, "method", http.StatusMethodNotAllowed)
+		}
+	})
+
+	mux.HandleFunc("/api/nodes/scheduling/qwen", func(w http.ResponseWriter, r *http.Request) {
+		switch r.Method {
+		case http.MethodGet:
+			_ = json.NewEncoder(w).Encode(map[string]any{"model_name": "qwen", "spread_all": true})
+		case http.MethodDelete:
+			w.WriteHeader(http.StatusNoContent)
+		default:
+			http.Error(w, "method", http.StatusMethodNotAllowed)
+		}
+	})
+
 	return httptest.NewServer(mux)
 }
 
@@ -197,6 +225,35 @@ var _ = Describe("httpapi.Client against the LocalAI admin REST surface", func()
 			Expect(bs[0].Installed).To(BeTrue())
 		})
 	})
+
+	Describe("Scheduling", func() {
+		It("lists scheduling configs", func() {
+			out, err := c.ListScheduling(ctx)
+			Expect(err).ToNot(HaveOccurred())
+			Expect(out).To(HaveLen(1))
+			Expect(out[0].ModelName).To(Equal("qwen"))
+			Expect(out[0].MinReplicas).To(Equal(1))
+		})
+
+		It("gets one scheduling config", func() {
+			out, err := c.GetScheduling(ctx, "qwen")
+			Expect(err).ToNot(HaveOccurred())
+			Expect(out.ModelName).To(Equal("qwen"))
+			Expect(out.SpreadAll).To(BeTrue())
+		})
+
+		It("sets a scheduling config", func() {
+			out, err := c.SetScheduling(ctx, localaitools.SetSchedulingRequest{ModelName: "qwen", MinReplicas: 1, MaxReplicas: 2})
+			Expect(err).ToNot(HaveOccurred())
+			Expect(out.ID).To(Equal("sched-1"))
+			Expect(out.ModelName).To(Equal("qwen"))
+			Expect(out.MaxReplicas).To(Equal(2))
+		})
+
+		It("deletes a scheduling config", func() {
+			Expect(c.DeleteScheduling(ctx, "qwen")).To(Succeed())
+		})
+	})
 })
 
 var _ = Describe("Model aliases", func() {