StubTokenCounter
count_tokens_from_messages
- messages (List[OpenAIMessage]): Message list with the chat history in OpenAI API format.
encode
- text (str): The text to encode.
decode
- token_ids (List[int]): List of token IDs to decode.
class StubTokenCounter(BaseTokenCounter):
def count_tokens_from_messages(self, messages: List[OpenAIMessage]):
def encode(self, text: str):
def decode(self, token_ids: List[int]):
class StubModel(BaseModelBackend):
def __init__(
self,
model_type: Union[ModelType, str],
model_config_dict: Optional[Dict[str, Any]] = None,
api_key: Optional[str] = None,
url: Optional[str] = None,
token_counter: Optional[BaseTokenCounter] = None,
timeout: Optional[float] = None,
max_retries: int = 3
):
def token_counter(self):
def _run(
self,
messages: List[OpenAIMessage],
response_format: Optional[Type[BaseModel]] = None,
tools: Optional[List[Dict[str, Any]]] = None
):