SEC Filing MCP Server

� ?f�h�� �
�dd
lZdd
l
Z
e
j�dej�ej�ej�e�
�
�
�
�
�
��
ddlmZ
ddl m Z
e d��
�

ddlmZm Z
Gd�d��Zed kr#ed ��
�
Ze�d�
�

d
Sd
S) �N)
�Pinecone)
�load_dotenvT)
�override)�embed�build_rerank_documentsc �<�eZ
dZd defd�Zd�Zddeded ed efd�ZdS)�Pinecone_DB�sec-embeddings� index_namec��|
|_d
|_
ttj�d�
�
��
�
|_dS)N�__default__�PINECONE_API_KEY)
�api_key)r� namespacer�os�environ�get�pc)�selfrs �/Users/sharhad/mcp/server/pc.py�__init__zPinecone_DB.__init__s5��$���&����R�Z�^�^�4F�%G�%G�H�H�H�����c
�R�t
|j
����
�

dS)
N)�printr�list_indexes)
rs
rrzPinecone_DB.list_indexess$�� �d�g�"�"�$�$�%�%�%�%�%r�F�query�top_k�top_n�rerankc ��
�g}t
|
�
�
}|j
�|j�
�
�|j||d
d���}|ri|j
j�d|
t|d�
�
|dg
dddi
� ��}|j D]$} |� | j| jd ��
�

�%n5|dD],} |� | dd| d d ��
�

�-|S)NFT)r�vectorr�include_values�include_metadatazbge-reranker-v2-m3�matches� chunk_text�truncate�END)�modelr� documentsr�rank_fields�return_documents� parameters)�document�score�metadata� original_textr/) rr�Indexrrr� inferencer r�data�appendr.r/)rrrrr � query_results�query_embed�results�ranked_results� ranked_result�results rrzPinecone_DB.querysF
��� ��E�l�l���'�-�-���0�0�6�6��� ��"�#�7� � ��� �!�W�.�5�5�,��2�7�9�3E�F�F��+�n�#'����6� � �N�"0�!4� � � ��$�$� -� 6�*�0�&�&����� �"�)�,� � ���$�$� &�z� 2�?� C�#�G�_�&�&������rN)
r )rrF) �__name__� __module__�__qualname__�strrr�int�boolr�rrr r s~������I
�I
�3�I
�I
�I
�I
�
&�
&�
&�!�!�3�!�s�!��!�$�!�!�!�!�!�!rr �__main__r )
rz'What is the latest 10-K for Apple Inc.?)r�sys�path�insert�dirname�abspath�__file__�pineconer�dotenvr� server.helperrrr r<rrrBrr�<module>rM
s
�


� � � � � � � � �����
�2�7�?�?�2�7�?�?�2�7�?�?�8�3L�3L�#M�#M�N�N�O�O�O���������������t�����7�7�7�7�7�7�7�7�*
�*
�*
�*
�*
�*
�*
�*
�X
�z��� ��"2� 3� 3� 3�B��H�H� 6�7�7�7�7�7��r