How do i trim messages?

[migracing]

I have LiteLLM running and I've been using it with no issues. I want to intercept the messages before they're sent to the api. How do I do this? There seems to be some proxy stuff built in but I can't figure it out.

I saw this as an option:

https://docs.litellm.ai/docs/completion/message_trimming

from litellm import completion
from litellm.utils import trim_messages

response = completion(
    model=model, 
    messages=trim_messages(messages, model, max_tokens=10), # trim_messages ensures tokens(messages) < max_tokens
) 

Where am i supposed to save this python file? What should the whole file look like? where is messages defined? Does this need to be a class? How do I tell LiteLLM that the files exists? I'm unfamiliar with LiteLLM and have spend hours trying out different things. Any direction and help is appreciated.

Thank you.