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

fix: Use AsyncSession in memory (#4665)

Browse source
This commit is contained in:
Christophe Bornet 2024-12-06 17:25:59 +01:00 • committed by GitHub
commit 79b03ba133
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
26 changed files with 610 additions and 173 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/tests/unit/custom/custom_component/test_component_events.py
View file

@ -52,7 +52,7 @@ async def test_component_message_sending():
)
# Send the message
sent_message = await asyncio.to_thread(component.send_message, message)
sent_message = await component.send_message(message)
# Verify the message was sent
assert sent_message.id is not None
@ -85,7 +85,7 @@ async def test_component_tool_output():
)
# Send the message
sent_message = await asyncio.to_thread(component.send_message, message)
sent_message = await component.send_message(message)
# Verify the message was stored and processed
assert sent_message.id is not None
@ -112,8 +112,7 @@ async def test_component_error_handling():
msg = "Test error"
raise CustomError(msg)
except CustomError as e:
sent_message = await asyncio.to_thread(
component.send_error,
sent_message = await component.send_error(
exception=e,
session_id="test_session",
trace_name="test_trace",
@ -227,7 +226,7 @@ async def test_component_streaming_message():
)
# Send the streaming message
sent_message = await asyncio.to_thread(component.send_message, message)
sent_message = await component.send_message(message)
# Verify the message
assert sent_message.id is not None