chore: import upstream snapshot with attribution

2026-07-13 12:32:26 +08:00
commit 1443d3fdf9
732 changed files with 196602 additions and 0 deletions
@@ -0,0 +1,111 @@
+from __future__ import annotations
+
+import json
+from contextlib import contextmanager
+
+import pytest
+import yuxi.models.providers.cache as cache_module
+from yuxi.models.providers.cache import REDIS_CACHE_KEY, ModelCache, ModelInfo
+
+pytestmark = pytest.mark.unit
+
+
+class _FakeRedis:
+    def __init__(self):
+        self.data: dict[str, str] = {}
+        self.get_calls = 0
+
+    def get(self, key: str) -> str | None:
+        self.get_calls += 1
+        return self.data.get(key)
+
+    def set(self, key: str, value: str) -> bool:
+        self.data[key] = value
+        return True
+
+
+def _patch_redis(monkeypatch: pytest.MonkeyPatch, redis: _FakeRedis) -> None:
+    @contextmanager
+    def fake_sync_redis_client(*args, **kwargs):
+        del args, kwargs
+        yield redis
+
+    monkeypatch.setattr(cache_module, "sync_redis_client", fake_sync_redis_client)
+
+
+def test_model_cache_prefers_model_base_url_override(monkeypatch):
+    saved_cache = {}
+
+    class Provider:
+        is_enabled = True
+        provider_id = "alibaba"
+        api_key = "sk-test"
+        api_key_env = None
+        provider_type = "openai"
+        base_url = "https://dashscope.aliyuncs.com/compatible-mode/v1"
+        embedding_base_url = "https://dashscope.aliyuncs.com/compatible-mode/v1/embeddings"
+        rerank_base_url = "https://dashscope.aliyuncs.com/compatible-api/v1/reranks"
+        headers_json = {}
+        extra_json = {}
+        enabled_models = [
+            {
+                "id": "qwen3-rerank",
+                "type": "rerank",
+                "display_name": "Qwen3 Rerank",
+                "base_url_override": "https://invalid.example/rerank",
+            }
+        ]
+
+    cache = ModelCache()
+    monkeypatch.setattr(cache, "_save_cache", lambda data: saved_cache.update(data))
+
+    cache.rebuild([Provider()])
+
+    assert saved_cache["alibaba:qwen3-rerank"].base_url == "https://invalid.example/rerank"
+
+
+def test_model_cache_loads_from_redis_and_uses_local_ttl(monkeypatch: pytest.MonkeyPatch):
+    redis = _FakeRedis()
+    _patch_redis(monkeypatch, redis)
+    redis.data[REDIS_CACHE_KEY] = json.dumps(
+        {
+            "provider:chat": {
+                "provider_id": "provider",
+                "model_id": "chat",
+                "model_type": "chat",
+                "display_name": "Chat",
+                "api_key": "sk-test",
+                "base_url": "https://example.com/v1",
+                "provider_type": "openai",
+            }
+        }
+    )
+    cache = ModelCache()
+
+    info = cache.get_model_info("provider:chat")
+    cached_info = cache.get_model_info("provider:chat")
+
+    assert info is not None
+    assert cached_info is info
+    assert info.base_url == "https://example.com/v1"
+    assert redis.get_calls == 1
+
+
+def test_model_cache_save_writes_redis_json(monkeypatch: pytest.MonkeyPatch):
+    redis = _FakeRedis()
+    _patch_redis(monkeypatch, redis)
+    cache = ModelCache()
+    info = ModelInfo(
+        provider_id="provider",
+        model_id="chat",
+        model_type="chat",
+        display_name="Chat",
+        api_key="sk-test",
+        base_url="https://example.com/v1",
+        provider_type="openai",
+    )
+
+    cache._save_cache({info.spec: info})
+
+    payload = json.loads(redis.data[REDIS_CACHE_KEY])
+    assert payload[info.spec]["base_url"] == "https://example.com/v1"