knight7561 commited on
Commit
a64fa62
·
verified ·
1 Parent(s): 10e12bb

Fix doc string

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,7 +24,7 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
24
  def arxiv_fetch_paper_tool(arxiv_id : str) -> str:
25
  """ An Tool that would fetch research papers from ArXiV collection and parse it and return the PDF text
26
  Args:
27
- arxiv_id : id for the arxiv paper
28
  Returns:
29
  text of the paper content and title of the paper
30
  """
@@ -84,7 +84,7 @@ with open("prompts.yaml", 'r') as stream:
84
 
85
  agent = CodeAgent(
86
  model=model,
87
- tools=[final_answer,arxiv_fetch_paper_tool], ## add your tools here (don't remove final answer)
88
  max_steps=6,
89
  verbosity_level=1,
90
  grammar=None,
 
24
  def arxiv_fetch_paper_tool(arxiv_id : str) -> str:
25
  """ An Tool that would fetch research papers from ArXiV collection and parse it and return the PDF text
26
  Args:
27
+ arxiv_id: id for the arxiv paper
28
  Returns:
29
  text of the paper content and title of the paper
30
  """
 
84
 
85
  agent = CodeAgent(
86
  model=model,
87
+ tools=[final_answer,arxiv_fetch_paper_tool,image_generation_tool], ## add your tools here (don't remove final answer)
88
  max_steps=6,
89
  verbosity_level=1,
90
  grammar=None,