Hiba03 commited on
Commit
7b88ea9
·
verified ·
1 Parent(s): d5ebeb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -93,14 +93,12 @@ def get_paper_content(link:str)->str:
93
 
94
 
95
  @tool
96
- def 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
-
100
  Args:
101
- title (str): The title of the query paper to find related papers for.
102
- max_results (int): The number of related papers to return.
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,related_papers], ## add your tools here (don't remove final answer)
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,