server.cpython-311.pyc•5.27 kB
�
l��g� � � � d dl Z d dlZd dlZd dlmZmZmZmZ d dlm Z d dl
m Z dZ ej
� dd� � Z e dd d
g� � � Ze� � � e d �
� � fdedefd�� � Ze� � � e d�
� � e d�
� � e dd�� � fdededeeeeef defd�� � Zdeeef defd�Zd� Zedk r e� � dS dS ) � N)�List�Dict�Any�Optional)�FastMCP)�Fieldzhttps://api.github-chat.com�GITHUB_API_KEY� zgithub-chat-mcp�requestszmcp[cli])� dependencieszNThe GitHub repository URL to index (format: https://github.com/username/repo).)� description�repo_url�returnc �Z � | st d� � �| � d� � st d� � �t j t � d�ddid| i�� � }|j d k r
d
|j � �S d | � d �S # t $ r+}d
t |� � pt |� � � �cY d}~S d}~ww xY w)zrIndex a GitHub repository to analyze its codebase. This must be done before asking questions about the repository.z Repository URL cannot be empty.�https://github.com/�FRepository URL must be in the format: https://github.com/username/repoz/verify� Content-Type�application/jsonr ��headers�json�� zError indexing repository: z!Successfully indexed repository: z2. You can now ask questions about this repository.�Error: N)
�
ValueError�
startswithr �post�GITHUB_CHAT_API_BASE� status_code�text� Exception�str�repr)r �response�es �3/Users/sheing/kagimcp/src/github_chat_mcp/server.py�index_repositoryr& s� � �-�� @� � >�?�?� ?��"�"�#8�9�9� g� � e�f�f� f� �=�#� ,� ,� ,�#�%7�8� �h�'�
�
�
�� � �3� &� &�@�� �@�@� @�o�8�o�o�o�o�� � -� -� -�,��Q���*�4��7�7�,�,�,�,�,�,�,�,�����-���s$ �A,A5 �/A5 �5
B*�? B%� B*�%B*zNThe GitHub repository URL to query (format: https://github.com/username/repo).z)The question to ask about the repository.z;Previous conversation history for multi-turn conversations.)r
�default�question�conversation_historyc �� � | r|st d� � �| � d� � st d� � �|pg }|� d|d�� � t j t
� d�ddi| |d ��
� � }|j d k r
d |j � �S |� � � }t |� � }|S # t $ r+}d
t |� � pt |� � � �cY d}~S d}~ww xY w)zpAsk questions about a GitHub repository and receive detailed AI responses. The repository must be indexed first.z,Repository URL and question cannot be empty.r r �user)�role�contentz/chat/completions/syncr r )r �messagesr r zError querying repository: r N)
r r �appendr r r r r r �format_chat_responser r! r" )r r( r) r. r# �result�formatted_responser$ s r% �query_repositoryr3 . sA � � -�� M�x� M� � K�L�L� L��"�"�#8�9�9� g� � e�f�f� f� (�-�2������H�=�=�>�>�>� �=�#� ;� ;� ;�#�%7�8� $� $�� �
�
�
�� � �3� &� &�@�� �@�@� @� � � � � �� 1�&� 9� 9��!�!�� � -� -� -�,��Q���*�4��7�7�,�,�,�,�,�,�,�,�����-���s$ �B B3 �$B3 �3
C(�= C#� C(�#C(r# c �� � d}d| v r|| d dz z
}d| v rN| d rF|dz
}t | d d� � D ]*\ }}d|v r!d|d v r||� d |d d � d
�z
}�+|� � � S ) z+Format the chat response in a readable way.r
�answerz
�contextsz Sources:
� � meta_data� file_pathz. �
)� enumerate�strip)r# � formatted�i�contexts r% r0 r0 ] s� � ��I��8����X�h�'�&�0�0� ��X� � �(�:�"6� ��\�!� �#�H�Z�$8�!�<�<� K� K�J�A�w��g�%�%�+���9M�*M�*M��� J� J�W�[�%9�+�%F� J� J� J�J� �� �?�?� � � � c �8 � t � � � d S )N)�mcp�run� r@ r% �mainrE m s � ��G�G�I�I�I�I�Ir@ �__main__)r �osr �typingr r r r �mcp.server.fastmcpr �pydanticr r �environ�get�API_KEYrB �toolr! r& r3 r0 rE �__name__rD r@ r% �<module>rP s �� � � � � � � � � ���� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,� &� &� &� &� &� &� � � � � � � 5� �
�*�.�.�)�2�
.�
.��
�g� �z�:�.F�G�G�G�� ������E�d�� � �-� -��-� � -� -� -� ��-�: ������E�d�� � � �E�?�� � � <A�5�Q�[_�<� <� <�+-� +-��+-� � +-� #�4��S�#�X��#7�8�+-� �+-� +-� +-� ��+-�\
�4��S��>�
�c�
�
�
�
� � � � �z����D�F�F�F�F�F� �r@