diff --git a/research_town/agent_base.py b/research_town/agent_base.py index 202abaf3..bdc2e310 100644 --- a/research_town/agent_base.py +++ b/research_town/agent_base.py @@ -1,13 +1,15 @@ import datetime -import requests +from typing import Any, Dict, List from xml.etree import ElementTree -from typing import List, Dict, Any + +import requests + from .utils import ( - summarize_research_direction, + bfs, + generate_ideas, get_bert_embedding, + summarize_research_direction, summarize_research_field, - generate_ideas, - bfs, ) diff --git a/research_town/env_base.py b/research_town/env_base.py index c0eac7cd..d263f49f 100644 --- a/research_town/env_base.py +++ b/research_town/env_base.py @@ -1,4 +1,5 @@ from typing import Dict + from .agent_base import BaseResearchAgent from .kb_base import BaseKnowledgeBase diff --git a/research_town/env_paper_rebuttal.py b/research_town/env_paper_rebuttal.py index 3621e303..63b6bdee 100644 --- a/research_town/env_paper_rebuttal.py +++ b/research_town/env_paper_rebuttal.py @@ -1,4 +1,5 @@ from typing import Dict + from .env_base import BaseMultiAgentEnv diff --git a/research_town/env_paper_submission.py b/research_town/env_paper_submission.py index 5030c153..901e7470 100644 --- a/research_town/env_paper_submission.py +++ b/research_town/env_paper_submission.py @@ -1,4 +1,5 @@ from typing import Dict + from .env_base import BaseMultiAgentEnv diff --git a/research_town/kb_base.py b/research_town/kb_base.py index c6c36ead..e31f3b98 100644 --- a/research_town/kb_base.py +++ b/research_town/kb_base.py @@ -1,4 +1,5 @@ from typing import Dict, List + from .utils import get_daily_papers diff --git a/research_town/utils.py b/research_town/utils.py index bcb405e5..ef305698 100644 --- a/research_town/utils.py +++ b/research_town/utils.py @@ -1,15 +1,16 @@ -import os -import json import datetime -import openai +import json +import os import time +from typing import Any, Dict, List, Tuple + +import arxiv import faiss -from transformers import BertTokenizer, BertModel +import openai import torch -import arxiv -from tqdm import tqdm -from typing import Dict, List, Tuple, Any from arxiv import Client, Search +from tqdm import tqdm +from transformers import BertModel, BertTokenizer KEY = "7a1821d4e4a3e41e3d523e97e0fd8950dedac2824aef99cb19d550500cb21a42" openai.api_base = "https://api.together.xyz"