melghorab/code-assistant
0
1o
2 ��gz � @ s` d dl mZmZ d dlZd dlZd dlmZ d dlZededefdd��Z dedefd d3�Z4dS ) � )�Tool�toolN)�Document� package_name�returnc 5 C s� t d| � �� d}zIzt�|� d| � d��}|�� |�� }W n tjy8 } z td| � dt|�� ���d}~ww t�|d d |d d6 d ��}t d |� �� |W S tyk } z
d
t|�� �W Y d}~S d}~ww )z�Search PyPI for Python package information. Input should be the package name.7 Args: 8 package_name: name of the package 9 zTool called for package: zhttps://pypi.org/pypi�/z/jsonz Error fetching PyPI info for z: N�info�name�summary)r r10 z
Tool result: z$Could not find package information: ) �print�requests�get�raise_for_status�json�RequestException� Exception�str�dumps)r �base_url�responser �e�result� r �F/home/melghorab/maven_nlp_course/code/Project/Code_Documenter/tools.py� search_pypi s* ��1112���r �documentation_textc C sz t � }| �d�}|D ])}|�d�r |j|dd� dd� q13|�d�r.|j|dd� dd� q14|�|� q15d }|�|� |S )16zZ17 Writes the AI-generated documentation to a .docx file and returns the file path.18 �19z# � N� )�levelz## � z generated_documentation.docx)r �split�20startswithZ add_headingZ
add_paragraph�save)r �doc�lines�line� file_pathr r r �
write_to_docx! s 212223 24r( ) �langchain_core.toolsr r r r �docxr �rer r r( r r r r �<module> s