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
):
def check_model_config(self):