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

6
src/backend/tests/integration/components/inputs/test_chat_input.py
View file

@ -1,5 +1,5 @@
from langflow.components.inputs import ChatInput
from langflow.memory import get_messages
from langflow.memory import aget_messages
from langflow.schema.message import Message
from tests.integration.utils import run_single_component
@ -38,7 +38,7 @@ async def test_do_not_store_messages():
assert outputs["message"].text == "hello"
assert outputs["message"].session_id == session_id
assert len(get_messages(session_id=session_id)) == 1
assert len(await aget_messages(session_id=session_id)) == 1
session_id = "test-session-id-another"
outputs = await run_single_component(
@ -48,4 +48,4 @@ async def test_do_not_store_messages():
assert outputs["message"].text == "hello"
assert outputs["message"].session_id == session_id
assert len(get_messages(session_id=session_id)) == 0
assert len(await aget_messages(session_id=session_id)) == 0

6
src/backend/tests/integration/components/outputs/test_chat_output.py
View file

@ -1,5 +1,5 @@
from langflow.components.outputs import ChatOutput
from langflow.memory import get_messages
from langflow.memory import aget_messages
from langflow.schema.message import Message
from tests.integration.utils import run_single_component
@ -29,7 +29,7 @@ async def test_do_not_store_message():
assert isinstance(outputs["message"], Message)
assert outputs["message"].text == "hello"
assert len(get_messages(session_id=session_id)) == 1
assert len(await aget_messages(session_id=session_id)) == 1
session_id = "test-session-id-another"
outputs = await run_single_component(
@ -38,4 +38,4 @@ async def test_do_not_store_message():
assert isinstance(outputs["message"], Message)
assert outputs["message"].text == "hello"
assert len(get_messages(session_id=session_id)) == 0
assert len(await aget_messages(session_id=session_id)) == 0