server.cpython-312.pyc•4.67 kB
�
l��g� � � � d dl Z d dlZd dlZd dlmZmZmZmZ d dlm Z d dl
m Z dZ ej j dd� Z e dd d
g� � Zej# � e d �
� fdedefd�� Zej# � e d�
� e d�
� e dd�� fdededeeeeef defd�� Zdeeef defd�Zd� Zedk( r e� yy) � 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 �F � | s t d� �| j d� s t d� �t j t � d�ddid| i�� }|j
d k7 rd
|j � �S d | � d �S # t $ r%}d
t |� xs t |� � �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� � e�f� f� �=�=�#�$�G� ,�#�%7�8� �h�'�
�� � � �3� &�0�� � ��@� @�2�8�*�<n�o�o�� � -���Q��*�4��7�+�,�,��-�s$ �A)A2 �,A2 �2 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|s t d� �| j d� s t d� �|xs g }|j d|d�� t j t
� d�ddi| |d ��
� }|j d k7 rd |j � �S |j � }t |� }|S # t $ r%}d
t |� xs t |� � �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 . s� � � -��x� � K�L� L��"�"�#8�9� � e�f� f� (�-�2������H�=�>� �=�=�#�$�$:� ;�#�%7�8� $� $��
�� � � �3� &�0�� � ��@� @� � � � �� 1�&� 9��!�!�� � -���Q��*�4��7�+�,�,��-�s$ �BB& � B& �& C�/C� C�Cr# c �� � d}d| v r || d dz z
}d| v r?| d r:|dz
}t | d d� D ]# \ }}d|v s� d|d v s�||� d |d d � d
�z
}�% |j � 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�J�A�w��g�%�+���9M�*M���s�"�W�[�%9�+�%F�$G�r� J�J� � K� �?�?� � � c �, � t j � y )N)�mcp�run� r@ r% �mainrE m s � ��G�G�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 s0 �� � � � ,� ,� &� � 5� �
�*�*�.�.�)�2�
.��
� �z�:�.F�G�� �����d��-��-� � -� �-�: �����d�� �?�� <A�Q�[_�<�+-��+-� � +-� #�4��S�#�X��#7�8�+-� �+-� �+-�\
�4��S��>�
�c�
� � �z���F� r@