AACT Clinical Trials MCP Server
by navisbio
- src
- mcp_server_aact
- __pycache__
o
�uTg�# � @ sT d dl Z d dlmZ d dlmZ d dlmZ ddlmZ e � d�Z
G dd� d�Z dS ) � N)�Any)�AnyUrl� )� AACTDatabasezmcp_aact_server.handlersc @ s� e Zd Zdefdd�Zdeej fdd�Zde de
fdd �Z deej fd
d �Z
d e
d
ee
e
f dB dejfdd�Zdeej fdd�Zd e
d
ee
ef dB deej fdd�ZdS )� MCPHandlers�dbc C s
|| _ d S )N)r )�selfr � r �C/Users/jonas/servers/src/ctgov-aact/src/mcp_server_aact/handlers.py�__init__
s
zMCPHandlers.__init__�returnc � s8 �t �d� tjtd�dddd�tjtd�dd dd�gS )
Nz Handling list_resources requestzmemo://landscapezClinical Trial LandscapezKKey findings about trial patterns, sponsor activity, and development trendsz
text/plain)�uri�name� description�mimeTypezmemo://metricsz
Trial MetricszKQuantitative metrics about trial phases, success rates, and temporal trends)�logger�debug�types�Resourcer �r r r r
�handle_list_resources
s �
���z!MCPHandlers.handle_list_resourcesr
c � s� �t �d|� �� |jdkr t �d|j� �� td|j� ���t|��dd�}|s2t �d� td��|dkr;| j�� S |dkrD| j� � S |d krM| j�
� S t �d
|� �� td
|� ���) Nz(Handling read_resource request for URI: �memozUnsupported URI scheme: zmemo://� zEmpty resource path�insightsZ landscape�metricszUnknown resource path: ) r r �scheme�error�
ValueError�str�replacer �get_insights_memo�get_landscape_memo�get_metrics_memo)r r
�pathr r r
�handle_read_resource s" �
z MCPHandlers.handle_read_resourcec � s, �t �d� tjddtjdddd�gd�gS ) Nz Handling list_prompts request�indication-landscapezpAnalyzes clinical trial patterns, development trends, and competitive dynamics within specific therapeutic areas�topiczWTherapeutic area or indication to analyze (e.g., 'multiple sclerosis', 'breast cancer')T)r r �required)r r � arguments)r r r �Prompt�PromptArgumentr r r r
�handle_list_prompts3 s �
����z MCPHandlers.handle_list_promptsr r( Nc � s� �ddl m} t�d|� d|� �� |dkr%t�d|� �� td|� ���|r+d|vr4t�d� td��|d }|j|d �}tjd
|� �tj d tj
d |� � d
�d�gd�S )Nr )�PROMPT_TEMPLATEz Handling get_prompt request for z with args r% zUnknown prompt: r& z Missing required argument: topic)r& z&Clinical trial landscape analysis for �user�text��typer. )�role�content)r �messages) �promptsr, r r r r �formatr �GetPromptResult�
PromptMessage� TextContent�strip)r r r( r, r& �promptr r r
�handle_get_promptC s& �
���z MCPHandlers.handle_get_promptc
� s� �t jddddddd�idgd�d �t jd
d di d �d �t jd
dddddd�idgd�d �t jddddddd�idgd�d �t jddddddd�idgd�d �t jdddddd d�idgd�d �gS ) N�
read-queryz;Execute a SELECT query on the AACT clinical trials database�object�query�stringzSELECT SQL query to execute)r0 r )r0 �
propertiesr' )r r � inputSchema� list-tablesz$List all tables in the AACT database)r0 r@ �describe-tablez7Get the schema information for a specific table in AACT�
table_namez Name of the table to describe�append-insightz"Add a business insight to the memo�insightz.Business insight discovered from data analysis�append-landscapez8Add findings about trial patterns and development trends�findingz/Analysis finding about trial patterns or trends�append-metricsz%Add quantitative metrics about trials�metricz*Quantitative metric or statistical finding)r �Toolr r r r
�handle_list_tools\ sl �
��� ��
���
���
���
����z MCPHandlers.handle_list_toolsc
� s� �z�|dvr
t d|� ���|s|dkrt d��|dkr,| j�d�}tjdt|�d�gW S |dkrNd |vr8t d
��| j�d d |d i�}tjdt|�d�gW S |d krt|d
�� �� �d�sat d��| j�|d
�}tjdt|�d�gW S |dkr�d|vr�t d��| j� |d � tjddd�gW S |dkr�d|vr�t d��| j�
|d � tjddd�gW S |dkr�d|vr�t d��| j� |d � tjddd�gW S W d S t y� } zt
|t �rۂ tjdd t|�� �d�gW Y d }~S d }~ww ) N> r< rI rE rB rG rC zUnknown tool: rB zMissing argumentsz�
SELECT table_name
FROM information_schema.tables
WHERE table_schema = 'ctgov'
ORDER BY table_name;
r. r/ rC rD zMissing table_name argumenta
SELECT column_name, data_type, character_maximum_length
FROM information_schema.columns
WHERE table_schema = 'ctgov'
AND table_name = %s
ORDER BY ordinal_position;
r< r> ZSELECTz.Only SELECT queries are allowed for read-queryrE rF zMissing insight argumentzInsight added to memorG rH zMissing finding argumentzLandscape finding addedrI rJ zMissing metric argumentz Metric addedzError: )r r �
execute_queryr r8 r r9 �upper�
startswith� add_insight�add_landscape_finding�add_metrics_finding� Exception�
isinstance)r r r( �results�er r r
�handle_call_tool� sV �
��
&��z MCPHandlers.handle_call_tool)�__name__�
__module__� __qualname__r r �listr r r r r r$ r) r+ �dictr6 r; rK rL r r8 rW r r r r
r s $,Cr ) �logging�typingr � mcp.typesr �pydanticr �databaser � getLoggerr r r r r r
�<module> s