mcp_server.cpython-310.pyc•7.52 kB
o
W�g�! � @ s d dl Z d dlZd dlZd dlmZmZmZmZmZ d dl Z d dl
Z
d dl m Z m
Z
d dlmZ d dlmZmZ d dlmZ d dlmZmZ ejejdd � e�d
�ZG d d � d e�Z G d
d� de�Z G dd� de�Z G dd� de�Z dd� Z dd� Z!d dd�Z"dd� Z#G d d � d �Z$dS ) � N)�Any�Dict�List�Optional�Union)�FastMCP�Server)�SseServerTransport)� BaseModel�Field)� Starlette)�Route�Mountz4%(asctime)s - %(name)s - %(levelname)s - %(message)s)�level�format�
mcp_serverc @ s6 e Zd ZU eed< eed< dZeed< dZeed< dS )�ApiParam�type�descF�requiredN�default) �__name__�
__module__� __qualname__�str�__annotations__r �boolr r � r r �RD:\Data\Money\2025\ai-fund-assistant\mcp-servers\http-for-mcp-server\mcp_server.pyr s
r c @ s& e Zd ZU eed< dZee ed< dS )�ApiResponseField�pathNr )r r r r r r r r r r r r s
r c @ s� e Zd ZU eed< eed< eed< dZeed< eeef ed< ee d�Z
eeef ed< ee d�Z eee ee
f f ed < d
Zeeeef ed < d
Zeeeef ed < d
S )
�ApiTool�name� description�url�GET�method�params)�default_factory�headers�responseN�response_transform�auth)r r r r r r& r r r �dictr) r* r r r+ r r r, r r r r r! # s
" r! c @ s e Zd ZU ee ed< dS )� ApiConfig�toolsN)r r r r r! r r r r r r. . s
r. c C s< |� d�}| }|D ]}t|t�r||v r|| }q d S |S )N�.)�split�
isinstancer- )�datar �keys�result�keyr r r �get_nested_value2 s
r7 c
C st z$t | ddd��
}t�|�}W d � n1 sw Y tdi |��W S ty9 } z t�d|� �� � d }~ww )N�r�utf-8��encodingz"Failed to load API configuration: r )�open�json�loadr. � Exception�logger�error)� config_path�fZ config_data�er r r �load_api_config< s ���rE � c
� s �i }| j �� D ]$\}}||v r|| ||< q|jd ur"|j||< q|jr,td|� ���q| j�� }| jrn| j�d�}|dkrKd| j�d�� �|d< n#|dkrn| j�dd�}| j�d d
�} | j�d �d krj| ||< n| ||< t �
� 4 I d H ��}
zo| j � � d
kr�|
j| j
|||d�I d H } nG| j � � dkr�|
j| j
|||d�I d H } n2| j � � dkr�|
j| j
|||d�I d H } n | j � � dkr�|
j| j
|||d�I d H } ntd| j � ���| �� | �� W W d �I d H S ty� } z t�d| � �� � d } ~ ww 1 I d H �sw Y d S )Nz Missing required parameter: r ZbearerzBearer �token�
AuthorizationZapi_key�key_name� key_value� �in�headerr% )r' r) �timeout�POST)r= r) rN �PUTZDELETEzUnsupported HTTP method: zHTTP request error: )r' �itemsr r �
ValueErrorr) �copyr, �get�httpxZ AsyncClientr& �upperr$ �post�put�deleteZraise_for_statusr= r? r@ rA )
�toolr' rN Zprocessed_paramsr" Z param_configr) Z auth_typerI rJ �clientr* rD r r r �make_http_requestE sz �
�
�����#�� �r\ c C sJ |s| S i }|� � D ]\}}t|t�r|}n|�d�}t| |�||< q
|S )Nr )rQ r2 r rT r7 )r3 �mappingr5 Z
output_key�sourcer r r r �transform_response� s
r_ c @ s@ e Zd Zddd�Zdd� Zdd� Zd d
� Zd d � Zddd�Zd
S )�
HttpMcpServer�config/config.json�config/apis.jsonc C s@ || _ || _| �� | _t|�| _td�| _| �� | � � | _
d S )NZhttp_mcp_server) rB � apis_path� _load_config�configrE �
api_configr �mcp�_register_tools�_create_starlette_app�app)�selfrB rc r r r �__init__� s
zHttpMcpServer.__init__c
C s� z!t | jddd��}t�|�W d � W S 1 sw Y W d S tyD } zt�d|� �� ddddd d
d �W Y d }~S d }~ww ) Nr8 r9 r: zFailed to load config: �0.0.0.0�@ F�info�
rF )�host�port�debug� log_levelZ max_workersrN )r< rB r= r> r? r@ rA )rk rC rD r r r rd � s (����zHttpMcpServer._load_configc C s | j jD ]}| �|� qd S �N)rf r/ �_register_tool�rk � tool_configr r r rh � s �z HttpMcpServer._register_toolsc s< � �fdd�}� j j�j�jd�|� t�d�j� �� d S )Nc
� s� �z t �| � j�dd�d�I d H }�jr t|�j�}|W S |W S tyC } zt�d�j� d|� �� dt |�iW Y d }~S d }~ww )NrN rF )rN zTool z execution failed: rA )
r\ re rT r* r_ r? r@ rA r" r )�kwargsZ
response_dataZtransformed_datarD rw r r � tool_handler� s � � ��z2HttpMcpServer._register_tool.<locals>.tool_handler)r" r# zRegistered API tool: )rg rZ r" r# r@ ro )rk rx rz r rw r rv � s ��z HttpMcpServer._register_toolc sd �j j� td��� �fdd�}�fdd�}t�j�dd�td|d �td�jd
�td |d �gd �}|S )
Nz
/messages/c � sl ��� | j| j| j�4 I d H �\}}� �||� �� �I d H W d �I d H d S 1 I d H s/w Y d S ru )Z connect_sse�scopeZreceiveZ_send�runZ create_initialization_options)�requestZ read_streamZ write_stream)r �sser r �
handle_sse� s � � �.�z7HttpMcpServer._create_starlette_app.<locals>.handle_ssec
� sf �zt � j�� _� �� ddd�W S ty2 } zt�d|� �� dt|�d�W Y d }~S d }~ww )N�successz#Configuration reloaded successfully)�status�messagezFailed to reload config: rA )rE rc rf rh r? r@ rA r )r} rD )rk r r �
reload_config� s � ��z:HttpMcpServer._create_starlette_app.<locals>.reload_configrs Fz/mcp/sse)Zendpoint)rj z
/admin/reload)rs Zroutes) rg Z _mcp_serverr r re rT r
r Zhandle_post_message)rk r r� rj r )r rk r~ r ri � s
�� z#HttpMcpServer._create_starlette_appNc C s\ |p| j �dd�}|p| j �dd�}t�d|� d|� �� tj| j||| j �dd�d � d S )
Nrq rm rr rn z.Starting HTTP-MCP middleware server at http://�:rt ro )rq rr rt )re rT r@ ro �uvicornr| rj )rk rq rr r r r r| � s
�zHttpMcpServer.run)ra rb )NN) r r r rl rd rh rv ri r| r r r r r` � s
$r` )rF )%r= Zlogging�os�typingr r r r r rU r� Z
mcp.serverr r Zmcp.server.sser Zpydanticr
r Zstarlette.applicationsr Zstarlette.routingr
r Z basicConfig�INFOZ getLoggerr@ r r r! r. r7 rE r\ r_ r` r r r r �<module> s2 �
A