server.cpython-310.pyc•3.23 kB
o
���g� � @ s d dl Z d dlZd dlZd dlmZmZmZmZ d dlm Z d dl
m Z dZ ej
�dd�Ze dd d
gd �Ze�� e d d
�fdedefdd��Ze�� e dd
�e dd
�e ddd�fdededeeeeef defdd��Zdeeef defdd �Zd d � Zed kr�e� dS dS ) � N)�List�Dict�Any�Optional)�FastMCP)�Fieldzhttps://api.github-chat.comZGITHUB_API_KEY� zgithub-chat-mcp�requestszmcp[cli])� dependencieszNThe GitHub repository URL to index (format: https://github.com/username/repo).)� description�repo_url�returnc
C s� z1| st d��| �d�st d��tjt� d�ddid| id�}|jd kr+d
|j� �W S d | � d �W S tyN } zd
t|�pAt |�� �W Y d}~S d}~ww )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�e� r" �3/Users/sheing/kagimcp/src/github_chat_mcp/server.py�index_repository s"
�
"��r$ 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
C s� zD| r|s t d��| �d�st d��|pg }|�d|d�� tjt� d�ddi| |d �d
�}|jd kr:d |j� �W S |�� }t |�}|W S t
ya } zd
t |�pTt |�� �W Y d}~S d}~ww )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 �resultZformatted_responser! r" r" r# �query_repository. s.
��
"��r/ r c C s� d}d| v r|| d d 7 }d| v r>| d r>|d7 }t | d d�D ] \}}d|v r=d|d v r=||� d |d d � d
�7 }q!|�� S ) z+Format the chat response in a readable way.r Zanswerz
�contextsz Sources:
� Z meta_data� file_pathz. �
)� enumerate�strip)r � formatted�i�contextr" r" r# r- ] s �r- c C s t �� d S )N)�mcp�runr" r" r" r# �mainm s r; �__main__)r �osr �typingr r r r �mcp.server.fastmcpr �pydanticr r �environ�getZAPI_KEYr9 �toolr r$ r/ r- r; �__name__r" r" r" r# �<module> sT ��� � �������
�.
�