From 8fcf54bcb8795f274cdf3f4db21f58589ea92722 Mon Sep 17 00:00:00 2001
From: Atharva J <me.atharvajadhav@gmail.com>
Date: Mon, 18 Dec 2023 01:04:37 +0530
Subject: [PATCH] minor - test check trial

---
 tests/test_endpoints.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/test_endpoints.py b/tests/test_endpoints.py
index 338224004..f4b65f4f7 100644
--- a/tests/test_endpoints.py
+++ b/tests/test_endpoints.py
@@ -11,6 +11,7 @@ from langflow.services.database.models.api_key.model import ApiKey
 from langflow.services.database.utils import session_getter
 from langflow.services.deps import get_db_service, get_settings_service
 from langflow.template.frontend_node.chains import TimeTravelGuideChainNode
+import os
 
 
 def run_post(client, flow_id, headers, post_data):
@@ -49,7 +50,7 @@ PROMPT_REQUEST = {
         "custom_fields": {},
         "output_types": [],
         "field_formatters": {
-            "formatters": {"openai_api_key": {}},
+            "formatters": {"openai_api_key": os.environ.get("OPENAI_API_KEY")},
             "base_formatters": {
                 "kwargs": {},
                 "optional": {},