src/research_assistant/pipeline/articleSearch.py

from research_assistant.app_logging import app_logger
from research_assistant.components.arxiv_search_api import ArxivApiWrap
from research_assistant.entity import ArticleSearchConfig


class ArticleSearchPipeline:
    def __init__(self, config: ArticleSearchConfig):
        self.config = config

    def get_article_list(self):
        arxiv_api = ArxivApiWrap(self.config)
        article_list = arxiv_api.get_article_search_result()
        if not article_list:
            app_logger.info("No articles were found for the given parameters.")
        return article_list