chore: Add GPTSCRIPT_HTTP_ENV_PREFIX to specify http env vars to send

gptscript-ai · Nov 11, 2024 · 176942a · 176942a
1 parent 02bcf6c
commit 176942a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pkg/engine/http.go b/pkg/engine/http.go
@@ -82,6 +82,16 @@ func (e *Engine) runHTTP(ctx context.Context, prg *types.Program, tool types.Too
 			req.Header.Add("X-GPTScript-Env", k+"="+envMap[k])
 		}
 	}
+	for _, prefix := range strings.Split(os.Getenv("GPTSCRIPT_HTTP_ENV_PREFIX"), ",") {
+		if prefix == "" {
+			continue
+		}
+		for _, k := range slices.Sorted(maps.Keys(envMap)) {
+			if strings.HasPrefix(k, prefix) {
+				req.Header.Add("X-GPTScript-Env", k+"="+envMap[k])
+			}
+		}
+	}
 
 	req.Header.Set("X-GPTScript-Tool-Name", tool.Parameters.Name)