formatting

Author: patricebechard

src/agentlab/llm/chat_api.py

@@ -6,12 +6,13 @@
 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:

src/agentlab/llm/llm_configs.py

@@ -1,3 +1,5 @@
+from openai import NOT_GIVEN
+
 from agentlab.llm.chat_api import (
     AnthropicModelArgs,
     AzureModelArgs,
@@ -6,7 +8,6 @@
     OpenRouterModelArgs,
     SelfHostedModelArgs,
 )
-from openai import NOT_GIVEN
 
 default_oss_llms_args = {
     "n_retry_server": 4,