OAI: Tokenize chat completion messages

Since chat completion messages are a structure, format the prompt
before checking in the tokenizer.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri
2024-04-15 14:17:16 -04:00
parent ed05f376d9
commit 515b3c2930
3 changed files with 26 additions and 5 deletions

View File

@@ -1,7 +1,7 @@
"""Tokenization types"""
from pydantic import BaseModel
from typing import List
from typing import Dict, List, Union
class CommonTokenRequest(BaseModel):
@@ -23,7 +23,7 @@ class CommonTokenRequest(BaseModel):
class TokenEncodeRequest(CommonTokenRequest):
"""Represents a tokenization request."""
text: str
text: Union[str, List[Dict[str, str]]]
class TokenEncodeResponse(BaseModel):