Fix incorrect API usage

PiperOrigin-RevId: 661191024 Change-Id: I04d468faf0b31970b019263284640b8c8cc4e1f6
google-deepmind · Aug 9, 2024 · 59ad795 · 59ad795
1 parent 267516b
commit 59ad795
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/concordia/language_model/mistral_model.py b/concordia/language_model/mistral_model.py
@@ -157,7 +157,7 @@ def _chat_text(
         temperature = sampling.dynamically_adjust_temperature(
             attempts, _MAX_CHAT_ATTEMPTS)
       try:
-        response = self._client.chat(
+        response = self._client.chat.complete(
             model=self._text_model_name,
             messages=messages,
             temperature=temperature,