53 lines
1.0 KiB
Python
53 lines
1.0 KiB
Python
from pydantic import BaseModel as PydanticBaseModel
|
|
from typing import List, Any, Dict,Union
|
|
from structures import Embedding,Word, Symbol,Phrase
|
|
|
|
|
|
|
|
class EmbedRequest(PydanticBaseModel):
|
|
stemmed_words: List[str]
|
|
tokens: dict
|
|
class EmbeddingResponse(PydanticBaseModel):
|
|
embeddings: List[Embedding]
|
|
|
|
|
|
class ResponseRequest(PydanticBaseModel):
|
|
text: str
|
|
context: Dict[str, Any]
|
|
|
|
|
|
class PreprocessRequest(PydanticBaseModel):
|
|
text: str
|
|
|
|
class PreprocessResponse(PydanticBaseModel):
|
|
raw_words: List[str]
|
|
tokens: Dict[str, Any]
|
|
|
|
|
|
class GetSymbolRequest(PydanticBaseModel):
|
|
lexeme_id: int
|
|
|
|
class GetSymbolResponse(PydanticBaseModel):
|
|
symbol: Symbol
|
|
|
|
class GetWordRequest(PydanticBaseModel):
|
|
word_text: str
|
|
stemmed: bool = False
|
|
|
|
|
|
class GetWordResponse(PydanticBaseModel):
|
|
word:Union[Word,None]
|
|
|
|
class GetSimilarWordRequest(PydanticBaseModel):
|
|
embedding: Embedding
|
|
|
|
class GetAllPhrasesResponse(PydanticBaseModel):
|
|
phrases: List[Phrase]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|