wehub-resource-sync
123 lines
4.0 KiB
Python
123 lines
4.0 KiB
Python
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from opensquilla.engine.runtime import TurnRunner
|
|
from opensquilla.engine.types import AgentConfig
|
|
from opensquilla.gateway.config import GatewayConfig
|
|
|
|
|
|
class _SessionConfigManager:
|
|
def __init__(self, config: object | None) -> None:
|
|
self.config = config
|
|
|
|
def get_session_config(self, session_key: str) -> object | None:
|
|
return self.config
|
|
|
|
|
|
def test_resolve_agent_request_timeout_prefers_explicit_value(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
monkeypatch.setenv("OPENSQUILLA_AGENT_REQUEST_TIMEOUT", "22")
|
|
runner = TurnRunner(
|
|
provider_selector=None,
|
|
session_manager=_SessionConfigManager(
|
|
SimpleNamespace(agent_request_timeout_seconds=11.0)
|
|
),
|
|
config=GatewayConfig(agent_request_timeout_seconds=33.0),
|
|
)
|
|
|
|
assert runner._resolve_agent_request_timeout("agent:main:test", 44.0) == 44.0
|
|
|
|
|
|
def test_resolve_agent_request_timeout_prefers_session_config(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
monkeypatch.setenv("OPENSQUILLA_AGENT_REQUEST_TIMEOUT", "22")
|
|
runner = TurnRunner(
|
|
provider_selector=None,
|
|
session_manager=_SessionConfigManager(
|
|
SimpleNamespace(agent_request_timeout_seconds=11.0)
|
|
),
|
|
config=GatewayConfig(agent_request_timeout_seconds=33.0),
|
|
)
|
|
|
|
assert runner._resolve_agent_request_timeout("agent:main:test") == 11.0
|
|
|
|
|
|
def test_resolve_agent_request_timeout_prefers_env_over_gateway_config(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
monkeypatch.setenv("OPENSQUILLA_AGENT_REQUEST_TIMEOUT", "22")
|
|
runner = TurnRunner(
|
|
provider_selector=None,
|
|
session_manager=_SessionConfigManager(None),
|
|
config=GatewayConfig(agent_request_timeout_seconds=33.0),
|
|
)
|
|
|
|
assert runner._resolve_agent_request_timeout("agent:main:test") == 22.0
|
|
|
|
|
|
def test_resolve_agent_request_timeout_uses_gateway_config(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
monkeypatch.delenv("OPENSQUILLA_AGENT_REQUEST_TIMEOUT", raising=False)
|
|
runner = TurnRunner(
|
|
provider_selector=None,
|
|
config=GatewayConfig(agent_request_timeout_seconds=33.0),
|
|
)
|
|
|
|
assert runner._resolve_agent_request_timeout("agent:main:test") == 33.0
|
|
|
|
|
|
def test_resolve_agent_request_timeout_falls_back_to_llm_request_timeout(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
"""When the new agent_request_timeout_seconds is unset, the legacy
|
|
llm_request_timeout_seconds must still apply so existing deployments
|
|
keep their tuned value. This is the regression-prevention test for
|
|
operators who set llm_request_timeout_seconds before the new knob
|
|
existed.
|
|
"""
|
|
monkeypatch.delenv("OPENSQUILLA_AGENT_REQUEST_TIMEOUT", raising=False)
|
|
runner = TurnRunner(
|
|
provider_selector=None,
|
|
config=GatewayConfig(llm_request_timeout_seconds=77.0),
|
|
)
|
|
|
|
assert runner._resolve_agent_request_timeout("agent:main:test") == 77.0
|
|
|
|
|
|
def test_resolve_agent_request_timeout_uses_agent_default_without_config(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
monkeypatch.delenv("OPENSQUILLA_AGENT_REQUEST_TIMEOUT", raising=False)
|
|
runner = TurnRunner(provider_selector=None, config=None)
|
|
|
|
assert (
|
|
runner._resolve_agent_request_timeout("agent:main:test")
|
|
== AgentConfig().request_timeout
|
|
)
|
|
|
|
|
|
def test_resolve_agent_request_timeout_invalid_env_falls_through(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
monkeypatch.setenv("OPENSQUILLA_AGENT_REQUEST_TIMEOUT", "not-a-float")
|
|
runner = TurnRunner(
|
|
provider_selector=None,
|
|
session_manager=_SessionConfigManager(None),
|
|
config=GatewayConfig(agent_request_timeout_seconds=33.0),
|
|
)
|
|
|
|
assert runner._resolve_agent_request_timeout("agent:main:test") == 33.0
|
|
|
|
|
|
def test_resolve_agent_request_timeout_rejects_invalid_explicit_value() -> None:
|
|
runner = TurnRunner(provider_selector=None, config=GatewayConfig())
|
|
|
|
with pytest.raises(ValueError, match="request_timeout"):
|
|
runner._resolve_agent_request_timeout("agent:main:test", 0.0)
|