Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/gateway/test_channel_overrides.py

@@ -0,0 +1,156 @@
+"""Tests for per-channel model and system prompt overrides (Fixes #1955)."""
+
+from unittest.mock import patch
+
+import pytest
+
+from gateway.config import (
+    ChannelOverride,
+    GatewayConfig,
+    Platform,
+    PlatformConfig,
+)
+from gateway.run import _get_channel_override, GatewayRunner
+from gateway.session import SessionSource
+
+
+class TestGetChannelOverride:
+
+
+    def test_no_override_when_channel_not_in_overrides(self):
+        config = GatewayConfig(
+            platforms={
+                Platform.DISCORD: PlatformConfig(
+                    enabled=True,
+                    channel_overrides={
+                        "999": ChannelOverride(model="openrouter/healer-alpha"),
+                    },
+                ),
+            },
+        )
+        assert _get_channel_override(config, Platform.DISCORD, "123") is None
+
+    def test_returns_override_when_channel_matches(self):
+        ov = ChannelOverride(
+            model="openrouter/healer-alpha",
+            provider="openrouter",
+            system_prompt="You are a summarizer.",
+        )
+        config = GatewayConfig(
+            platforms={
+                Platform.DISCORD: PlatformConfig(
+                    enabled=True,
+                    channel_overrides={"1234567890": ov},
+                ),
+            },
+        )
+        result = _get_channel_override(config, Platform.DISCORD, "1234567890")
+        assert result is not None
+        assert result.model == "openrouter/healer-alpha"
+        assert result.provider == "openrouter"
+        assert result.system_prompt == "You are a summarizer."
+
+
+    def test_thread_id_lookup_when_chat_id_misses(self):
+        config = GatewayConfig(
+            platforms={
+                Platform.DISCORD: PlatformConfig(
+                    enabled=True,
+                    channel_overrides={
+                        "thread_99": ChannelOverride(model="topic-model"),
+                    },
+                ),
+            },
+        )
+        result = _get_channel_override(
+            config, Platform.DISCORD, "parent_chan", thread_id="thread_99"
+        )
+        assert result is not None
+        assert result.model == "topic-model"
+
+
+class TestResolveModelForChannel:
+    def test_uses_channel_override_when_present(self):
+        config = GatewayConfig(
+            platforms={
+                Platform.DISCORD: PlatformConfig(
+                    enabled=True,
+                    channel_overrides={
+                        "chan_1": ChannelOverride(model="anthropic/claude-opus-4.6"),
+                    },
+                ),
+            },
+        )
+        runner = object.__new__(GatewayRunner)
+        runner.config = config
+        model = runner._resolve_model_for_channel(Platform.DISCORD, "chan_1")
+        assert model == "anthropic/claude-opus-4.6"
+
+
+class TestGetSystemPromptForChannel:
+    def test_uses_channel_override_when_present(self):
+        config = GatewayConfig(
+            platforms={
+                Platform.DISCORD: PlatformConfig(
+                    enabled=True,
+                    channel_overrides={
+                        "chan_1": ChannelOverride(system_prompt="You are a coding assistant."),
+                    },
+                ),
+            },
+        )
+        runner = object.__new__(GatewayRunner)
+        runner.config = config
+        runner._ephemeral_system_prompt = "Global prompt"
+        prompt = runner._get_system_prompt_for_channel(Platform.DISCORD, "chan_1")
+        assert prompt == "You are a coding assistant."
+
+
+class TestResolveSessionAgentRuntimePriority:
+    """Model/runtime priority: session /model → channel_overrides → global."""
+
+    def test_channel_override_beats_global(self):
+        runner = object.__new__(GatewayRunner)
+        runner._session_model_overrides = {}
+        runner.config = GatewayConfig(
+            platforms={
+                Platform.DISCORD: PlatformConfig(
+                    enabled=True,
+                    channel_overrides={
+                        "chan_1": ChannelOverride(
+                            model="channel/model",
+                            provider="openrouter",
+                        ),
+                    },
+                ),
+            },
+        )
+        source = SessionSource(
+            platform=Platform.DISCORD,
+            chat_id="chan_1",
+            user_id="u1",
+        )
+        with patch("gateway.run._resolve_gateway_model", return_value="global/model"), \
+             patch("gateway.run._resolve_runtime_agent_kwargs", return_value={
+                 "provider": "anthropic",
+                 "api_key": "k",
+                 "base_url": "https://api.anthropic.com",
+                 "api_mode": "chat_completions",
+             }), \
+             patch(
+                 "gateway.run._resolve_runtime_agent_kwargs_for_provider",
+                 return_value={
+                     "provider": "openrouter",
+                     "api_key": "k2",
+                     "base_url": "https://openrouter.ai/api/v1",
+                     "api_mode": "chat_completions",
+                 },
+             ):
+            model, runtime = runner._resolve_session_agent_runtime(
+                source=source,
+                user_config={"model": {"default": "global/model"}},
+            )
+        assert model == "channel/model"
+        assert runtime["provider"] == "openrouter"
+
+