chore: import upstream snapshot with attribution

2026-07-13 13:12:33 +08:00
commit aacb60a4af
3387 changed files with 981307 additions and 0 deletions
@@ -0,0 +1,122 @@
+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)