SEC Filing MCP Server

� ry�h� ��
�dd
lZdd
l
Z
dd
lZddlmZ
ddlmZ
ddlmZ
ddl m Z
ddlmZ
ddl mZ
ddlmZ
ed � �
�

e
j�dej�ej�ej�e�
�
�
�
�
�
��
ddlmZ
Gd�d ��Zdefd�Zedkrejed�
�
�
�

d
Sd
S)�N)
�Agent)
�AsyncOpenAI)
�MCPServerStreamableHTTP)
� FallbackModel)
�OpenAIProvider)
�OpenAIResponsesModel)
�load_dotenvT)
�override)
�QUERY_SYSTEM_PROMPTc�$�eZ
dZd
�Zdedefd�ZdS)�Clientc
�D
�t
tj�d
�
�
��
�
|_td�
�
|_ttdt|j��
�
���tdt|j��
�
�����|_ dS)N�OPENAI_API_KEY)
�api_keyzhttp://127.0.0.1:8000/mcpzgpt-5)
� openai_client)� model_name�providerzgpt-4.1)r�os�environ�getrr�serverrrr�model)
�selfs
�*/Users/sharhad/mcp/client/client_runner.py�__init__zClient.__init__s���(�2�:�>�>�BR�3S�3S�T�T�T���-�.I�J�J���"� �$�)��8J�K�K�K� � � � !�&�)�$�:L�M�M�M� � � �  �  �� � � ��query�returnc��K
�t
|j
d
t|jg
���}|�|
�
�
�d{V��S)N�query_agent)�name� system_prompt�toolsets)rrrr�run)rr�agents rrzClient.query!sR������J� �/���}�  � � ���Y�Y�u�%�%�%�%�%�%�%�%�%rN)�__name__� __module__�__qualname__r�strr�rrr r sE������ � � �&��&��&�&�&�&�&�&rr rc
��K
�t
��}
|
�
|�
�
�d{V��}t|j�
�

dS)
N)r r�print�output)r�client�results r�mainr0*sH���� �X�X�F��<�<��&�&� &� &� &� &� &� &�F� �&�-�����r�__main__z'What is the latest 10-K for Apple Inc.?)r�sys�asyncio�pydantic_air�openair�pydantic_ai.mcpr�pydantic_ai.models.fallbackr�pydantic_ai.providers.openair�pydantic_ai.models.openair�dotenvr �path�insert�dirname�abspath�__file__� client.promptrr r)r0r&r$r*rr�<module>rA
s~
�


� � � � � � � � �����������������3�3�3�3�3�3�5�5�5�5�5�5�7�7�7�7�7�7�:�:�:�:�:�:���������t���������
�2�7�?�?�2�7�?�?�2�7�?�?�8�3L�3L�#M�#M�N�N�O�O�O�-�-�-�-�-�-�
&�
&�
&�
&�
&�
&�
&�
&�.
�c�
�
�
�
� �z����G�K���>�?�?�@�@�@�@�@��r