🎨 增加并发访问

2026-03-05 21:28:41 +08:00
commit 84c66ccaa7
114 changed files with 35396 additions and 0 deletions
--- a/tools/test_empty_content.py
+++ b/tools/test_empty_content.py
@@ -0,0 +1,145 @@
+#!/usr/bin/env python3
+"""测试空消息内容和 prefill 处理的改进"""
+
+import json
+import requests
+
+BASE_URL = "http://localhost:8080"
+API_KEY = "test-key"
+
+def safe_print_response(response):
+    """安全打印响应，处理非 JSON 情况"""
+    try:
+        data = response.json()
+        print(f"响应: {json.dumps(data, indent=2, ensure_ascii=False)}")
+        return data
+    except (json.JSONDecodeError, ValueError):
+        print(f"响应 (非 JSON): {response.text}")
+        return None
+
+def test_empty_content():
+    """测试空消息内容应返回 400 错误"""
+    print("测试 1: 空消息内容")
+    response = requests.post(
+        f"{BASE_URL}/v1/messages",
+        headers={
+            "x-api-key": API_KEY,
+            "anthropic-version": "2023-06-01",
+            "content-type": "application/json",
+        },
+        json={
+            "model": "claude-sonnet-4",
+            "max_tokens": 1024,
+            "messages": [
+                {"role": "user", "content": ""}
+            ]
+        }
+    )
+    print(f"状态码: {response.status_code}")
+    data = safe_print_response(response)
+    assert response.status_code == 400, "应返回 400 错误"
+    if data:
+        assert "消息内容为空" in data.get("error", {}).get("message", ""), "错误消息应包含'消息内容为空'"
+    print("✓ 测试通过\n")
+
+def test_empty_text_blocks():
+    """测试仅包含空白文本块的消息"""
+    print("测试 2: 仅包含空白文本块")
+    response = requests.post(
+        f"{BASE_URL}/v1/messages",
+        headers={
+            "x-api-key": API_KEY,
+            "anthropic-version": "2023-06-01",
+            "content-type": "application/json",
+        },
+        json={
+            "model": "claude-sonnet-4",
+            "max_tokens": 1024,
+            "messages": [
+                {
+                    "role": "user",
+                    "content": [
+                        {"type": "text", "text": "   "},
+                        {"type": "text", "text": "\n\t"}
+                    ]
+                }
+            ]
+        }
+    )
+    print(f"状态码: {response.status_code}")
+    data = safe_print_response(response)
+    assert response.status_code == 400, "应返回 400 错误"
+    if data:
+        assert "消息内容为空" in data.get("error", {}).get("message", ""), "错误消息应包含'消息内容为空'"
+    print("✓ 测试通过\n")
+
+def test_prefill_with_empty_user():
+    """测试 prefill 场景下空 user 消息"""
+    print("测试 3: Prefill 场景下空 user 消息")
+    response = requests.post(
+        f"{BASE_URL}/v1/messages",
+        headers={
+            "x-api-key": API_KEY,
+            "anthropic-version": "2023-06-01",
+            "content-type": "application/json",
+        },
+        json={
+            "model": "claude-sonnet-4",
+            "max_tokens": 1024,
+            "messages": [
+                {"role": "user", "content": ""},
+                {"role": "assistant", "content": "Hi there"}
+            ]
+        }
+    )
+    print(f"状态码: {response.status_code}")
+    data = safe_print_response(response)
+    assert response.status_code == 400, "应返回 400 错误"
+    if data:
+        assert "消息内容为空" in data.get("error", {}).get("message", ""), "错误消息应包含'消息内容为空'"
+    print("✓ 测试通过\n")
+
+def test_valid_message():
+    """测试正常消息应该成功"""
+    print("测试 4: 正常消息（对照组）")
+    response = requests.post(
+        f"{BASE_URL}/v1/messages",
+        headers={
+            "x-api-key": API_KEY,
+            "anthropic-version": "2023-06-01",
+            "content-type": "application/json",
+        },
+        json={
+            "model": "claude-sonnet-4",
+            "max_tokens": 50,
+            "messages": [
+                {"role": "user", "content": "Say 'test' only"}
+            ]
+        }
+    )
+    print(f"状态码: {response.status_code}")
+    if response.status_code == 200:
+        print("✓ 测试通过：正常消息处理成功\n")
+    else:
+        safe_print_response(response)
+        print()
+
+if __name__ == "__main__":
+    print("=" * 60)
+    print("空消息内容验证测试")
+    print("=" * 60 + "\n")
+
+    try:
+        test_empty_content()
+        test_empty_text_blocks()
+        test_prefill_with_empty_user()
+        test_valid_message()
+        print("=" * 60)
+        print("所有测试通过！")
+        print("=" * 60)
+    except AssertionError as e:
+        print(f"\n✗ 测试失败: {e}")
+    except requests.exceptions.ConnectionError:
+        print("\n✗ 无法连接到服务器，请确保服务正在运行")
+    except Exception as e:
+        print(f"\n✗ 发生错误: {e}")