15 lines
494 B
Python
15 lines
494 B
Python
import pytest
|
|
import httpx
|
|
|
|
|
|
@pytest.mark.parametrize("case", list(range(120)))
|
|
def test_embeddings(api_client, respx_mock, case):
|
|
respx_mock.post("http://llama.test/v1/embeddings").mock(
|
|
return_value=httpx.Response(200, json={"data": [{"embedding": [0.1, 0.2]}]})
|
|
)
|
|
payload = {"model": "model-a.gguf", "input": f"text-{case}"}
|
|
resp = api_client.post("/v1/embeddings", json=payload)
|
|
assert resp.status_code == 200
|
|
data = resp.json()
|
|
assert "data" in data
|