Rememberizer MCP Server for Common Knowledge

� m��g� ����SS
KrSSK
Jr SS
KrSSKJr SSKJrJrJr SSK J r SSKJr \R"\5
r\"5 SrSrS rS rSrSrS rSrSr"SS\5r"SS5rSrg
)�N)
�Enum)
�load_dotenv)� HTTPError�HTTPStatusError�Timeout)
�McpError�
�AnyUrl�mcp_server_rememberizerzaccount/z documents/zdocuments/{id}/contents/z1discussions/{id}/contents/?integration_type=slackzdocuments/search/zdocuments/agentic_search/z integrations/zdocuments/memorize/c
�,�\r
SrS
rSrSrSrSrSrSr Sr g ) �RememberizerTools��0retrieve_semantically_similar_internal_knowledge�smart_search_internal_knowledge�list_internal_knowledge_systems� rememberizer_account_information�&list_personal_team_knowledge_documents� remember_this�N)�__name__� __module__�__qualname__�__firstlineno__�SEARCH�AGENTIC_SEARCH�LIST_INTEGRATIONS�ACCOUNT_INFORMATION�LIST_DOCUMENTS�MEMORIZE�__static_attributes__r��P/Users/cuongpm/Code/mcp-server-rememberizer/src/mcp_server_rememberizer/utils.pyr r s"�� ?�F�6�N�9��<��=�N��Hr!r c�R�\r
SrS
rS\S\4SjrSS\S\4Sjj
rSS \S\4S jj
rSr g) � APIClient�"�base_url�api_keyc �T�[R"U
[S
S
SSS9SUS.S9Ulg)NgN@g@)�connect�read�write�poolzapplication/json)zContent-Typez X-API-Key)r&�timeout�headers)�httpx�AsyncClientr�http_client)�selfr&r's r"�__init__�APIClient.__init__#s1�� �,�,���D�t�3�S�I� 2�$�� ��r!N�path�paramsc��X# �[RS
U
35
URRXS9IShv
�NnURS:Xa[S5
e
UR 5 UR5$N?![adn[RSURRSU
S[U5
3SS 9 [S U
SURR35
e
SnAf
[a7n[RSU
S[U5
3SS 9 [S U
S 35
e
SnAf
f=f
7f
)Nz Fetching )
r6�
�=Error: Unauthorized. Please check your REMEMBERIZER API token�HTTP z error while fetching �: T�
�exc_infozFailed to fetch � . Status: z Connection error while fetching �. Connection error.) �logger�debugr1�get�status_coder�raise_for_status�jsonr�error�response�strr)r2r5r6rG�excs r"rB� APIClient.get-s=
��� I
��L�L�9�T�F�+�,�!�-�-�1�1�$�1�F�F�H��#�#�s�*��S��� �%�%�'��=�=�?�"� G
��� ��L�L�����0�0�1�1G��v�R�PS�TW�PX�z�Z�� � ��"�4�&� �3�<�<�3K�3K�2L�M�� ��� I
��L�L�2�4�&��3�s�8�*�E�PT� � ��-�d�V�3F�G�H�H��  I
�sE�
D*
�4A:�
A8�>A:�7
D*
�8
A:�: D'�AC#�# D'�02D"�"D'�'D*
�datac��Z# �[RS
U
35
URRXUS9IShv
�NnURS:Xa[S5
e
UR 5 UR5$N?![adn[RSURRSU
S[U5
3SS 9 [S U
SURR35
e
SnAf
[a7n[RSU
S[U5
3SS 9 [S U
S 35
e
SnAf
f=f
7f
)NzPosting to )rEr6r8r9r:z error while posting to r;Tr<zFailed to post to r>z"Connection error while posting to r?) r@rAr1�postrCrrDrErrFrGrHr)r2r5rKr6rGrIs r"rM�APIClient.postEs@
��� K
��L�L�;�t�f�-�.�!�-�-�2�2�4�6�2�R�R�H��#�#�s�*��S��� �%�%�'��=�=�?�"� S
��� ��L�L�����0�0�1�1I�$��r�RU�VY�RZ�Q[�\�� � ��$�T�F�*�S�\�\�5M�5M�4N�O�� ��� K
��L�L�4�T�F�"�S��X�J�G�RV� � ��/��v�5H�I�J�J��  K
�sE�
D+
�5A;�
A9�>A;�8
D+
�9
A;�; D(�AC$�$ D(�12D#�#D(�(D+
)
r1)
N) rrrrrHr3�dictrBrMr rr!r"r$r$"sG�� �� �s� �I
�c�I
�4�I
�0K
�T�K
�4�K
�K
r!r$c
�D�US
S:XaSO
Sn
[
SU
SUS35
$)N�integration_type�slack�documentzrememberizer://�
/�pkr )rS�hosts r"�get_document_urirW^s3���1�2�g�=�7�:�D��O�D�6��8�D�>�*:�;�<�<r!)�logging�enumrr/�dotenvrrrr�mcpr�pydanticr � getLoggerrr@�APP_NAME�ACCOUNT_INFORMATION_PATH�LIST_DOCUMENTS_PATH�RETRIEVE_DOCUMENT_PATH�RETRIEVE_SLACK_PATH�SEARCH_PATH�AGENTIC_SEARCH_PATH�LIST_INTEGRATIONS_PATH� MEMORIZE_PATHr r$rWrr!r"�<module>rg
s��


�����5�5��� � � �8� $��� �$��%��"��3��I��!��1��(��%� �
��
�9
K
�9
K
�x

=r!