From 06e29f15c84a34f96b37e60d78c68e5f78cb0897 Mon Sep 17 00:00:00 2001
From: DragonAssassin-one <2819637808@qq.com>
Date: Mon, 11 May 2026 11:28:05 +0800
Subject: [PATCH] test: Add deterministic encoding stub for prompt prefix
 affinity check tests

---
 .../test_prompt_prefix_affinity_check.py      | 29 +++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/tests/test_litellm/router_utils/pre_call_checks/test_prompt_prefix_affinity_check.py b/tests/test_litellm/router_utils/pre_call_checks/test_prompt_prefix_affinity_check.py
index d7a0155c5d7..222d0c20f23 100644
--- a/tests/test_litellm/router_utils/pre_call_checks/test_prompt_prefix_affinity_check.py
+++ b/tests/test_litellm/router_utils/pre_call_checks/test_prompt_prefix_affinity_check.py
@@ -1,15 +1,44 @@
 import os
 import sys
 
+import hashlib
+import re
+
 import pytest
 
 sys.path.insert(0, os.path.abspath("../.."))
 
+import litellm.router_utils.pre_call_checks.prompt_prefix_affinity_check as affinity_module
+
 from litellm.router_utils.pre_call_checks.prompt_prefix_affinity_check import (
     PromptPrefixAffinityCheck,
 )
 
 
+def _fake_encode(*, model: str, text: str):
+    """
+    Deterministic, offline tokenizer stub.
+
+    Tests in `tests/test_litellm/` must not make network calls. The real
+    `litellm.utils.encode()` can trigger a tiktoken download on fresh CI runners.
+    We stub it with a pure function that produces stable token IDs derived from
+    token content, so prefix routing behavior remains meaningful.
+    """
+
+    _ = model  # model selection is irrelevant for this stub
+    tokens = re.findall(r"\w+|[^\w\s]", text, flags=re.UNICODE)
+    token_ids = []
+    for token in tokens:
+        digest = hashlib.sha256(token.encode("utf-8")).digest()
+        token_ids.append(int.from_bytes(digest[:4], "big"))
+    return token_ids
+
+
+@pytest.fixture(autouse=True)
+def _stub_encode(monkeypatch):
+    monkeypatch.setattr(affinity_module, "encode", _fake_encode)
+
+
 def _deployments():
     return [
         {"model_info": {"id": "deployment-a"}},