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