diff --git a/vizro-ai/src/vizro_ai/_llm_models.py b/vizro-ai/src/vizro_ai/_llm_models.py index 74ba126db..553ebb4ef 100644 --- a/vizro-ai/src/vizro_ai/_llm_models.py +++ b/vizro-ai/src/vizro_ai/_llm_models.py @@ -88,6 +88,6 @@ def _get_model_name(model: BaseChatModel) -> str: if __name__ == "__main__": - llm_chat_openai = _get_llm_model(model="gpt-3.5-turbo") + llm_chat_openai = _get_llm_model(model="gpt-4o-mini") print(repr(llm_chat_openai)) # noqa: T201 print(llm_chat_openai.model_name) # noqa: T201