diff --git a/tests/test_endpoints.py b/tests/test_endpoints.py index f4b65f4f7..338224004 100644 --- a/tests/test_endpoints.py +++ b/tests/test_endpoints.py @@ -11,7 +11,6 @@ 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): @@ -50,7 +49,7 @@ PROMPT_REQUEST = { "custom_fields": {}, "output_types": [], "field_formatters": { - "formatters": {"openai_api_key": os.environ.get("OPENAI_API_KEY")}, + "formatters": {"openai_api_key": {}}, "base_formatters": { "kwargs": {}, "optional": {},