From bde97648e67fdba8cde8a773e3ff0b88607b61f6 Mon Sep 17 00:00:00 2001
From: stuartgo <ottersenstuart@gmail.com>
Date: Sat, 13 Dec 2025 15:19:44 +0100
Subject: [PATCH] reverted to using embedding class

---
 anansyTypes/schemas.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/anansyTypes/schemas.py b/anansyTypes/schemas.py
index 04e8e91..2cd5dee 100644
--- a/anansyTypes/schemas.py
+++ b/anansyTypes/schemas.py
@@ -36,7 +36,7 @@ class EmbeddingRequest(ModifiedBaseModel):
     word_ids: List[int]
 
 class EmbeddingResponse(ModifiedBaseModel):
-    embeddings: List[Tensor[torch.Tensor, tuple, Literal["float32", "float64"]]]
+    embeddings: List[Embedding]
 
 class ResponseRequest(ModifiedBaseModel):
     text: str