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

9
tests/test_initial_setup.py
View file

@ -1,16 +1,16 @@
from datetime import datetime
from pathlib import Path
from sqlmodel import select
from langflow.initial_setup.setup import (
STARTER_FOLDER_NAME,
create_or_update_starter_projects,
get_project_data,
load_starter_projects,
)
from langflow.services.database.models.flow.model import Flow
from langflow.services.database.models.folder.model import Folder
from langflow.services.deps import session_scope
from sqlalchemy import func
from sqlmodel import select
def test_load_starter_projects():
@ -50,7 +50,8 @@ def test_create_or_update_starter_projects(client):
num_projects = len(load_starter_projects())
# Get the number of projects in the database
num_db_projects = session.exec(select(func.count(Flow.id)).where(Flow.folder == STARTER_FOLDER_NAME)).one()
folder = session.exec(select(Folder).where(Folder.name == STARTER_FOLDER_NAME)).first()
num_db_projects = len(folder.flows)
# Check that the number of projects in the database is the same as the number of projects returned by load_starter_projects
assert num_db_projects == num_projects