From 2d26ff04453686090b89dfadde34a913d68bdea9 Mon Sep 17 00:00:00 2001 From: stuartgo Date: Sat, 13 Dec 2025 14:48:22 +0100 Subject: [PATCH] change so embeddingresponse returns list of tensors --- anansyTypes/schemas.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/anansyTypes/schemas.py b/anansyTypes/schemas.py index b86d90a..b5beaa0 100644 --- a/anansyTypes/schemas.py +++ b/anansyTypes/schemas.py @@ -36,7 +36,7 @@ class EmbeddingRequest(ModifiedBaseModel): word_ids: List[int] class EmbeddingResponse(ModifiedBaseModel): - embeddings: Tensor[torch.Tensor, tuple, Literal["float32", "float64"]] + embeddings: List[Tensor[torch.Tensor, tuple, Literal["float32", "float64"]]] class ResponseRequest(ModifiedBaseModel): text: str