diff --git a/research_town/agents/agent_prompting.py b/research_town/agents/agent_prompting.py index 7a43e2c2..657e7cb1 100644 --- a/research_town/agents/agent_prompting.py +++ b/research_town/agents/agent_prompting.py @@ -130,4 +130,4 @@ def summarize_research_direction(personal_info: str) -> List[str]: top_p=0, ) content = completion.choices[0].message["content"] - return [content] \ No newline at end of file + return [content] diff --git a/research_town/utils/paper_collection.py b/research_town/utils/paper_collection.py index 44bab089..5ef2f854 100644 --- a/research_town/utils/paper_collection.py +++ b/research_town/utils/paper_collection.py @@ -45,12 +45,14 @@ def neiborhood_search( def get_daily_papers( topic: str, query: str = "slam", max_results: int = 2 ) -> Tuple[Dict[str, Dict[str, List[str]]], str]: - search_engine = arxiv.Search( + client = arxiv.Client() + search = arxiv.Search( query=query, max_results=max_results, sort_by=arxiv.SortCriterion.SubmittedDate ) + results = client.results(search) content: Dict[str, Dict[str, List[str]]] = {} newest_day = "" - for result in search_engine.results(): + for result in results: paper_title = result.title paper_url = result.entry_id paper_abstract = result.summary.replace("\n", " ")