Api: Add token endpoints

Support for encoding and decoding with various parameters.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri
2023-11-14 22:43:37 -05:00
parent 2d741653c3
commit 8fea5391a8
3 changed files with 60 additions and 1 deletions

30
OAI/types/token.py Normal file
View File

@@ -0,0 +1,30 @@
from pydantic import BaseModel
from typing import List
class CommonTokenRequest(BaseModel):
add_bos: bool = True
encode_special_tokens: bool = True
decode_special_tokens: bool = True
def get_params(self):
return {
"add_bos": self.add_bos,
"encode_special_tokens": self.encode_special_tokens,
"decode_special_tokens": self.decode_special_tokens
}
class TokenEncodeRequest(CommonTokenRequest):
text: str
class TokenEncodeResponse(BaseModel):
tokens: List[int]
length: int
class TokenDecodeRequest(CommonTokenRequest):
tokens: List[int]
class TokenDecodeResponse(BaseModel):
text: str
class TokenCountResponse(BaseModel):
length: int