dep fix

private/buf/buflsp/buflsp.go

@@ -24,7 +24,6 @@ import (
 	"sync"
 	"sync/atomic"
 
-	"github.com/bufbuild/buf/private/buf/bufcli"
 	"github.com/bufbuild/buf/private/buf/bufctl"
 	"github.com/bufbuild/buf/private/bufpkg/bufcheck"
 	"github.com/bufbuild/buf/private/pkg/app/appext"
@@ -41,6 +40,7 @@ import (
 // Returns a context for managing the server.
 func Serve(
 	ctx context.Context,
+	wktBucket storage.ReadBucket,
 	container appext.Container,
 	controller bufctl.Controller,
 	checkClient bufcheck.Client,
@@ -58,15 +58,6 @@ func Serve(
 		return nil, err
 	}
 
-	wktStore, err := bufcli.NewWKTStore(container)
-	if err != nil {
-		return nil, err
-	}
-	wktBucket, err := wktStore.GetBucket(ctx)
-	if err != nil {
-		return nil, err
-	}
-
 	conn := jsonrpc2.NewConn(stream)
 	lsp := &lsp{
 		conn: conn,

private/buf/cmd/buf/command/beta/lsp/lsp.go

@@ -89,6 +89,15 @@ func run(
 		return err
 	}
 
+	wktStore, err := bufcli.NewWKTStore(container)
+	if err != nil {
+		return err
+	}
+	wktBucket, err := wktStore.GetBucket(ctx)
+	if err != nil {
+		return err
+	}
+
 	controller, err := bufcli.NewController(container)
 	if err != nil {
 		return err
@@ -114,7 +123,7 @@ func run(
 		return err
 	}
 
-	conn, err := buflsp.Serve(ctx, container, controller, checkClient, jsonrpc2.NewStream(transport))
+	conn, err := buflsp.Serve(ctx, wktBucket, container, controller, checkClient, jsonrpc2.NewStream(transport))
 	if err != nil {
 		return err
 	}