diff --git a/anansyTypes/schemas.py b/anansyTypes/schemas.py index cad7a3a..b858b7c 100644 --- a/anansyTypes/schemas.py +++ b/anansyTypes/schemas.py @@ -31,6 +31,7 @@ class ModifiedBaseModel(PydanticBaseModel): class EmbeddingRequest(ModifiedBaseModel): stemmed_words: List[str] tokens: dict + token_ids: List[int] class EmbeddingResponse(ModifiedBaseModel): embeddings: List[dict]