PR feedback

Signed-off-by: Grant Linville <[email protected]>

Author: g-linville

pkg/gptscript/gptscript.go

@@ -32,8 +32,6 @@ import (
 
 var log = mvl.Package()
 
-const defaultDatasetToolRepo = "github.com/gptscript-ai/datasets"
-
 type GPTScript struct {
 	Registry               *llm.Registry
 	Runner                 *runner.Runner
@@ -84,9 +82,6 @@ func Complete(opts ...Options) Options {
 	if len(result.CredentialContexts) == 0 {
 		result.CredentialContexts = []string{credentials.DefaultCredentialContext}
 	}
-	if result.DatasetToolRepo == "" {
-		result.DatasetToolRepo = defaultDatasetToolRepo
-	}
 
 	return result
 }

pkg/sdkserver/server.go

@@ -23,6 +23,8 @@ import (
 	"github.com/rs/cors"
 )
 
+const defaultDatasetToolRepo = "github.com/gptscript-ai/datasets"
+
 type Options struct {
 	gptscript.Options
 
@@ -165,5 +167,9 @@ func complete(opts ...Options) Options {
 		result.ListenAddress = "127.0.0.1:0"
 	}
 
+	if result.DatasetToolRepo == "" {
+		result.DatasetToolRepo = defaultDatasetToolRepo
+	}
+
 	return result
 }