102 lines
3.3 KiB
Python
102 lines
3.3 KiB
Python
import httpx
|
|
import pytest
|
|
from openai import APITimeoutError, AuthenticationError, InternalServerError, RateLimitError
|
|
|
|
from apps.modelhub.crypto import encrypt_secret
|
|
from apps.modelhub.models import ModelDeployment, ModelInvocation, ModelProvider
|
|
from apps.modelhub.services import CompletionResult, execute_invocation
|
|
|
|
|
|
def create_invocation():
|
|
provider = ModelProvider.objects.create(
|
|
name="模型渠道",
|
|
base_url="https://models.example.com/v1",
|
|
api_key_encrypted=encrypt_secret("secret"),
|
|
)
|
|
deployment = ModelDeployment.objects.create(
|
|
provider=provider,
|
|
display_name="支持模型",
|
|
model_name="support-model",
|
|
)
|
|
return ModelInvocation.objects.create(
|
|
deployment=deployment,
|
|
input_snapshot={"messages": [{"role": "user", "content": "hello"}]},
|
|
)
|
|
|
|
|
|
def test_transient_error_retries_once(db, monkeypatch):
|
|
invocation = create_invocation()
|
|
calls = []
|
|
|
|
def request(*args, **kwargs):
|
|
calls.append(1)
|
|
if len(calls) == 1:
|
|
raise APITimeoutError(request=None)
|
|
return CompletionResult("ok", 3, 1)
|
|
|
|
monkeypatch.setattr("apps.modelhub.services._request_completion", request)
|
|
monkeypatch.setattr("apps.modelhub.services.sleep", lambda *_: None)
|
|
|
|
execute_invocation(invocation, invocation.input_snapshot["messages"])
|
|
|
|
invocation.refresh_from_db()
|
|
assert len(calls) == 2
|
|
assert invocation.status == "succeeded"
|
|
assert invocation.retry_count == 1
|
|
|
|
|
|
def test_non_transient_error_fails_without_exposing_raw_error(db, monkeypatch):
|
|
invocation = create_invocation()
|
|
|
|
def request(*args, **kwargs):
|
|
raise ValueError("secret provider detail")
|
|
|
|
monkeypatch.setattr("apps.modelhub.services._request_completion", request)
|
|
|
|
execute_invocation(invocation, invocation.input_snapshot["messages"])
|
|
|
|
invocation.refresh_from_db()
|
|
assert invocation.status == "failed"
|
|
assert invocation.error_code == "unexpected_error"
|
|
assert "secret provider detail" not in invocation.error_message
|
|
|
|
|
|
def provider_error(error_type, status_code):
|
|
request = httpx.Request("POST", "https://models.example.com/v1/chat/completions")
|
|
response = httpx.Response(status_code, request=request)
|
|
return error_type("provider raw error", response=response, body={})
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("error", "expected_code", "expected_calls"),
|
|
[
|
|
(provider_error(AuthenticationError, 401), "authentication_failed", 1),
|
|
(provider_error(RateLimitError, 429), "rate_limited", 2),
|
|
(provider_error(InternalServerError, 500), "provider_unavailable", 2),
|
|
],
|
|
)
|
|
def test_provider_error_classification_and_retry(
|
|
db,
|
|
monkeypatch,
|
|
error,
|
|
expected_code,
|
|
expected_calls,
|
|
):
|
|
invocation = create_invocation()
|
|
calls = []
|
|
|
|
def request(*args, **kwargs):
|
|
calls.append(1)
|
|
raise error
|
|
|
|
monkeypatch.setattr("apps.modelhub.services._request_completion", request)
|
|
monkeypatch.setattr("apps.modelhub.services.sleep", lambda *_: None)
|
|
|
|
execute_invocation(invocation, invocation.input_snapshot["messages"])
|
|
|
|
invocation.refresh_from_db()
|
|
assert invocation.status == "failed"
|
|
assert invocation.error_code == expected_code
|
|
assert len(calls) == expected_calls
|
|
assert "provider raw error" not in invocation.error_message
|