o
    áÇh†  ã                   @   sè   d dl Z ddd„ZedkrpdZeeƒZede› d�ƒ eeƒD ]R\ZZd	ev r3ed
ed	 › �ƒ  dS dev r@eed ƒ  dS eed › ded › �ƒ edd 	ed ¡› �ƒ eded › �ƒ eded › �ƒ edƒ qdS dS )é    Né   c              
   C   sÊ   g }zAt  ¡ }t j| |t jjd�}t| |¡ƒ}|s#dd| › �igW S |D ]}| |jdd„ |j	D ƒ|j
t|jƒ|j|jdœ¡ q%W |S  tyd } ztd|› �ƒ dd	|› �igW  Y d
}~S d
}~ww )z3Searches academic papers on arXiv based on a query.)ÚqueryÚmax_resultsÚsort_byÚmessagezNo arXiv papers found for: c                 S   s   g | ]}t |ƒ‘qS © )Ústr)Ú.0Úauthorr   r   úAC:\Users\shubh\Videos\agent\agent_preprod\modules\search\arxiv.pyÚ
<listcomp>   s    zfetch_arxiv.<locals>.<listcomp>)ÚtitleÚauthorsÚsummaryÚ	publishedÚpdf_urlZ	arxiv_urlzError fetching arXiv results: ÚerrorzarXiv API Error: N)ÚarxivÚClientÚSearchÚSortCriterionÚ	RelevanceÚlistÚresultsÚappendr   r   r   r   r   r   Úentry_idÚ	ExceptionÚprint)r   r   r   ÚclientÚsearchZpapersZpaperÚer   r   r   Úfetch_arxiv   s6   ý
úÿü€þr!   Ú__main__zquantum computing algorithmszarXiv results for 'z':r   zError: r   é   z. r   z   Authors: z, r   z   Published: r   z   PDF: r   z---)r   )
r   r!   Ú__name__Úsearch_queryZarxiv_resultsr   Ú	enumerateÚiÚresultÚjoinr   r   r   r   Ú<module>   s(   
$
ð