From 3164e2fde797a8122aa61afaf53c52d9b14c73b7 Mon Sep 17 00:00:00 2001 From: stuartgo Date: Sat, 13 Dec 2025 14:42:27 +0100 Subject: [PATCH] changed embeddingresponse to use tensor --- anansyTypes/schemas.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/anansyTypes/schemas.py b/anansyTypes/schemas.py index 87adaf3..3ed687d 100644 --- a/anansyTypes/schemas.py +++ b/anansyTypes/schemas.py @@ -2,7 +2,9 @@ from pydantic import BaseModel as PydanticBaseModel from typing import List, Any, Dict,Union from .structures import Embedding,Word, Symbol,Phrase from pydantic import field_validator, ConfigDict - +from pydantic_tensor import Tensor +import torch +from typing import Any, Literal class ModifiedBaseModel(PydanticBaseModel): model_config = ConfigDict(arbitrary_types_allowed=True) @@ -34,7 +36,7 @@ class EmbeddingRequest(ModifiedBaseModel): word_ids: List[int] class EmbeddingResponse(ModifiedBaseModel): - embeddings: List[dict] + embeddings: Tensor[torch.Tensor, tuple, Literal["int32", "int64"]] class ResponseRequest(ModifiedBaseModel): text: str