WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-28 20:45:05 +02:00
Code Issues Packages Projects Releases Wiki Activity

Simpler init syntax for pubsub

Browse Source
This commit is contained in:
Nuno Campos
2023-09-19 11:12:02 +01:00
parent c2a5f54121
commit 62f27eb8c4
2 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_invoke.py
+4 -4
View File
@@ -16,7 +16,7 @@ def test_invoke_single_process_in_out(mocker: MockerFixture):
assert chain.invoke(2) == 3
conn = InMemoryPubSubConnection()
pubsub = PubSub(processes=(chain,), connection=conn)
pubsub = PubSub(chain, connection=conn)
# Using in-memory conn internals to make assertions about pubsub
# If we start with 0 listeners
@@ -38,7 +38,7 @@ def test_invoke_two_processes_in_out(mocker: MockerFixture):
assert chain_two.invoke(2) == 3
conn = InMemoryPubSubConnection()
pubsub = PubSub(processes=(chain_one, chain_two), connection=conn)
pubsub = PubSub(chain_one, chain_two, connection=conn)
# Using in-memory conn internals to make assertions about pubsub
# If we start with 0 listeners
@@ -60,8 +60,8 @@ def test_invoke_two_processes_in_out_interrupt(mocker: MockerFixture):
assert chain_two.invoke(2) == 3
conn = InMemoryPubSubConnection(clear_on_disconnect=False)
pubsub_one = PubSub(processes=(chain_one,), connection=conn)
pubsub_two = PubSub(processes=(chain_two,), connection=conn)
pubsub_one = PubSub(chain_one, connection=conn)
pubsub_two = PubSub(chain_two, connection=conn)
# Using in-memory conn internals to make assertions about pubsub
# If we start with 0 listeners