mirror of
https://hubproxy.babadafafafafa.cn/https://github.com/jxxghp/MoviePilot.git
synced 2026-09-20 08:03:34 +08:00
102 lines
4.4 KiB
Python
102 lines
4.4 KiB
Python
"""OpenCode 官方端点的会话路由请求头回归测试。"""
|
|
|
|
import asyncio
|
|
from unittest.mock import AsyncMock
|
|
|
|
import httpx
|
|
import pytest
|
|
|
|
from app.agent.llm import helper
|
|
from app.agent.llm.helper import LLMHelper
|
|
|
|
|
|
@pytest.mark.parametrize("base_url", [
|
|
"https://opencode.ai/zen/v1",
|
|
"https://opencode.ai/zen/go/v1",
|
|
])
|
|
@pytest.mark.parametrize("cache_key", ["moviepilot-agent-private-hash", None])
|
|
def test_opencode_model_sends_stable_session_on_every_request(monkeypatch, base_url, cache_key):
|
|
"""主对话与独立调用经过真实 SDK 后仍携带稳定标识，工具绑定不丢失请求头。"""
|
|
requests = []
|
|
|
|
def respond(request):
|
|
"""在本地捕获 SDK 请求并返回最小聊天响应，禁止真实出站。"""
|
|
requests.append(request)
|
|
return httpx.Response(200, json={
|
|
"id": "chatcmpl-test",
|
|
"object": "chat.completion",
|
|
"created": 0,
|
|
"model": "test-model",
|
|
"choices": [{"index": 0, "message": {"role": "assistant", "content": "OK"},
|
|
"finish_reason": "stop"}],
|
|
})
|
|
|
|
transport = httpx.MockTransport(respond)
|
|
with httpx.Client(transport=transport) as client:
|
|
async def invoke():
|
|
"""使用离线运行时创建模型，并验证同步与异步请求使用同一标识。"""
|
|
async with httpx.AsyncClient(transport=transport) as async_client:
|
|
monkeypatch.setattr(
|
|
helper, "_build_httpx_client",
|
|
lambda _proxy, **kwargs: async_client if kwargs.get("async_client") else client,
|
|
)
|
|
runtime = AsyncMock()
|
|
runtime.resolve_runtime.return_value = {
|
|
"runtime": "openai_compatible", "model_id": "test-model",
|
|
"api_key": "test-key", "base_url": base_url,
|
|
}
|
|
model = await LLMHelper.get_llm(
|
|
provider="opencode", model="test-model", user_agent="",
|
|
use_proxy=False, api_protocol="chat_completions", web_search_mode="disabled",
|
|
prompt_cache_key=cache_key, provider_runtime=runtime,
|
|
)
|
|
bound = model.bind_tools([{
|
|
"type": "function", "function": {"name": "example", "description": "测试工具",
|
|
"parameters": {"type": "object", "properties": {}}},
|
|
}])
|
|
assert bound.invoke("hello").content == "OK"
|
|
assert (await bound.ainvoke("again")).content == "OK"
|
|
|
|
asyncio.run(invoke())
|
|
|
|
assert len(requests) == 2
|
|
session = requests[0].headers["x-opencode-session"]
|
|
assert session
|
|
assert requests[1].headers["x-opencode-session"] == session
|
|
if cache_key:
|
|
assert session == cache_key
|
|
assert requests[0].headers["user-agent"] == "MoviePilot"
|
|
|
|
|
|
@pytest.mark.parametrize("base_url", [
|
|
"https://opencode.ai/zen/go/v1", "https://OPENCODE.AI/zen/v1",
|
|
])
|
|
def test_opencode_independent_models_have_distinct_sessions(base_url):
|
|
"""没有对话标识的独立模型互不共用会话，并保留自定义 UA 和模型参数。"""
|
|
original = {"user-agent": "custom-client/1.0", "X-Test": "value"}
|
|
options = {"extra_body": {"test": True}}
|
|
headers = [LLMHelper._build_openai_prompt_cache_options(
|
|
provider="custom", base_url=base_url, use_responses_api=True,
|
|
prompt_cache_key=None, default_headers=original, model_kwargs=options,
|
|
) for _ in range(2)]
|
|
assert headers[0][0]["x-opencode-session"] != headers[1][0]["x-opencode-session"]
|
|
assert headers[0][0]["user-agent"] == "custom-client/1.0"
|
|
assert "User-Agent" not in headers[0][0]
|
|
assert headers[0][0]["X-Test"] == "value"
|
|
assert headers[0][1] == options
|
|
assert "x-opencode-session" not in original
|
|
|
|
|
|
@pytest.mark.parametrize("base_url", [
|
|
"https://opencode.ai.example/zen/go/v1", "https://proxy.example/opencode.ai",
|
|
"https://opencode.ai@proxy.example/v1", "https://[invalid", None,
|
|
])
|
|
def test_opencode_headers_do_not_leak_to_other_hosts(base_url):
|
|
"""供应商名称与路径不能代替官方主机校验，兼容端点保持原有参数。"""
|
|
headers, kwargs = LLMHelper._build_openai_prompt_cache_options(
|
|
provider="opencode", base_url=base_url, use_responses_api=False,
|
|
prompt_cache_key="private-cache-key", default_headers=None, model_kwargs={},
|
|
)
|
|
assert headers is None
|
|
assert kwargs == {}
|