diff --git a/vision_agent/lmm/lmm.py b/vision_agent/lmm/lmm.py index cd0729bd..ddcb495b 100644 --- a/vision_agent/lmm/lmm.py +++ b/vision_agent/lmm/lmm.py @@ -381,6 +381,7 @@ def generate( class ClaudeSonnetLMM(LMM): r"""An LMM class for Anthropic's Claude Sonnet model.""" + def __init__( self, api_key: Optional[str] = None, @@ -464,4 +465,4 @@ def generate( messages=[{"role": "user", "content": content}], **self.kwargs, ) - return cast(str, response.content[0].text) \ No newline at end of file + return cast(str, response.content[0].text)