feat(cli): add restart command

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: ganisback

README.md

@@ -157,6 +157,7 @@ All apps auto-configure to use csghub-lite's OpenAI-compatible API endpoint with
 | `csghub-lite ps` | List currently running models and their keep-alive |
 | `csghub-lite stop <model>` | Stop/unload a running model |
 | `csghub-lite serve` | Start the API server (auto-started by `run`) |
+| `csghub-lite restart` | Restart the background API server |
 | `csghub-lite pull <model>` | Download a model from CSGHub |
 | `csghub-lite list` / `ls` | List locally downloaded models |
 | `csghub-lite show <model>` | Show model details (format, size, files) |

docs/README.md

@@ -15,6 +15,7 @@ csghub-lite 是一个轻量级的本地大语言模型运行工具，基于 [CSG
 - [run](cli/run.md) — 自动下载并交互对话
 - [chat](cli/chat.md) — 与本地模型交互对话
 - [serve](cli/serve.md) — 启动 REST API 服务
+- [restart](cli/restart.md) — 重启后台 API 服务
 - [pull](cli/pull.md) — 下载模型
 - [list](cli/list.md) — 列出本地模型
 - [show](cli/show.md) — 查看模型详情

docs/cli/overview.md

@@ -26,6 +26,7 @@ csghub-lite 提供以下命令：
 |------|------|
 | [`ps`](ps.md) | 列出服务器上运行中的模型 |
 | [`stop <model>`](stop.md) | 停止/卸载服务器上运行中的模型 |
+| [`restart`](restart.md) | 重启后台 API 服务 |
 
 ## 配置与认证
 

docs/cli/restart.md

@@ -0,0 +1,26 @@
+# csghub-lite restart
+
+重启后台运行的 csghub-lite API 服务。如果当前没有可用的后台服务，也会直接启动一个新的后台服务。
+
+## 用法
+
+```bash
+csghub-lite restart
+```
+
+## 说明
+
+该命令会先尝试优雅停止现有服务，然后重新启动并等待健康检查通过。它使用当前配置中的 `listen_addr`，默认地址为 `:11435`。
+
+可用别名：
+
+- `restart-service`
+- `restart-server`
+- `reload`
+
+## 示例
+
+```bash
+# 重启默认后台服务
+csghub-lite restart
+```

internal/cli/cli_test.go

@@ -19,7 +19,7 @@ func TestNewRootCmd(t *testing.T) {
 
 	expected := []string{
 		"serve", "apps", "run MODEL", "chat MODEL", "pull NAME", "list",
-		"show MODEL", "ps", "stop MODEL", "stop-service",
+		"show MODEL", "ps", "stop MODEL", "stop-service", "restart",
 		"rm NAME", "login", "search QUERY", "config", "uninstall",
 	}
 	for _, name := range expected {

internal/cli/restart.go

@@ -0,0 +1,55 @@
+package cli
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/opencsgs/csghub-lite/internal/config"
+	"github.com/spf13/cobra"
+)
+
+var (
+	stopBackgroundServiceForRestart = stopBackgroundServiceIfRunning
+	startBackgroundServerForRestart = startBackgroundServer
+	waitForServerForRestart         = waitForServer
+)
+
+func newRestartCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:     "restart",
+		Aliases: []string{"restart-service", "restart-server", "reload"},
+		Short:   "Restart the background csghub-lite service",
+		Long:    "Restart the background csghub-lite API service, starting it if it is not already running.",
+		Args:    cobra.NoArgs,
+		RunE:    runRestart,
+	}
+	return cmd
+}
+
+func runRestart(cmd *cobra.Command, args []string) error {
+	cfg, err := config.Load()
+	if err != nil {
+		return fmt.Errorf("loading config: %w", err)
+	}
+	return restartBackgroundService(cfg)
+}
+
+func restartBackgroundService(cfg *config.Config) error {
+	if cfg == nil {
+		return fmt.Errorf("missing config")
+	}
+
+	baseURL := serverBaseURL(cfg)
+	fmt.Println("Restarting csghub-lite service...")
+	if err := stopBackgroundServiceForRestart(); err != nil {
+		return fmt.Errorf("stopping existing service: %w", err)
+	}
+	if err := startBackgroundServerForRestart(cfg); err != nil {
+		return fmt.Errorf("starting service: %w", err)
+	}
+	if err := waitForServerForRestart(baseURL, 15*time.Second); err != nil {
+		return err
+	}
+	fmt.Printf("Restarted csghub-lite service at %s\n", baseURL)
+	return nil
+}

internal/cli/restart_test.go

@@ -0,0 +1,83 @@
+package cli
+
+import (
+	"errors"
+	"reflect"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/opencsgs/csghub-lite/internal/config"
+)
+
+func TestRestartBackgroundServiceRestartsAndWaits(t *testing.T) {
+	oldStop := stopBackgroundServiceForRestart
+	oldStart := startBackgroundServerForRestart
+	oldWait := waitForServerForRestart
+	defer func() {
+		stopBackgroundServiceForRestart = oldStop
+		startBackgroundServerForRestart = oldStart
+		waitForServerForRestart = oldWait
+	}()
+
+	cfg := &config.Config{ListenAddr: ":14567"}
+	var calls []string
+
+	stopBackgroundServiceForRestart = func() error {
+		calls = append(calls, "stop")
+		return nil
+	}
+	startBackgroundServerForRestart = func(got *config.Config) error {
+		if got != cfg {
+			t.Fatalf("start config = %#v, want original config", got)
+		}
+		calls = append(calls, "start")
+		return nil
+	}
+	waitForServerForRestart = func(baseURL string, timeout time.Duration) error {
+		if baseURL != "http://127.0.0.1:14567" {
+			t.Fatalf("baseURL = %q, want http://127.0.0.1:14567", baseURL)
+		}
+		if timeout != 15*time.Second {
+			t.Fatalf("timeout = %s, want 15s", timeout)
+		}
+		calls = append(calls, "wait")
+		return nil
+	}
+
+	if err := restartBackgroundService(cfg); err != nil {
+		t.Fatalf("restartBackgroundService returned error: %v", err)
+	}
+
+	if want := []string{"stop", "start", "wait"}; !reflect.DeepEqual(calls, want) {
+		t.Fatalf("calls = %#v, want %#v", calls, want)
+	}
+}
+
+func TestRestartBackgroundServiceWrapsStopError(t *testing.T) {
+	oldStop := stopBackgroundServiceForRestart
+	oldStart := startBackgroundServerForRestart
+	oldWait := waitForServerForRestart
+	defer func() {
+		stopBackgroundServiceForRestart = oldStop
+		startBackgroundServerForRestart = oldStart
+		waitForServerForRestart = oldWait
+	}()
+
+	stopBackgroundServiceForRestart = func() error {
+		return errors.New("boom")
+	}
+	startBackgroundServerForRestart = func(*config.Config) error {
+		t.Fatal("start should not be called after stop failure")
+		return nil
+	}
+	waitForServerForRestart = func(string, time.Duration) error {
+		t.Fatal("wait should not be called after stop failure")
+		return nil
+	}
+
+	err := restartBackgroundService(&config.Config{ListenAddr: config.DefaultListenAddr})
+	if err == nil || !strings.Contains(err.Error(), "stopping existing service: boom") {
+		t.Fatalf("error = %v, want wrapped stop error", err)
+	}
+}

internal/cli/root.go

@@ -31,6 +31,7 @@ and the full CSGHub platform.`,
 		newPsCmd(),
 		newStopCmd(),
 		newStopServiceCmd(),
+		newRestartCmd(),
 		newRmCmd(),
 		newLoginCmd(),
 		newSearchCmd(),