mcp_handlers.cpython-313.pyc•7.42 kB
�
��h � �| � S SK r S SKrS SKrS SKrS SKJr S SKJrJr S SK J
r
SSK J r Sr
\" 5 r\R S5 S 5 r\R S
5 S \4S j5 r\R S
5 S\4S j5 rS \\\4 4S jrS \\\4 4S jrS \\\4 4S jrS \\\4 4S jrS \\\4 4S jrg)� N)� APIRouter)�Dict�Any)�Path� )�JSONRPCResponse�mock_hdf5_dataz/listResourcesc � � # � S/ SQSS.$ 7f)N�2.0)� resource1� resource2� resource3r ��jsonrpc�result�id� r � �:/Users/jafaralzoubi/Desktop/mcp-server/src/mcp_handlers.py�list_resourcesr s � � � �9�� � �s � z /callTool�paramsc � � # � U R S5 nU R S5 nU R S5 nSSU SU SU S3S S
.$ 7f) N� tool_name� script_path�
core_countr zJob submitted to z
with script z on z cores� r )�get)r r r r s r � call_toolr sZ � � ��
�
�;�'�I��*�*�]�+�K����L�)�J� �%�i�[�
�k�]�$�z�l�Z`�a�� � �s �AAz/jsonrpc�requestc � �4 # � [ U 5 I S h v�N $ N7f�N)�handle_request)r s r �json_rpcr# $ s � � � ��(� (�(� (�s ���c � � ^ # � S/n[ U 4S jU 5 5 ( d [ SSS.S9R 5 $ T S n US:X a [ T 5 I S h v�N $ US:X a [ T 5 I S h v�N $ US :X a [ T 5 I S h v�N $ US
:X a [
T 5 I S h v�N $ [ SS S.S9R 5 $ Ni NR N; N$! [ a- n[ S [ U5 S.S9R 5 s S nA$ S nAff = f7f)
N�actionc 3 �, ># � U H oT; v � M g 7fr! r ��.0�pr s �r � <genexpr>�!handle_request.<locals>.<genexpr>+ � �� � �4�O�q�F�{�O�� �馀��zAction parameter required��code�message��error�list�read�metadata� submitJobzInvalid action� ���) �allr �dict� _handle_list� _handle_read�_handle_metadata�_handle_job_submission� Exception�str)r �required_paramsr% �es ` r r" r" ) s �� � � �j�O� �4�O�4� 4� 4� �!�.I�J�
�
�$�&� � �H�
�F�� �V� �%�f�-�-� -�
�v�
�%�f�-�-� -�
�z�
!�)�&�1�1� 1�
�{�
"�/��7�7� 7�"� %�2B�C���d�f�
� .�-�1�7��
� � �!�c�!�f�5�
�
�$�&� ���s� �=D�C �B?�C �D�C �.C�/C �2D�3C �C�C � D� C � C�!C �$D�%C �>D�?C �C �C �C �
C>�"C9�3C>�4D�9C>�>Dc � � # � U R SS5 nU R SS5 n[ R " [ SS9 / SQnU H> n[ [ R
R
[ U5 5 R 5 M@ [ R
R
[ X5 n[ R " U5 nU Vs/ s H' n[ R
R U[ 5 PM) sn[ U5 UUS.S .$ s snf 7f)
N�path� �patternz*.h5T)�exist_ok)zsimulation_run_123.h5zexperiment_data_456.h5zresults/analysis_output.h5)�
searchPathrF )�files�countr6 ) r �os�makedirs�
MOCK_DATA_DIRr rD �join�touch�glob�relpath�len)r rD rF �
mock_files�f� search_pathrI s r r; r; E s� � � � �:�:�f�b� !�D��j�j��F�+�G� �K�K�
��-��J�
�� �R�W�W�\�\�-��
+�,�2�2�4� � �'�'�,�,�}�d�<�K� �I�I�k� "�E� >C�C�U��"�'�'�/�/�!�]�3�U�C��U��%� �
� � ��C�s �CD �.D�4D c � �� # � U R S5 nU R S5 nU( a U( d [ S5 e[ R R [
U5 nUSS/S/ SQS.SS US
.S .$ 7f) N�filePath�datasetz1Both filePath and dataset parameters are required�d �float32)g�������?g �?g�������?)�shape�dtype�
sample_values�KelvinzSimulated temperature data)�units� description�file)rX �datar6 )r �
ValueErrorrK rD rN rM )r � file_pathrX � full_paths r r< r< a sw � � ��
�
�:�&�I��j�j��#�G� �G��L�M�M���� � �]�I�6�I� ��3�Z� � ,�
� �7� �
� � �s �A2A4c � �d # � U R S5 nU( d [ S5 eUSSSSS.S.$ 7f) NrW z filePath parameter is required�HDF5z1.10z2023-01-01T00:00:00Zz1.2GB)�format�version�created�size)ra r6 )r rc )r rd s r r= r= z sB � � ��
�
�:�&�I� ��9�:�:� � � �-��
� � �s �.0c � � ^ # � SS/n[ U 4S jU 5 5 ( d, [ SSSR U5 3S.S9R 5 $ T S nT S nT R S S
5 nT R S S 5 n[
R R U5 ( d [ S
U 35 e[ U[ 5 ( a US: a [ S5 e[ [ R " 5 5 nSSU 3SU 3SU 3U/nUSSR U5 SSS/ S.S.$ ! [ a- n[ S[ U5 S.S9R 5 s S nA$ S nAff = f7f) N�script�coresc 3 �, ># � U H oT; v � M g 7fr! r r' s �r r* �)_handle_job_submission.<locals>.<genexpr>� r, r- r. zRequired parameters: z, r/ r2 �memory�4GB�jobName�mcp_jobzScript file not found: r z Cores must be a positive integer�sbatchz --job-name=z --ntasks=z--mem=�PENDING� z2023-01-01T12:00:00Zz2023-01-01T12:05:00Z�normal)�estimatedStart�queue�allocatedNodes)�jobId�status�command�submissionTimer6 r8 )r9 r rN r: r rK rD �existsrc �
isinstance�intr@ �uuid�uuid4r? ) r rA r rn rq �job_name�job_id� mock_commandrB s ` r r> r> � sl �� � � ��)�O� �4�O�4� 4� 4� �!�0E�d�i�i�P_�F`�Ea�.b�c�
�
�$�&� �'� �X�&� ��w������H�e�,���:�:�i��3�� �w�w�~�~�k�*�*� � 6�{�m� D�E� E��%��%�%���� � ?�@� @� �T�Z�Z�\�"�� ��(�� $���w� ��V�H� � �
� � � ��x�x� �-� 4�"8�!�"$��
�
�� � � �!�c�!�f�5�
�
�$�&� ���s7 �A E�CD � E�
E�("E�
E� E�E�E)r� �
subprocessrK rP �fastapir �typingr r �pathlibr �modelsr rM �router�postr r: r r# r@ r" r; r< r= r> r r r �<module>r� s� �� � � � � � � � #� �
� ��� ���
� �� �� ���[�� �D� � � � ���Z��)�D� )� �)���c�3�h� � �8�t�C��H�~� �8�t�C��H�~� �2
�4��S��>�
� .��c�3�h� � .r