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

fix: Fix test_get_and_cache_all_types_dict (#5054)

Browse source
This commit is contained in:
Christophe Bornet 2024-12-05 14:15:21 +01:00 • committed by GitHub
commit c1bd7c89c3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 16 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

2
pyproject.toml
View file

@ -176,7 +176,7 @@ dev-dependencies = [
"asgi-lifespan>=2.1.0",
"pytest-github-actions-annotate-failures>=0.2.0",
"pytest-codspeed>=3.0.0",
"blockbuster>=1.1.1,<1.2",
"blockbuster>=1.3.1,<1.4",
"types-aiofiles>=24.1.0.20240626",
]

2
src/backend/base/langflow/main.py
View file

@ -101,7 +101,7 @@ def get_lifespan(*, fix_migration=False, version=None):
rprint("[bold green]Starting Langflow...[/bold green]")
try:
await initialize_services(fix_migration=fix_migration)
await asyncio.to_thread(setup_llm_caching)
setup_llm_caching()
await initialize_super_user_if_needed()
all_types_dict = await get_and_cache_all_types_dict(get_settings_service())
await asyncio.to_thread(create_or_update_starter_projects, all_types_dict)

1
src/backend/tests/conftest.py
View file

@ -57,6 +57,7 @@ def blockbuster():
for func in bb.functions:
if func.startswith("sqlite3."):
bb.functions[func].deactivate()
bb.functions["threading.Lock.acquire"].deactivate()
yield bb

18
src/backend/tests/performance/test_server_init.py
View file

@ -25,7 +25,7 @@ async def test_initialize_services():
from langflow.services.utils import initialize_services
await initialize_services(fix_migration=False)
settings_service = await asyncio.to_thread(get_settings_service)
settings_service = get_settings_service()
assert "test_performance.db" in settings_service.settings.database_url
@ -34,8 +34,8 @@ async def test_setup_llm_caching():
"""Benchmark LLM caching setup."""
from langflow.interface.utils import setup_llm_caching
await asyncio.to_thread(setup_llm_caching)
settings_service = await asyncio.to_thread(get_settings_service)
setup_llm_caching()
settings_service = get_settings_service()
assert "test_performance.db" in settings_service.settings.database_url
@ -47,7 +47,7 @@ async def test_initialize_super_user():
await initialize_services(fix_migration=False)
await initialize_super_user_if_needed()
settings_service = await asyncio.to_thread(get_settings_service)
settings_service = get_settings_service()
assert "test_performance.db" in settings_service.settings.database_url
@ -56,9 +56,9 @@ async def test_get_and_cache_all_types_dict():
"""Benchmark get_and_cache_all_types_dict function."""
from langflow.interface.types import get_and_cache_all_types_dict
settings_service = await asyncio.to_thread(get_settings_service)
result = await asyncio.to_thread(get_and_cache_all_types_dict, settings_service)
assert result is not None
settings_service = get_settings_service()
result = await get_and_cache_all_types_dict(settings_service)
assert "vectorstores" in result
assert "test_performance.db" in settings_service.settings.database_url
@ -70,7 +70,7 @@ async def test_create_starter_projects():
from langflow.services.utils import initialize_services
await initialize_services(fix_migration=False)
settings_service = await asyncio.to_thread(get_settings_service)
settings_service = get_settings_service()
types_dict = await get_and_cache_all_types_dict(settings_service)
await asyncio.to_thread(create_or_update_starter_projects, types_dict)
assert "test_performance.db" in settings_service.settings.database_url
@ -82,5 +82,5 @@ async def test_load_flows():
from langflow.initial_setup.setup import load_flows_from_directory
await load_flows_from_directory()
settings_service = await asyncio.to_thread(get_settings_service)
settings_service = get_settings_service()
assert "test_performance.db" in settings_service.settings.database_url

8
uv.lock generated
View file

@ -461,14 +461,14 @@ wheels = [
[[package]]
name = "blockbuster"
version = "1.1.1"
version = "1.3.1"
source = { registry = "https://pypi.org/simple" }
dependencies = [
{ name = "forbiddenfruit" },
]
sdist = { url = "https://files.pythonhosted.org/packages/ec/75/b220b51bf86f0ac04c2d5d37c14bde16d2f88efaa2487cdfea89e9523c06/blockbuster-1.1.1.tar.gz", hash = "sha256:7b9adc2a6cff91d8e671458bf8fe24bd9b78472f28095d677666e178e3f4faef", size = 9080 }
sdist = { url = "https://files.pythonhosted.org/packages/6a/8c/e7e61396951b9ab1c39c35fa3a9737270a62da41d0b3ff6a15207747e5eb/blockbuster-1.3.1.tar.gz", hash = "sha256:59984286e1cd274d406129f93a42bd62b98fe53f484d544e6e743a5d6d7e42d5", size = 9409 }
wheels = [
{ url = "https://files.pythonhosted.org/packages/7e/67/3fc297f8b3a1af3d43ec2252dbc59e710fb003427f63f2bfc9974fa68ccc/blockbuster-1.1.1-py3-none-any.whl", hash = "sha256:cf75011d3cb5266c618be292337f3f3ecbc00e0d056849a64a183ba1cb55518b", size = 7902 },
{ url = "https://files.pythonhosted.org/packages/a2/de/28b5c37969037ad7dcf08c18dba255e1ba3a42539cf04cae3ef627f706ae/blockbuster-1.3.1-py3-none-any.whl", hash = "sha256:883292a02c1ae9dbd76d637ea8b604a1cc75aebb0544d566aed90491e056963e", size = 8111 },
]
[[package]]
@ -3760,7 +3760,7 @@ requires-dist = [
[package.metadata.requires-dev]
dev = [
{ name = "asgi-lifespan", specifier = ">=2.1.0" },
{ name = "blockbuster", specifier = ">=1.1.1,<1.2" },
{ name = "blockbuster", specifier = ">=1.3.1,<1.4" },
{ name = "dictdiffer", specifier = ">=0.9.0" },
{ name = "httpx", specifier = ">=0.27.0" },
{ name = "ipykernel", specifier = ">=6.29.0" },