enhance: set sdk cred override env var for ui

njhale · njhale · commit 966b49375e6d · 2024-07-02T17:06:03.000-04:00
Pass credential overrides to the UI by setting the UI tool's `GPTSCRIPT_SDKSERVER_CREDENTIAL_OVERRIDE`
environment variable before running it.

Signed-off-by: Nick Hale &lt;4175918+njhale@users.noreply.github.com&gt;
diff --git a/pkg/cli/gptscript.go b/pkg/cli/gptscript.go
@@ -363,10 +363,13 @@ func (r *GPTScript) Run(cmd *cobra.Command, args []string) (retErr error) {
 				gptOpt.Env = append(gptOpt.Env, system.BinEnvVar+"="+system.Bin())
 			}
 
-			// If the DefaultModel is set, then pass the correct environment variable.
+			// Pass the corrected environment variables for SDK server options
 			if r.DefaultModel != "" {
 				gptOpt.Env = append(gptOpt.Env, "GPTSCRIPT_SDKSERVER_DEFAULT_MODEL="+r.DefaultModel)
 			}
+			if len(r.CredentialOverride) > 0 {
+				gptOpt.Env = append(gptOpt.Env, "GPTSCRIPT_SDKSERVER_CREDENTIAL_OVERRIDE="+strings.Join(r.CredentialOverride, ","))
+			}
 
 			args = append([]string{args[0]}, "--file="+file)
 

Original file line number	Diff line number	Diff line change
`@@ -363,10 +363,13 @@ func (r GPTScript) Run(cmd cobra.Command, args []string) (retErr error) {`
`363`	`363`	`gptOpt.Env = append(gptOpt.Env, system.BinEnvVar+"="+system.Bin())`
`364`	`364`	`}`
`365`	`365`
`366`		`- // If the DefaultModel is set, then pass the correct environment variable.`
	`366`	`+ // Pass the corrected environment variables for SDK server options`
`367`	`367`	`if r.DefaultModel != "" {`
`368`	`368`	`gptOpt.Env = append(gptOpt.Env, "GPTSCRIPT_SDKSERVER_DEFAULT_MODEL="+r.DefaultModel)`
`369`	`369`	`}`
	`370`	`+ if len(r.CredentialOverride) > 0 {`
	`371`	`+ gptOpt.Env = append(gptOpt.Env, "GPTSCRIPT_SDKSERVER_CREDENTIAL_OVERRIDE="+strings.Join(r.CredentialOverride, ","))`
	`372`	`+ }`
`370`	`373`
`371`	`374`	`args = append([]string{args[0]}, "--file="+file)`
`372`	`375`