🚑 Fix loading the repositories properly from pinocchio

go-go-golems · Dec 31, 2024 · 579e0c0 · 579e0c0
1 parent 095428a
commit 579e0c0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/cmd/prompto/cmds/serve.go b/cmd/prompto/cmds/serve.go
@@ -30,5 +30,5 @@ func (s *ServeCommand) run(cmd *cobra.Command, args []string) error {
 	port, _ := cmd.Flags().GetInt("port")
 	watching, _ := cmd.Flags().GetBool("watching")
 
-	return server.Serve(port, watching)
+	return server.Serve(port, watching, s.repositories)
 }
diff --git a/pkg/server/serve.go b/pkg/server/serve.go
@@ -13,7 +13,6 @@ import (
 	"github.com/go-go-golems/glazed/pkg/helpers/templating"
 	"github.com/go-go-golems/prompto/pkg"
 	"github.com/rs/zerolog/log"
-	"github.com/spf13/viper"
 )
 
 type ServerState struct {
@@ -34,7 +33,6 @@ func (s *ServerState) LoadRepositories() error {
 	s.mu.Lock()
 	defer s.mu.Unlock()
 
-	s.Repositories = viper.GetStringSlice("repositories")
 	for _, repoPath := range s.Repositories {
 		repo := pkg.NewRepository(repoPath)
 		err := repo.LoadPromptos()
@@ -179,8 +177,9 @@ func (s *ServerState) WatchRepositories(ctx context.Context) error {
 	return nil
 }
 
-func Serve(port int, watching bool) error {
+func Serve(port int, watching bool, repositories []string) error {
 	state := NewServerState(watching)
+	state.Repositories = repositories
 	if err := state.LoadRepositories(); err != nil {
 		return fmt.Errorf("error loading repositories: %w", err)
 	}