formatting

Author: patricebechard

src/agentlab/llm/chat_api.py

@@ -6,13 +6,12 @@
 from functools import partial
 from typing import Optional
 
+import agentlab.llm.tracking as tracking
 import anthropic
 import openai
-from openai import NOT_GIVEN, OpenAI
-
-import agentlab.llm.tracking as tracking
 from agentlab.llm.base_api import AbstractChatModel, BaseModelArgs
 from agentlab.llm.llm_utils import AIMessage, Discussion
+from openai import NOT_GIVEN, OpenAI
 
 
 def make_system_message(content: str) -> dict:
@@ -584,6 +583,7 @@ def __init__(
             aws_secret_key=os.getenv("AWS_SECRET_KEY"),
         )
 
+
 @dataclass
 class BedrockModelArgs(BaseModelArgs):
     def make_model(self):

src/agentlab/llm/llm_configs.py

@@ -1,13 +1,12 @@
-from openai import NOT_GIVEN
-
 from agentlab.llm.chat_api import (
     AnthropicModelArgs,
     AzureModelArgs,
+    BedrockModelArgs,
     OpenAIModelArgs,
     OpenRouterModelArgs,
     SelfHostedModelArgs,
-    BedrockModelArgs,
 )
+from openai import NOT_GIVEN
 
 default_oss_llms_args = {
     "n_retry_server": 4,
@@ -211,17 +210,17 @@
         temperature=1e-1,
     ),
     # ------------ Anthropic / Bedrock ------------#
-    "bedrock/claude-3.7-sonnet": BedrockModelArgs(
+    "bedrock/claude-3-7-sonnet": BedrockModelArgs(
         model_name="us.anthropic.claude-3-7-sonnet-20250219-v1:0",
         max_new_tokens=16_384,
         temperature=1e-1,
     ),
-    "bedrock/claude-4.0-sonnet": BedrockModelArgs(
+    "bedrock/claude-4-0-sonnet": BedrockModelArgs(
         model_name="us.anthropic.claude-sonnet-4-20250514-v1:0",
         max_new_tokens=16_384,
         temperature=1e-1,
     ),
-    "bedrock/claude-4.5-sonnet": BedrockModelArgs(
+    "bedrock/claude-4-5-sonnet": BedrockModelArgs(
         model_name="us.anthropic.claude-sonnet-4-5-20250929-v1:0",
         max_new_tokens=16_384,
         temperature=1e-1,