| |
| |
| |
| |
| |
| |
| """API-layer tests: the /text-embedding and /image-embedding endpoints share one request/response |
| shape. The model-backed services are mocked so no weights are loaded.""" |
|
|
| from fastapi.testclient import TestClient |
| from src import main |
| from src.embeddings import EmbeddingData, EmbeddingsRequest, EmbeddingsResponse |
|
|
| client = TestClient(main.app) |
|
|
|
|
| def test_text_embedding_returns_list_envelope(monkeypatch): |
| async def fake(request: EmbeddingsRequest) -> EmbeddingsResponse: |
| return EmbeddingsResponse( |
| data=[EmbeddingData(index=0, embedding=[0.1, 0.2, 0.3])], |
| model=request.model, |
| ) |
|
|
| monkeypatch.setattr(main.text_embedding_service, "generate_embeddings", fake) |
|
|
| response = client.post("/text-embedding", json={"model": "nomic-text", "input": "hello"}) |
|
|
| assert response.status_code == 200 |
| payload = response.json() |
| assert payload["object"] == "list" |
| assert payload["model"] == "nomic-text" |
| assert payload["data"][0]["embedding"] == [0.1, 0.2, 0.3] |
|
|
|
|
| def test_image_embedding_returns_list_envelope(monkeypatch): |
| async def fake(request: EmbeddingsRequest) -> EmbeddingsResponse: |
| return EmbeddingsResponse( |
| data=[EmbeddingData(index=0, embedding=[1.0, 2.0])], |
| model=request.model, |
| ) |
|
|
| monkeypatch.setattr(main.image_embedding_service, "generate_embeddings", fake) |
|
|
| response = client.post("/image-embedding", json={"model": "nomic-vision", "input": "BASE64IMG"}) |
|
|
| assert response.status_code == 200 |
| assert response.json()["data"][0]["embedding"] == [1.0, 2.0] |
|
|
|
|
| def test_text_embedding_accepts_batch_input(monkeypatch): |
| async def fake(request: EmbeddingsRequest) -> EmbeddingsResponse: |
| items = request.input if isinstance(request.input, list) else [request.input] |
| return EmbeddingsResponse( |
| data=[EmbeddingData(index=i, embedding=[float(i)]) for i in range(len(items))], |
| model=request.model, |
| ) |
|
|
| monkeypatch.setattr(main.text_embedding_service, "generate_embeddings", fake) |
|
|
| response = client.post("/text-embedding", json={"model": "m", "input": ["a", "b"]}) |
|
|
| assert response.status_code == 200 |
| assert [d["index"] for d in response.json()["data"]] == [0, 1] |
|
|
|
|
| def test_text_embedding_rejects_missing_model(): |
| response = client.post("/text-embedding", json={"input": "hello"}) |
|
|
| assert response.status_code == 422 |
|
|
