imports

zitro677 · Oct 19, 2024 · 3411dc0 · 3411dc0
1 parent f7c62fa
commit 3411dc0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/ai_scientist/generate_ideas.py b/ai_scientist/generate_ideas.py
@@ -1,11 +1,12 @@
-import backoff
 import json
 import os
 import os.path as osp
-import requests
 import time
 from typing import List, Dict, Union
 
+import backoff
+import requests
+
 from ai_scientist.llm import get_response_from_llm, extract_json_between_markers, create_client, AVAILABLE_LLMS
 
 S2_API_KEY = os.getenv("S2_API_KEY")

diff --git a/ai_scientist/llm.py b/ai_scientist/llm.py
@@ -1,8 +1,11 @@
-import backoff
 import json
-import openai
+import os
 import re
 
+import anthropic
+import backoff
+import openai
+
 MAX_NUM_TOKENS = 4096
 
 AVAILABLE_LLMS = [