View source on GitHub |
Calls the API to calculate the number of tokens used in the prompt.
google.generativeai.count_message_tokens(
*,
prompt: discuss_types.MessagePromptOptions = None,
context: (str | None) = None,
examples: (discuss_types.ExamplesOptions | None) = None,
messages: (discuss_types.MessagesOptions | None) = None,
model: model_types.AnyModelNameOptions = DEFAULT_DISCUSS_MODEL,
client: (glm.DiscussServiceAsyncClient | None) = None,
request_options: (helper_types.RequestOptionsType | None) = None
) -> discuss_types.TokenCount