helper.cpython-311.pyc•2.59 kB
�
�{�hw � �^ � d dl Z d dlZd dlZdedefd�Zdedefd�Zdedefd�Zdedefd �ZdS )
� N�ticker�returnc �� � t j � t j � t � � � � }t j � |dd| � � }t
j |� d�� � }|S )N�..�data�/*.txt)�os�path�dirname�abspath�__file__�join�glob)r �
script_dir�data_dir�filess �#/Users/sharhad/mcp/server/helper.py� get_filesr sY � ��� � ��� � ��!:�!:�;�;�J��w�|�|�J��f�f�=�=�H� �I��)�)�)� *� *�E� �L� c �� � t j � t j � t � � � � }t j � |dd| � � }t
j |� d�� � }g }|D ]`}t j � |� � }|� d� � r�7|� |�
dd� � � � �a|S )Nr r r z _summary.txtz.txt� ) r r
r r r
r r �basename�endswith�append�replace)r r r r �result�file�filenames r �get_sec_filing_filesr s� � ��� � ��� � ��!:�!:�;�;�J��w�|�|�J��f�f�=�=�H� �I��)�)�)� *� *�E�
�F�� 4� 4���7�#�#�D�)�)�� � � �^� ,� ,� � ��
�
�h�&�&�v�r�2�2�3�3�3�3� �Mr �queryc �j � t j � | dd�� � }|j d j S )Nztext-embedding-3-smalli )�input�model�
dimensionsr )�openai�
embeddings�creater � embedding)r �ress r �embedr* s9 � �
�
�
"�
"��(�� #� � �C�
�8�A�;� � r �resultsc �h � g }| D ],}|� |d |d d d�� � �-|S )N�id�metadata�
original_text)r- �
chunk_text)r )r+ � documentsr s r �build_rerank_documentsr2 ! sY � ��I�� � �������,� ��,�_�=�
�
� � � � � �r ) r r r% �strr �listr r* r2 � r r �<module>r6 s� �� � � � � � � � �
�
�
�
��c� �c� � � � � �� �� � � � � !�� !�� !� !� !� !��D� �T� � � � � � r