From 39f39965106b6b7a31fa07e1d5bc8a2e975e6721 Mon Sep 17 00:00:00 2001 From: Dustin Ngo Date: Wed, 20 Mar 2024 14:25:23 -0400 Subject: [PATCH] =?UTF-8?q?Ruff=20=F0=9F=90=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- packages/phoenix-evals/src/phoenix/evals/models/mistralai.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/packages/phoenix-evals/src/phoenix/evals/models/mistralai.py b/packages/phoenix-evals/src/phoenix/evals/models/mistralai.py index 6b4c93af7e..c340bf1ee1 100644 --- a/packages/phoenix-evals/src/phoenix/evals/models/mistralai.py +++ b/packages/phoenix-evals/src/phoenix/evals/models/mistralai.py @@ -110,7 +110,7 @@ async def _async_generate(self, prompt: str, **kwargs: Dict[str, Any]) -> str: invocation_parameters = self.invocation_parameters() invocation_parameters.update(kwargs) response = await self._async_rate_limited_completion( - model=self.model, + model=self.model, messages=self._format_prompt(prompt), **invocation_parameters, ) @@ -127,7 +127,7 @@ async def _async_completion(**kwargs: Any) -> Any: if http_status and http_status == 429: raise MistralRateLimitError() from exc raise exc - + return response.choices[0].message.content return await _async_completion(**kwargs)