From 027fd0658b2a4fba3a59a1053370a2a550ec8b22 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 1 Apr 2024 09:00:53 -0300 Subject: [PATCH] Remove unused code in test_endpoints.py --- tests/test_endpoints.py | 14 -------------- 1 file changed, 14 deletions(-) diff --git a/tests/test_endpoints.py b/tests/test_endpoints.py index 92ff35566..e2f14a36e 100644 --- a/tests/test_endpoints.py +++ b/tests/test_endpoints.py @@ -4,7 +4,6 @@ from uuid import uuid4 import pytest from fastapi import status from fastapi.testclient import TestClient - from langflow.interface.custom.directory_reader.directory_reader import DirectoryReader from langflow.services.deps import get_settings_service from langflow.template.frontend_node.chains import TimeTravelGuideChainNode @@ -424,19 +423,6 @@ def test_build_vertex_invalid_vertex_id(client, added_flow_with_prompt_and_histo assert response.status_code == 500 -# Helper function to poll task status -def poll_task_status(client, headers, href, max_attempts=20, sleep_time=1): - for _ in range(max_attempts): - task_status_response = client.get( - href, - headers=headers, - ) - if task_status_response.status_code == 200 and task_status_response.json()["status"] == "SUCCESS": - return task_status_response.json() - time.sleep(sleep_time) - return None # Return None if task did not complete in time - - def test_successful_run(client, starter_project, created_api_key): headers = {"x-api-key": created_api_key.api_key} flow_id = starter_project["id"]