My recommendation is use model name as you find in the documentations (.e.g gemini-2.0-flash-lite)

Why? Pricing differs for each model and Model clarity essential during debugging & model evaluation purposes.

Suggestion Don't hard-code the model names. For your app, define a external CLI variables and fetch it using os.getenv

For Google Cloud, check google-cloud-retired-models.