github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

Make RunContext internal (#386)

Browse Source 
* Make RunContext internal

* Mypy
This commit is contained in:
Eric Zhu
2024-08-21 13:59:59 -07:00
committed by GitHub
parent 09ceef4b4a
commit ed0890525d
40 changed files with 2360 additions and 2329 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
python/tests/test_intervention.py
View File

@@ -21,11 +21,11 @@ async def test_intervention_count_messages() -> None:
runtime = SingleThreadedAgentRuntime(intervention_handler=handler)
await runtime.register("name", LoopbackAgent)
loopback = AgentId("name", key="default")
run_context = runtime.start()
runtime.start()
_response = await runtime.send_message(MessageType(), recipient=loopback)
await run_context.stop()
await runtime.stop()
assert handler.num_messages == 1
loopback_agent = await runtime.try_get_underlying_agent_instance(loopback, type=LoopbackAgent)
@@ -43,12 +43,12 @@ async def test_intervention_drop_send() -> None:
await runtime.register("name", LoopbackAgent)
loopback = AgentId("name", key="default")
run_context = runtime.start()
runtime.start()
with pytest.raises(MessageDroppedException):
_response = await runtime.send_message(MessageType(), recipient=loopback)
await run_context.stop()
await runtime.stop()
loopback_agent = await runtime.try_get_underlying_agent_instance(loopback, type=LoopbackAgent)
assert loopback_agent.num_calls == 0
@@ -66,12 +66,12 @@ async def test_intervention_drop_response() -> None:
await runtime.register("name", LoopbackAgent)
loopback = AgentId("name", key="default")
run_context = runtime.start()
runtime.start()
with pytest.raises(MessageDroppedException):
_response = await runtime.send_message(MessageType(), recipient=loopback)
await run_context.stop()
await runtime.stop()
@pytest.mark.asyncio
@@ -89,12 +89,12 @@ async def test_intervention_raise_exception_on_send() -> None:
await runtime.register("name", LoopbackAgent)
loopback = AgentId("name", key="default")
run_context = runtime.start()
runtime.start()
with pytest.raises(InterventionException):
_response = await runtime.send_message(MessageType(), recipient=loopback)
await run_context.stop()
await runtime.stop()
long_running_agent = await runtime.try_get_underlying_agent_instance(loopback, type=LoopbackAgent)
assert long_running_agent.num_calls == 0
@@ -114,11 +114,11 @@ async def test_intervention_raise_exception_on_respond() -> None:
await runtime.register("name", LoopbackAgent)
loopback = AgentId("name", key="default")
run_context = runtime.start()
runtime.start()
with pytest.raises(InterventionException):
_response = await runtime.send_message(MessageType(), recipient=loopback)
await run_context.stop()
await runtime.stop()
long_running_agent = await runtime.try_get_underlying_agent_instance(loopback, type=LoopbackAgent)
assert long_running_agent.num_calls == 1