[Bug] ModuleNotFoundError: No module named 'mistralai.models.chat_completion'

[WahahaZeng]

Describe the bug

Traceback (most recent call last):
File "/home/xyz/HarmBench/generate_completions.py", line 9, in
from api_models import api_models_map
File "/home/xyz/HarmBench/api_models.py", line 11, in
from mistralai.models.chat_completion import ChatMessage
ModuleNotFoundError: No module named 'mistralai.models.chat_completion'

The version is am using is the latest one
$ pip list |grep mistralai
mistralai 1.2.5

I also tried some old version like "mistralai==1.2.0", "mistralai==1.0.1", but the bug is still there

- OS: Linux
- Python version: 3.10.15

This bug was reported before in #640