From 778993d64c73fb643e38d032cc1ca0317a967ab9 Mon Sep 17 00:00:00 2001
From: Edwin Jose <edwin.jose@datastax.com>
Date: Fri, 26 Sep 2025 10:35:39 -0400
Subject: [PATCH] Update env vars and header keys for Langflow service

Added OWNER, OWNER_NAME, OWNER_EMAIL, and CONNECTOR_TYPE environment variables to docker-compose.yml. Updated LANGFLOW_VARIABLES_TO_GET_FROM_ENVIRONMENT to match. Changed header keys in langflow_file_service.py to uppercase and ensured values are stringified for consistency.
---
 docker-compose.yml                    |  6 +++++-
 src/services/langflow_file_service.py | 14 +++++++-------
 2 files changed, 12 insertions(+), 8 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index be863df3..daa921ae 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -100,9 +100,13 @@ services:
       - LANGFLOW_LOAD_FLOWS_PATH=/app/flows
       - LANGFLOW_SECRET_KEY=${LANGFLOW_SECRET_KEY}
       - JWT="dummy"
+      - OWNER=None
+      - OWNER_NAME=None
+      - OWNER_EMAIL=None
+      - CONNECTOR_TYPE=system
       - OPENRAG-QUERY-FILTER="{}"
       - OPENSEARCH_PASSWORD=${OPENSEARCH_PASSWORD}
-      - LANGFLOW_VARIABLES_TO_GET_FROM_ENVIRONMENT=JWT,OPENRAG-QUERY-FILTER,OPENSEARCH_PASSWORD,OWNER,OWNER_NAME,OWNER_EMAIL,CONNECTOR_TYPE,SESSION_ID,FILE_PATH
+      - LANGFLOW_VARIABLES_TO_GET_FROM_ENVIRONMENT=JWT,OPENRAG-QUERY-FILTER,OPENSEARCH_PASSWORD,OWNER,OWNER_NAME,OWNER_EMAIL,CONNECTOR_TYPE
       - LANGFLOW_LOG_LEVEL=DEBUG
       - LANGFLOW_AUTO_LOGIN=${LANGFLOW_AUTO_LOGIN}
       - LANGFLOW_SUPERUSER=${LANGFLOW_SUPERUSER}
diff --git a/src/services/langflow_file_service.py b/src/services/langflow_file_service.py
index 63353874..39c5eac7 100644
--- a/src/services/langflow_file_service.py
+++ b/src/services/langflow_file_service.py
@@ -130,13 +130,13 @@ class LangflowFileService:
         )
 
         # Avoid logging full payload to prevent leaking sensitive data (e.g., JWT)
-        headers = {
-            "X-Langflow-Global-Var-JWT": jwt_token,
-            "X-Langflow-Global-Var-Owner": owner,
-            "X-Langflow-Global-Var-Owner-Name": owner_name,
-            "X-Langflow-Global-Var-Owner-Email": owner_email,
-            "X-Langflow-Global-Var-Connector-Type": connector_type,
-        }
+        headers={
+                "X-Langflow-Global-Var-JWT": str(jwt_token),
+                "X-Langflow-Global-Var-OWNER": str(owner),
+                "X-Langflow-Global-Var-OWNER_NAME": str(owner_name),
+                "X-Langflow-Global-Var-OWNER_EMAIL": str(owner_email),
+                "X-Langflow-Global-Var-CONNECTOR_TYPE": str(connector_type),
+            }
 
         resp = await clients.langflow_request(
             "POST", f"/api/v1/run/{self.flow_id_ingest}", json=payload, headers=headers