Joseph Doherty
ac6e74ab4c
Implements embed() on FeatherlessClient. Featherless's OpenAI- compatible surface does NOT expose /v1/embeddings at the time of writing, so this implementation raises NotImplementedError rather than issuing a request that would 404. The chat.services.embeddings.generate_embedding wrapper (T112.3) catches the exception and degrades to the zero-vector fallback path (plus the existing T107 warning) — misconfigured callers fail loudly in logs while the request path keeps working. If/when Featherless ships embeddings, swap the body for self._client.embeddings.create(model=..., input=...) guarded by the existing 2-conn semaphore (mirrors generate/stream). The Protocol seam in T112.1 is already wired so no other code needs to change. Adds tests/test_featherless.py pinning the NotImplementedError contract.
33 lines
1.3 KiB
Python
33 lines
1.3 KiB
Python
"""Tests for FeatherlessClient (Phase 4.5+).
|
|
|
|
Phase 4.5 adds an ``embed()`` method to the LLMClient Protocol (T112).
|
|
Featherless does not expose an OpenAI-compatible ``/v1/embeddings``
|
|
endpoint, so its implementation deliberately raises
|
|
``NotImplementedError`` to surface the gap clearly. The
|
|
``generate_embedding`` wrapper catches this and degrades to the
|
|
zero-vector fallback (the existing T107 warning path).
|
|
|
|
If/when Featherless ships embeddings, swap the body for a real call to
|
|
``/v1/embeddings`` and update this test to mock the HTTP layer.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from chat.llm.featherless import FeatherlessClient
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_featherless_embed_raises_not_implemented():
|
|
"""Featherless does not expose ``/v1/embeddings`` — embed() must
|
|
raise ``NotImplementedError`` so callers (``generate_embedding``)
|
|
can degrade to the fallback zero vector + warning rather than
|
|
silently producing useless output."""
|
|
client = FeatherlessClient(api_key="test-key")
|
|
with pytest.raises(NotImplementedError) as excinfo:
|
|
await client.embed("hello world", model="bge-small-en-v1.5")
|
|
# Message should hint at the cause so operators see why their
|
|
# real-model swap fell back.
|
|
assert "embeddings" in str(excinfo.value).lower()
|