Update tools/arxiv_tool.py
Browse files- 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': {
|
14 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 = {
|
|
|
|
|
|
|
|
|
|
|
|
|
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:
|