Is this package capable of calculating tokens for OpenAI assistant mode and more advanced chats?

[jasonsu123]

Hello, I noticed that the code package you wrote is very impressive. However, is it only capable of counting tokens for regular simple chats?

I saw your code requires the input prompt to include "role", "user", and "content" strings.....

message_prompt = [{ "role": "user", "content": "Hello world"}]

If using the assistant mode with instructions, file search, and uploading files to vector stores for RAG, the calculation might be more complex.

Are the token calculation methods for gpt-4-1106-preview and gpt4o the same? I checked the tokenizer on the official website, but the tokenizer for gpt4o is not yet available: https://platform.openai.com/tokenizer

Currently, my code for calculating tokens is as follows. Is this correct? Thank you.

import tiktoken encoding = tiktoken.encoding_for_model("gpt-4-1106-preview") token_contents = len(encoding.encode(contents))