main.cpython-313.pyc•4.11 kB
�
���g�
� �t � S SK r S SKrS SKJrJr S SKJr S SKJrJ r J
r
J r S SK J
r
S SKJr S SKJr S SKr\
" 5 \" \ R( " SS 5 5 r\ R( " S
S 5 r\ R( " S S
5 r\ R( " SS5 r\" SS9rSqSq " S S\5 r " S S\5 r \R= S5 S 5 r \RA \RB 5 \RE S\ S9S\ 4S j5 r#\RE S 5 S#S \$4S jj5 r%\&S :X a S SK'r'\'RP " S \\S!S"9 gg)$� N)�FastAPI�
HTTPException)� BaseModel)�List�Dict�Any�Optional)� load_dotenv)�FAISSVectorStore)�DocumentProcessor�PORT�8000�HOSTz0.0.0.0�
INDEX_FILEz./data/faiss_index.bin�EMBEDDING_MODELzall-MiniLM-L6-v2zMCP RAG Server)�titlec �. � \ rS rSr% \\S' Sr\\S' Srg)� QueryRequest� �query� �top_k� N) �__name__�
__module__� __qualname__�__firstlineno__�str�__annotations__r �int�__static_attributes__r � �'/Users/kz/vcs/sui-ai/mcp_server/main.pyr r s � � �J��E�3�Nr" r c �N � \ rS rSr% \\\\4 \S' Sr \
\\\4 \S' Sr g)�
QueryResponse� � documentsN�metadatar ) r r r r r r r r r r( r r! r r" r# r% r% s, � ��D��c��N�#�#�)-�H�h�t�C��H�~�&�-r" r% �startupc � �8 # � [ [ S9q[ 5 q[
R R [ 5 ( a- [ S[ S35 [ R [ 5 O [ S5 [ R " [ [ 5 g 7f)N)�
model_namez"Loading existing FAISS index from z...z<No existing FAISS index found. Please index documents first.)
r r �
doc_processorr � vector_store�os�path�existsr �print�load�mcp_api� configurer r" r# �
startup_eventr5 # sk � � �
&� �A�M� $�%�L�
�w�w�~�~�j�!�!�
�2�:�,�c�B�C����*�%�
�L�M� ���l�M�2�s �BBz/query)�response_model�requestc � � # � [ c
[ SSS9e[ R U R 5 n[ R XR 5 nSU0$ 7f)N�� z Vector store not initialized�� status_code�detailr' )r- r r, �
get_embeddingr �searchr )r7 �query_embedding�resultss r# �query_documentsrA : sP � � �����4R�S�S� $�1�1�'�-�-�@�O� �!�!�/�=�=�A�G� �� !�!�s �AAz/index�directory_pathc � �� # � [ b [ c
[ SSS9e[ R U 5 n[ R U5 [ R [ 5 SS[ U5 S3S.$ 7f)Nr9 z!Server components not initializedr: �successzIndexed z
documents)�status�message)r- r, r �process_documents�index_documents�saver �len)rB r' s r# rH rH G sk � � � ��}� 4���4W�X�X� �/�/��?�I� � � ��+� ���j�!� �h�s�9�~�6F�j�,Q� R�R�s �A.A0�__main__zmain:appT)�host�port�reload)�docs))r. �json�fastapir r �pydanticr �typingr r r r �dotenvr
�models.vector_storer �utils.document_processorr r3 r �getenvr
r r r �appr- r, r r% �on_eventr5 �include_router�router�postrA r rH r �uvicorn�runr r" r# �<module>r_ sE �� � � *� � ,� ,� � 0� 6� � �
� �2�9�9�V�V� $�%�� �y�y���#��
�Y�Y�|�%=�
>�
��)�)� -�/A�B��
�$�%�� � ��
��9� �.�I� .� � � �i��3� �3�( � � �7�>�>� "����(�=��1�
"�<�
"� 2�
"� ���(��S�#� S� �S�" �z��� �K�K�
��D��>� r"