wehub-resource-sync
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
"""Tests for llms.adapters.base — ABC enforcement."""
|
|
|
|
from collections.abc import AsyncIterator
|
|
from typing import Any
|
|
|
|
import pytest
|
|
|
|
from omnigent.llms.adapters.base import BaseAdapter
|
|
|
|
|
|
def test_cannot_instantiate_base_adapter() -> None:
|
|
"""BaseAdapter is abstract and cannot be instantiated directly."""
|
|
with pytest.raises(TypeError, match="abstract"):
|
|
BaseAdapter() # type: ignore[abstract]
|
|
|
|
|
|
def test_subclass_must_implement_chat_completions() -> None:
|
|
"""A subclass that does not implement chat_completions cannot be instantiated."""
|
|
|
|
class IncompleteAdapter(BaseAdapter):
|
|
pass
|
|
|
|
with pytest.raises(TypeError, match="abstract"):
|
|
IncompleteAdapter() # type: ignore[abstract]
|
|
|
|
|
|
def test_concrete_subclass_can_be_instantiated() -> None:
|
|
"""A complete subclass that implements chat_completions can be instantiated."""
|
|
|
|
class ConcreteAdapter(BaseAdapter):
|
|
async def chat_completions(
|
|
self,
|
|
messages: list[dict[str, Any]],
|
|
model: str,
|
|
tools: list[dict[str, Any]] | None,
|
|
stream: bool,
|
|
extra: dict[str, Any],
|
|
*,
|
|
connection_params: dict[str, str] | None = None,
|
|
timeout: int | None = None,
|
|
) -> dict[str, Any] | AsyncIterator[dict[str, Any]]:
|
|
return {"choices": []}
|
|
|
|
adapter = ConcreteAdapter()
|
|
assert isinstance(adapter, BaseAdapter)
|