Ferocious0xide commited on
Commit
9dbae6e
·
verified ·
1 Parent(s): daaf1a3

Update tools/arxiv_tool.py

Browse files
Files changed (1) hide show
  1. tools/arxiv_tool.py +17 -3
tools/arxiv_tool.py CHANGED
@@ -10,8 +10,16 @@ class ArxivSearchTool(Tool):
10
  name = "search_arxiv"
11
  description = "Search ArXiv for papers matching the query"
12
  inputs = {
13
- 'query': {'type': 'string', 'description': 'The search query for papers'},
14
- 'max_results': {'type': 'integer', 'description': 'Maximum number of results to return'}
 
 
 
 
 
 
 
 
15
  }
16
  output_type = "string"
17
 
@@ -69,7 +77,13 @@ class ArxivSearchTool(Tool):
69
  class LatestPapersTool(Tool):
70
  name = "get_latest_papers"
71
  description = "Get papers from the last N days from saved results"
72
- inputs = {'days_back': {'type': 'integer', 'description': 'Number of days to look back'}}
 
 
 
 
 
 
73
  output_type = "string"
74
 
75
  def forward(self, days_back: int = 1) -> str:
 
10
  name = "search_arxiv"
11
  description = "Search ArXiv for papers matching the query"
12
  inputs = {
13
+ 'query': {
14
+ 'type': 'string',
15
+ 'description': 'The search query for papers',
16
+ 'nullable': True
17
+ },
18
+ 'max_results': {
19
+ 'type': 'integer',
20
+ 'description': 'Maximum number of results to return',
21
+ 'nullable': True
22
+ }
23
  }
24
  output_type = "string"
25
 
 
77
  class LatestPapersTool(Tool):
78
  name = "get_latest_papers"
79
  description = "Get papers from the last N days from saved results"
80
+ inputs = {
81
+ 'days_back': {
82
+ 'type': 'integer',
83
+ 'description': 'Number of days to look back',
84
+ 'nullable': True
85
+ }
86
+ }
87
  output_type = "string"
88
 
89
  def forward(self, days_back: int = 1) -> str: