From 8f9afa0950197c66ab0635187c33e14c00fbfc0f Mon Sep 17 00:00:00 2001
From: phact <estevezsebastian@gmail.com>
Date: Wed, 27 Aug 2025 23:09:24 -0400
Subject: [PATCH] x-api-key fix

---
 src/agent.py    | 7 ++++++-
 src/api/chat.py | 3 +++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/agent.py b/src/agent.py
index 93a2ea86..a7e69d8d 100644
--- a/src/agent.py
+++ b/src/agent.py
@@ -57,9 +57,14 @@ async def async_response_stream(client, prompt: str, model: str, extra_headers:
         }
         if previous_response_id is not None:
             request_params["previous_response_id"] = previous_response_id
+
+        if "x-api-key" not in client.default_headers:
+            if hasattr(client, 'api_key') and extra_headers is not None:
+                extra_headers["x-api-key"] = client.api_key
+
         if extra_headers:
             request_params["extra_headers"] = extra_headers
-        
+
         response = await client.responses.create(**request_params)
         
         full_response = ""
diff --git a/src/api/chat.py b/src/api/chat.py
index 9fde3d1a..112bfbf8 100644
--- a/src/api/chat.py
+++ b/src/api/chat.py
@@ -89,6 +89,9 @@ async def langflow_endpoint(request: Request, chat_service, session_manager):
             return JSONResponse(result)
         
     except Exception as e:
+        import traceback
+        traceback.print_exc()
+        print(f"[ERROR] Langflow request failed: {str(e)}")
         return JSONResponse({"error": f"Langflow request failed: {str(e)}"}, status_code=500)
 
 async def chat_history_endpoint(request: Request, chat_service, session_manager):