959a391334
Some checks failed
publish docs / publish-docs (push) Has been cancelled
release-please / release-please (push) Has been cancelled
tests / setup (push) Has been cancelled
tests / ${{ matrix.quality-command }} (black) (push) Has been cancelled
tests / ${{ matrix.quality-command }} (mypy) (push) Has been cancelled
tests / ${{ matrix.quality-command }} (ruff) (push) Has been cancelled
tests / test (push) Has been cancelled
tests / all_checks_passed (push) Has been cancelled
Mark stale issues and pull requests / stale (push) Has been cancelled
17 lines
553 B
Python
17 lines
553 B
Python
from fastapi.testclient import TestClient
|
|
|
|
from private_gpt.server.embeddings.embeddings_router import (
|
|
EmbeddingsBody,
|
|
EmbeddingsResponse,
|
|
)
|
|
|
|
|
|
def test_embeddings_generation(test_client: TestClient) -> None:
|
|
body = EmbeddingsBody(input="Embed me")
|
|
response = test_client.post("/v1/embeddings", json=body.model_dump())
|
|
|
|
assert response.status_code == 200
|
|
embedding_response = EmbeddingsResponse.model_validate(response.json())
|
|
assert len(embedding_response.data) > 0
|
|
assert len(embedding_response.data[0].embedding) > 0
|