Skip to content

Commit 7d21bf8

Browse files
committed
fix arxiv param
1 parent 95765a8 commit 7d21bf8

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

python/graphy/graph/nodes/pdf_extract_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def execute(
6464
if self.arxiv_fetch_paper:
6565
paper_title = paper_metadata.get("title", "")
6666
if len(paper_title) > 0:
67-
result = self.arxiv_fetcher.fetch_paper(paper_title, 5)
67+
result = self.arxiv_fetcher.fetch_paper(paper_title, 20)
6868
if result:
6969
paper_metadata.update(result)
7070

python/graphy/utils/arxiv_fetcher.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ def __init__(
100100
:param timeout: The maximum time (in seconds) allowed for each paper fetching operation.
101101
:param download_folder: The folder where the fetched papers will be downloaded.
102102
"""
103-
self.client = arxiv.Client(delay_seconds=0.5, page_size=6, num_retries=1)
103+
self.client = arxiv.Client(delay_seconds=0.5, page_size=20, num_retries=1)
104104
self.timeout = timeout
105105
self.download_folder = download_folder
106106
self.bib_search_arxiv = BibSearchArxiv(
@@ -139,7 +139,7 @@ def find_paper_with_arxiv_id(self, name):
139139

140140
def find_paper_from_arxiv(self, name, max_results):
141141
new_names = sorted(
142-
[s.strip() for s in re.split(r"[.\\/]", name.strip()) if len(s) >= 20],
142+
[s.strip() for s in re.split(r"[.\\/]", name.strip()) if len(s) < 100],
143143
key=len,
144144
reverse=True,
145145
)

0 commit comments

Comments
 (0)