SEC Filing MCP Server

� �x�h����dd
lZdd
l
Z
dd
lZddlmZ
ddlmZ
ddlmZ
ddl m Z
ddlmZ
ddl mZ
ddlmZ
Gd �d ��Zdefd�Zed krejed�
�
�
�

d
Sd
S)�N)
�Agent)
�AsyncOpenAI)
�MCPServerStreamableHTTP)
� FallbackModel)
�OpenAIProvider)
�OpenAIResponsesModel)
�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.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)rrr r�run)rr�agents rrzClient.querysR������J� �/���}�  � � ���Y�Y�u�%�%�%�%�%�%�%�%�%rN)�__name__� __module__�__qualname__r�strr�rrrr sE������ � � �&��&��&�&�&�&�&�&rrrc
��K
�t
��}
|
�
|�
�
�d{V��}t|j�
�

dS)
N)rr�print�output)r�client�results r�mainr.$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� client.promptr rr'r.r$r"r(rr�<module>r9
s
�


� � � � � � � � �����������������3�3�3�3�3�3�5�5�5�5�5�5�7�7�7�7�7�7�:�:�:�:�:�:�-�-�-�-�-�-�
&�
&�
&�
&�
&�
&�
&�
&�.
�c�
�
�
�
� �z����G�K���>�?�?�@�@�@�@�@��r