getsimilarwordrequest now uses tensor

This commit is contained in:
stuart committed 2025-12-13 15:08:05 +01:00
1 parent 2d26ff0445
commit 2473c9895f
1 file changed
+1 -1
+1 -1
View File
@@ -64,7 +64,7 @@ class GetWordResponse(ModifiedBaseModel):
word: Union[dict, None]
class GetSimilarWordRequest(ModifiedBaseModel):
embedding: dict
embedding: Tensor[torch.Tensor, tuple, Literal["float32", "float64"]]
class GetAllPhrasesResponse(ModifiedBaseModel):
phrases: List[dict]