Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -93,14 +93,12 @@ def get_paper_content(link:str)->str:
|
|
| 93 |
|
| 94 |
|
| 95 |
@tool
|
| 96 |
-
def
|
| 97 |
"""
|
| 98 |
A tool that searches for related papers on arXiv based on the title of the query paper.
|
| 99 |
-
|
| 100 |
Args:
|
| 101 |
-
title
|
| 102 |
-
max_results
|
| 103 |
-
|
| 104 |
Returns:
|
| 105 |
list: A list of dictionaries, each containing a related paper's title and URL.
|
| 106 |
"""
|
|
@@ -139,7 +137,7 @@ with open("prompts.yaml", 'r') as stream:
|
|
| 139 |
|
| 140 |
agent = CodeAgent(
|
| 141 |
model=model,
|
| 142 |
-
tools=[final_answer,get_top_paper,get_paper_link,get_paper_content,
|
| 143 |
max_steps=6,
|
| 144 |
verbosity_level=1,
|
| 145 |
grammar=None,
|
|
|
|
| 93 |
|
| 94 |
|
| 95 |
@tool
|
| 96 |
+
def get_related_papers(title:str, max_results:int)->list:
|
| 97 |
"""
|
| 98 |
A tool that searches for related papers on arXiv based on the title of the query paper.
|
|
|
|
| 99 |
Args:
|
| 100 |
+
title: A string representing the title of the query paper to find related papers for.
|
| 101 |
+
max_results: A integer representing the number of related papers to return.
|
|
|
|
| 102 |
Returns:
|
| 103 |
list: A list of dictionaries, each containing a related paper's title and URL.
|
| 104 |
"""
|
|
|
|
| 137 |
|
| 138 |
agent = CodeAgent(
|
| 139 |
model=model,
|
| 140 |
+
tools=[final_answer,get_top_paper,get_paper_link,get_paper_content,get_related_papers], ## add your tools here (don't remove final answer)
|
| 141 |
max_steps=6,
|
| 142 |
verbosity_level=1,
|
| 143 |
grammar=None,
|