Correct model_id

nachollorca · Jan 20, 2025 · 0a74fab · 0a74fab
1 parent 283522e
commit 0a74fab
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/src/lamine/providers/anthropic.py b/src/lamine/providers/anthropic.py
@@ -4,20 +4,14 @@
 
 
 class Anthropic(Provider):
-    model_ids = [
-        "claude-3.5-sonnet-latest",
-        "claude-3-5-haiku-latest",
-        "claude-3-haiku-20240307"
-    ]
+    model_ids = ["claude-3-5-sonnet-latest", "claude-3-5-haiku-latest", "claude-3-haiku-20240307"]
 
     def get_answer(self, model: Model, conversation: list[Message], **kwargs) -> Answer:
         client = Anthropic_source()
         messages = [message.dict for message in conversation]
-        response = client.messages.create(
-            model=model.id, messages=messages, max_tokens=4096, **kwargs
-        )
+        response = client.messages.create(model=model.id, messages=messages, max_tokens=4096, **kwargs)
         return Answer(
             content=response.content[0].text,
             tokens_in=response.usage.input_tokens,
             tokens_out=response.usage.output_tokens,
-        )
+        )