jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Fix lint and tests (#1959)

Browse source 
* refactor(chat.py): rename flow_id variable to flow_id_str for better clarity
refactor(chat.py): update flow_id usage to flow_id_str in build_vertex function for consistency
refactor(chat.py): change flow_id to flow_id_str in build_vertex function for consistency
refactor(chat.py): replace flow_id with flow_id_str in build_vertex function for consistency
refactor(chat.py): update flow_id to flow_id_str in build_vertex_stream function for consistency
refactor(chat.py): change flow_id to flow_id_str in build_vertex_stream function for consistency
refactor(chat.py): replace flow_id with flow_id_str in build_vertex_stream function for consistency
refactor(chat.py): update flow_id to flow_id_str in build_vertex_stream function for consistency
refactor(endpoints.py): replace flow_id with flow_id_str in simplified_run_flow function for consistency
refactor(endpoints.py): update flow_id to flow_id_str in simplified_run_flow function for consistency
refactor(endpoints.py): change flow_id to flow_id_str in simplified_run_flow function for consistency
refactor(endpoints.py): replace flow_id with flow_id_str in simplified_run_flow function for consistency
refactor(endpoints.py): update flow_id to flow_id_str in simplified_run_flow function for consistency
refactor(endpoints.py): change flow_id to flow_id_str in simplified_run_flow function for consistency
refactor(endpoints.py): replace flow_id with flow_id_str in simplified_run_flow function for consistency
refactor(endpoints.py): update flow_id to flow_id_str in simplified_run_flow function for consistency
refactor(endpoints.py): change flow_id to flow_id_str in experimental_run_flow function for consistency
refactor(endpoints.py): replace flow_id with flow_id_str in experimental_run_flow function for consistency
refactor(endpoints.py): update flow_id to flow_id_str in experimental_run_flow function for consistency
refactor(endpoints.py): change flow_id to flow_id_str in experimental_run_flow function for consistency
refactor(endpoints.py): replace flow_id with flow_id_str in experimental_run_flow function for consistency
refactor(endpoints.py): update flow_id to flow_id_str in experimental_run_flow function for consistency
refactor(endpoints.py): change flow_id to flow_id_str in experimental_run_flow function for consistency
refactor(endpoints.py): replace flow_id with flow_id_str in experimental_run_flow function for consistency
refactor(endpoints.py): update flow_id

fix(files.py): update variable names from flow_id to flow_id_str for consistency
feat(files.py): use flow_id_str instead of flow_id for file operations to ensure correct folder paths
feat(files.py): improve error handling and response messages in file operations
feat(flows.py): optimize query for fetching example flows
feat(folders.py): refactor create_folder function for better readability and maintainability
feat(folders.py): update_folder function to handle excluded_flows and concat_folder_components
feat(folders.py): improve error handling and response messages in folder operations
feat(folders.py): optimize query for fetching folder in download_file function
feat(schemas.py): update FlowListReadWithFolderName schema field names to name and description

feat(OpenAIConversationalAgent.py): add support for using pydantic SecretStr for storing API key securely
feat(ChatOpenAISpecs.py): add support for using pydantic SecretStr for storing API key securely
refactor(RecordsOutput.py): change base class from TextComponent to CustomComponent
refactor(base.py): remove unused import and log_transaction call
refactor(model.py): change components and flows fields to have default_factory=list
refactor(utils.py): update SQLModel import and use Field for components and flows fields
refactor(service.py): change table_map declaration to use type hints for better readability

* feat(tests): add import statement for Folder model in conftest.py to support new functionality
refactor(tests): update query in conftest.py to use nested query for Folder name
refactor(tests): update import statement for UUID in test_endpoints.py
refactor(tests): update UUID usage in test_endpoints.py for consistency and clarity
refactor(tests): update UUID usage in test_endpoints.py for consistency and clarity
refactor(tests): update UUID usage in test_endpoints.py for consistency and clarity
refactor(tests): update UUID usage in test_endpoints.py for consistency and clarity
feat(tests): add import statement for Folder model in test_initial_setup.py to support new functionality
refactor(tests): update query in test_initial_setup.py to use Folder model for better readability and maintainability

* new lock
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-05-23 07:55:06 -07:00 • committed by GitHub
commit c7f5e4f843
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
18 changed files with 498 additions and 482 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/test_endpoints.py
View file

@ -1,5 +1,5 @@
import time
from uuid import uuid4
from uuid import UUID, uuid4
import pytest
from fastapi import status
@ -393,8 +393,9 @@ def test_various_prompts(client, prompt, expected_input_variables):
def test_get_vertices_flow_not_found(client, logged_in_headers):
response = client.post("/api/v1/build/nonexistent_id/vertices", headers=logged_in_headers)
assert response.status_code == 500 # Or whatever status code you've set for invalid ID
uuid = uuid4()
response = client.post(f"/api/v1/build/{uuid}/vertices", headers=logged_in_headers)
assert response.status_code == 500
def test_get_vertices(client, added_flow_with_prompt_and_history, logged_in_headers):
@ -414,7 +415,8 @@ def test_get_vertices(client, added_flow_with_prompt_and_history, logged_in_head
def test_build_vertex_invalid_flow_id(client, logged_in_headers):
response = client.post("/api/v1/build/nonexistent_id/vertices/vertex_id", headers=logged_in_headers)
uuid = uuid4()
response = client.post(f"/api/v1/build/{uuid}/vertices/vertex_id", headers=logged_in_headers)
assert response.status_code == 500
@ -652,7 +654,11 @@ def test_invalid_flow_id(client, created_api_key):
headers = {"x-api-key": created_api_key.api_key}
flow_id = "invalid-flow-id"
response = client.post(f"/api/v1/run/{flow_id}", headers=headers)
assert response.status_code == status.HTTP_404_NOT_FOUND
assert response.status_code == status.HTTP_422_UNPROCESSABLE_ENTITY, response.text
headers = {"x-api-key": created_api_key.api_key}
flow_id = UUID(int=0)
response = client.post(f"/api/v1/run/{flow_id}", headers=headers)
assert response.status_code == status.HTTP_404_NOT_FOUND, response.text
# Check if the error detail is as expected