import tiktoken

def count_tokens(prompt, model="gpt-4o-mini"):
    encoding = tiktoken.encoding_for_model(model)
    return len(encoding.encode(prompt))

print(count_tokens("Simple prompt example..."))