models/study.py

from pydantic import BaseModel
from typing import List

class WordList(BaseModel):
    words: List[str]

class WordAnalysis(BaseModel):
    word: str
    n_neighbors: int = 20