Spaces:
Running
Running
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,
|