arXiv title search

query_arxiv.py

import arxiv

# ref: https://note.nkmk.me/python-arxiv-api-download-rss/
def query_arxiv(search_word):
    arxiv_client = arxiv.Client()
    keyword_list = search_word.split()
    search_query = ' AND '.join([f'ti:"{keyword}"' for keyword in keyword_list])
    response = arxiv_client.results(
        search=arxiv.Search(
            query=search_query,
            max_results=100,
        )
    )
    cnt = 0
    for item in response:
        print(item.title)
        cnt += 1
    print(f"Found {cnt} papars")

if __name__ == "__main__":
    query_arxiv("LoRA image generation")

The gist was created with the purpose of modifying the implementation from the following blog:

https://blog.shikoan.com/arxiv_rag/