PR feedback

Signed-off-by: Grant Linville <[email protected]>
gptscript-ai · Oct 10, 2024 · 955ad9c · 955ad9c
1 parent 2a63506
commit 955ad9c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/pkg/gptscript/gptscript.go b/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
 }

diff --git a/pkg/sdkserver/server.go b/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
 }