From c66e289a95f9db585e22c276685c42de00b578c7 Mon Sep 17 00:00:00 2001
From: "tembo[bot]" <208362400+tembo[bot]@users.noreply.github.com>
Date: Fri, 12 Dec 2025 13:48:41 +0000
Subject: [PATCH] feat(mcp): Add workflow ID header and fix env var header
 format

Co-authored-by: Darren <dmb.programmer@gmail.com>
---
 src/commands/mcp.ts | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/src/commands/mcp.ts b/src/commands/mcp.ts
index 6417128..bcfb929 100644
--- a/src/commands/mcp.ts
+++ b/src/commands/mcp.ts
@@ -20,16 +20,25 @@ mcp.action(async (options) => {
   const temboEnvVars = getTemboEnvVars();
 
   // Convert temboEnvVars to headers, X-Tembo-<key>: <value>
-  const headers = Object.entries(temboEnvVars).map(([key, value]) => ({
-    "X-Tembo-": `${key}: ${value}`,
-  }));
+  const temboHeaders: Record<string, string> = {};
+  for (const [key, value] of Object.entries(temboEnvVars)) {
+    if (value !== undefined) {
+      temboHeaders[`X-Tembo-${key}`] = value;
+    }
+  }
+
+  // Add workflow ID header if present (for automation context)
+  const workflowId = process.env.TEMBO_WORKFLOW_ID;
+  if (workflowId) {
+    temboHeaders["X-Tembo-Workflow-Id"] = workflowId;
+  }
 
   streamableHttpToStdio({
     streamableHttpUrl: options.sseUrl,
     logger: console,
     headers: {
       Authorization: `Bearer ${authToken}`,
-      ...headers,
+      ...temboHeaders,
     },
   });
 });