diff --git a/lmm_tools/lmm/lmm.py b/lmm_tools/lmm/lmm.py index c2efde5d..e6f954a4 100644 --- a/lmm_tools/lmm/lmm.py +++ b/lmm_tools/lmm/lmm.py @@ -58,7 +58,7 @@ def generate(self, prompt: str, image: Optional[Union[str, Path]]) -> str: ) response = self.client.chat.completions.create( - model="gpt-4-vision-preview", message=message + model="gpt-4-vision-preview", messages=message # type: ignore ) return cast(str, response.choices[0].message.content)