wehub-resource-sync
63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
"""The shared OpenAI-chat provider owns explicit request preflight."""
|
|
|
|
from collections.abc import AsyncIterator
|
|
|
|
import pytest
|
|
|
|
from free_claude_code.core.anthropic.models import Message, MessagesRequest
|
|
from free_claude_code.providers.base import BaseProvider, ProviderConfig
|
|
from free_claude_code.providers.openai_chat import OpenAIChatProvider
|
|
|
|
|
|
class RecordingOpenAIProvider(OpenAIChatProvider):
|
|
def __init__(self) -> None:
|
|
self.build_calls: list[tuple[MessagesRequest, bool | None]] = []
|
|
|
|
def _build_request_body(
|
|
self, request: MessagesRequest, thinking_enabled: bool | None = None
|
|
) -> dict:
|
|
self.build_calls.append((request, thinking_enabled))
|
|
return {}
|
|
|
|
|
|
class ProviderWithoutPreflight(BaseProvider):
|
|
async def cleanup(self) -> None:
|
|
return None
|
|
|
|
async def list_model_ids(self) -> frozenset[str]:
|
|
return frozenset()
|
|
|
|
async def stream_response(
|
|
self,
|
|
request: MessagesRequest,
|
|
input_tokens: int = 0,
|
|
*,
|
|
request_id: str | None = None,
|
|
thinking_enabled: bool | None = None,
|
|
) -> AsyncIterator[str]:
|
|
if False:
|
|
yield ""
|
|
|
|
|
|
def test_provider_base_requires_an_explicit_preflight_implementation() -> None:
|
|
with pytest.raises(TypeError, match="preflight_stream"):
|
|
ProviderWithoutPreflight(
|
|
ProviderConfig(api_key="test", base_url="https://test.invalid")
|
|
)
|
|
|
|
|
|
def test_openai_provider_owns_preflight() -> None:
|
|
assert OpenAIChatProvider.preflight_stream is not BaseProvider.preflight_stream
|
|
|
|
|
|
def test_provider_preflight_calls_builder_and_preserves_false() -> None:
|
|
provider = RecordingOpenAIProvider()
|
|
request = MessagesRequest(
|
|
model="test-model",
|
|
messages=[Message(role="user", content="hello")],
|
|
)
|
|
|
|
provider.preflight_stream(request, thinking_enabled=False)
|
|
|
|
assert provider.build_calls == [(request, False)]
|