__main__.cpython-310.pyc•4.42 kB
o
�{�g� � @ s� d Z ddlZddlZddlZddlZddlmZmZ ddlm Z ddl
m
Z dee ef fdd �Z
dd e d e d
eddfdd�Zdd� Zddd�ZedkrQe� dS dS )z>
Main entry point for running the persistent-code MCP server.
� N)�Dict�Any� )�PersistentCodeMCP)�config�returnc C s: t jdd�} | jddd�}|jddd�}|jd d
d d d
� |jddddd
� |jddddd� |jddd�}|jd d
d d d
� |jddddd
� |jddddd gd d � |jd td d!d"� |jd#d$d�}|jd%d&d'd(gd)d*� |jd+d,d-d� |jd.d/td0d1� |jd2ddd3d� | �� }|js�| �� t �
d4� t |�S )5zSParse command line arguments.
Returns:
args: Parsed arguments
zPersistent-Code MCP Server)� description�commandzCommand to run)�dest�help�initzInitialize a new project)r z--project-namez-p�defaultzName of the project)r
r z
--storage-dirz-sNz"Directory to store persistent dataz--disable-llama-indexz-d�
store_truez Disable LlamaIndex integration)�actionr �servezStart the MCP serverz --transportz-t�stdio�httpzTransport protocol to use)r
�choicesr z--porti@ z Port to use for HTTP transport)�typer
r r zConfigure settingsz
--llama-indexz-l�enable�disablez(Enable or disable LlamaIndex integration)r r z--embedding-modelz-ez&Set the embedding model for LlamaIndexz--similarity-top-kz-kz0Set the number of similar components to retrieve)r r z
--show-configzShow current configurationr ) �argparse�ArgumentParser�add_subparsers�
add_parser� add_argument�int�
parse_argsr �
print_help�sys�exit�vars)�parser�
subparsersZ init_parserZ serve_parserZ
config_parser�args� r% �S/Users/sparshdrolia/Sparsh personal/persistent-code-mcp/persistent_code/__main__.pyr s� ������������
r F� project_name� storage_dir�disable_llama_indexc C sr |p
t j�t �� d�}t j|dd� t j�|| �}t j|dd� |r-t�ddd� td� td| � d |� �� d
S ) z�Initialize a new project.
Args:
project_name: Name of the project
storage_dir: Directory to store persistent data
disable_llama_index: Whether to disable LlamaIndex integration
�storageT)�exist_ok� llama_index�enabledFz LlamaIndex integration disabledzInitialized project 'z' in N)�os�path�join�getcwd�makedirs�config_instance�set�print)r' r( r) � project_dirr% r% r& � init_projectf s r7 c C s | � d�r8td� tdt�� � �� tdt�� � �� tdt�� � �� tdt�� � �� tdt�� � �� dS | � d �rU| d d
k}t�d d |� td |rPd nd
� �� | � d�rlt�d d| d � td| d � �� | � d�r�t�dd| d � td| d � �� dS dS )zLConfigure settings.
Args:
args: Command-line arguments
� show_configzCurrent configuration:z LlamaIndex enabled: z Embedding model: z Similarity top-k: z Max tokens per component: z Logging level: Nr, r r- zLlamaIndex integration �disabled�embedding_modelzEmbedding model set to: �similarity_top_k�advancedzSimilarity top-k set to: ) �getr5 r3 �is_llama_index_enabled�get_embedding_model�get_similarity_top_k� get_max_tokens_per_component�get_logging_levelr4 )r$ r- r% r% r& �configure_settings} s&
�rC c C s� t � } | d dkrt| d | d | �dd�� dS | d dkr]t| d | d d�}td | d � d
�� td | d � �� | d d
krKtd| d � �� | d d
krUtd��|jdd� dS | d dkrit| � dS dS )zMain entry point.r r r' r( r) Fr )r' r( z1Starting persistent-code MCP server for project '�'z Transport: � transportr zPort: �portz"HTTP transport not yet implementedr )rE r N)r r7 r= r r5 �NotImplementedError�runrC )r$ �serverr% r% r& �main� s$ � �rJ �__main__)NF)r N)�__doc__r. r r �logging�typingr r Z
mcp_serverr r r3 �strr �boolr7 rC rJ �__name__r% r% r% r& �<module> s X
"
�