MCP RAG Server

� q��g���
�SS
KJ
r
JrJr SSKJr SSKJrJrJ r J r \
"SS/
S9r"SS\5r"S S \5r SqSqSr\R#S \ S9\"S5
4
S\4Sjj
5rS\\\ 4S\ 4SjrS\\\ 4S\ 4Sjrg)�)� APIRouter� HTTPException�Body)
� BaseModel)�List�Dict�Any�Optionalz/mcp�MCP)�prefix�tagsc�8�\r
SrS
r%Sr\\S'\\\4\S'Sr g)� MCPAction�zModel for an MCP action request�action_type�payload�N) �__name__� __module__�__qualname__�__firstlineno__�__doc__�str�__annotations__rr �__static_attributes__r��*/Users/kz/vcs/sui-ai/mcp_server/mcp_api.pyrrs��)��� �#�s�(�^�rrc�L�\r
SrS
r%Sr\\S'Sr\\ \S'Sr \\\S'Srg)�MCPResponse� zModel for an MCP response�statusN�data�errorr)rrrrrrrr"r r r#rrrrrr s&��#��K��D�(�3�-���E�8�C�=�rrNc
��UqU
q
g
)zAConfigure the router with the vector store and document processorN)�vector_store� doc_processor)�vs�dps r� configurer)s���L��Mrz/action)
�response_model.�actionc
��z
# �[b[c [SSS9e
URS:Xa[ UR 5
IS
hv
�N$URS:Xa[ UR 5
IS
hv
�N$[SSUR3S 9$NHN![an
[S[U
5
S 9sS
n
A
$S
n
A
f
f=f
7f
) zHandle MCP action requestsNi�
zServer not properly initialized)�status_code�detail�retrieve_documents�index_documentsr#zUnknown action type: �r!r#) r%r&rrr/rr0r� Exceptionr)r+�
es r� handle_actionr4s������}�4���4U�V�V� ����!5�5�+�F�N�N�;�;�;� � � �#4� 4�(����8�8�8���-�f�.@�.@�-A�B�� � <�8��� ����a�&� � �� �sp�B;
�(B�
B �B�
B;
� (B�1
B�2B�5
B;
�6B�
B;
�
B�
B� B8�B3�-
B8�.B;
�3B8�8B;
r�returnc
���# �UR
S
5
n
UR
SS5nU
(d [SSS9$[RU
5
n[RX25n[SUU
[ U5
S.S 9$7f
) z#Retrieve documents based on a query�query�top_k�r#z/Query is required for retrieve_documents actionr1�success)� documentsr7� total_results�r!r")�getrr&� get_embeddingr%�search�len)rr7r8�query_embedding�resultss rr/r/3s����K�K�� �E��K�K���#�E����C� � �$�1�1�%�8�O��!�!�/�9�G��� �� ��\� ���s�A6A8
c
��
# �UR
S
S5n
[RU
5
nU(d [SSS0
SS9$[RU5
[R S5
[S [U5
S [U5
S3S.S 9$7f
)z Index documents from a directory� directory�docs�warning� indexed_countrzNo documents found to index)r!r"r#z./data/faiss_index.binr:zSuccessfully indexed z documents)rH�messager=)r>r&�process_documentsrr%r0�saverA)rrEr;s rr0r0Ms�������K��0�I��/�/� �:�I����!�1�%�/� � �� � ��+����.�/��� ��^�.�s�9�~�.>�j�I� ���s�B B
)�fastapirrr�pydanticr�typingrrr r �routerrrr%r&r)�postr4rr/r0rrr�<module>rQ
s��


�2�2��,�,� �&��w� /��
� �
� 
�)�
���� �
����Y�{��
3�,0��I�
� �
�4�
�*
�d�3��8�n�
��
�4
�4��S��>�
�k�
r