import tiktoken
def count_tokens(text: str) -> int:
return len(tiktoken.encoding_for_model("gpt-4o").encode(text))