From 6f3e95ba238be37b388be7179fb6431f86711162 Mon Sep 17 00:00:00 2001 From: Moutasem Akkad Date: Thu, 18 Jul 2024 14:28:30 -0700 Subject: [PATCH] linting --- vision_agent/lmm/lmm.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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)