diff --git a/anansyTypes/schemas.py b/anansyTypes/schemas.py index 2cd5dee..f62c672 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[Embedding] + embeddings: List[dict] class ResponseRequest(ModifiedBaseModel): text: str