move prompting into agents

ulab-uiuc · May 9, 2024 · 4904390 · 4904390
1 parent a1b73b4
commit 4904390
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/research_town/agents/agent_prompting.py b/research_town/agents/agent_prompting.py
@@ -3,7 +3,10 @@
 
 import openai
 
-from ..utils.paper_collection import get_bert_embedding, neiborhood_search
+from ..utils.paper_collection import (
+    get_bert_embedding,
+    neiborhood_search,
+)
 
 KEY = "7a1821d4e4a3e41e3d523e97e0fd8950dedac2824aef99cb19d550500cb21a42"
 openai.api_base = "https://api.together.xyz"