mirror of
https://github.com/Alishahryar1/free-claude-code.git
synced 2026-06-02 06:13:46 +02:00
Alishahryar1
fbb1d6586d
Route these providers through POST /messages with vendor headers and bases (including Kimi model list on OpenAI /v1/models). Remove Z.ai from OpenAI-chat server-tool rejection; extend tests and README.
122 lines
3.5 KiB
Python
122 lines
3.5 KiB
Python
"""Tests for Kimi (Moonshot) native Anthropic Messages provider."""
|
|
|
|
from contextlib import asynccontextmanager
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from api.models.anthropic import Message, MessagesRequest
|
|
from config.constants import ANTHROPIC_DEFAULT_MAX_OUTPUT_TOKENS
|
|
from providers.base import ProviderConfig
|
|
from providers.defaults import KIMI_DEFAULT_BASE
|
|
from providers.exceptions import InvalidRequestError
|
|
from providers.kimi import KimiProvider
|
|
|
|
|
|
@pytest.fixture
|
|
def kimi_config():
|
|
return ProviderConfig(
|
|
api_key="test_kimi_key",
|
|
base_url=KIMI_DEFAULT_BASE,
|
|
rate_limit=10,
|
|
rate_window=60,
|
|
enable_thinking=True,
|
|
)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def mock_rate_limiter():
|
|
@asynccontextmanager
|
|
async def _slot():
|
|
yield
|
|
|
|
with patch("providers.anthropic_messages.GlobalRateLimiter") as mock:
|
|
instance = mock.get_scoped_instance.return_value
|
|
|
|
async def _passthrough(fn, *args, **kwargs):
|
|
return await fn(*args, **kwargs)
|
|
|
|
instance.execute_with_retry = AsyncMock(side_effect=_passthrough)
|
|
instance.concurrency_slot.side_effect = _slot
|
|
yield instance
|
|
|
|
|
|
@pytest.fixture
|
|
def kimi_provider(kimi_config):
|
|
return KimiProvider(kimi_config)
|
|
|
|
|
|
def test_init(kimi_config):
|
|
with patch("httpx.AsyncClient") as mock_client:
|
|
provider = KimiProvider(kimi_config)
|
|
assert provider._api_key == "test_kimi_key"
|
|
assert provider._base_url == KIMI_DEFAULT_BASE
|
|
assert mock_client.called
|
|
|
|
|
|
def test_request_headers(kimi_provider):
|
|
h = kimi_provider._request_headers()
|
|
assert h["Authorization"] == "Bearer test_kimi_key"
|
|
assert h["anthropic-version"] == "2023-06-01"
|
|
|
|
|
|
def test_build_request_body_native(kimi_provider):
|
|
request = MessagesRequest(
|
|
model="kimi-k2.5",
|
|
max_tokens=50,
|
|
messages=[Message(role="user", content="hi")],
|
|
)
|
|
body = kimi_provider._build_request_body(request)
|
|
assert body["model"] == "kimi-k2.5"
|
|
assert body["stream"] is True
|
|
assert body["messages"][0]["role"] == "user"
|
|
|
|
|
|
def test_build_request_body_default_max_tokens(kimi_provider):
|
|
request = MessagesRequest(
|
|
model="m",
|
|
messages=[Message(role="user", content="x")],
|
|
)
|
|
body = kimi_provider._build_request_body(request)
|
|
assert body["max_tokens"] == ANTHROPIC_DEFAULT_MAX_OUTPUT_TOKENS
|
|
|
|
|
|
def test_build_request_body_rejects_extra_body(kimi_provider):
|
|
request = MessagesRequest.model_validate(
|
|
{
|
|
"model": "m",
|
|
"messages": [{"role": "user", "content": "x"}],
|
|
"extra_body": {"x": 1},
|
|
}
|
|
)
|
|
with pytest.raises(InvalidRequestError, match="does not support extra_body"):
|
|
kimi_provider._build_request_body(request)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_model_list_uses_moonshot_openai_url(kimi_provider):
|
|
called: dict[str, str] = {}
|
|
|
|
async def fake_get(url: str, **_k):
|
|
called["url"] = url
|
|
mock_resp = MagicMock()
|
|
mock_resp.raise_for_status = lambda: None
|
|
mock_resp.json = lambda: {"data": [{"id": "kimi-k2.5"}]}
|
|
mock_resp.aclose = AsyncMock()
|
|
return mock_resp
|
|
|
|
kimi_provider._client.get = fake_get
|
|
|
|
await kimi_provider.list_model_infos()
|
|
|
|
assert called["url"] == "https://api.moonshot.ai/v1/models"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_cleanup_aclose(kimi_provider):
|
|
kimi_provider._client = AsyncMock()
|
|
|
|
await kimi_provider.cleanup()
|
|
|
|
kimi_provider._client.aclose.assert_awaited_once()
|