mirror of
https://github.com/HKUDS/nanobot.git
synced 2026-09-04 02:01:48 +03:00
fix(agent): default request concurrency to unlimited (#5572)
* fix(agent): default request concurrency to unlimited * test(agent): clarify session serialization coverage
This commit is contained in:
@@ -0,0 +1,47 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from types import SimpleNamespace
|
||||
from unittest.mock import MagicMock
|
||||
|
||||
import pytest
|
||||
|
||||
|
||||
def _provider() -> MagicMock:
|
||||
provider = MagicMock()
|
||||
provider.get_default_model.return_value = "test-model"
|
||||
provider.generation = SimpleNamespace(
|
||||
max_tokens=4096,
|
||||
temperature=0.1,
|
||||
reasoning_effort=None,
|
||||
)
|
||||
return provider
|
||||
|
||||
|
||||
def test_request_concurrency_is_unlimited_by_default(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
loop_factory,
|
||||
) -> None:
|
||||
monkeypatch.delenv("NANOBOT_MAX_CONCURRENT_REQUESTS", raising=False)
|
||||
|
||||
loop = loop_factory(provider=_provider(), patch_deps=True)
|
||||
|
||||
assert loop._concurrency_gate is None
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_positive_request_concurrency_keeps_explicit_cap(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
loop_factory,
|
||||
) -> None:
|
||||
monkeypatch.setenv("NANOBOT_MAX_CONCURRENT_REQUESTS", "2")
|
||||
loop = loop_factory(provider=_provider(), patch_deps=True)
|
||||
gate = loop._concurrency_gate
|
||||
|
||||
assert gate is not None
|
||||
for _ in range(2):
|
||||
await gate.acquire()
|
||||
try:
|
||||
assert gate.locked()
|
||||
finally:
|
||||
for _ in range(2):
|
||||
gate.release()
|
||||
@@ -254,7 +254,7 @@ class TestDispatch:
|
||||
assert isinstance(second.event, StreamEndEvent)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_processing_lock_serializes(self):
|
||||
async def test_same_session_dispatches_serialize(self):
|
||||
from nanobot.bus.events import InboundMessage, OutboundMessage
|
||||
|
||||
loop, bus = _make_loop()
|
||||
|
||||
Reference in New Issue
Block a user