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

fix: make sure tests run async (#5842)

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2025-01-21 13:16:33 -03:00 • committed by GitHub
commit 86b83b0f64
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 13 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/backend/tests/unit/api/v1/test_endpoints.py
View file

@ -1,3 +1,4 @@
import asyncio
import inspect
from typing import Any
@ -65,7 +66,7 @@ async def test_update_component_model_name_options(client: AsyncClient, logged_i
# load the code from the file at langflow.components.agents.agent.py asynchronously
# we are at str/backend/tests/unit/api/v1/test_endpoints.py
# find the file by using the class AgentComponent
agent_component_file = inspect.getsourcefile(AgentComponent)
agent_component_file = await asyncio.to_thread(inspect.getsourcefile, AgentComponent)
async with async_open(agent_component_file, encoding="utf-8") as f:
code = await f.read()
@ -87,12 +88,12 @@ async def test_update_component_model_name_options(client: AsyncClient, logged_i
assert "template" in result
assert "model_name" in result["template"]
assert isinstance(result["template"]["model_name"]["options"], list)
assert len(result["template"]["model_name"]["options"]) > 0, (
f"Model names: {result['template']['model_name']['options']}"
)
assert current_model_names != result["template"]["model_name"]["options"], (
f"Current model names: {current_model_names}, New model names: {result['template']['model_name']['options']}"
)
assert (
len(result["template"]["model_name"]["options"]) > 0
), f"Model names: {result['template']['model_name']['options']}"
assert (
current_model_names != result["template"]["model_name"]["options"]
), f"Current model names: {current_model_names}, New model names: {result['template']['model_name']['options']}"
# Now test with Custom provider
template["agent_llm"]["value"] = "Custom"
request.field_value = "Custom"