Also do it for HTTP headers

i'm not super satisfied by the verbosity but...

Author: julien-c

packages/tiny-agents/src/cli.ts

@@ -113,6 +113,13 @@ async function main() {
 						}
 					}
 				}
+				if ((server.type === "http" || server.type === "sse") && server.config.options?.requestInit?.headers) {
+					for (const [key, value] of Object.entries(server.config.options.requestInit.headers)) {
+						if (value.includes(envSpecialValue)) {
+							inputVars.add(key);
+						}
+					}
+				}
 			}
 
 			if (inputVars.size === 0) {
@@ -156,6 +163,29 @@ async function main() {
 						}
 					}
 				}
+				if ((server.type === "http" || server.type === "sse") && server.config.options?.requestInit?.headers) {
+					for (const [key, value] of Object.entries(server.config.options.requestInit.headers)) {
+						if (value.includes(envSpecialValue)) {
+							if (userInput) {
+								server.config.options.requestInit.headers[key] = value.replace(envSpecialValue, userInput);
+							} else {
+								const valueFromEnv = process.env[key] || "";
+								server.config.options.requestInit.headers[key] = value.replace(envSpecialValue, valueFromEnv);
+								if (valueFromEnv) {
+									stdout.write(ANSI.GREEN);
+									stdout.write(`Value successfully loaded from '${key}'`);
+									stdout.write(ANSI.RESET);
+									stdout.write("\n");
+								} else {
+									stdout.write(ANSI.YELLOW);
+									stdout.write(`No value found for '${key}' in environment variables. Continuing.`);
+									stdout.write(ANSI.RESET);
+									stdout.write("\n");
+								}
+							}
+						}
+					}
+				}
 			}
 		}
 

packages/tiny-agents/src/lib/types.ts

@@ -21,6 +21,14 @@ export const ServerConfigSchema = z.discriminatedUnion("type", [
 			url: z.union([z.string(), z.string().url()]),
 			options: z
 				.object({
+					/**
+					 * Customizes HTTP requests to the server.
+					 */
+					requestInit: z
+						.object({
+							headers: z.record(z.string()).optional(),
+						})
+						.optional(),
 					/**
 					 * Session ID for the connection. This is used to identify the session on the server.
 					 * When not provided and connecting to a server that supports session IDs, the server will generate a new session ID.
@@ -34,7 +42,18 @@ export const ServerConfigSchema = z.discriminatedUnion("type", [
 		type: z.literal("sse"),
 		config: z.object({
 			url: z.union([z.string(), z.string().url()]),
-			options: z.object({}).optional(),
+			options: z
+				.object({
+					/**
+					 * Customizes HTTP requests to the server.
+					 */
+					requestInit: z
+						.object({
+							headers: z.record(z.string()).optional(),
+						})
+						.optional(),
+				})
+				.optional(),
 		}),
 	}),
 ]);