From f443d56882e448f4ed56e8303442ffc93a9e06f7 Mon Sep 17 00:00:00 2001 From: stuartgo Date: Sat, 13 Dec 2025 14:44:21 +0100 Subject: [PATCH] changed types in embeddingresponse --- anansyTypes/schemas.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/anansyTypes/schemas.py b/anansyTypes/schemas.py index 3ed687d..b86d90a 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["int32", "int64"]] + embeddings: Tensor[torch.Tensor, tuple, Literal["float32", "float64"]] class ResponseRequest(ModifiedBaseModel): text: str